From 40a570c9575fbcbc08a7954ebe7018b6e45c3669 Mon Sep 17 00:00:00 2001
From: Daniel Chalef <131175+danielchalef@users.noreply.github.com>
Date: Mon, 25 Aug 2025 16:38:28 -0700
Subject: [PATCH] chore: Update MCP server configuration and documentation
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Updates MCP server factories, pyproject.toml, and README.md to improve configuration handling and documentation clarity.

🤖 Generated with [Claude Code](https://claude.ai/code)

Co-Authored-By: Claude <noreply@anthropic.com>
---
 .../utils/maintenance/graph_data_operations.py |  2 +-
 mcp_server/README.md                           | 18 ++++++++++++++++++
 mcp_server/factories.py                        | 10 +++++-----
 mcp_server/pyproject.toml                      |  4 ++++
 4 files changed, 28 insertions(+), 6 deletions(-)

diff --git a/graphiti_core/utils/maintenance/graph_data_operations.py b/graphiti_core/utils/maintenance/graph_data_operations.py
index 66dc55e4..9c1b5976 100644
--- a/graphiti_core/utils/maintenance/graph_data_operations.py
+++ b/graphiti_core/utils/maintenance/graph_data_operations.py
@@ -89,7 +89,7 @@ async def build_indices_and_constraints(driver: GraphDriver, delete_existing: bo
 
 
 async def clear_data(driver: GraphDriver, group_ids: list[str] | None = None):
-    async with driver.session() as session:
+    with driver.session() as session:
 
         async def delete_all(tx):
             await tx.run('MATCH (n) DETACH DELETE n')
diff --git a/mcp_server/README.md b/mcp_server/README.md
index 999f2258..f39a1f21 100644
--- a/mcp_server/README.md
+++ b/mcp_server/README.md
@@ -97,6 +97,24 @@ database:
   provider: "neo4j"  # or "falkordb" (requires additional setup)
 ```
 
+### Using Ollama for Local LLM
+
+To use Ollama with the MCP server, configure it as an OpenAI-compatible endpoint:
+
+```yaml
+llm:
+  provider: "openai"
+  model: "llama3.2"  # or your preferred Ollama model
+  api_base: "http://localhost:11434/v1"
+  api_key: "ollama"  # dummy key required
+
+embedder:
+  provider: "sentence_transformers"  # recommended for local setup
+  model: "all-MiniLM-L6-v2"
+```
+
+Make sure Ollama is running locally with: `ollama serve`
+
 ### Environment Variables
 
 The `config.yaml` file supports environment variable expansion using `${VAR_NAME}` or `${VAR_NAME:default}` syntax. Key variables:
diff --git a/mcp_server/factories.py b/mcp_server/factories.py
index 48a9011d..9facd0d6 100644
--- a/mcp_server/factories.py
+++ b/mcp_server/factories.py
@@ -25,14 +25,14 @@ except ImportError:
     HAS_AZURE_EMBEDDER = False
 
 try:
-    from graphiti_core.embedder import GeminiEmbedder
+    from graphiti_core.embedder.gemini_embedder import GeminiEmbedder
 
     HAS_GEMINI_EMBEDDER = True
 except ImportError:
     HAS_GEMINI_EMBEDDER = False
 
 try:
-    from graphiti_core.embedder import VoyageEmbedder
+    from graphiti_core.embedder.voyage_embedder import VoyageEmbedder
 
     HAS_VOYAGE_EMBEDDER = True
 except ImportError:
@@ -46,21 +46,21 @@ except ImportError:
     HAS_AZURE_LLM = False
 
 try:
-    from graphiti_core.llm_client import AnthropicClient
+    from graphiti_core.llm_client.anthropic_client import AnthropicClient
 
     HAS_ANTHROPIC = True
 except ImportError:
     HAS_ANTHROPIC = False
 
 try:
-    from graphiti_core.llm_client import GeminiClient
+    from graphiti_core.llm_client.gemini_client import GeminiClient
 
     HAS_GEMINI = True
 except ImportError:
     HAS_GEMINI = False
 
 try:
-    from graphiti_core.llm_client import GroqClient
+    from graphiti_core.llm_client.groq_client import GroqClient
 
     HAS_GROQ = True
 except ImportError:
diff --git a/mcp_server/pyproject.toml b/mcp_server/pyproject.toml
index 298a881a..60aaa6d3 100644
--- a/mcp_server/pyproject.toml
+++ b/mcp_server/pyproject.toml
@@ -11,6 +11,10 @@ dependencies = [
     "azure-identity>=1.21.0",
     "pydantic-settings>=2.0.0",
     "pyyaml>=6.0",
+    "google-genai>=1.8.0",
+    "anthropic>=0.49.0",
+    "groq>=0.2.0",
+    "voyageai>=0.2.3",
 ]
 
 [dependency-groups]