mcp-memgraph 0.1.0__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -0,0 +1,174 @@
1
+ # Byte-compiled / optimized / DLL files
2
+ __pycache__/
3
+ *.py[cod]
4
+ *$py.class
5
+
6
+ # C extensions
7
+ *.so
8
+
9
+ # Distribution / packaging
10
+ .Python
11
+ build/
12
+ develop-eggs/
13
+ dist/
14
+ downloads/
15
+ eggs/
16
+ .eggs/
17
+ lib/
18
+ lib64/
19
+ parts/
20
+ sdist/
21
+ var/
22
+ wheels/
23
+ share/python-wheels/
24
+ *.egg-info/
25
+ .installed.cfg
26
+ *.egg
27
+ MANIFEST
28
+
29
+ # PyInstaller
30
+ # Usually these files are written by a python script from a template
31
+ # before PyInstaller builds the exe, so as to inject date/other infos into it.
32
+ *.manifest
33
+ *.spec
34
+
35
+ # Installer logs
36
+ pip-log.txt
37
+ pip-delete-this-directory.txt
38
+
39
+ # Unit test / coverage reports
40
+ htmlcov/
41
+ .tox/
42
+ .nox/
43
+ .coverage
44
+ .coverage.*
45
+ .cache
46
+ nosetests.xml
47
+ coverage.xml
48
+ *.cover
49
+ *.py,cover
50
+ .hypothesis/
51
+ .pytest_cache/
52
+ cover/
53
+
54
+ # Translations
55
+ *.mo
56
+ *.pot
57
+
58
+ # Django stuff:
59
+ *.log
60
+ local_settings.py
61
+ db.sqlite3
62
+ db.sqlite3-journal
63
+
64
+ # Flask stuff:
65
+ instance/
66
+ .webassets-cache
67
+
68
+ # Scrapy stuff:
69
+ .scrapy
70
+
71
+ # Sphinx documentation
72
+ docs/_build/
73
+
74
+ # PyBuilder
75
+ .pybuilder/
76
+ target/
77
+
78
+ # Jupyter Notebook
79
+ .ipynb_checkpoints
80
+
81
+ # IPython
82
+ profile_default/
83
+ ipython_config.py
84
+
85
+ # pyenv
86
+ # For a library or package, you might want to ignore these files since the code is
87
+ # intended to run in multiple environments; otherwise, check them in:
88
+ # .python-version
89
+
90
+ # pipenv
91
+ # According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
92
+ # However, in case of collaboration, if having platform-specific dependencies or dependencies
93
+ # having no cross-platform support, pipenv may install dependencies that don't work, or not
94
+ # install all needed dependencies.
95
+ #Pipfile.lock
96
+
97
+ # UV
98
+ # Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
99
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
100
+ # commonly ignored for libraries.
101
+ #uv.lock
102
+
103
+ # poetry
104
+ # Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
105
+ # This is especially recommended for binary packages to ensure reproducibility, and is more
106
+ # commonly ignored for libraries.
107
+ # https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
108
+ #poetry.lock
109
+
110
+ # pdm
111
+ # Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
112
+ #pdm.lock
113
+ # pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
114
+ # in version control.
115
+ # https://pdm.fming.dev/latest/usage/project/#working-with-version-control
116
+ .pdm.toml
117
+ .pdm-python
118
+ .pdm-build/
119
+
120
+ # PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
121
+ __pypackages__/
122
+
123
+ # Celery stuff
124
+ celerybeat-schedule
125
+ celerybeat.pid
126
+
127
+ # SageMath parsed files
128
+ *.sage.py
129
+
130
+ # Environments
131
+ .env
132
+ .venv
133
+ env/
134
+ venv/
135
+ ENV/
136
+ env.bak/
137
+ venv.bak/
138
+
139
+ # Spyder project settings
140
+ .spyderproject
141
+ .spyproject
142
+
143
+ # Rope project settings
144
+ .ropeproject
145
+
146
+ # mkdocs documentation
147
+ /site
148
+
149
+ # mypy
150
+ .mypy_cache/
151
+ .dmypy.json
152
+ dmypy.json
153
+
154
+ # Pyre type checker
155
+ .pyre/
156
+
157
+ # pytype static type analyzer
158
+ .pytype/
159
+
160
+ # Cython debug symbols
161
+ cython_debug/
162
+
163
+ # PyCharm
164
+ # JetBrains specific template is maintained in a separate JetBrains.gitignore that can
165
+ # be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
166
+ # and can be added to the global gitignore or merged into this file. For a more nuclear
167
+ # option (not recommended) you can uncomment the following to ignore the entire idea folder.
168
+ #.idea/
169
+
170
+ # Ruff stuff:
171
+ .ruff_cache/
172
+
173
+ # PyPI configuration file
174
+ .pypirc
@@ -0,0 +1 @@
1
+ 3.10
@@ -0,0 +1,21 @@
1
+ MIT License
2
+
3
+ Copyright (c) 2025 Memgraph
4
+
5
+ Permission is hereby granted, free of charge, to any person obtaining a copy
6
+ of this software and associated documentation files (the "Software"), to deal
7
+ in the Software without restriction, including without limitation the rights
8
+ to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
9
+ copies of the Software, and to permit persons to whom the Software is
10
+ furnished to do so, subject to the following conditions:
11
+
12
+ The above copyright notice and this permission notice shall be included in all
13
+ copies or substantial portions of the Software.
14
+
15
+ THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16
+ IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17
+ FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18
+ AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19
+ LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20
+ OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21
+ SOFTWARE.
@@ -0,0 +1,108 @@
1
+ Metadata-Version: 2.4
2
+ Name: mcp-memgraph
3
+ Version: 0.1.0
4
+ Summary: MCP integration and utilities for Memgraph MCP server
5
+ Project-URL: Homepage, https://github.com/memgraph/ai-toolkit
6
+ Project-URL: Source, https://github.com/memgraph/ai-toolkit/tree/main/integrations/mcp-memgraph
7
+ Project-URL: Issues, https://github.com/memgraph/ai-toolkit/issues
8
+ Author-email: antejavor <ante.javor@memgraph.io>
9
+ License: MIT
10
+ License-File: LICENSE
11
+ Keywords: graph,integration,mcp,memgraph,toolkit
12
+ Classifier: Intended Audience :: Developers
13
+ Classifier: License :: OSI Approved :: MIT License
14
+ Classifier: Programming Language :: Python :: 3
15
+ Classifier: Programming Language :: Python :: 3.10
16
+ Classifier: Programming Language :: Python :: 3.11
17
+ Classifier: Programming Language :: Python :: 3.12
18
+ Requires-Python: >=3.10
19
+ Requires-Dist: httpx>=0.28.1
20
+ Requires-Dist: mcp[cli]>=1.3.0
21
+ Requires-Dist: memgraph-toolbox
22
+ Requires-Dist: neo4j>=5.28.1
23
+ Provides-Extra: test
24
+ Requires-Dist: anthropic; extra == 'test'
25
+ Requires-Dist: mcp[cli]>=1.3.0; extra == 'test'
26
+ Requires-Dist: pytest-asyncio>=0.20.3; extra == 'test'
27
+ Requires-Dist: pytest>=8.3.5; extra == 'test'
28
+ Requires-Dist: python-dotenv>=1.0.1; extra == 'test'
29
+ Description-Content-Type: text/markdown
30
+
31
+ # 🚀 Memgraph MCP Server
32
+
33
+ Memgraph MCP Server is a lightweight server implementation of the Model Context Protocol (MCP) designed to connect Memgraph with LLMs.
34
+
35
+ ![mcp-server](./mcp-server.png)
36
+
37
+ ## ⚡ Quick start
38
+
39
+ > 📹 [Memgraph MCP Server Quick Start video](https://www.youtube.com/watch?v=0Tjw5QWj_qY)
40
+
41
+ ### 1. Run Memgraph MCP Server
42
+
43
+ 1. Install [`uv`](https://docs.astral.sh/uv/getting-started/installation/) and create `venv` with `uv venv`. Activate virtual environment with `.venv\Scripts\activate`.
44
+ 2. Install dependencies: `uv add "mcp[cli]" httpx`
45
+ 3. Run Memgraph MCP server: `uv run server.py`.
46
+
47
+ ### 2. Run MCP Client
48
+
49
+ 1. Install [Claude for Desktop](https://claude.ai/download).
50
+ 2. Add the Memgraph server to Claude config:
51
+
52
+ **MacOS/Linux**
53
+
54
+ ```
55
+ code ~/Library/Application\ Support/Claude/claude_desktop_config.json
56
+ ```
57
+
58
+ **Windows**
59
+
60
+ ```
61
+ code $env:AppData\Claude\claude_desktop_config.json
62
+ ```
63
+
64
+ Example config:
65
+
66
+ ```
67
+ {
68
+ "mcpServers": {
69
+ "mpc-memgraph": {
70
+ "command": "/Users/katelatte/.local/bin/uv",
71
+ "args": [
72
+ "--directory",
73
+ "/Users/katelatte/projects/mcp-memgraph",
74
+ "run",
75
+ "server.py"
76
+ ]
77
+ }
78
+ }
79
+ }
80
+ ```
81
+
82
+ > [!NOTE]
83
+ > 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.
84
+
85
+ ### 3. Chat with the database
86
+
87
+ 1. Run Memgraph MAGE:
88
+ ```
89
+ docker run -p 7687:7687 memgraph/memgraph-mage --schema-info-enabled=True
90
+ ```
91
+ The `--schema-info-enabled` configuration setting is set to `True` to allow LLM to run `SHOW SCHEMA INFO` query.
92
+ 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)
93
+
94
+ ## 🔧Tools
95
+
96
+ ### run_query()
97
+
98
+ Run a Cypher query against Memgraph.
99
+
100
+ ## 🗃️ Resources
101
+
102
+ ### get_schema()
103
+
104
+ Get Memgraph schema information (prerequisite: `--schema-info-enabled=True`).
105
+
106
+ ## 🗺️ Roadmap
107
+
108
+ 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.
@@ -0,0 +1,78 @@
1
+ # 🚀 Memgraph MCP Server
2
+
3
+ Memgraph MCP Server is a lightweight server implementation of the Model Context Protocol (MCP) designed to connect Memgraph with LLMs.
4
+
5
+ ![mcp-server](./mcp-server.png)
6
+
7
+ ## ⚡ Quick start
8
+
9
+ > 📹 [Memgraph MCP Server Quick Start video](https://www.youtube.com/watch?v=0Tjw5QWj_qY)
10
+
11
+ ### 1. Run Memgraph MCP Server
12
+
13
+ 1. Install [`uv`](https://docs.astral.sh/uv/getting-started/installation/) and create `venv` with `uv venv`. Activate virtual environment with `.venv\Scripts\activate`.
14
+ 2. Install dependencies: `uv add "mcp[cli]" httpx`
15
+ 3. Run Memgraph MCP server: `uv run server.py`.
16
+
17
+ ### 2. Run MCP Client
18
+
19
+ 1. Install [Claude for Desktop](https://claude.ai/download).
20
+ 2. Add the Memgraph server to Claude config:
21
+
22
+ **MacOS/Linux**
23
+
24
+ ```
25
+ code ~/Library/Application\ Support/Claude/claude_desktop_config.json
26
+ ```
27
+
28
+ **Windows**
29
+
30
+ ```
31
+ code $env:AppData\Claude\claude_desktop_config.json
32
+ ```
33
+
34
+ Example config:
35
+
36
+ ```
37
+ {
38
+ "mcpServers": {
39
+ "mpc-memgraph": {
40
+ "command": "/Users/katelatte/.local/bin/uv",
41
+ "args": [
42
+ "--directory",
43
+ "/Users/katelatte/projects/mcp-memgraph",
44
+ "run",
45
+ "server.py"
46
+ ]
47
+ }
48
+ }
49
+ }
50
+ ```
51
+
52
+ > [!NOTE]
53
+ > 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.
54
+
55
+ ### 3. Chat with the database
56
+
57
+ 1. Run Memgraph MAGE:
58
+ ```
59
+ docker run -p 7687:7687 memgraph/memgraph-mage --schema-info-enabled=True
60
+ ```
61
+ The `--schema-info-enabled` configuration setting is set to `True` to allow LLM to run `SHOW SCHEMA INFO` query.
62
+ 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)
63
+
64
+ ## 🔧Tools
65
+
66
+ ### run_query()
67
+
68
+ Run a Cypher query against Memgraph.
69
+
70
+ ## 🗃️ Resources
71
+
72
+ ### get_schema()
73
+
74
+ Get Memgraph schema information (prerequisite: `--schema-info-enabled=True`).
75
+
76
+ ## 🗺️ Roadmap
77
+
78
+ 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.
@@ -0,0 +1,56 @@
1
+ [project]
2
+ name = "mcp-memgraph"
3
+ version = "0.1.0"
4
+ description = "MCP integration and utilities for Memgraph MCP server"
5
+ readme = "README.md"
6
+ requires-python = ">=3.10"
7
+
8
+ authors = [
9
+ { name = "antejavor", email = "ante.javor@memgraph.io" }
10
+ ]
11
+
12
+ license = { text = "MIT" }
13
+
14
+ keywords = ["memgraph", "mcp", "graph", "integration", "toolkit"]
15
+
16
+ classifiers = [
17
+ "Intended Audience :: Developers",
18
+ "License :: OSI Approved :: MIT License",
19
+ "Programming Language :: Python :: 3",
20
+ "Programming Language :: Python :: 3.10",
21
+ "Programming Language :: Python :: 3.11",
22
+ "Programming Language :: Python :: 3.12",
23
+ ]
24
+
25
+ dependencies = [
26
+ "httpx>=0.28.1",
27
+ "mcp[cli]>=1.3.0",
28
+ "neo4j>=5.28.1",
29
+ "memgraph-toolbox"
30
+ ]
31
+
32
+ [project.optional-dependencies]
33
+ test = [
34
+ "pytest>=8.3.5",
35
+ "python-dotenv>=1.0.1",
36
+ "anthropic",
37
+ "mcp[cli]>=1.3.0",
38
+ "pytest-asyncio>=0.20.3"
39
+ ]
40
+
41
+ [project.scripts]
42
+ mcp-memgraph = "mcp_memgraph.server:main"
43
+
44
+ [project.urls]
45
+ "Homepage" = "https://github.com/memgraph/ai-toolkit"
46
+ "Source" = "https://github.com/memgraph/ai-toolkit/tree/main/integrations/mcp-memgraph"
47
+ "Issues" = "https://github.com/memgraph/ai-toolkit/issues"
48
+
49
+ [tool.pytest.ini_options]
50
+ pythonpath = ["src"]
51
+ asyncio_mode = "strict"
52
+ asyncio_default_fixture_loop_scope = "function"
53
+
54
+ [build-system]
55
+ requires = ["hatchling", "build", "setuptools"]
56
+ build-backend = "hatchling.build"
@@ -0,0 +1,23 @@
1
+ from .server import (
2
+ run_query,
3
+ get_configuration,
4
+ get_index,
5
+ get_constraint,
6
+ get_schema,
7
+ get_storage,
8
+ get_triggers,
9
+ get_betweenness_centrality,
10
+ get_page_rank,
11
+ )
12
+
13
+ __all__ = [
14
+ "run_query",
15
+ "get_configuration",
16
+ "get_index",
17
+ "get_constraint",
18
+ "get_schema",
19
+ "get_storage",
20
+ "get_triggers",
21
+ "get_betweenness_centrality",
22
+ "get_page_rank",
23
+ ]
@@ -0,0 +1,137 @@
1
+ from mcp.server.fastmcp import FastMCP
2
+
3
+ from memgraph_toolbox.api.memgraph import Memgraph
4
+ from memgraph_toolbox.tools.cypher import CypherTool
5
+ from memgraph_toolbox.tools.config import ShowConfigTool
6
+ from memgraph_toolbox.tools.index import ShowIndexInfoTool
7
+ from memgraph_toolbox.tools.constraint import ShowConstraintInfoTool
8
+ from memgraph_toolbox.tools.schema import ShowSchemaInfoTool
9
+ from memgraph_toolbox.tools.storage import ShowStorageInfoTool
10
+ from memgraph_toolbox.tools.trigger import ShowTriggersTool
11
+ from memgraph_toolbox.tools.betweenness_centrality import BetweennessCentralityTool
12
+ from memgraph_toolbox.tools.page_rank import PageRankTool
13
+ from memgraph_toolbox.utils.logging import logger_init
14
+
15
+
16
+ from typing import Any, Dict, List
17
+
18
+ # Configure logging
19
+ logger = logger_init("mcp-memgraph")
20
+
21
+ # Initialize FastMCP server
22
+ mcp = FastMCP("mcp-memgraph")
23
+
24
+ MEMGRAPH_URI = "bolt://localhost:7687"
25
+ MEMGRAPH_USER = ""
26
+ MEMGRAPH_PASSWORD = ""
27
+
28
+ # Initialize Memgraph client
29
+ db = Memgraph(
30
+ uri=MEMGRAPH_URI,
31
+ username=MEMGRAPH_USER,
32
+ password=MEMGRAPH_PASSWORD,
33
+ )
34
+
35
+
36
+ @mcp.tool()
37
+ def run_query(query: str) -> List[Dict[str, Any]]:
38
+ """Run a Cypher query on Memgraph"""
39
+ logger.info(f"Running query: {query}")
40
+ try:
41
+ result = CypherTool(db=db).call({"query": query})
42
+ return result
43
+ except Exception as e:
44
+ return [f"Error running query: {str(e)}"]
45
+
46
+
47
+ @mcp.tool()
48
+ def get_configuration() -> List[Dict[str, Any]]:
49
+ """Get Memgraph configuration information"""
50
+ logger.info("Fetching Memgraph configuration...")
51
+ try:
52
+ config = ShowConfigTool(db=db).call({})
53
+ return config
54
+ except Exception as e:
55
+ return [f"Error fetching configuration: {str(e)}"]
56
+
57
+
58
+ @mcp.tool()
59
+ def get_index() -> List[Dict[str, Any]]:
60
+ """Get Memgraph index information"""
61
+ logger.info("Fetching Memgraph index...")
62
+ try:
63
+ index = ShowIndexInfoTool(db=db).call({})
64
+ return index
65
+ except Exception as e:
66
+ return [f"Error fetching index: {str(e)}"]
67
+
68
+
69
+ @mcp.tool()
70
+ def get_constraint() -> List[Dict[str, Any]]:
71
+ """Get Memgraph constraint information"""
72
+ logger.info("Fetching Memgraph constraint...")
73
+ try:
74
+ constraint = ShowConstraintInfoTool(db=db).call({})
75
+ return constraint
76
+ except Exception as e:
77
+ return [f"Error fetching constraint: {str(e)}"]
78
+
79
+
80
+ @mcp.tool()
81
+ def get_schema() -> List[Dict[str, Any]]:
82
+ """Get Memgraph schema information"""
83
+ logger.info("Fetching Memgraph schema...")
84
+ try:
85
+ schema = ShowSchemaInfoTool(db=db).call({})
86
+ return schema
87
+ except Exception as e:
88
+ return [f"Error fetching schema: {str(e)}"]
89
+
90
+
91
+ @mcp.tool()
92
+ def get_storage() -> List[Dict[str, Any]]:
93
+ """Get Memgraph storage information"""
94
+ logger.info("Fetching Memgraph storage...")
95
+ try:
96
+ storage = ShowStorageInfoTool(db=db).call({})
97
+ return storage
98
+ except Exception as e:
99
+ return [f"Error fetching storage: {str(e)}"]
100
+
101
+
102
+ @mcp.tool()
103
+ def get_triggers() -> List[Dict[str, Any]]:
104
+ """Get Memgraph triggers information"""
105
+ logger.info("Fetching Memgraph triggers...")
106
+ try:
107
+ triggers = ShowTriggersTool(db=db).call({})
108
+ return triggers
109
+ except Exception as e:
110
+ return [f"Error fetching triggers: {str(e)}"]
111
+
112
+
113
+ @mcp.tool()
114
+ def get_betweenness_centrality() -> List[Dict[str, Any]]:
115
+ """Get betweenness centrality information"""
116
+ logger.info("Fetching betweenness centrality...")
117
+ try:
118
+ betweenness = BetweennessCentralityTool(db=db).call({})
119
+ return betweenness
120
+ except Exception as e:
121
+ return [f"Error fetching betweenness centrality: {str(e)}"]
122
+
123
+
124
+ @mcp.tool()
125
+ def get_page_rank() -> List[Dict[str, Any]]:
126
+ """Get page rank information"""
127
+ logger.info("Fetching page rank...")
128
+ try:
129
+ page_rank = PageRankTool(db=db).call({})
130
+ return page_rank
131
+ except Exception as e:
132
+ return [f"Error fetching page rank: {str(e)}"]
133
+
134
+
135
+ if __name__ == "__main__":
136
+ logger.info("Starting FastMCP server...")
137
+ mcp.run(transport="stdio")
@@ -0,0 +1,65 @@
1
+ import asyncio
2
+ from typing import Optional
3
+ from contextlib import AsyncExitStack
4
+
5
+ from mcp import ClientSession, StdioServerParameters
6
+ from mcp.client.stdio import stdio_client
7
+
8
+ from anthropic import Anthropic
9
+ from dotenv import load_dotenv
10
+
11
+ import pytest
12
+
13
+ load_dotenv() # load environment variables from .env
14
+
15
+
16
+ class MCPClient:
17
+ """Client for connecting to an MCP server."""
18
+
19
+ def __init__(self):
20
+ # Initialize session and client objects
21
+ self.session: Optional[ClientSession] = None
22
+ self.exit_stack = AsyncExitStack()
23
+ self.anthropic = Anthropic()
24
+
25
+ async def connect_to_server(self, server_script_path: str):
26
+ """Connect to an MCP server
27
+
28
+ Args:
29
+ server_script_path: Path to the server script (.py or .js)
30
+ """
31
+ is_python = server_script_path.endswith(".py")
32
+ is_js = server_script_path.endswith(".js")
33
+ if not (is_python or is_js):
34
+ raise ValueError("Server script must be a .py or .js file")
35
+
36
+ command = "python" if is_python else "node"
37
+ server_params = StdioServerParameters(
38
+ command=command, args=[server_script_path], env=None
39
+ )
40
+
41
+ stdio_transport = await self.exit_stack.enter_async_context(
42
+ stdio_client(server_params)
43
+ )
44
+ self.stdio, self.write = stdio_transport
45
+ self.session = await self.exit_stack.enter_async_context(
46
+ ClientSession(self.stdio, self.write)
47
+ )
48
+
49
+ await self.session.initialize()
50
+
51
+ # List available tools
52
+ response = await self.session.list_tools()
53
+ tools = response.tools
54
+ print("\nConnected to server with tools:", [tool.name for tool in tools])
55
+
56
+
57
+ def test_mcp_client():
58
+ """Test the MCP client connection to the server."""
59
+ server_script_path = "integrations/mcp-memgraph/src/mcp_memgraph/server.py"
60
+ client = MCPClient()
61
+ asyncio.run(client.connect_to_server(server_script_path))
62
+ print("MCP session:", client.session)
63
+
64
+
65
+ test_mcp_client()
Binary file
@@ -0,0 +1,136 @@
1
+ import asyncio
2
+ from typing import Optional
3
+ from contextlib import AsyncExitStack
4
+
5
+ from mcp import ClientSession, StdioServerParameters
6
+ from mcp.client.stdio import stdio_client
7
+
8
+ from anthropic import Anthropic
9
+ from dotenv import load_dotenv
10
+
11
+ from mcp_memgraph import run_query, get_schema
12
+ import pytest
13
+
14
+ pytestmark = pytest.mark.asyncio # Mark all tests in this file as asyncio-compatible
15
+
16
+ load_dotenv() # Load environment variables from .env
17
+
18
+
19
+ class MCPClient:
20
+ """Client for connecting to an MCP server."""
21
+
22
+ def __init__(self):
23
+ # Initialize session and client objects
24
+ self.session: Optional[ClientSession] = None
25
+ self.exit_stack = AsyncExitStack()
26
+ self.anthropic = Anthropic()
27
+
28
+ async def connect_to_server(self, server_script_path: str):
29
+ """Connect to an MCP server
30
+
31
+ Args:
32
+ server_script_path: Path to the server script (.py or .js)
33
+ """
34
+ is_python = server_script_path.endswith(".py")
35
+ is_js = server_script_path.endswith(".js")
36
+ if not (is_python or is_js):
37
+ raise ValueError("Server script must be a .py or .js file")
38
+
39
+ command = "python" if is_python else "node"
40
+ server_params = StdioServerParameters(
41
+ command=command, args=[server_script_path], env=None
42
+ )
43
+
44
+ stdio_transport = await self.exit_stack.enter_async_context(
45
+ stdio_client(server_params)
46
+ )
47
+ self.stdio, self.write = stdio_transport
48
+ self.session = await self.exit_stack.enter_async_context(
49
+ ClientSession(self.stdio, self.write)
50
+ )
51
+
52
+ await self.session.initialize()
53
+
54
+
55
+ @pytest.mark.asyncio
56
+ async def test_mcp_client():
57
+ """Test the MCP client connection to the server."""
58
+ server_script_path = "integrations/mcp-memgraph/src/mcp_memgraph/server.py"
59
+ client = MCPClient()
60
+ try:
61
+ await client.connect_to_server(server_script_path)
62
+ response = await client.session.list_tools()
63
+ tools = response.tools
64
+ print("\nConnected to server with tools:", [tool.name for tool in tools])
65
+
66
+ assert client.session is not None, "Session should be initialized"
67
+ assert client.stdio is not None, "Stdio transport should be initialized"
68
+ assert client.write is not None, "Write transport should be initialized"
69
+
70
+ finally:
71
+ await client.exit_stack.aclose()
72
+
73
+
74
+ @pytest.mark.asyncio
75
+ async def test_run_query():
76
+ """Test the run_query tool."""
77
+ query = "MATCH (n) RETURN n LIMIT 1;"
78
+ response = run_query(query)
79
+ assert isinstance(response, list), "Expected response to be a list"
80
+ assert len(response) >= 0, "Expected response to have at least 0 results"
81
+ # Add more assertions based on expected query results
82
+
83
+
84
+ @pytest.mark.asyncio
85
+ async def test_get_schema():
86
+ """Test the get_schema tool."""
87
+ response = get_schema()
88
+ assert isinstance(response, list), "Expected response to be a list"
89
+ assert len(response) >= 0, "Expected response to have at least 0 results"
90
+ # Add more assertions based on expected schema information
91
+
92
+
93
+ @pytest.mark.asyncio
94
+ async def test_tools_and_resources():
95
+ """Test that all tools and resources are present in the MCP server."""
96
+ server_script_path = "integrations/mcp-memgraph/src/mcp_memgraph/server.py"
97
+ client = MCPClient()
98
+ try:
99
+ await client.connect_to_server(server_script_path)
100
+
101
+ # TODO(@antejavor): Add this dynamically.
102
+ expected_tools = [
103
+ "run_query",
104
+ "get_configuration",
105
+ "get_index",
106
+ "get_constraint",
107
+ "get_schema",
108
+ "get_storage",
109
+ "get_triggers",
110
+ "get_betweenness_centrality",
111
+ "get_page_rank",
112
+ ]
113
+ expected_resources = []
114
+
115
+ response = await client.session.list_tools()
116
+ available_tools = [tool.name for tool in response.tools]
117
+
118
+ response = await client.session.list_resources()
119
+ available_resources = [str(resource.uri) for resource in response.resources]
120
+
121
+ assert len(available_tools) == len(
122
+ expected_tools
123
+ ), "Mismatch in number of tools"
124
+ for tool in expected_tools:
125
+ assert tool in available_tools, f"Tool '{tool}' is missing from the server"
126
+
127
+ assert len(available_resources) == len(
128
+ expected_resources
129
+ ), "Mismatch in number of resources"
130
+ for resource in expected_resources:
131
+ assert (
132
+ resource in available_resources
133
+ ), f"Resource '{resource}' is missing from the server"
134
+
135
+ finally:
136
+ await client.exit_stack.aclose()
@@ -0,0 +1,357 @@
1
+ version = 1
2
+ revision = 1
3
+ requires-python = ">=3.13"
4
+
5
+ [[package]]
6
+ name = "annotated-types"
7
+ version = "0.7.0"
8
+ source = { registry = "https://pypi.org/simple" }
9
+ sdist = { url = "https://files.pythonhosted.org/packages/ee/67/531ea369ba64dcff5ec9c3402f9f51bf748cec26dde048a2f973a4eea7f5/annotated_types-0.7.0.tar.gz", hash = "sha256:aff07c09a53a08bc8cfccb9c85b05f1aa9a2a6f23728d790723543408344ce89", size = 16081 }
10
+ wheels = [
11
+ { url = "https://files.pythonhosted.org/packages/78/b6/6307fbef88d9b5ee7421e68d78a9f162e0da4900bc5f5793f6d3d0e34fb8/annotated_types-0.7.0-py3-none-any.whl", hash = "sha256:1f02e8b43a8fbbc3f3e0d4f0f4bfc8131bcb4eebe8849b8e5c773f3a1c582a53", size = 13643 },
12
+ ]
13
+
14
+ [[package]]
15
+ name = "anyio"
16
+ version = "4.8.0"
17
+ source = { registry = "https://pypi.org/simple" }
18
+ dependencies = [
19
+ { name = "idna" },
20
+ { name = "sniffio" },
21
+ ]
22
+ sdist = { url = "https://files.pythonhosted.org/packages/a3/73/199a98fc2dae33535d6b8e8e6ec01f8c1d76c9adb096c6b7d64823038cde/anyio-4.8.0.tar.gz", hash = "sha256:1d9fe889df5212298c0c0723fa20479d1b94883a2df44bd3897aa91083316f7a", size = 181126 }
23
+ wheels = [
24
+ { url = "https://files.pythonhosted.org/packages/46/eb/e7f063ad1fec6b3178a3cd82d1a3c4de82cccf283fc42746168188e1cdd5/anyio-4.8.0-py3-none-any.whl", hash = "sha256:b5011f270ab5eb0abf13385f851315585cc37ef330dd88e27ec3d34d651fd47a", size = 96041 },
25
+ ]
26
+
27
+ [[package]]
28
+ name = "certifi"
29
+ version = "2025.1.31"
30
+ source = { registry = "https://pypi.org/simple" }
31
+ sdist = { url = "https://files.pythonhosted.org/packages/1c/ab/c9f1e32b7b1bf505bf26f0ef697775960db7932abeb7b516de930ba2705f/certifi-2025.1.31.tar.gz", hash = "sha256:3d5da6925056f6f18f119200434a4780a94263f10d1c21d032a6f6b2baa20651", size = 167577 }
32
+ wheels = [
33
+ { url = "https://files.pythonhosted.org/packages/38/fc/bce832fd4fd99766c04d1ee0eead6b0ec6486fb100ae5e74c1d91292b982/certifi-2025.1.31-py3-none-any.whl", hash = "sha256:ca78db4565a652026a4db2bcdf68f2fb589ea80d0be70e03929ed730746b84fe", size = 166393 },
34
+ ]
35
+
36
+ [[package]]
37
+ name = "click"
38
+ version = "8.1.8"
39
+ source = { registry = "https://pypi.org/simple" }
40
+ dependencies = [
41
+ { name = "colorama", marker = "sys_platform == 'win32'" },
42
+ ]
43
+ sdist = { url = "https://files.pythonhosted.org/packages/b9/2e/0090cbf739cee7d23781ad4b89a9894a41538e4fcf4c31dcdd705b78eb8b/click-8.1.8.tar.gz", hash = "sha256:ed53c9d8990d83c2a27deae68e4ee337473f6330c040a31d4225c9574d16096a", size = 226593 }
44
+ wheels = [
45
+ { url = "https://files.pythonhosted.org/packages/7e/d4/7ebdbd03970677812aac39c869717059dbb71a4cfc033ca6e5221787892c/click-8.1.8-py3-none-any.whl", hash = "sha256:63c132bbbed01578a06712a2d1f497bb62d9c1c0d329b7903a866228027263b2", size = 98188 },
46
+ ]
47
+
48
+ [[package]]
49
+ name = "colorama"
50
+ version = "0.4.6"
51
+ source = { registry = "https://pypi.org/simple" }
52
+ sdist = { url = "https://files.pythonhosted.org/packages/d8/53/6f443c9a4a8358a93a6792e2acffb9d9d5cb0a5cfd8802644b7b1c9a02e4/colorama-0.4.6.tar.gz", hash = "sha256:08695f5cb7ed6e0531a20572697297273c47b8cae5a63ffc6d6ed5c201be6e44", size = 27697 }
53
+ wheels = [
54
+ { url = "https://files.pythonhosted.org/packages/d1/d6/3965ed04c63042e047cb6a3e6ed1a63a35087b6a609aa3a15ed8ac56c221/colorama-0.4.6-py2.py3-none-any.whl", hash = "sha256:4f1d9991f5acc0ca119f9d443620b77f9d6b33703e51011c16baf57afb285fc6", size = 25335 },
55
+ ]
56
+
57
+ [[package]]
58
+ name = "h11"
59
+ version = "0.14.0"
60
+ source = { registry = "https://pypi.org/simple" }
61
+ sdist = { url = "https://files.pythonhosted.org/packages/f5/38/3af3d3633a34a3316095b39c8e8fb4853a28a536e55d347bd8d8e9a14b03/h11-0.14.0.tar.gz", hash = "sha256:8f19fbbe99e72420ff35c00b27a34cb9937e902a8b810e2c88300c6f0a3b699d", size = 100418 }
62
+ wheels = [
63
+ { url = "https://files.pythonhosted.org/packages/95/04/ff642e65ad6b90db43e668d70ffb6736436c7ce41fcc549f4e9472234127/h11-0.14.0-py3-none-any.whl", hash = "sha256:e3fe4ac4b851c468cc8363d500db52c2ead036020723024a109d37346efaa761", size = 58259 },
64
+ ]
65
+
66
+ [[package]]
67
+ name = "httpcore"
68
+ version = "1.0.7"
69
+ source = { registry = "https://pypi.org/simple" }
70
+ dependencies = [
71
+ { name = "certifi" },
72
+ { name = "h11" },
73
+ ]
74
+ sdist = { url = "https://files.pythonhosted.org/packages/6a/41/d7d0a89eb493922c37d343b607bc1b5da7f5be7e383740b4753ad8943e90/httpcore-1.0.7.tar.gz", hash = "sha256:8551cb62a169ec7162ac7be8d4817d561f60e08eaa485234898414bb5a8a0b4c", size = 85196 }
75
+ wheels = [
76
+ { url = "https://files.pythonhosted.org/packages/87/f5/72347bc88306acb359581ac4d52f23c0ef445b57157adedb9aee0cd689d2/httpcore-1.0.7-py3-none-any.whl", hash = "sha256:a3fff8f43dc260d5bd363d9f9cf1830fa3a458b332856f34282de498ed420edd", size = 78551 },
77
+ ]
78
+
79
+ [[package]]
80
+ name = "httpx"
81
+ version = "0.28.1"
82
+ source = { registry = "https://pypi.org/simple" }
83
+ dependencies = [
84
+ { name = "anyio" },
85
+ { name = "certifi" },
86
+ { name = "httpcore" },
87
+ { name = "idna" },
88
+ ]
89
+ sdist = { url = "https://files.pythonhosted.org/packages/b1/df/48c586a5fe32a0f01324ee087459e112ebb7224f646c0b5023f5e79e9956/httpx-0.28.1.tar.gz", hash = "sha256:75e98c5f16b0f35b567856f597f06ff2270a374470a5c2392242528e3e3e42fc", size = 141406 }
90
+ wheels = [
91
+ { url = "https://files.pythonhosted.org/packages/2a/39/e50c7c3a983047577ee07d2a9e53faf5a69493943ec3f6a384bdc792deb2/httpx-0.28.1-py3-none-any.whl", hash = "sha256:d909fcccc110f8c7faf814ca82a9a4d816bc5a6dbfea25d6591d6985b8ba59ad", size = 73517 },
92
+ ]
93
+
94
+ [[package]]
95
+ name = "httpx-sse"
96
+ version = "0.4.0"
97
+ source = { registry = "https://pypi.org/simple" }
98
+ sdist = { url = "https://files.pythonhosted.org/packages/4c/60/8f4281fa9bbf3c8034fd54c0e7412e66edbab6bc74c4996bd616f8d0406e/httpx-sse-0.4.0.tar.gz", hash = "sha256:1e81a3a3070ce322add1d3529ed42eb5f70817f45ed6ec915ab753f961139721", size = 12624 }
99
+ wheels = [
100
+ { url = "https://files.pythonhosted.org/packages/e1/9b/a181f281f65d776426002f330c31849b86b31fc9d848db62e16f03ff739f/httpx_sse-0.4.0-py3-none-any.whl", hash = "sha256:f329af6eae57eaa2bdfd962b42524764af68075ea87370a2de920af5341e318f", size = 7819 },
101
+ ]
102
+
103
+ [[package]]
104
+ name = "idna"
105
+ version = "3.10"
106
+ source = { registry = "https://pypi.org/simple" }
107
+ sdist = { url = "https://files.pythonhosted.org/packages/f1/70/7703c29685631f5a7590aa73f1f1d3fa9a380e654b86af429e0934a32f7d/idna-3.10.tar.gz", hash = "sha256:12f65c9b470abda6dc35cf8e63cc574b1c52b11df2c86030af0ac09b01b13ea9", size = 190490 }
108
+ wheels = [
109
+ { url = "https://files.pythonhosted.org/packages/76/c6/c88e154df9c4e1a2a66ccf0005a88dfb2650c1dffb6f5ce603dfbd452ce3/idna-3.10-py3-none-any.whl", hash = "sha256:946d195a0d259cbba61165e88e65941f16e9b36ea6ddb97f00452bae8b1287d3", size = 70442 },
110
+ ]
111
+
112
+ [[package]]
113
+ name = "markdown-it-py"
114
+ version = "3.0.0"
115
+ source = { registry = "https://pypi.org/simple" }
116
+ dependencies = [
117
+ { name = "mdurl" },
118
+ ]
119
+ sdist = { url = "https://files.pythonhosted.org/packages/38/71/3b932df36c1a044d397a1f92d1cf91ee0a503d91e470cbd670aa66b07ed0/markdown-it-py-3.0.0.tar.gz", hash = "sha256:e3f60a94fa066dc52ec76661e37c851cb232d92f9886b15cb560aaada2df8feb", size = 74596 }
120
+ wheels = [
121
+ { url = "https://files.pythonhosted.org/packages/42/d7/1ec15b46af6af88f19b8e5ffea08fa375d433c998b8a7639e76935c14f1f/markdown_it_py-3.0.0-py3-none-any.whl", hash = "sha256:355216845c60bd96232cd8d8c40e8f9765cc86f46880e43a8fd22dc1a1a8cab1", size = 87528 },
122
+ ]
123
+
124
+ [[package]]
125
+ name = "mcp"
126
+ version = "1.3.0"
127
+ source = { registry = "https://pypi.org/simple" }
128
+ dependencies = [
129
+ { name = "anyio" },
130
+ { name = "httpx" },
131
+ { name = "httpx-sse" },
132
+ { name = "pydantic" },
133
+ { name = "pydantic-settings" },
134
+ { name = "sse-starlette" },
135
+ { name = "starlette" },
136
+ { name = "uvicorn" },
137
+ ]
138
+ sdist = { url = "https://files.pythonhosted.org/packages/6b/b6/81e5f2490290351fc97bf46c24ff935128cb7d34d68e3987b522f26f7ada/mcp-1.3.0.tar.gz", hash = "sha256:f409ae4482ce9d53e7ac03f3f7808bcab735bdfc0fba937453782efb43882d45", size = 150235 }
139
+ wheels = [
140
+ { url = "https://files.pythonhosted.org/packages/d0/d2/a9e87b506b2094f5aa9becc1af5178842701b27217fa43877353da2577e3/mcp-1.3.0-py3-none-any.whl", hash = "sha256:2829d67ce339a249f803f22eba5e90385eafcac45c94b00cab6cef7e8f217211", size = 70672 },
141
+ ]
142
+
143
+ [package.optional-dependencies]
144
+ cli = [
145
+ { name = "python-dotenv" },
146
+ { name = "typer" },
147
+ ]
148
+
149
+ [[package]]
150
+ name = "mcp-memgraph"
151
+ version = "0.1.0"
152
+ source = { virtual = "." }
153
+ dependencies = [
154
+ { name = "httpx" },
155
+ { name = "mcp", extra = ["cli"] },
156
+ { name = "neo4j" },
157
+ ]
158
+
159
+ [package.metadata]
160
+ requires-dist = [
161
+ { name = "httpx", specifier = ">=0.28.1" },
162
+ { name = "mcp", extras = ["cli"], specifier = ">=1.3.0" },
163
+ { name = "neo4j", specifier = ">=5.28.1" },
164
+ ]
165
+
166
+ [[package]]
167
+ name = "mdurl"
168
+ version = "0.1.2"
169
+ source = { registry = "https://pypi.org/simple" }
170
+ sdist = { url = "https://files.pythonhosted.org/packages/d6/54/cfe61301667036ec958cb99bd3efefba235e65cdeb9c84d24a8293ba1d90/mdurl-0.1.2.tar.gz", hash = "sha256:bb413d29f5eea38f31dd4754dd7377d4465116fb207585f97bf925588687c1ba", size = 8729 }
171
+ wheels = [
172
+ { url = "https://files.pythonhosted.org/packages/b3/38/89ba8ad64ae25be8de66a6d463314cf1eb366222074cfda9ee839c56a4b4/mdurl-0.1.2-py3-none-any.whl", hash = "sha256:84008a41e51615a49fc9966191ff91509e3c40b939176e643fd50a5c2196b8f8", size = 9979 },
173
+ ]
174
+
175
+ [[package]]
176
+ name = "neo4j"
177
+ version = "5.28.1"
178
+ source = { registry = "https://pypi.org/simple" }
179
+ dependencies = [
180
+ { name = "pytz" },
181
+ ]
182
+ sdist = { url = "https://files.pythonhosted.org/packages/4b/20/733dac16f7cedc80b23093415822c9763302519cba0e7c8bcdb5c01fc512/neo4j-5.28.1.tar.gz", hash = "sha256:ae8e37a1d895099062c75bc359b2cce62099baac7be768d0eba7180c1298e214", size = 231094 }
183
+ wheels = [
184
+ { url = "https://files.pythonhosted.org/packages/6a/57/94225fe5e9dabdc0ff60c88cbfcedf11277f4b34e7ab1373d3e62dbdd207/neo4j-5.28.1-py3-none-any.whl", hash = "sha256:6755ef9e5f4e14b403aef1138fb6315b120631a0075c138b5ddb2a06b87b09fd", size = 312258 },
185
+ ]
186
+
187
+ [[package]]
188
+ name = "pydantic"
189
+ version = "2.10.6"
190
+ source = { registry = "https://pypi.org/simple" }
191
+ dependencies = [
192
+ { name = "annotated-types" },
193
+ { name = "pydantic-core" },
194
+ { name = "typing-extensions" },
195
+ ]
196
+ sdist = { url = "https://files.pythonhosted.org/packages/b7/ae/d5220c5c52b158b1de7ca89fc5edb72f304a70a4c540c84c8844bf4008de/pydantic-2.10.6.tar.gz", hash = "sha256:ca5daa827cce33de7a42be142548b0096bf05a7e7b365aebfa5f8eeec7128236", size = 761681 }
197
+ wheels = [
198
+ { url = "https://files.pythonhosted.org/packages/f4/3c/8cc1cc84deffa6e25d2d0c688ebb80635dfdbf1dbea3e30c541c8cf4d860/pydantic-2.10.6-py3-none-any.whl", hash = "sha256:427d664bf0b8a2b34ff5dd0f5a18df00591adcee7198fbd71981054cef37b584", size = 431696 },
199
+ ]
200
+
201
+ [[package]]
202
+ name = "pydantic-core"
203
+ version = "2.27.2"
204
+ source = { registry = "https://pypi.org/simple" }
205
+ dependencies = [
206
+ { name = "typing-extensions" },
207
+ ]
208
+ sdist = { url = "https://files.pythonhosted.org/packages/fc/01/f3e5ac5e7c25833db5eb555f7b7ab24cd6f8c322d3a3ad2d67a952dc0abc/pydantic_core-2.27.2.tar.gz", hash = "sha256:eb026e5a4c1fee05726072337ff51d1efb6f59090b7da90d30ea58625b1ffb39", size = 413443 }
209
+ wheels = [
210
+ { url = "https://files.pythonhosted.org/packages/41/b1/9bc383f48f8002f99104e3acff6cba1231b29ef76cfa45d1506a5cad1f84/pydantic_core-2.27.2-cp313-cp313-macosx_10_12_x86_64.whl", hash = "sha256:7d14bd329640e63852364c306f4d23eb744e0f8193148d4044dd3dacdaacbd8b", size = 1892709 },
211
+ { url = "https://files.pythonhosted.org/packages/10/6c/e62b8657b834f3eb2961b49ec8e301eb99946245e70bf42c8817350cbefc/pydantic_core-2.27.2-cp313-cp313-macosx_11_0_arm64.whl", hash = "sha256:82f91663004eb8ed30ff478d77c4d1179b3563df6cdb15c0817cd1cdaf34d154", size = 1811273 },
212
+ { url = "https://files.pythonhosted.org/packages/ba/15/52cfe49c8c986e081b863b102d6b859d9defc63446b642ccbbb3742bf371/pydantic_core-2.27.2-cp313-cp313-manylinux_2_17_aarch64.manylinux2014_aarch64.whl", hash = "sha256:71b24c7d61131bb83df10cc7e687433609963a944ccf45190cfc21e0887b08c9", size = 1823027 },
213
+ { url = "https://files.pythonhosted.org/packages/b1/1c/b6f402cfc18ec0024120602bdbcebc7bdd5b856528c013bd4d13865ca473/pydantic_core-2.27.2-cp313-cp313-manylinux_2_17_armv7l.manylinux2014_armv7l.whl", hash = "sha256:fa8e459d4954f608fa26116118bb67f56b93b209c39b008277ace29937453dc9", size = 1868888 },
214
+ { url = "https://files.pythonhosted.org/packages/bd/7b/8cb75b66ac37bc2975a3b7de99f3c6f355fcc4d89820b61dffa8f1e81677/pydantic_core-2.27.2-cp313-cp313-manylinux_2_17_ppc64le.manylinux2014_ppc64le.whl", hash = "sha256:ce8918cbebc8da707ba805b7fd0b382816858728ae7fe19a942080c24e5b7cd1", size = 2037738 },
215
+ { url = "https://files.pythonhosted.org/packages/c8/f1/786d8fe78970a06f61df22cba58e365ce304bf9b9f46cc71c8c424e0c334/pydantic_core-2.27.2-cp313-cp313-manylinux_2_17_s390x.manylinux2014_s390x.whl", hash = "sha256:eda3f5c2a021bbc5d976107bb302e0131351c2ba54343f8a496dc8783d3d3a6a", size = 2685138 },
216
+ { url = "https://files.pythonhosted.org/packages/a6/74/d12b2cd841d8724dc8ffb13fc5cef86566a53ed358103150209ecd5d1999/pydantic_core-2.27.2-cp313-cp313-manylinux_2_17_x86_64.manylinux2014_x86_64.whl", hash = "sha256:bd8086fa684c4775c27f03f062cbb9eaa6e17f064307e86b21b9e0abc9c0f02e", size = 1997025 },
217
+ { url = "https://files.pythonhosted.org/packages/a0/6e/940bcd631bc4d9a06c9539b51f070b66e8f370ed0933f392db6ff350d873/pydantic_core-2.27.2-cp313-cp313-manylinux_2_5_i686.manylinux1_i686.whl", hash = "sha256:8d9b3388db186ba0c099a6d20f0604a44eabdeef1777ddd94786cdae158729e4", size = 2004633 },
218
+ { url = "https://files.pythonhosted.org/packages/50/cc/a46b34f1708d82498c227d5d80ce615b2dd502ddcfd8376fc14a36655af1/pydantic_core-2.27.2-cp313-cp313-musllinux_1_1_aarch64.whl", hash = "sha256:7a66efda2387de898c8f38c0cf7f14fca0b51a8ef0b24bfea5849f1b3c95af27", size = 1999404 },
219
+ { url = "https://files.pythonhosted.org/packages/ca/2d/c365cfa930ed23bc58c41463bae347d1005537dc8db79e998af8ba28d35e/pydantic_core-2.27.2-cp313-cp313-musllinux_1_1_armv7l.whl", hash = "sha256:18a101c168e4e092ab40dbc2503bdc0f62010e95d292b27827871dc85450d7ee", size = 2130130 },
220
+ { url = "https://files.pythonhosted.org/packages/f4/d7/eb64d015c350b7cdb371145b54d96c919d4db516817f31cd1c650cae3b21/pydantic_core-2.27.2-cp313-cp313-musllinux_1_1_x86_64.whl", hash = "sha256:ba5dd002f88b78a4215ed2f8ddbdf85e8513382820ba15ad5ad8955ce0ca19a1", size = 2157946 },
221
+ { url = "https://files.pythonhosted.org/packages/a4/99/bddde3ddde76c03b65dfd5a66ab436c4e58ffc42927d4ff1198ffbf96f5f/pydantic_core-2.27.2-cp313-cp313-win32.whl", hash = "sha256:1ebaf1d0481914d004a573394f4be3a7616334be70261007e47c2a6fe7e50130", size = 1834387 },
222
+ { url = "https://files.pythonhosted.org/packages/71/47/82b5e846e01b26ac6f1893d3c5f9f3a2eb6ba79be26eef0b759b4fe72946/pydantic_core-2.27.2-cp313-cp313-win_amd64.whl", hash = "sha256:953101387ecf2f5652883208769a79e48db18c6df442568a0b5ccd8c2723abee", size = 1990453 },
223
+ { url = "https://files.pythonhosted.org/packages/51/b2/b2b50d5ecf21acf870190ae5d093602d95f66c9c31f9d5de6062eb329ad1/pydantic_core-2.27.2-cp313-cp313-win_arm64.whl", hash = "sha256:ac4dbfd1691affb8f48c2c13241a2e3b60ff23247cbcf981759c768b6633cf8b", size = 1885186 },
224
+ ]
225
+
226
+ [[package]]
227
+ name = "pydantic-settings"
228
+ version = "2.8.1"
229
+ source = { registry = "https://pypi.org/simple" }
230
+ dependencies = [
231
+ { name = "pydantic" },
232
+ { name = "python-dotenv" },
233
+ ]
234
+ sdist = { url = "https://files.pythonhosted.org/packages/88/82/c79424d7d8c29b994fb01d277da57b0a9b09cc03c3ff875f9bd8a86b2145/pydantic_settings-2.8.1.tar.gz", hash = "sha256:d5c663dfbe9db9d5e1c646b2e161da12f0d734d422ee56f567d0ea2cee4e8585", size = 83550 }
235
+ wheels = [
236
+ { url = "https://files.pythonhosted.org/packages/0b/53/a64f03044927dc47aafe029c42a5b7aabc38dfb813475e0e1bf71c4a59d0/pydantic_settings-2.8.1-py3-none-any.whl", hash = "sha256:81942d5ac3d905f7f3ee1a70df5dfb62d5569c12f51a5a647defc1c3d9ee2e9c", size = 30839 },
237
+ ]
238
+
239
+ [[package]]
240
+ name = "pygments"
241
+ version = "2.19.1"
242
+ source = { registry = "https://pypi.org/simple" }
243
+ sdist = { url = "https://files.pythonhosted.org/packages/7c/2d/c3338d48ea6cc0feb8446d8e6937e1408088a72a39937982cc6111d17f84/pygments-2.19.1.tar.gz", hash = "sha256:61c16d2a8576dc0649d9f39e089b5f02bcd27fba10d8fb4dcc28173f7a45151f", size = 4968581 }
244
+ wheels = [
245
+ { url = "https://files.pythonhosted.org/packages/8a/0b/9fcc47d19c48b59121088dd6da2488a49d5f72dacf8262e2790a1d2c7d15/pygments-2.19.1-py3-none-any.whl", hash = "sha256:9ea1544ad55cecf4b8242fab6dd35a93bbce657034b0611ee383099054ab6d8c", size = 1225293 },
246
+ ]
247
+
248
+ [[package]]
249
+ name = "python-dotenv"
250
+ version = "1.0.1"
251
+ source = { registry = "https://pypi.org/simple" }
252
+ sdist = { url = "https://files.pythonhosted.org/packages/bc/57/e84d88dfe0aec03b7a2d4327012c1627ab5f03652216c63d49846d7a6c58/python-dotenv-1.0.1.tar.gz", hash = "sha256:e324ee90a023d808f1959c46bcbc04446a10ced277783dc6ee09987c37ec10ca", size = 39115 }
253
+ wheels = [
254
+ { url = "https://files.pythonhosted.org/packages/6a/3e/b68c118422ec867fa7ab88444e1274aa40681c606d59ac27de5a5588f082/python_dotenv-1.0.1-py3-none-any.whl", hash = "sha256:f7b63ef50f1b690dddf550d03497b66d609393b40b564ed0d674909a68ebf16a", size = 19863 },
255
+ ]
256
+
257
+ [[package]]
258
+ name = "pytz"
259
+ version = "2025.1"
260
+ source = { registry = "https://pypi.org/simple" }
261
+ sdist = { url = "https://files.pythonhosted.org/packages/5f/57/df1c9157c8d5a05117e455d66fd7cf6dbc46974f832b1058ed4856785d8a/pytz-2025.1.tar.gz", hash = "sha256:c2db42be2a2518b28e65f9207c4d05e6ff547d1efa4086469ef855e4ab70178e", size = 319617 }
262
+ wheels = [
263
+ { url = "https://files.pythonhosted.org/packages/eb/38/ac33370d784287baa1c3d538978b5e2ea064d4c1b93ffbd12826c190dd10/pytz-2025.1-py2.py3-none-any.whl", hash = "sha256:89dd22dca55b46eac6eda23b2d72721bf1bdfef212645d81513ef5d03038de57", size = 507930 },
264
+ ]
265
+
266
+ [[package]]
267
+ name = "rich"
268
+ version = "13.9.4"
269
+ source = { registry = "https://pypi.org/simple" }
270
+ dependencies = [
271
+ { name = "markdown-it-py" },
272
+ { name = "pygments" },
273
+ ]
274
+ sdist = { url = "https://files.pythonhosted.org/packages/ab/3a/0316b28d0761c6734d6bc14e770d85506c986c85ffb239e688eeaab2c2bc/rich-13.9.4.tar.gz", hash = "sha256:439594978a49a09530cff7ebc4b5c7103ef57baf48d5ea3184f21d9a2befa098", size = 223149 }
275
+ wheels = [
276
+ { url = "https://files.pythonhosted.org/packages/19/71/39c7c0d87f8d4e6c020a393182060eaefeeae6c01dab6a84ec346f2567df/rich-13.9.4-py3-none-any.whl", hash = "sha256:6049d5e6ec054bf2779ab3358186963bac2ea89175919d699e378b99738c2a90", size = 242424 },
277
+ ]
278
+
279
+ [[package]]
280
+ name = "shellingham"
281
+ version = "1.5.4"
282
+ source = { registry = "https://pypi.org/simple" }
283
+ sdist = { url = "https://files.pythonhosted.org/packages/58/15/8b3609fd3830ef7b27b655beb4b4e9c62313a4e8da8c676e142cc210d58e/shellingham-1.5.4.tar.gz", hash = "sha256:8dbca0739d487e5bd35ab3ca4b36e11c4078f3a234bfce294b0a0291363404de", size = 10310 }
284
+ wheels = [
285
+ { url = "https://files.pythonhosted.org/packages/e0/f9/0595336914c5619e5f28a1fb793285925a8cd4b432c9da0a987836c7f822/shellingham-1.5.4-py2.py3-none-any.whl", hash = "sha256:7ecfff8f2fd72616f7481040475a65b2bf8af90a56c89140852d1120324e8686", size = 9755 },
286
+ ]
287
+
288
+ [[package]]
289
+ name = "sniffio"
290
+ version = "1.3.1"
291
+ source = { registry = "https://pypi.org/simple" }
292
+ sdist = { url = "https://files.pythonhosted.org/packages/a2/87/a6771e1546d97e7e041b6ae58d80074f81b7d5121207425c964ddf5cfdbd/sniffio-1.3.1.tar.gz", hash = "sha256:f4324edc670a0f49750a81b895f35c3adb843cca46f0530f79fc1babb23789dc", size = 20372 }
293
+ wheels = [
294
+ { url = "https://files.pythonhosted.org/packages/e9/44/75a9c9421471a6c4805dbf2356f7c181a29c1879239abab1ea2cc8f38b40/sniffio-1.3.1-py3-none-any.whl", hash = "sha256:2f6da418d1f1e0fddd844478f41680e794e6051915791a034ff65e5f100525a2", size = 10235 },
295
+ ]
296
+
297
+ [[package]]
298
+ name = "sse-starlette"
299
+ version = "2.2.1"
300
+ source = { registry = "https://pypi.org/simple" }
301
+ dependencies = [
302
+ { name = "anyio" },
303
+ { name = "starlette" },
304
+ ]
305
+ sdist = { url = "https://files.pythonhosted.org/packages/71/a4/80d2a11af59fe75b48230846989e93979c892d3a20016b42bb44edb9e398/sse_starlette-2.2.1.tar.gz", hash = "sha256:54470d5f19274aeed6b2d473430b08b4b379ea851d953b11d7f1c4a2c118b419", size = 17376 }
306
+ wheels = [
307
+ { url = "https://files.pythonhosted.org/packages/d9/e0/5b8bd393f27f4a62461c5cf2479c75a2cc2ffa330976f9f00f5f6e4f50eb/sse_starlette-2.2.1-py3-none-any.whl", hash = "sha256:6410a3d3ba0c89e7675d4c273a301d64649c03a5ef1ca101f10b47f895fd0e99", size = 10120 },
308
+ ]
309
+
310
+ [[package]]
311
+ name = "starlette"
312
+ version = "0.46.1"
313
+ source = { registry = "https://pypi.org/simple" }
314
+ dependencies = [
315
+ { name = "anyio" },
316
+ ]
317
+ sdist = { url = "https://files.pythonhosted.org/packages/04/1b/52b27f2e13ceedc79a908e29eac426a63465a1a01248e5f24aa36a62aeb3/starlette-0.46.1.tar.gz", hash = "sha256:3c88d58ee4bd1bb807c0d1acb381838afc7752f9ddaec81bbe4383611d833230", size = 2580102 }
318
+ wheels = [
319
+ { url = "https://files.pythonhosted.org/packages/a0/4b/528ccf7a982216885a1ff4908e886b8fb5f19862d1962f56a3fce2435a70/starlette-0.46.1-py3-none-any.whl", hash = "sha256:77c74ed9d2720138b25875133f3a2dae6d854af2ec37dceb56aef370c1d8a227", size = 71995 },
320
+ ]
321
+
322
+ [[package]]
323
+ name = "typer"
324
+ version = "0.15.2"
325
+ source = { registry = "https://pypi.org/simple" }
326
+ dependencies = [
327
+ { name = "click" },
328
+ { name = "rich" },
329
+ { name = "shellingham" },
330
+ { name = "typing-extensions" },
331
+ ]
332
+ sdist = { url = "https://files.pythonhosted.org/packages/8b/6f/3991f0f1c7fcb2df31aef28e0594d8d54b05393a0e4e34c65e475c2a5d41/typer-0.15.2.tar.gz", hash = "sha256:ab2fab47533a813c49fe1f16b1a370fd5819099c00b119e0633df65f22144ba5", size = 100711 }
333
+ wheels = [
334
+ { url = "https://files.pythonhosted.org/packages/7f/fc/5b29fea8cee020515ca82cc68e3b8e1e34bb19a3535ad854cac9257b414c/typer-0.15.2-py3-none-any.whl", hash = "sha256:46a499c6107d645a9c13f7ee46c5d5096cae6f5fc57dd11eccbbb9ae3e44ddfc", size = 45061 },
335
+ ]
336
+
337
+ [[package]]
338
+ name = "typing-extensions"
339
+ version = "4.12.2"
340
+ source = { registry = "https://pypi.org/simple" }
341
+ sdist = { url = "https://files.pythonhosted.org/packages/df/db/f35a00659bc03fec321ba8bce9420de607a1d37f8342eee1863174c69557/typing_extensions-4.12.2.tar.gz", hash = "sha256:1a7ead55c7e559dd4dee8856e3a88b41225abfe1ce8df57b7c13915fe121ffb8", size = 85321 }
342
+ wheels = [
343
+ { url = "https://files.pythonhosted.org/packages/26/9f/ad63fc0248c5379346306f8668cda6e2e2e9c95e01216d2b8ffd9ff037d0/typing_extensions-4.12.2-py3-none-any.whl", hash = "sha256:04e5ca0351e0f3f85c6853954072df659d0d13fac324d0072316b67d7794700d", size = 37438 },
344
+ ]
345
+
346
+ [[package]]
347
+ name = "uvicorn"
348
+ version = "0.34.0"
349
+ source = { registry = "https://pypi.org/simple" }
350
+ dependencies = [
351
+ { name = "click" },
352
+ { name = "h11" },
353
+ ]
354
+ sdist = { url = "https://files.pythonhosted.org/packages/4b/4d/938bd85e5bf2edeec766267a5015ad969730bb91e31b44021dfe8b22df6c/uvicorn-0.34.0.tar.gz", hash = "sha256:404051050cd7e905de2c9a7e61790943440b3416f49cb409f965d9dcd0fa73e9", size = 76568 }
355
+ wheels = [
356
+ { url = "https://files.pythonhosted.org/packages/61/14/33a3a1352cfa71812a3a21e8c9bfb83f60b0011f5e36f2b1399d51928209/uvicorn-0.34.0-py3-none-any.whl", hash = "sha256:023dc038422502fa28a09c7a30bf2b6991512da7dcdb8fd35fe57cfc154126f4", size = 62315 },
357
+ ]