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

adbpg_mcp_server-2.0.0/PKG-INFO

@@ -0,0 +1,278 @@
+Metadata-Version: 2.4
+Name: adbpg-mcp-server
+Version: 2.0.0
+Summary: AnalyticDB PostgreSQL MCP Server serves as a universal interface between AI Agents and AnalyticDB PostgreSQL databases. It enables seamless communication between AI Agents and AnalyticDB PostgreSQL, helping AI Agents retrieve database metadata and execute SQL operations.
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: mcp>=1.13.1
+Requires-Dist: psycopg>=3.2.10
+Requires-Dist: requests>=2.32.5
+
+# AnalyticDB PostgreSQL MCP Server
+
+AnalyticDB PostgreSQL MCP Server serves as a universal interface between AI Agents and AnalyticDB PostgreSQL databases. It enables seamless communication between AI Agents and AnalyticDB PostgreSQL, helping AI Agents retrieve database metadata and execute SQL operations.
+
+## Installation
+
+You can set up the server either from the source code for development or by installing it from PyPI for direct use.
+
+### Option 1: From Source (for Development)
+
+This method is recommended if you want to modify or contribute to the server.
+
+```shell
+# 1. Clone the repository
+git clone https://github.com/aliyun/alibabacloud-adbpg-mcp-server.git
+cd alibabacloud-adbpg-mcp-server
+
+# 2. Create and activate a virtual environment using uv
+uv venv .venv
+source .venv/bin/activate  # On Linux/macOS
+# .\.venv\Scripts\activate  # On Windows
+
+# 3. Install the project in editable mode
+uv pip install -e .
+```
+
+### Option 2: From PyPI (for Production/Usage)
+
+This is the simplest way to install the server for direct use within your projects.
+
+```shell
+pip install adbpg_mcp_server
+```
+
+## Running the Server
+
+The server can be run in two transport modes: `stdio` (default) for integration with MCP clients, and `http` for direct API access or debugging.
+
+Make sure you have set up the required [Environment Variables](#environment-variables) before running the server.
+
+### Stdio Mode (Default)
+
+This is the standard mode for communication with an MCP client.
+
+```bash
+# Run using the default transport (stdio)
+uv run adbpg-mcp-server
+
+# Or explicitly specify the transport
+uv run adbpg-mcp-server --transport stdio
+```
+
+### Streamable-HTTP Mode
+
+This mode exposes an HTTP server, which is useful for testing, debugging, or direct integration via REST APIs.
+
+```bash
+# Run the server in HTTP mode on the default host and port (127.0.0.1:3000)
+uv run adbpg-mcp-server --transport http
+
+# Specify a custom host and port
+uv run adbpg-mcp-server --transport http --host 0.0.0.0 --port 3000
+```
+
+## MCP Integration
+
+To integrate this server with a parent MCP client, add the following configuration to the client's configuration file. The arguments in the `args` array will depend on the transport protocol you choose.
+
+### Example for Stdio Transport 
+
+```json
+"mcpServers": {
+  "adbpg-mcp-server": {
+    "command": "uv",
+    "args": [
+      "run",
+      "adbpg-mcp-server",
+      "--transport",
+      "stdio"
+    ],
+    "env": {
+      "ADBPG_HOST": "host",
+      "ADBPG_PORT": "port",
+      "ADBPG_USER": "username",
+      "ADBPG_PASSWORD": "password",
+      "ADBPG_DATABASE": "database",
+      "GRAPHRAG_API_KEY": "graphrag llm api key",
+      "GRAPHRAG_BASE_URL": "graphrag llm base url",
+      "GRAPHRAG_LLM_MODEL": "graphrag llm model name",
+      "GRAPHRAG_EMBEDDING_MODEL": "graphrag embedding model name",
+      "GRAPHRAG_EMBEDDING_API_KEY": "graphrag embedding api key",
+      "GRAPHRAG_EMBEDDING_BASE_URL": "graphrag embedding url",
+      "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_ENABLE_GRAPH": "enable graph engine for llm memory (Default: false)"
+    }
+  }
+}
+```
+> **Note:** Since `stdio` is the default, you can optionally omit `"--transport", "stdio"` from the `args` array.
+
+### Example for Streamable-HTTP Transport
+
+```json
+"mcpServers": {
+  "adbpg-mcp-server": {
+    "command": "uv",
+    "args": [
+      "run",
+      "adbpg-mcp-server",
+      "--transport",
+      "http",
+      "--port",
+      "3000"
+    ],
+    "env": {
+      "ADBPG_HOST": "host",
+      "ADBPG_PORT": "port",
+      "ADBPG_USER": "username",
+      "ADBPG_PASSWORD": "password",
+      "ADBPG_DATABASE": "database",
+      "GRAPHRAG_API_KEY": "graphrag llm api key",
+      "GRAPHRAG_BASE_URL": "graphrag llm base url",
+      "GRAPHRAG_LLM_MODEL": "graphrag llm model name",
+      "GRAPHRAG_EMBEDDING_MODEL": "graphrag embedding model name",
+      "GRAPHRAG_EMBEDDING_API_KEY": "graphrag embedding api key",
+      "GRAPHRAG_EMBEDDING_BASE_URL": "graphrag embedding url",
+      "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_ENABLE_GRAPH": "enable graph engine for llm memory (Default: false)"
+    }
+  }
+}
+```
+
+
+### Tools
+
+* `execute_select_sql`: Execute SELECT SQL queries on the AnalyticDB PostgreSQL server
+* `execute_dml_sql`: Execute DML (INSERT, UPDATE, DELETE) SQL queries on the AnalyticDB PostgreSQL server
+* `execute_ddl_sql`: Execute DDL (CREATE, ALTER, DROP) SQL queries on the AnalyticDB PostgreSQL server
+* `analyze_table`: Collect table statistics
+* `explain_query`: Get query execution plan
+
+* `adbpg_graphrag_upload`
+    - **Description:** Upload a text file (with its name) and file content to graphrag to generate a knowledge graph.
+    - **Parameters:**
+        - `filename` (`text`): The name of the file to be uploaded.
+        - `context` (`text`): The textual content of the file.
+
+* `adbpg_graphrag_query`
+    - **Description:** Query the graphrag using the specified query string and mode。
+    - **Parameters:**
+        - `query_str` (`text`): the query content.
+        - `query_mode` (`text`): The query mode, choose from `[bypass, naive, local, global, hybrid, mix]`. If null, defaults to `mix`.
+
+* `adbpg_graphrag.upload_decision_tree(context text, root_node text)`  
+    - **Description:** Upload a decision tree with the specified `root_node`. If the `root_node` does not exist, a new decision tree will be created.
+    - **Parameters:**
+        - `context` (`text`): The textual representation of the decision tree.
+        - `root_node` (`text`): The content of the root node.
+
+* `adbpg_graphrag.append_decision_tree(context text, root_node_id text)`  
+    - **Description:** Append a subtree to an existing decision tree at the node specified by `root_node_id`.
+    - **Parameters:**
+        - `context` (`text`): The textual representation of the subtree.
+        - `root_node_id` (`text`): The ID of the node to which the subtree will be appended.
+
+* `adbpg_graphrag.delete_decision_tree(root_node_entity text)`  
+    - **Description:** Delete a sub-decision tree under the node specified by `root_node_entity`.
+    - **Parameters:**
+        - `root_node_entity` (`text`): The ID of the root node of the sub-decision tree to be deleted.
+
+
+
+
+* `adbpg_llm_memory_add`
+    - **Description:** Add LLM long memory.
+    - **Parameters:**
+        - `messages` (`json`): The name of the file to be uploaded.
+        - `user_id` (`text`): The user id.
+        - `run_id` (`text`): The run id.
+        - `agent_id` (`text`): The agent id.
+        - `metadata` (`json`): The metadata json(optional).
+        - `memory_type` (`text`): The memory type(optional).
+        - `prompt` (`text`): The prompt(optional).
+        **Note:**  
+        At least one of `user_id`, `run_id`, or `agent_id` should be provided.
+
+* `adbpg_llm_memory_get_all`
+    - **Description:** Retrieves all memory records associated with a specific user, run or agent.
+    - **Parameters:**
+        - `user_id` (`text`): User ID (optional). If provided, fetch all memories for this user.
+        - `run_id` (`text`): Run ID (optional).
+        - `agent_id` (`text`): Agent ID (optional). If provided, fetch all memories for this agent.
+        **Note:**  
+        At least one of `user_id`, `run_id`, or `agent_id` should be provided.
+
+* `adbpg_llm_memory_search`
+    - **Description:**  Retrieves memories relevant to the given query for a specific user, run, or agent.
+    - **Parameters:**
+        - `query` (`text`): The search query string.
+        - `user_id` (`text`): User ID (optional). If provided, fetch all memories for this user.
+        - `run_id` (`text`): Run ID (optional).
+        - `agent_id` (`text`): Agent ID (optional). If provided, fetch all memories for this agent.
+        - `filter` (`json`): Additional filter conditions in JSON format (optional).
+        **Note:**  
+        At least one of `user_id`, `run_id`, or `agent_id` should be provided.
+
+* `adbpg_llm_memory_delete_all`:
+    - **Description:** Delete all memory records associated with a specific user, run or agent.
+    - **Parameters:**
+        - `user_id` (`text`): User ID (optional). If provided, fetch all memories for this user.
+        - `run_id` (`text`): Run ID (optional).
+        - `agent_id` (`text`): Agent ID (optional). If provided, fetch all memories for this agent.
+        **Note:**  
+        At least one of `user_id`, `run_id`, or `agent_id` should be provided.
+
+### Resources
+
+#### Built-in Resources
+
+* `adbpg:///schemas`: Get all schemas in the database
+
+#### Resource Templates
+
+* `adbpg:///{schema}/tables`: List all tables in a specific schema
+* `adbpg:///{schema}/{table}/ddl`: Get table DDL
+* `adbpg:///{schema}/{table}/statistics`: Show table statistics
+
+## Environment Variables
+
+MCP Server requires the following environment variables to connect to AnalyticDB PostgreSQL instance:
+
+- `ADBPG_HOST`: Database host address
+- `ADBPG_PORT`: Database port
+- `ADBPG_USER`: Database username
+- `ADBPG_PASSWORD`: Database password
+- `ADBPG_DATABASE`: Database name
+
+MCP Server requires the following environment variables to initialize graphRAG and llm memory server：
+
+- `API_KEY`: API key for LLM provider or embedding API
+- `BASE_URL`: Base URL for LLM or embedding service endpoint
+- `LLM_MODEL`: LLM model name or identifier
+- `EMBEDDING_MODEL`: Embedding model name or identifier
+
+### Resources
+
+#### Built-in Resources
+
+*   `adbpg:///schemas`: Get all schemas in the database
+
+#### Resource Templates
+
+*   `adbpg:///{schema}/tables`: List all tables in a specific schema
+*   `adbpg:///{schema}/{table}/ddl`: Get table DDL
+*   `adbpg:///{schema}/{table}/statistics`: Show table statistics
+
+
+## Dependencies
+
+*   Python 3.11 or higher
+*   `uv` (for environment and package management)

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

@@ -2,31 +2,81 @@
 
 AnalyticDB PostgreSQL MCP Server serves as a universal interface between AI Agents and AnalyticDB PostgreSQL databases. It enables seamless communication between AI Agents and AnalyticDB PostgreSQL, helping AI Agents retrieve database metadata and execute SQL operations.
 
-## Configuration
+## Installation
 
+You can set up the server either from the source code for development or by installing it from PyPI for direct use.
 
+### Option 1: From Source (for Development)
 
-### Mode 1: Download
-
-Download from Github
+This method is recommended if you want to modify or contribute to the server.
 
 ```shell
+# 1. Clone the repository
 git clone https://github.com/aliyun/alibabacloud-adbpg-mcp-server.git
+cd alibabacloud-adbpg-mcp-server
+
+# 2. Create and activate a virtual environment using uv
+uv venv .venv
+source .venv/bin/activate  # On Linux/macOS
+# .\.venv\Scripts\activate  # On Windows
+
+# 3. Install the project in editable mode
+uv pip install -e .
 ```
 
-#### MCP Integration
+### Option 2: From PyPI (for Production/Usage)
+
+This is the simplest way to install the server for direct use within your projects.
+
+```shell
+pip install adbpg_mcp_server
+```
+
+## Running the Server
+
+The server can be run in two transport modes: `stdio` (default) for integration with MCP clients, and `http` for direct API access or debugging.
+
+Make sure you have set up the required [Environment Variables](#environment-variables) before running the server.
 
-Add the following configuration to the MCP client configuration file:
+### Stdio Mode (Default)
+
+This is the standard mode for communication with an MCP client.
+
+```bash
+# Run using the default transport (stdio)
+uv run adbpg-mcp-server
+
+# Or explicitly specify the transport
+uv run adbpg-mcp-server --transport stdio
+```
+
+### Streamable-HTTP Mode
+
+This mode exposes an HTTP server, which is useful for testing, debugging, or direct integration via REST APIs.
+
+```bash
+# Run the server in HTTP mode on the default host and port (127.0.0.1:3000)
+uv run adbpg-mcp-server --transport http
+
+# Specify a custom host and port
+uv run adbpg-mcp-server --transport http --host 0.0.0.0 --port 3000
+```
+
+## MCP Integration
+
+To integrate this server with a parent MCP client, add the following configuration to the client's configuration file. The arguments in the `args` array will depend on the transport protocol you choose.
+
+### Example for Stdio Transport 
 
 ```json
 "mcpServers": {
   "adbpg-mcp-server": {
     "command": "uv",
     "args": [
-      "--directory",
-      "/path/to/adbpg-mcp-server",
       "run",
-      "adbpg-mcp-server"
+      "adbpg-mcp-server",
+      "--transport",
+      "stdio"
     ],
     "env": {
       "ADBPG_HOST": "host",
@@ -43,24 +93,27 @@ 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)"
     }
   }
 }
 ```
+> **Note:** Since `stdio` is the default, you can optionally omit `"--transport", "stdio"` from the `args` array.
 
-### Mode 2: Using pip
+### Example for Streamable-HTTP Transport
 
-```
-pip install adbpg_mcp_server
-```
-#### MCP Integration
 ```json
 "mcpServers": {
   "adbpg-mcp-server": {
-    "command": "uvx",
+    "command": "uv",
     "args": [
-      "adbpg_mcp_server"
+      "run",
+      "adbpg-mcp-server",
+      "--transport",
+      "http",
+      "--port",
+      "3000"
     ],
     "env": {
       "ADBPG_HOST": "host",
@@ -68,22 +121,22 @@ pip install adbpg_mcp_server
       "ADBPG_USER": "username",
       "ADBPG_PASSWORD": "password",
       "ADBPG_DATABASE": "database",
-      "GRAPHRAG_API_KEY": "graphrag api_key",
-      "GRAPHRAG_BASE_URL": "graphrag base_url",
-      "GRAPHRAG_LLM_MODEL": "graphrag model name",
+      "GRAPHRAG_API_KEY": "graphrag llm api key",
+      "GRAPHRAG_BASE_URL": "graphrag llm base url",
+      "GRAPHRAG_LLM_MODEL": "graphrag llm model name",
       "GRAPHRAG_EMBEDDING_MODEL": "graphrag embedding model name",
       "GRAPHRAG_EMBEDDING_API_KEY": "graphrag embedding api key",
       "GRAPHRAG_EMBEDDING_BASE_URL": "graphrag embedding url",
       "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)"
     }
   }
 }
 ```
 
-## Components
 
 ### Tools
 
@@ -100,49 +153,50 @@ pip install adbpg_mcp_server
         - `context` (`text`): The textual content of the file.
 
 * `adbpg_graphrag_query`
-    - **Description:** Query the graphrag using the specified query string and mode.
+    - **Description:** Query the graphrag using the specified query string and mode。
     - **Parameters:**
-        - `query_str` (`text`): The query content.
+        - `query_str` (`text`): the query content.
         - `query_mode` (`text`): The query mode, choose from `[bypass, naive, local, global, hybrid, mix]`. If null, defaults to `mix`.
 
-* `adbpg_graphrag_upload_decision_tree`  
+* `adbpg_graphrag.upload_decision_tree(context text, root_node text)`  
     - **Description:** Upload a decision tree with the specified `root_node`. If the `root_node` does not exist, a new decision tree will be created.
     - **Parameters:**
         - `context` (`text`): The textual representation of the decision tree.
         - `root_node` (`text`): The content of the root node.
 
-* `adbpg_graphrag_append_decision_tree`  
+* `adbpg_graphrag.append_decision_tree(context text, root_node_id text)`  
     - **Description:** Append a subtree to an existing decision tree at the node specified by `root_node_id`.
     - **Parameters:**
         - `context` (`text`): The textual representation of the subtree.
         - `root_node_id` (`text`): The ID of the node to which the subtree will be appended.
 
-* `adbpg_graphrag_delete_decision_tree`  
+* `adbpg_graphrag.delete_decision_tree(root_node_entity text)`  
     - **Description:** Delete a sub-decision tree under the node specified by `root_node_entity`.
     - **Parameters:**
         - `root_node_entity` (`text`): The ID of the root node of the sub-decision tree to be deleted.
 
+
+
+
 * `adbpg_llm_memory_add`
-    - **Description:** Add LLM long memory with a specific user, run or agent.
+    - **Description:** Add LLM long memory.
     - **Parameters:**
-        - `messages` (`json`): The messages.
-        - `user_id` (`text`): User id.
-        - `run_id` (`text`): Run id.
-        - `agent_id` (`text`): Agent id.
+        - `messages` (`json`): The name of the file to be uploaded.
+        - `user_id` (`text`): The user id.
+        - `run_id` (`text`): The run id.
+        - `agent_id` (`text`): The agent id.
         - `metadata` (`json`): The metadata json(optional).
         - `memory_type` (`text`): The memory type(optional).
         - `prompt` (`text`): The prompt(optional).
-        
         **Note:**  
         At least one of `user_id`, `run_id`, or `agent_id` should be provided.
 
 * `adbpg_llm_memory_get_all`
     - **Description:** Retrieves all memory records associated with a specific user, run or agent.
     - **Parameters:**
-        - `user_id` (`text`): User ID.
-        - `run_id` (`text`): Run ID.
-        - `agent_id` (`text`): Agent ID.
-        
+        - `user_id` (`text`): User ID (optional). If provided, fetch all memories for this user.
+        - `run_id` (`text`): Run ID (optional).
+        - `agent_id` (`text`): Agent ID (optional). If provided, fetch all memories for this agent.
         **Note:**  
         At least one of `user_id`, `run_id`, or `agent_id` should be provided.
 
@@ -150,21 +204,19 @@ pip install adbpg_mcp_server
     - **Description:**  Retrieves memories relevant to the given query for a specific user, run, or agent.
     - **Parameters:**
         - `query` (`text`): The search query string.
-        - `user_id` (`text`): User ID.
-        - `run_id` (`text`): Run ID.
-        - `agent_id` (`text`): Agent ID.
+        - `user_id` (`text`): User ID (optional). If provided, fetch all memories for this user.
+        - `run_id` (`text`): Run ID (optional).
+        - `agent_id` (`text`): Agent ID (optional). If provided, fetch all memories for this agent.
         - `filter` (`json`): Additional filter conditions in JSON format (optional).
-        
         **Note:**  
         At least one of `user_id`, `run_id`, or `agent_id` should be provided.
 
 * `adbpg_llm_memory_delete_all`:
     - **Description:** Delete all memory records associated with a specific user, run or agent.
     - **Parameters:**
-        - `user_id` (`text`): User ID.
-        - `run_id` (`text`): Run ID.
-        - `agent_id` (`text`): Agent ID.
-       
+        - `user_id` (`text`): User ID (optional). If provided, fetch all memories for this user.
+        - `run_id` (`text`): Run ID (optional).
+        - `agent_id` (`text`): Agent ID (optional). If provided, fetch all memories for this agent.
         **Note:**  
         At least one of `user_id`, `run_id`, or `agent_id` should be provided.
 
@@ -192,44 +244,25 @@ MCP Server requires the following environment variables to connect to AnalyticDB
 
 MCP Server requires the following environment variables to initialize graphRAG and llm memory server：
 
-- `GRAPHRAG_API_KEY`: API key for LLM provider
-- `GRAPHRAG_BASE_URL`: Base URL for LLM service endpoint
-- `GRAPHRAG_LLM_MODEL`: LLM model name or identifier
-- `GRAPHRAG_EMBEDDING_MODEL`: Embedding model name or identifier
-- `GRAPHRAG_EMBEDDING_API_KEY`: API key for embedding model provider
-- `GRAPHRAG_EMBEDDING_BASE_URL`: Base URL for embedding service endpoint
-- `GRAPHRAG_LANGUAGE`: (Optional)The language used by graphrag. Defaults to English if not set.
-- `GRAPHRAG_ENTITY_TYPES`: (Optional) Specifies the types of entity nodes to be extracted when parsing the document to generate the knowledge graph.
-- `GRAPHRAG_RELATIONSHIP_TYPES`: (Optional) Specifies the types of relationship edges to be extracted when parsing the document to generate the knowledge graph.
-
-- `LLMEMORY_API_KEY`: API key for LLM provider or embedding API
-- `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
+- `API_KEY`: API key for LLM provider or embedding API
+- `BASE_URL`: Base URL for LLM or embedding service endpoint
+- `LLM_MODEL`: LLM model name or identifier
+- `EMBEDDING_MODEL`: Embedding model name or identifier
 
-## Dependencies
+### Resources
 
-- Python 3.10 or higher
-- Required packages:
-  - mcp >= 1.4.0
-  - psycopg >= 3.1.0
-  - python-dotenv >= 1.0.0
-  - pydantic >= 2.0.0
+#### Built-in Resources
 
-## Running
+*   `adbpg:///schemas`: Get all schemas in the database
 
-```bash
-# Create and activate virtual environment
-uv venv .venv
-source .venv/bin/activate  # Linux/Mac
-# or
-.venv\Scripts\activate     # Windows
+#### Resource Templates
 
-# Install dependencies
-uv pip install -e .
+*   `adbpg:///{schema}/tables`: List all tables in a specific schema
+*   `adbpg:///{schema}/{table}/ddl`: Get table DDL
+*   `adbpg:///{schema}/{table}/statistics`: Show table statistics
 
-# Run server
-uv run adbpg-mcp-server
-```
 
+## Dependencies
 
+*   Python 3.11 or higher
+*   `uv` (for environment and package management)

adbpg_mcp_server-2.0.0/pyproject.toml

@@ -0,0 +1,24 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "adbpg-mcp-server"
+version = "2.0.0"
+description = "AnalyticDB PostgreSQL MCP Server serves as a universal interface between AI Agents and AnalyticDB PostgreSQL databases. It enables seamless communication between AI Agents and AnalyticDB PostgreSQL, helping AI Agents retrieve database metadata and execute SQL operations."
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "mcp>=1.13.1",
+    "psycopg>=3.2.10",
+    "requests>=2.32.5",
+]
+
+[project.scripts]
+adbpg-mcp-server = "adbpg_mcp_server.__main__:main"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+
+[tool.setuptools.packages.find]
+where = ["src"]

adbpg_mcp_server-2.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

adbpg_mcp_server-2.0.0/src/adbpg_mcp_server/__main__.py

@@ -0,0 +1,45 @@
+import argparse
+import sys
+
+from .server_http import run_http_server
+from .server_stdio import run_stdio_server
+
+def main():
+    """
+    通过参数选择使用 stdio 或 http 服务器
+    """
+    parser = argparse.ArgumentParser(
+        description="ADBPG Management & Control Plane MCP Server"
+    )
+    parser.add_argument(
+        "--transport",
+        choices = ["stdio", "http"],
+        default = "stdio",
+        help = "Transport protocol to use (stdio or streamable-http)"
+    )
+    http_group = parser.add_argument_group("HTTP Server Options")
+    http_group.add_argument(
+        "--host",
+        default="127.0.0.1",
+        help = "Host to bind the HTTP server to (default: 127.0.0.1)."
+    )
+    http_group.add_argument(
+        "--port",
+        type = int,
+        default = 3000,
+        help = "Port to bind the HTTP server to (default: 3000)."
+    )
+    
+    args = parser.parse_args()
+    print(f"Starting ADBPG MCP Server with transport {args.transport}")
+
+    if args.transport == "http":
+        run_http_server(host=args.host, port=args.port)
+    elif args.transport == "stdio":
+        run_stdio_server()
+    else:
+        raise ValueError(f"Invalid transport: {args.transport}")
+
+if __name__ == "__main__":
+    main()
+