Skip to main content

A graph-based memory system for LLMs with intelligent retrieval using knowledge graphs, hybrid search, and semantic embeddings

Project description

MemoGraph ๐Ÿง 

PyPI version Python Version License MCP Registry MCP Code style: ruff pre-commit Type checked: mypy Tests Code Quality

A graph-based memory system for LLMs with intelligent retrieval. MemoGraph provides a powerful solution to the LLM memory problem by combining knowledge graphs, hybrid retrieval, and semantic search.

๐Ÿ“Š Project Status: MemoGraph is production-ready! See docs/PROJECT_STATUS.md for current status and docs/FUTURE_ENHANCEMENTS.md for optional improvements.

โœจ Features

  • ๐Ÿค– Smart Auto-Organization Engine: Automatically extract structured information from memories using LLMs
    • Topics, subtopics, and recurring themes
    • People with roles and organizations
    • Action items with assignees and deadlines
    • Decisions, questions, and sentiment analysis
    • Risks, ideas, and timeline events
  • ๐Ÿท๏ธ AI-Powered Tag Suggestions: Automatically suggest relevant tags using semantic analysis and content structure detection
  • ๐Ÿ”— AI-Powered Link Suggestions: Intelligently recommend wikilinks to related notes using semantic similarity and graph analysis
  • Graph-Based Memory: Navigate knowledge using bidirectional wikilinks and backlinks
  • Hybrid Retrieval: Combines keyword matching, graph traversal, and optional vector embeddings
  • Markdown-Native: Human-readable markdown files with YAML frontmatter
  • Memory Types: Support for episodic, semantic, procedural, and fact-based memories
  • Smart Indexing: Efficient caching system that only re-indexes changed files
  • CLI & Python API: Use via command line or integrate into your Python applications
  • Multiple LLM Providers: Works with Ollama, Claude, and OpenAI
  • Context Compression: Intelligent token budgeting for optimal context windows
  • Salience Scoring: Memory importance ranking for better retrieval

๐Ÿš€ Quick Start

Installation

pip install memograph

Install with optional dependencies:

# For OpenAI support
pip install memograph[openai]

# For Anthropic Claude support
pip install memograph[anthropic]

# For Ollama support
pip install memograph[ollama]

# For embedding support
pip install memograph[embeddings]

# Install everything
pip install memograph[all]

Python Usage

from memograph import MemoryKernel, MemoryType

# Initialize the kernel attached to your vault path
kernel = MemoryKernel("~/my-vault")

# Ingest all notes in the vault
stats = kernel.ingest()
print(f"Indexed {stats['indexed']} memories.")

# Programmatically add a new memory
kernel.remember(
    title="Meeting Note",
    content="Decided to use BFS graph traversal for retrieval.",
    memory_type=MemoryType.EPISODIC,
    tags=["design", "retrieval"]
)

# Retrieve context for an LLM query
context = kernel.context_window(
    query="how does retrieval work?",
    tags=["retrieval"],
    depth=2,
    top_k=8
)

print(context)

๐Ÿ”Œ MCP Server (Model Context Protocol)

MemoGraph includes a full-featured MCP server for seamless integration with AI assistants like Cline and Claude Desktop.

๐Ÿ“– New to MemoGraph MCP? See the MCP User Guide for practical usage instructions and examples!

๐Ÿšจ Having connection issues? See Setup & Troubleshooting Guide - Common fixes for "cannot connect" errors!

19 Available Tools

Category Tools Description
Search search_vault, query_with_context Semantic search and context retrieval
Create create_memory, import_document Add memories and import documents
Read list_memories, get_memory, get_vault_info Browse and retrieve memories
Update update_memory Modify existing memories
Delete delete_memory Remove memories by ID
Analytics get_vault_stats Vault statistics and insights
Discovery list_available_tools List all available tools
Autonomous auto_hook_query, auto_hook_response, configure_autonomous_mode, get_autonomous_config Autonomous memory management
Graph relate_memories, search_by_graph, find_path Graph-native linking and traversal
Bulk bulk_create Create multiple memories in one call

Quick Setup for Cline

Add to your ~/.cline/mcp_settings.json:

{
  "mcp": {
    "servers": {
      "memograph": {
        "command": "python",
        "args": ["-m", "memograph.mcp.run_server"],
        "env": {
          "MEMOGRAPH_VAULT": "/path/to/your/vault"
        }
      }
    }
  }
}

Quick Setup for Claude Desktop

Add to your claude_desktop_config.json:

{
  "mcpServers": {
    "memograph": {
      "command": "python",
      "args": ["-m", "memograph.mcp.run_server", "--vault", "/path/to/your/vault"]
    }
  }
}

Install from MCP Registry

NEW: MemoGraph is now available in the official MCP Registry! ๐ŸŽ‰

Registry URL: https://github.com/modelcontextprotocol/servers/tree/main/src/memograph

Step 1: Install MemoGraph

First, install the Python package:

pip install memograph

Step 2: Configure in Your MCP Client

The MCP Registry provides the configuration template. Add to your client's config file:

For Cline (~/.cline/mcp_settings.json):

{
  "mcp": {
    "servers": {
      "memograph": {
        "command": "python",
        "args": ["-m", "memograph.mcp.run_server"],
        "env": {
          "MEMOGRAPH_VAULT": "/path/to/your/vault"
        }
      }
    }
  }
}

For Claude Desktop (claude_desktop_config.json):

{
  "mcpServers": {
    "memograph": {
      "command": "python",
      "args": ["-m", "memograph.mcp.run_server"],
      "env": {
        "MEMOGRAPH_VAULT": "/path/to/your/vault"
      }
    }
  }
}

Benefits of MCP Registry Listing:

  • โœ… Official registry backed by Anthropic, GitHub, and Microsoft
  • โœ… Discoverable by all MCP-compatible clients
  • โœ… Verified server card and metadata
  • โœ… Direct link from PyPI package
  • โœ… Trusted by the MCP community

Note: The registry uses the PyPI package version. When you pip install memograph, you automatically get the latest registry-listed version.

See MCP_REGISTRY_GUIDE.md for complete submission and configuration guide.

Usage Examples

Once configured, use natural language with your AI assistant:

"Search my vault for memories about Python"
"Create a memory titled 'Project Ideas' with content '...'"
"Update memory abc-123 to have salience 0.9"
"Delete memory xyz-456"
"What tools are available?"
"Get vault statistics"

See CONFIG_REFERENCE.md for complete MCP configuration guide.

Using Auto-Save Hooks

MemoGraph provides autonomous hooks to save conversations automatically:

  • โš ๏ธ Important: Hooks are passive tools - see Autonomous Hooks Guide for setup
  • ๐Ÿ“ Quick fix: Add custom instructions to Claude Desktop (instructions in guide)
  • ๐Ÿ”ง Configure with MEMOGRAPH_AUTONOMOUS_MODE=true

Read the full Autonomous Hooks User Guide โ†’

๐ŸŽฏ CLI Usage

MemoGraph comes with a powerful CLI for managing your vault and chatting with it.

Ingest

Index your markdown files into the graph database:

memograph --vault ~/my-vault ingest

Force re-indexing all files:

memograph --vault ~/my-vault ingest --force

Remember

Quickly add a memory from the command line:

memograph --vault ~/my-vault remember \
    --title "Team Sync" \
    --content "Discussed Q3 goals." \
    --tags planning q3

Context Window

Generate context for a query:

memograph --vault ~/my-vault context \
    --query "What did we decide about the database?" \
    --tags architecture \
    --depth 2 \
    --top-k 5

Ask (Interactive Chat)

Start an interactive chat session with your vault context:

memograph --vault ~/my-vault ask --chat --provider ollama --model llama3

Or ask a single question:

memograph --vault ~/my-vault ask \
    --query "Summarize our design decisions" \
    --provider claude \
    --model claude-3-5-sonnet-20240620

Diagnostics

Check your environment and connection to LLM providers:

memograph --vault ~/my-vault doctor

### Import Documents

Import documents (TXT, PDF, DOCX) and convert them to markdown:

```bash
# Import a single file
memograph --vault ~/my-vault import document.pdf --type episodic

# Import entire folder
memograph --vault ~/my-vault import ~/Documents --recursive

# Preview files without importing (dry run)
memograph --vault ~/my-vault import ~/Documents --dry-run

# Auto-ingest after import
memograph --vault ~/my-vault import document.pdf --auto-ingest

Batch Operations

Efficiently manage multiple memories at once:

# Bulk create memories from JSON/CSV
memograph --vault ~/my-vault batch-create memories.json

# Bulk update memories by filter
memograph --vault ~/my-vault batch-update \
    --filter-tags outdated \
    --add-tags reviewed \
    --salience 0.8

# Bulk delete with safety checks
memograph --vault ~/my-vault batch-delete \
    --filter-type episodic \
    --filter-max-salience 0.3 \
    --dry-run

Data Management

Export, backup, and restore your vault:

# Export vault to JSON/CSV/Markdown
memograph --vault ~/my-vault export --format json --output backup.json

# Create timestamped backup
memograph --vault ~/my-vault backup --output ./backups

# Restore from backup
memograph --vault ~/my-vault import-backup backup.zip

Configuration & Statistics

Manage settings and view vault analytics:

# View vault statistics
memograph --vault ~/my-vault stats

# Configure settings
memograph config set embedding_provider openai
memograph config get embedding_provider
memograph config list

# Manage profiles
memograph config profile create work --vault ~/work-vault
memograph config profile use work

MCP Setup

Interactive wizard to configure MCP server for Claude Desktop or Cline:

# Run interactive setup wizard
memograph setup-mcp

# Verify MCP configuration
memograph verify-mcp

๐Ÿ“– Complete CLI Documentation: See CLI Usage Guide for detailed documentation with 200+ examples covering all 24 commands.

๐Ÿค– AI Features

MemoGraph includes powerful AI-powered features to enhance your knowledge management workflow. See AI Features Guide for complete documentation.

๐Ÿท๏ธ AutoTagger - Intelligent Tag Suggestions

Automatically suggest relevant tags using semantic analysis, content structure, and existing patterns:

# Suggest tags for a note
memograph suggest-tags note.md

# Apply high-confidence suggestions automatically
memograph suggest-tags note.md --apply

# Adjust confidence threshold and limit
memograph suggest-tags note.md --min-confidence 0.5 --max-suggestions 10

Features: Frequency-based extraction โ€ข Semantic similarity โ€ข Structure detection โ€ข Pattern learning โ€ข Confidence scoring

๐Ÿ”— LinkSuggester - Smart Wikilink Recommendations

Intelligently recommend wikilinks to related notes using semantic similarity and graph analysis:

# Suggest links for a note
memograph suggest-links note.md

# Apply suggestions automatically
memograph suggest-links note.md --apply

# Show bidirectional link opportunities
memograph suggest-links note.md --show-bidirectional

Features: Semantic search โ€ข Keyword matching โ€ข Graph-based suggestions โ€ข Bidirectional detection โ€ข Target previews

๐Ÿ” GapDetector - Knowledge Base Analysis

Identify missing topics, weak coverage, and isolated notes in your vault:

# Detect all gaps
memograph detect-gaps

# Focus on high-severity gaps
memograph detect-gaps --min-severity 0.7

# Export results to JSON
memograph detect-gaps --output json > gaps.json

Gap Types: Missing Topics โ€ข Weak Coverage โ€ข Isolated Notes โ€ข Missing Links

๐Ÿ“Š Knowledge Analysis - Comprehensive Insights

Get comprehensive analysis of your entire knowledge base:

# Full analysis with all features
memograph analyze-knowledge

# Export detailed report to JSON
memograph analyze-knowledge --output json > analysis.json

Analysis Includes: Vault statistics โ€ข Topic clustering โ€ข Learning paths โ€ข Gap detection โ€ข Connection analysis

Python API for AI Features

from memograph import MemoryKernel
from memograph.ai import AutoTagger, LinkSuggester, GapDetector

kernel = MemoryKernel("~/my-vault")
kernel.ingest()

# Get tag suggestions
tagger = AutoTagger(kernel, min_confidence=0.4)
suggestions = await tagger.suggest_tags(
    content="Python is great for data science",
    title="Data Science with Python"
)

# Get link suggestions
suggester = LinkSuggester(kernel, min_confidence=0.5)
links = await suggester.suggest_links(
    content="Python async programming tutorial",
    title="Async Python"
)

# Detect knowledge gaps
detector = GapDetector(kernel, min_severity=0.5)
gaps = await detector.detect_gaps()

# Comprehensive analysis
analysis = await detector.analyze_knowledge_base()

๐Ÿ“– Complete Documentation:

๐Ÿ’ก Use Cases: Auto-organize notes โ€ข Discover connections โ€ข Identify gaps โ€ข Maintain consistency โ€ข Build learning paths

๐Ÿ“– Core Concepts

Memory Types

MemoGraph supports different types of memories inspired by cognitive science:

  • Episodic: Personal experiences and events (e.g., meeting notes)
  • Semantic: Facts and general knowledge (e.g., documentation)
  • Procedural: How-to knowledge and processes (e.g., tutorials)
  • Fact: Discrete factual information (e.g., configuration values)

Graph Traversal

The library uses BFS (Breadth-First Search) to traverse your knowledge graph:

# Retrieve nodes with depth=2 (2 hops from seed nodes)
nodes = kernel.retrieve_nodes(
    query="graph algorithms",
    depth=2,  # Traverse up to 2 levels deep
    top_k=10  # Return top 10 relevant memories
)

Salience Scoring

Each memory has a salience score (0.0-1.0) that represents its importance:

---
title: "Critical Architecture Decision"
salience: 0.9
memory_type: semantic
---

We decided to use PostgreSQL for better ACID guarantees...

๐Ÿ—๏ธ Project Structure

MemoGraph/
โ”œโ”€โ”€ memograph/          # Main package
โ”‚   โ”œโ”€โ”€ core/           # Core functionality
โ”‚   โ”‚   โ”œโ”€โ”€ kernel.py   # Memory kernel
โ”‚   โ”‚   โ”œโ”€โ”€ graph.py    # Graph implementation
โ”‚   โ”‚   โ”œโ”€โ”€ retriever.py # Hybrid retrieval
โ”‚   โ”‚   โ”œโ”€โ”€ indexer.py  # File indexing
โ”‚   โ”‚   โ””โ”€โ”€ parser.py   # Markdown parsing
โ”‚   โ”œโ”€โ”€ adapters/       # LLM and embedding adapters
โ”‚   โ”‚   โ”œโ”€โ”€ embeddings/ # Embedding providers
โ”‚   โ”‚   โ”œโ”€โ”€ frameworks/ # Framework integrations
โ”‚   โ”‚   โ””โ”€โ”€ llm/        # LLM providers
โ”‚   โ”œโ”€โ”€ storage/        # Storage and caching
โ”‚   โ”œโ”€โ”€ mcp/            # MCP server implementation
โ”‚   โ””โ”€โ”€ cli.py          # CLI implementation
โ”œโ”€โ”€ tests/              # Test suite
โ”œโ”€โ”€ examples/           # Example usage
โ””โ”€โ”€ scripts/            # Utility scripts

๐Ÿค Contributing

We welcome contributions! Please see our Contributing Guide for details.

Development Setup

  1. Clone the repository:

    git clone https://github.com/Indhar01/MemoGraph.git
    cd MemoGraph
    
  2. Install in development mode:

    pip install -e ".[all,dev]"
    
  3. Install pre-commit hooks:

    pre-commit install
    
  4. Run tests:

    pytest
    

Code Quality

We maintain high code quality standards:

  • Linting: Ruff for fast Python linting
  • Formatting: Ruff formatter for consistent code style
  • Type Checking: MyPy for static type analysis
  • Testing: Pytest with comprehensive test coverage
  • Pre-commit Hooks: Automated checks before each commit

๐Ÿ“š Documentation

Getting Started

For Developers & Contributors

๐Ÿ”’ Security

See our Security Policy for reporting vulnerabilities.

๐Ÿ“„ License

This project is licensed under the MIT License - see the LICENSE file for details.

๐ŸŒŸ Acknowledgments

Inspired by the need for better memory management in LLM applications. Built with:

  • Graph-based knowledge representation
  • Hybrid retrieval strategies
  • Cognitive science principles

๐Ÿ“ฌ Contact & Support

๐Ÿ“ฃ Community & Feedback

We value community feedback and contributions! Here's how to get involved:

Report Issues

Found a bug or have a feature request? Open an issue on GitHub.

Discussions

Join the conversation in GitHub Discussions:

  • Ask questions
  • Share use cases
  • Suggest improvements
  • Show what you've built

Contributing

We welcome contributions! See our Contributing Guide for details on:

  • Code contributions
  • Documentation improvements
  • Bug reports and feature requests
  • Community support

Stay Updated

  • โญ Star the repository on GitHub
  • ๐Ÿ‘๏ธ Watch for updates and releases
  • ๐Ÿ“ฆ Follow the project on PyPI
  • ๐Ÿ”— Check out the MCP Registry listing

๐Ÿšฆ Status

Current Version: 0.1.1 (Alpha - Marketplace Ready)

This project is in active development with a focus on code quality and stability:

  • โœ… Core functionality is stable and tested
  • โœ… All linter checks passing (Ruff)
  • โœ… Type checking configured (MyPy)
  • โœ… Pre-commit hooks enabled
  • โœ… Comprehensive test suite
  • โš ๏ธ API may change in minor versions until v1.0.0

Recent Improvements:

  • ๐ŸŽ‰ Published to official MCP Registry (io.github.indhar01/memograph)
  • ๐Ÿ“ฆ Version 0.1.1 Released with registry integration improvements
  • Enhanced code quality with Ruff linting and formatting
  • Added comprehensive type checking with MyPy
  • Improved project structure and organization
  • Updated MCP server with 19 tools including autonomous features and graph operations
  • Added AGENTS.md for AI assistant integration
  • Created comprehensive MCP Registry submission guide
  • Improved documentation with accurate installation instructions

Made with โค๏ธ for better LLM memory management

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

memograph-0.3.0.tar.gz (334.9 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

memograph-0.3.0-py3-none-any.whl (302.4 kB view details)

Uploaded Python 3

File details

Details for the file memograph-0.3.0.tar.gz.

File metadata

  • Download URL: memograph-0.3.0.tar.gz
  • Upload date:
  • Size: 334.9 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for memograph-0.3.0.tar.gz
Algorithm Hash digest
SHA256 9e522b61ad14c86341c22c9f78bb6b08a62b54f372560361358281cc7dda315d
MD5 97425d53c29c44cdd8a0e56395c8560d
BLAKE2b-256 88c40ef69d4b1253ca4c20dd9458292023cd889607deefa0c23f1b31dd859ccf

See more details on using hashes here.

Provenance

The following attestation bundles were made for memograph-0.3.0.tar.gz:

Publisher: publish.yml on Indhar01/MemoGraph

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file memograph-0.3.0-py3-none-any.whl.

File metadata

  • Download URL: memograph-0.3.0-py3-none-any.whl
  • Upload date:
  • Size: 302.4 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.12

File hashes

Hashes for memograph-0.3.0-py3-none-any.whl
Algorithm Hash digest
SHA256 2e8b2695245f23554afa3c46eab36a75dbd3d10dd3bd5c05a819c6f47abb6b7a
MD5 514cffe12e64c1350cb17beec81a84da
BLAKE2b-256 201127864197a87e94f99eb8e11f9e8fe5630c39962860b1379a007c6963ef1c

See more details on using hashes here.

Provenance

The following attestation bundles were made for memograph-0.3.0-py3-none-any.whl:

Publisher: publish.yml on Indhar01/MemoGraph

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page