Support Host (#30)

* Support Host

Author: zhaoshuaibo-china

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "alibaba-cloud-ops-mcp-server"
-version = "0.8.8"
+version = "0.8.9"
 description = "A MCP server for Alibaba Cloud"
 readme = "README.md"
 authors = [

src/alibaba_cloud_ops_mcp_server/server.py

@@ -21,11 +21,18 @@
     default=8000,
     help="Port number",
 )
-def main(transport: str, port: int):
+@click.option(
+    "--host",
+    type=str,
+    default="127.0.0.1",
+    help="Host",
+)
+def main(transport: str, port: int, host: str):
     # Create an MCP server
     mcp = FastMCP(
         name="alibaba-cloud-ops-mcp-server",
-        port=port
+        port=port,
+        host=host
     )
     for tool in oos_tools.tools:
         mcp.add_tool(tool)

tests/test_server.py

@@ -11,8 +11,8 @@ def test_main_run(mock_create_api_tools, mock_FastMCP):
         mcp = MagicMock()
         mock_FastMCP.return_value = mcp
         # 调用main函数
-        server.main.callback(transport='stdio', port=12345)
-        mock_FastMCP.assert_called_once_with(name='alibaba-cloud-ops-mcp-server', port=12345)
+        server.main.callback(transport='stdio', port=12345, host='127.0.0.1')
+        mock_FastMCP.assert_called_once_with(name='alibaba-cloud-ops-mcp-server', port=12345, host='127.0.0.1')
         assert mcp.add_tool.call_count == 3  # oss/oos/cms 各1
         mock_create_api_tools.assert_called_once()
         mcp.run.assert_called_once_with(transport='stdio')