From 57ad247666fb9fd4f15473c652d3b288e2d803ae Mon Sep 17 00:00:00 2001 From: Daniel Chalef <131175+danielchalef@users.noreply.github.com> Date: Thu, 30 Oct 2025 21:41:35 -0700 Subject: [PATCH] fix: Remove obsolete KuzuDB check from status endpoint - Remove dead code checking for 'kuzu' provider (was removed) - Simplify status check to use configured database provider directly - Status now correctly reports neo4j or falkordb based on config --- mcp_server/src/graphiti_mcp_server.py | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/mcp_server/src/graphiti_mcp_server.py b/mcp_server/src/graphiti_mcp_server.py index 14e18cde..baa41baf 100644 --- a/mcp_server/src/graphiti_mcp_server.py +++ b/mcp_server/src/graphiti_mcp_server.py @@ -716,24 +716,18 @@ async def get_status() -> StatusResponse: try: client = await graphiti_service.get_client() - # For KuzuDB, just check if the client exists (it's in-memory) - if config.database.provider.lower() == 'kuzu': - provider_info = f'{config.database.provider} database (in-memory)' - return StatusResponse( - status='ok', - message=f'Graphiti MCP server is running and connected to {provider_info}', - ) - - # For Neo4j and FalkorDB, test connection with a simple query + # Test database connection with a simple query async with client.driver.session() as session: result = await session.run('MATCH (n) RETURN count(n) as count') # Consume the result to verify query execution if result: _ = [record async for record in result] - provider_info = f'{config.database.provider} database' + # Use the actual configured provider from the global config + provider_name = config.database.provider return StatusResponse( - status='ok', message=f'Graphiti MCP server is running and connected to {provider_info}' + status='ok', + message=f'Graphiti MCP server is running and connected to {provider_name} database', ) except Exception as e: error_msg = str(e)