memgraph-toolbox 0.2.2__tar.gz → 0.2.3__tar.gz

{memgraph_toolbox-0.2.2 → memgraph_toolbox-0.2.3}/.gitignore

@@ -187,3 +187,6 @@ cython_debug/
 
 # deepeval writes run caches and telemetry here; build artifacts, not source.
 .deepeval/
+
+# Eval runs hold benchmark text (questions, answers, retrieval); only published baselines are committed.
+context-graph/eval/runs/

{memgraph_toolbox-0.2.2 → memgraph_toolbox-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: memgraph-toolbox
-Version: 0.2.2
+Version: 0.2.3
 Summary: Memgraph toolbox library for Memgraph AI tools and utilities
 Project-URL: Homepage, https://github.com/memgraph/ai-toolkit
 Project-URL: Source, https://github.com/memgraph/ai-toolkit

{memgraph_toolbox-0.2.2 → memgraph_toolbox-0.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "memgraph-toolbox"
-version = "0.2.2"
+version = "0.2.3"
 description = "Memgraph toolbox library for Memgraph AI tools and utilities"
 readme = "README.md"
 authors = [

{memgraph_toolbox-0.2.2 → memgraph_toolbox-0.2.3}/src/memgraph_toolbox/api/memgraph.py

@@ -112,6 +112,16 @@ def merge_with_retry(db: Any, query: str, params: dict | None = None, *, attempt
     raise ValueError(f"attempts must be at least 1, got {attempts}")
 
 
+def _driver_database(name: str) -> str | None:
+    """The database to hand the driver: *name*, or None for an empty one (the login's main database).
+
+    Over a direct ``bolt://`` connection ``""`` and None both leave the database to the
+    server. A routing ``neo4j://`` driver instead asks the cluster for the routing table
+    of a database named ``""``, which has no writer, so every write fails.
+    """
+    return name or None
+
+
 class Memgraph:
     """
     Base Memgraph client for interacting with Memgraph database.
@@ -141,7 +151,7 @@ class Memgraph:
             url: The Memgraph connection URL
             username: Username for authentication
             password: Password for authentication
-            database: The database name to connect to (default: "memgraph")
+            database: The database name to connect to (default: "memgraph"); "" for the login's main database
             driver_config: Additional Neo4j driver configuration
             user_agent: Client name sent to the server (e.g. "mcp-memgraph", "langchain-memgraph", "sql2graph", etc.)
         """
@@ -157,7 +167,7 @@ class Memgraph:
 
         self.driver = GraphDatabase.driver(url, auth=(username, password), **config)
 
-        self.database = env["MEMGRAPH_DATABASE"]
+        self.database = _driver_database(env["MEMGRAPH_DATABASE"])
 
         try:
             import neo4j
@@ -251,7 +261,7 @@ class AsyncMemgraph:
             url: The Memgraph connection URL
             username: Username for authentication
             password: Password for authentication
-            database: The database name to connect to (default: "memgraph")
+            database: The database name to connect to (default: "memgraph"); "" for the login's main database
             driver_config: Additional Neo4j driver configuration
             user_agent: Client name sent to the server (e.g. "mcp-memgraph", "langchain-memgraph", "sql2graph", etc.)
         """
@@ -266,7 +276,7 @@ class AsyncMemgraph:
         config.setdefault("user_agent", user_agent or self.DEFAULT_USER_AGENT)
 
         self.driver = AsyncGraphDatabase.driver(url, auth=(username, password), **config)
-        self.database = env["MEMGRAPH_DATABASE"]
+        self.database = _driver_database(env["MEMGRAPH_DATABASE"])
 
     async def verify_connectivity(self) -> None:
         """

{memgraph_toolbox-0.2.2 → memgraph_toolbox-0.2.3}/src/memgraph_toolbox/client/mcp_prompt.py

@@ -6,7 +6,7 @@ import sys
 from typing import cast
 
 try:
-    from litellm import acompletion, experimental_mcp_client
+    from litellm import ModelResponse, acompletion, experimental_mcp_client
     from mcp import ClientSession, StdioServerParameters
     from mcp.client.stdio import stdio_client
 except ImportError as e:
@@ -69,10 +69,13 @@ async def prompt_with_tools(
         logger.info(f"{len(tools)} MCP tools loaded")
 
         messages = [*tool_selection_system_messages, {"role": "user", "content": prompt}]
-        resp = await acompletion(
-            model=tool_selection_model_name,
-            messages=messages,
-            tools=tools,
+        resp = cast(
+            "ModelResponse",
+            await acompletion(
+                model=tool_selection_model_name,
+                messages=messages,
+                tools=tools,
+            ),
         )
         msg = resp["choices"][0]["message"]
         tool_calls = msg.get("tool_calls", [])
@@ -147,9 +150,12 @@ async def prompt_with_tools(
                     )
 
             messages = messages + response_system_messages
-            final_resp = await acompletion(
-                model=response_model_name,
-                messages=messages,
+            final_resp = cast(
+                "ModelResponse",
+                await acompletion(
+                    model=response_model_name,
+                    messages=messages,
+                ),
             )
             return final_resp["choices"][0]["message"].content
 

{memgraph_toolbox-0.2.2 → memgraph_toolbox-0.2.3}/src/memgraph_toolbox/tests/test_async_memgraph.py

@@ -32,6 +32,11 @@ def test_async_memgraph_resolves_config_from_env(monkeypatch):
     assert explicit.database == "arg-db"
 
 
+def test_an_empty_database_reaches_the_driver_as_none():
+    """An empty name means the login's main database; a routing driver would look up a database named ""."""
+    assert AsyncMemgraph(url="neo4j://localhost:7687", database="").database is None
+
+
 def test_async_memgraph_mirrors_sync_defaults():
     """AsyncMemgraph shares the sync client's user-agent default."""
     assert AsyncMemgraph.DEFAULT_USER_AGENT == Memgraph.DEFAULT_USER_AGENT