diff --git a/mcp_server/Dockerfile b/mcp_server/Dockerfile index 683634d8..83c51949 100644 --- a/mcp_server/Dockerfile +++ b/mcp_server/Dockerfile @@ -10,6 +10,8 @@ RUN apt-get update && apt-get install -y curl && \ # Add uv to PATH ENV PATH="/root/.local/bin:${PATH}" +ENV MCP_SERVER_HOST="0.0.0.0" + # Copy pyproject.toml and install dependencies COPY pyproject.toml . RUN uv sync diff --git a/mcp_server/graphiti_mcp_server.py b/mcp_server/graphiti_mcp_server.py index d7a07f40..9b8330b4 100644 --- a/mcp_server/graphiti_mcp_server.py +++ b/mcp_server/graphiti_mcp_server.py @@ -1166,6 +1166,11 @@ async def initialize_server() -> MCPConfig: action='store_true', help='Enable entity extraction using the predefined ENTITY_TYPES', ) + parser.add_argument( + '--host', + default=os.environ.get('MCP_SERVER_HOST'), + help='Host to bind the MCP server to (default: MCP_SERVER_HOST environment variable)', + ) args = parser.parse_args() @@ -1187,6 +1192,11 @@ async def initialize_server() -> MCPConfig: # Initialize Graphiti await initialize_graphiti() + if args.host: + logger.info(f'Setting MCP server host to: {args.host}') + # Set MCP server host from CLI or env + mcp.settings.host = args.host + # Return MCP configuration return MCPConfig.from_cli(args)