Code-Graph-RAG

A graph-based RAG system that parses multi-language codebases with Tree-sitter, builds knowledge graphs in Memgraph, and enables natural language querying, editing, and optimization.

Install

pip install code-graph-rag

With all Tree-sitter grammars (Python, JS, TS, Rust, Go, Java, Scala, C++, Lua):

pip install 'code-graph-rag[treesitter-full]'

With semantic code search (UniXcoder embeddings):

pip install 'code-graph-rag[semantic]'

Prerequisites

  • Python 3.12+
  • Docker (for Memgraph)
  • cmake (for building pymgclient)
  • ripgrep (rg) (for shell command text searching)

CLI Quick Start

The package installs a cgr command.

Start Memgraph, parse a repo, and query it:

cgr daemon up                              # start Memgraph + Qdrant
cgr start --repo-path ./my-project \
          --update-graph --clean           # parse & launch interactive chat

Index to protobuf for offline use:

cgr index -o ./index-output --repo-path ./my-project

Export knowledge graph to JSON:

cgr export -o graph.json

AI-guided optimization:

cgr optimize python --repo-path ./my-project

Run as an MCP server (for Claude Code):

cgr mcp-server

Check your setup:

cgr doctor

Python SDK

The cgr package provides short imports for programmatic use.

Load and query an exported graph

from cgr import load_graph

graph = load_graph("graph.json")
print(graph.summary())

functions = graph.find_nodes_by_label("Function")
for fn in functions[:5]:
    rels = graph.get_relationships_for_node(fn.node_id)
    print(f"{fn.properties['name']}: {len(rels)} relationships")

Query Memgraph with Cypher

from cgr import MemgraphIngestor

with MemgraphIngestor(host="localhost", port=7687) as db:
    rows = db.fetch_all("MATCH (f:Function) RETURN f.name LIMIT 10")
    for row in rows:
        print(row)

Generate Cypher from natural language

import asyncio
from cgr import CypherGenerator

async def main():
    gen = CypherGenerator()
    cypher = await gen.generate("Find all classes that inherit from BaseModel")
    print(cypher)

asyncio.run(main())

Semantic code search

Requires the semantic extra.

from cgr import embed_code

embedding = embed_code("def authenticate(user, password): ...")
print(f"Embedding dimension: {len(embedding)}")

Configuration

from cgr import settings

settings.set_orchestrator("openai", "gpt-4o", api_key="sk-...")
settings.set_cypher("google", "gemini-2.5-flash", api_key="your-key")

Environment Variables

Configure via .env or environment variables:

Variable Default Description
MEMGRAPH_HOST localhost Memgraph hostname
MEMGRAPH_PORT 7687 Memgraph port
ORCHESTRATOR_PROVIDER Provider: google, openai, ollama
ORCHESTRATOR_MODEL Model ID (e.g. gpt-4o, gemini-2.5-pro)
ORCHESTRATOR_API_KEY API key for the provider (not needed for ollama)
CYPHER_PROVIDER Provider for Cypher generation
CYPHER_MODEL Model ID for Cypher generation (e.g. codellama, gpt-4o-mini)
CYPHER_API_KEY API key for Cypher provider (not needed for ollama)
TARGET_REPO_PATH . Default repository path

Documentation

Full documentation, architecture details, and contribution guide: docs.code-graph-rag.com

License

MIT

Key dates

PyPI data
Data sourced directly from PyPI's database.
  • Released:
Newer release available (0.0.589)

1 maintainer

PyPI data
Data sourced directly from PyPI's database.
Avatar for vitali87 from gravatar.com vitali87

License expression

MIT
View SPDX License List

Requires

Python >=3.12

Provides Extra

test treesitter-full semantic

Tags

rag retrieval-augmented-generation knowledge-graph code-analysis tree-sitter mcp mcp-server llm graph-database semantic-search codebase memgraph developer-tools monorepo

Classifiers

Report project as malware

Download files

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

Source Distribution

code_graph_rag-0.0.246.tar.gz (358.3 kB view details)

Uploaded Source

Built Distribution

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

code_graph_rag-0.0.246-py3-none-any.whl (401.8 kB view details)

Uploaded Python 3

File details

Details for the file code_graph_rag-0.0.246.tar.gz.

File metadata

  • Download URL: code_graph_rag-0.0.246.tar.gz
  • Upload date:
  • Size: 358.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for code_graph_rag-0.0.246.tar.gz
Algorithm Hash digest
SHA256 b434e68675857066fd7aaa029ded5f05e95f15d35686af5cad8be77c2a6a5d8d
MD5 cf4070a60fe78a898e77321483aa3f5b
BLAKE2b-256 046cde4e9c78b34b96d3e6351d59bd04a41d9be6535e2dd0cb425c30c6ac8b0f

See more details on using hashes here.

Provenance

The following attestation bundles were made for code_graph_rag-0.0.246.tar.gz:

Publisher: publish.yml on vitali87/code-graph-rag

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

File details

Details for the file code_graph_rag-0.0.246-py3-none-any.whl.

File metadata

File hashes

Hashes for code_graph_rag-0.0.246-py3-none-any.whl
Algorithm Hash digest
SHA256 2e4177e44fe15aaf64edf9b8f1a21be8378b40139c00eef79d2baee7b5e9f8ee
MD5 23948be74d06e64a2caacf8739f283f3
BLAKE2b-256 bd2d04860076e8afa7cae4d92da0db1b6a9595ecb3468c9801f5da9839e06701

See more details on using hashes here.

Provenance

The following attestation bundles were made for code_graph_rag-0.0.246-py3-none-any.whl:

Publisher: publish.yml on vitali87/code-graph-rag

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