From 8332ec3aa9b5907a7f164d6a34a3fe3d7e2bd582 Mon Sep 17 00:00:00 2001
From: Daniel Chalef <131175+danielchalef@users.noreply.github.com>
Date: Thu, 30 Oct 2025 07:22:58 -0700
Subject: [PATCH] conductor-checkpoint-msg_0127MeSvxWk8BLXjB5k3wDJY

---
 mcp_server/src/graphiti_mcp_server.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/mcp_server/src/graphiti_mcp_server.py b/mcp_server/src/graphiti_mcp_server.py
index 5bb4e502..e57b8ed5 100644
--- a/mcp_server/src/graphiti_mcp_server.py
+++ b/mcp_server/src/graphiti_mcp_server.py
@@ -235,9 +235,13 @@ class GraphitiService:
                     max_coroutines=self.semaphore_limit,
                 )
 
-            # Test connection (only Neo4j has verify_connectivity)
+            # Test connection based on database provider
             if self.config.database.provider.lower() == 'neo4j':
                 await self.client.driver.client.verify_connectivity()  # type: ignore
+            elif self.config.database.provider.lower() == 'falkordb':
+                # FalkorDB uses ping() method to test connectivity
+                self.client.driver.client.ping()  # type: ignore
+            # KuzuDB doesn't need a connectivity test (in-memory)
 
             # Build indices
             await self.client.build_indices_and_constraints()