adbpg-mcp-server 1.0.7__tar.gz → 1.0.8__tar.gz

{adbpg_mcp_server-1.0.7 → adbpg_mcp_server-1.0.8}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: adbpg-mcp-server
-Version: 1.0.7
-Summary: MCP server for AnalyticDB PostgreSQL
+Version: 1.0.8
+Summary: MCP Server for AnalyticDB PostgreSQL
 License-File: LICENSE
 Requires-Python: >=3.10
 Requires-Dist: mcp>=1.4.0

{adbpg_mcp_server-1.0.7 → adbpg_mcp_server-1.0.8}/README.md

@@ -43,7 +43,8 @@ Add the following configuration to the MCP client configuration file:
       "LLMEMORY_API_KEY": "llm memory api_key",
       "LLMEMORY_BASE_URL": "llm memory base_url",
       "LLMEMORY_LLM_MODEL": "llm memory model name",
-      "LLMEMORY_EMBEDDING_MODEL": "llm memory embedding model name"
+      "LLMEMORY_EMBEDDING_MODEL": "llm memory embedding model name",
+      "LLMEMORY_ENABLE_GRAPH": "enable graph engine for llm memory (Default: false)"
     }
   }
 }
@@ -78,6 +79,7 @@ pip install adbpg_mcp_server
       "LLMEMORY_BASE_URL": "llm memory base_url",
       "LLMEMORY_LLM_MODEL": "llm memory model name",
       "LLMEMORY_EMBEDDING_MODEL": "llm memory embedding model name"
+      "LLMEMORY_ENABLE_GRAPH": "enable graph engine for llm memory (Default: false)"
     }
   }
 }
@@ -206,6 +208,7 @@ MCP Server requires the following environment variables to initialize graphRAG a
 - `LLMEMORY_BASE_URL`: Base URL for LLM or embedding service endpoint
 - `LLMEMORY_LLM_MODEL`: LLM model name or identifier
 - `LLMEMORY_EMBEDDING_MODEL`: Embedding model name or identifier
+- `LLMEMORY_ENABLE_GRAPH`: (Optional) Enable graph engine for llm memory (Default: false)
 
 ## Dependencies
 

{adbpg_mcp_server-1.0.7 → adbpg_mcp_server-1.0.8}/adbpg_mcp_server.py

@@ -70,7 +70,7 @@ except Exception as e:
     logger.error(f"Error loading environment variables: {e}")
     sys.exit(1)
 
-SERVER_VERSION = "0.1.0"
+SERVER_VERSION = "0.2.0"
 
 
 # 获得 graphrag 初始化配置
@@ -91,6 +91,22 @@ def get_graphrag_config():
     
 # 获得llmemory 初始化配置
 def get_llmemory_config():
+    config = get_db_config()
+    port = 3000
+    sql = """
+        select port from gp_segment_configuration where content = -1 and role = 'p';
+    """
+    try:
+        with psycopg.connect(**config) as conn:
+            conn.autocommit = True
+            with conn.cursor() as cursor:
+                cursor.execute(sql)
+                port = cursor.fetchone()[0]
+    except Error as e:
+        raise RuntimeError(f"Database error: {str(e)}")
+    llmemory_enable_graph = os.getenv("LLMEMORY_ENABLE_GRAPH", "False") 
+    
+
     llm_memory_config = {
         "llm": {
             "provider": "openai",
@@ -116,10 +132,22 @@ def get_llmemory_config():
                 "password": os.getenv("ADBPG_PASSWORD"),
                 "dbname": os.getenv("ADBPG_DATABASE"),
                 "hnsw": "True",
-                "embedding_model_dims": os.getenv("LLMEMORY_EMBEDDING_DIMS", 1024)
+                "embedding_model_dims": os.getenv("LLMEMORY_EMBEDDING_DIMS", 1024),
+                "port": port
             }
         }
     }
+    if llmemory_enable_graph == "True" or llmemory_enable_graph == "true":
+        llm_memory_config["graph_store"] = {
+            "provider": "adbpg",
+            "config": {
+                "url": "http://localhost",
+                "username": os.getenv("ADBPG_USER"),
+                "password": os.getenv("ADBPG_PASSWORD"),
+                "database": os.getenv("ADBPG_DATABASE"),
+                "port": port
+            }
+        }
     return llm_memory_config
 
 def get_db_config():
@@ -870,14 +898,18 @@ async def call_tool(name: str, arguments: dict) -> list[TextContent]:
         if not query_mode:
             # default mode
             query_mode = "mix"
-        if not start_search_node_id:
-            start_search_node_id = None
-
-        # 命令拼接
+            
         wrapped_sql = f"""
-                SELECT adbpg_graphrag.query(%s::text, %s::text, %s::text)
-            """
-        params = [query_str, query_mode, start_search_node_id]
+            SELECT adbpg_graphrag.query(%s::text, %s::text)
+        """
+        params = [query_str, query_mode]
+
+        if start_search_node_id:
+            wrapped_sql = f"""
+                    SELECT adbpg_graphrag.query(%s::text, %s::text, %s::text)
+                """
+            params = [query_str, query_mode, start_search_node_id]
+        
         return get_graphrag_tool_result(wrapped_sql, params)
     
     elif name == "adbpg_graphrag_reset_tree_query":

{adbpg_mcp_server-1.0.7 → adbpg_mcp_server-1.0.8}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "hatchling.build"
 
 [project]
 name = "adbpg-mcp-server"
-version = "1.0.7"
-description = "MCP server for AnalyticDB PostgreSQL"
+version = "1.0.8"
+description = "MCP Server for AnalyticDB PostgreSQL"
 requires-python = ">=3.10"
 dependencies = [
     "mcp>=1.4.0",