mcp-memgraph 0.1.0__py3-none-any.whl

mcp_memgraph/__init__.py

@@ -0,0 +1,23 @@
+from .server import (
+    run_query,
+    get_configuration,
+    get_index,
+    get_constraint,
+    get_schema,
+    get_storage,
+    get_triggers,
+    get_betweenness_centrality,
+    get_page_rank,
+)
+
+__all__ = [
+    "run_query",
+    "get_configuration",
+    "get_index",
+    "get_constraint",
+    "get_schema",
+    "get_storage",
+    "get_triggers",
+    "get_betweenness_centrality",
+    "get_page_rank",
+]

mcp_memgraph/server.py

@@ -0,0 +1,137 @@
+from mcp.server.fastmcp import FastMCP
+
+from memgraph_toolbox.api.memgraph import Memgraph
+from memgraph_toolbox.tools.cypher import CypherTool
+from memgraph_toolbox.tools.config import ShowConfigTool
+from memgraph_toolbox.tools.index import ShowIndexInfoTool
+from memgraph_toolbox.tools.constraint import ShowConstraintInfoTool
+from memgraph_toolbox.tools.schema import ShowSchemaInfoTool
+from memgraph_toolbox.tools.storage import ShowStorageInfoTool
+from memgraph_toolbox.tools.trigger import ShowTriggersTool
+from memgraph_toolbox.tools.betweenness_centrality import BetweennessCentralityTool
+from memgraph_toolbox.tools.page_rank import PageRankTool
+from memgraph_toolbox.utils.logging import logger_init
+
+
+from typing import Any, Dict, List
+
+# Configure logging
+logger = logger_init("mcp-memgraph")
+
+# Initialize FastMCP server
+mcp = FastMCP("mcp-memgraph")
+
+MEMGRAPH_URI = "bolt://localhost:7687"
+MEMGRAPH_USER = ""
+MEMGRAPH_PASSWORD = ""
+
+# Initialize Memgraph client
+db = Memgraph(
+    uri=MEMGRAPH_URI,
+    username=MEMGRAPH_USER,
+    password=MEMGRAPH_PASSWORD,
+)
+
+
+@mcp.tool()
+def run_query(query: str) -> List[Dict[str, Any]]:
+    """Run a Cypher query on Memgraph"""
+    logger.info(f"Running query: {query}")
+    try:
+        result = CypherTool(db=db).call({"query": query})
+        return result
+    except Exception as e:
+        return [f"Error running query: {str(e)}"]
+
+
+@mcp.tool()
+def get_configuration() -> List[Dict[str, Any]]:
+    """Get Memgraph configuration information"""
+    logger.info("Fetching Memgraph configuration...")
+    try:
+        config = ShowConfigTool(db=db).call({})
+        return config
+    except Exception as e:
+        return [f"Error fetching configuration: {str(e)}"]
+
+
+@mcp.tool()
+def get_index() -> List[Dict[str, Any]]:
+    """Get Memgraph index information"""
+    logger.info("Fetching Memgraph index...")
+    try:
+        index = ShowIndexInfoTool(db=db).call({})
+        return index
+    except Exception as e:
+        return [f"Error fetching index: {str(e)}"]
+
+
+@mcp.tool()
+def get_constraint() -> List[Dict[str, Any]]:
+    """Get Memgraph constraint information"""
+    logger.info("Fetching Memgraph constraint...")
+    try:
+        constraint = ShowConstraintInfoTool(db=db).call({})
+        return constraint
+    except Exception as e:
+        return [f"Error fetching constraint: {str(e)}"]
+
+
+@mcp.tool()
+def get_schema() -> List[Dict[str, Any]]:
+    """Get Memgraph schema information"""
+    logger.info("Fetching Memgraph schema...")
+    try:
+        schema = ShowSchemaInfoTool(db=db).call({})
+        return schema
+    except Exception as e:
+        return [f"Error fetching schema: {str(e)}"]
+
+
+@mcp.tool()
+def get_storage() -> List[Dict[str, Any]]:
+    """Get Memgraph storage information"""
+    logger.info("Fetching Memgraph storage...")
+    try:
+        storage = ShowStorageInfoTool(db=db).call({})
+        return storage
+    except Exception as e:
+        return [f"Error fetching storage: {str(e)}"]
+
+
+@mcp.tool()
+def get_triggers() -> List[Dict[str, Any]]:
+    """Get Memgraph triggers information"""
+    logger.info("Fetching Memgraph triggers...")
+    try:
+        triggers = ShowTriggersTool(db=db).call({})
+        return triggers
+    except Exception as e:
+        return [f"Error fetching triggers: {str(e)}"]
+
+
+@mcp.tool()
+def get_betweenness_centrality() -> List[Dict[str, Any]]:
+    """Get betweenness centrality information"""
+    logger.info("Fetching betweenness centrality...")
+    try:
+        betweenness = BetweennessCentralityTool(db=db).call({})
+        return betweenness
+    except Exception as e:
+        return [f"Error fetching betweenness centrality: {str(e)}"]
+
+
+@mcp.tool()
+def get_page_rank() -> List[Dict[str, Any]]:
+    """Get page rank information"""
+    logger.info("Fetching page rank...")
+    try:
+        page_rank = PageRankTool(db=db).call({})
+        return page_rank
+    except Exception as e:
+        return [f"Error fetching page rank: {str(e)}"]
+
+
+if __name__ == "__main__":
+    logger.info("Starting FastMCP server...")
+    mcp.run(transport="stdio")

mcp_memgraph-0.1.0.dist-info/METADATA

@@ -0,0 +1,108 @@
+Metadata-Version: 2.4
+Name: mcp-memgraph
+Version: 0.1.0
+Summary: MCP integration and utilities for Memgraph MCP server
+Project-URL: Homepage, https://github.com/memgraph/ai-toolkit
+Project-URL: Source, https://github.com/memgraph/ai-toolkit/tree/main/integrations/mcp-memgraph
+Project-URL: Issues, https://github.com/memgraph/ai-toolkit/issues
+Author-email: antejavor <ante.javor@memgraph.io>
+License: MIT
+License-File: LICENSE
+Keywords: graph,integration,mcp,memgraph,toolkit
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Python: >=3.10
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: mcp[cli]>=1.3.0
+Requires-Dist: memgraph-toolbox
+Requires-Dist: neo4j>=5.28.1
+Provides-Extra: test
+Requires-Dist: anthropic; extra == 'test'
+Requires-Dist: mcp[cli]>=1.3.0; extra == 'test'
+Requires-Dist: pytest-asyncio>=0.20.3; extra == 'test'
+Requires-Dist: pytest>=8.3.5; extra == 'test'
+Requires-Dist: python-dotenv>=1.0.1; extra == 'test'
+Description-Content-Type: text/markdown
+
+# 🚀 Memgraph MCP Server
+
+Memgraph MCP Server is a lightweight server implementation of the Model Context Protocol (MCP) designed to connect Memgraph with LLMs.
+
+![mcp-server](./mcp-server.png)
+
+## ⚡ Quick start
+
+> 📹 [Memgraph MCP Server Quick Start video](https://www.youtube.com/watch?v=0Tjw5QWj_qY)
+
+### 1. Run Memgraph MCP Server
+
+1. Install [`uv`](https://docs.astral.sh/uv/getting-started/installation/) and create `venv` with `uv venv`. Activate virtual environment with `.venv\Scripts\activate`.
+2. Install dependencies: `uv add "mcp[cli]" httpx`
+3. Run Memgraph MCP server: `uv run server.py`.
+
+### 2. Run MCP Client
+
+1. Install [Claude for Desktop](https://claude.ai/download).
+2. Add the Memgraph server to Claude config:
+
+**MacOS/Linux**
+
+```
+code ~/Library/Application\ Support/Claude/claude_desktop_config.json
+```
+
+**Windows**
+
+```
+code $env:AppData\Claude\claude_desktop_config.json
+```
+
+Example config:
+
+```
+{
+    "mcpServers": {
+      "mpc-memgraph": {
+        "command": "/Users/katelatte/.local/bin/uv",
+        "args": [
+            "--directory",
+            "/Users/katelatte/projects/mcp-memgraph",
+            "run",
+            "server.py"
+        ]
+     }
+   }
+}
+```
+
+> [!NOTE]  
+> You may need to put the full path to the uv executable in the command field. You can get this by running `which uv` on MacOS/Linux or `where uv` on Windows. Make sure you pass in the absolute path to your server.
+
+### 3. Chat with the database
+
+1. Run Memgraph MAGE:
+   ```
+   docker run -p 7687:7687 memgraph/memgraph-mage --schema-info-enabled=True
+   ```
+   The `--schema-info-enabled` configuration setting is set to `True` to allow LLM to run `SHOW SCHEMA INFO` query.
+2. Open Claude Desktop and see the Memgraph tools and resources listed. Try it out! (You can load dummy data from [Memgraph Lab](https://memgraph.com/docs/data-visualization) Datasets)
+
+## 🔧Tools
+
+### run_query()
+
+Run a Cypher query against Memgraph.
+
+## 🗃️ Resources
+
+### get_schema()
+
+Get Memgraph schema information (prerequisite: `--schema-info-enabled=True`).
+
+## 🗺️ Roadmap
+
+The Memgraph MCP Server is just at its beginnings. We're actively working on expanding its capabilities and making it even easier to integrate Memgraph into modern AI workflows.

mcp_memgraph-0.1.0.dist-info/RECORD

@@ -0,0 +1,7 @@
+mcp_memgraph/__init__.py,sha256=Kpg32aj6VDEIBvqIzDkLG8mbs5kQWqkOj5qLdeq7zps,407
+mcp_memgraph/server.py,sha256=kJOfhbo9S2e6Ib-X78aqTeJojtQtkIIfEV6OPaneRxY,4071
+mcp_memgraph-0.1.0.dist-info/METADATA,sha256=8grufJGuCXWqNkvMmtrA74aB-fomev573Woh-20x1ek,3486
+mcp_memgraph-0.1.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+mcp_memgraph-0.1.0.dist-info/entry_points.txt,sha256=iOKnAtGNhVtiUjWaULMDWXTsuisdmnHadzKC0FQjQuM,58
+mcp_memgraph-0.1.0.dist-info/licenses/LICENSE,sha256=Zs4MZAXqusfO2KGQoAOuIqEUtojbyIYIkiJQ0BTt45o,1065
+mcp_memgraph-0.1.0.dist-info/RECORD,,

mcp_memgraph-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.27.0
+Root-Is-Purelib: true
+Tag: py3-none-any

mcp_memgraph-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+mcp-memgraph = mcp_memgraph.server:main

mcp_memgraph-0.1.0.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 Memgraph
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.