by memgraph
Unified mono‑repo offering core Python utilities, a LangChain integration, an MCP server, and an experimental migration agent for building AI‑powered graph applications on Memgraph.
Memgraph AI Toolkit provides a set of libraries that enable developers to create, query, and manage AI‑driven graph applications on the Memgraph graph database. It bundles core Python utilities, a LangChain integration, an MCP server for lightweight communication, and an experimental agent that automates MySQL‑to‑Memgraph migrations using LLMs.
uv pip install -e <package-path>[test]
.pytest -s <package>/tests
).memgraph-toolbox
for direct Python queries and CLI utilities.langchain_memgraph
to expose Memgraph operations as LangChain tools.mcp-memgraph
and integrating it with your application.uv run main.py
inside integrations/agents
after configuring .env
with MySQL and Memgraph credentials.pytest
and environment‑based configurations.Q: Do I need a Memgraph license to use the toolkit? A: The toolkit works with any Memgraph instance, including the community edition.
Q: Which Python package manager is recommended?
A: The README uses uv
, but pip
works as well.
Q: How do I provide my OpenAI API key for LangChain tests?
A: Create a .env
file with OPENAI_API_KEY=<your‑key>
in the integrations/langchain-memgraph
directory.
Q: Is the migration agent production‑ready? A: It is marked as experimental; it is suitable for testing and proof‑of‑concept migrations.
Q: Can I run the MCP server as a separate service?
A: Yes, after installing mcp-memgraph
, you can import and start the server from your application code.
A unified mono-repo for integrating AI-powered graph tools on top of Memgraph.
This repository contains the following libraries:
memgraph-toolbox Core Python utilities and CLI tools for querying and analyzing a Memgraph database. The package is available on the PyPi
langchain-memgraph A LangChain integration package, exposing Memgraph operations as LangChain tools and toolkits. The package is available on the PyPi
mcp-memgraph An MCP (Model Context Protocol) server implementation, exposing Memgraph tools over a lightweight STDIO protocol. The package is available on the PyPi
agents ⚡ Experimental An intelligent database migration agent that automates the process of migrating from MySQL to Memgraph using LLM-powered graph modeling and analysis. Features automated schema analysis, intelligent graph modeling with interactive refinement, and data migration with validation.
For individual examples on how to use the toolbox, LangChain, MCP, or agents, refer to our documentation:
You can build and test each package directly from your repo. First, start a Memgraph MAGE instance with schema info enabled:
docker run -p 7687:7687 \
--name memgraph \
memgraph/memgraph-mage:latest \
--schema-info-enabled=true
Once Memgraph is running, install any package in “editable” mode and run its test suite. For example, to test the core toolbox:
uv pip install -e memgraph-toolbox[test]
pytest -s memgraph-toolbox/src/memgraph_toolbox/tests
To test the core toolbox, just run:
uv pip install -e memgraph-toolbox[test]
pytest -s memgraph-toolbox/src/memgraph_toolbox/tests
To run the LangChain tests, create a .env file with your OPENAI_API_KEY, as the tests depend on LLM calls:
uv pip install -e integrations/langchain-memgraph[test]
pytest -s integrations/langchain-memgraph/tests
uv pip install -e integrations/mcp-memgraph[test]
pytest -s integrations/mcp-memgraph/tests
uv pip install -e integrations/agents[test]
pytest -s integrations/agents/tests
To run a complete migration workflow with the agent:
cd integrations/agents
uv run main.py
Note: The agent requires both MySQL and Memgraph connections. Set up your environment variables in .env
based on .env.example
.
If you are running any test on MacOS in zsh, add ""
to the command:
uv pip install -e memgraph-toolbox"[test]"
Please log in to share your review and rating for this MCP.
Discover more MCP servers with similar functionality and use cases
by zed-industries
Provides real-time collaborative editing powered by Rust, enabling developers to edit code instantly across machines with a responsive, GPU-accelerated UI.
by cline
Provides autonomous coding assistance directly in the IDE, enabling file creation, editing, terminal command execution, browser interactions, and tool extension with user approval at each step.
by continuedev
Provides continuous AI assistance across IDEs, terminals, and CI pipelines, offering agents, chat, inline editing, and autocomplete to accelerate software development.
by github
Enables AI agents, assistants, and chatbots to interact with GitHub via natural‑language commands, providing read‑write access to repositories, issues, pull requests, workflows, security data and team activity.
by block
Automates engineering tasks by installing, executing, editing, and testing code using any large language model, providing end‑to‑end project building, debugging, workflow orchestration, and external API interaction.
by RooCodeInc
An autonomous coding agent that lives inside VS Code, capable of generating, refactoring, debugging code, managing files, running terminal commands, controlling a browser, and adapting its behavior through custom modes and instructions.
by lastmile-ai
A lightweight, composable framework for building AI agents using Model Context Protocol and simple workflow patterns.
by firebase
Provides a command‑line interface to manage, test, and deploy Firebase projects, covering hosting, databases, authentication, cloud functions, extensions, and CI/CD workflows.
by gptme
Empowers large language models to act as personal AI assistants directly inside the terminal, providing capabilities such as code execution, file manipulation, web browsing, vision, and interactive tool usage.