sourcesage

MCP.Pizza Chef: sarathsp06

SourceSage is an MCP server that efficiently caches and memorizes codebases as knowledge graphs, capturing logic, style, and standards. It supports multiple programming languages, enables incremental updates, and optimizes token usage for fast retrieval and dynamic code analysis using LLMs. This server enhances code understanding and memory for AI workflows.

Use This MCP server To

Cache large codebases as structured knowledge graphs Enable fast retrieval of code logic and style patterns Support multi-language code analysis and memory Incrementally update code knowledge on changes Optimize token usage for efficient LLM memory Provide LLM-driven insights on code relationships Facilitate AI-assisted code review and refactoring Integrate with developer tools for real-time code context

README

SourceSage: Efficient Code Memory for LLMs

SourceSage is an MCP (Model Context Protocol) server that efficiently memorizes key aspects of a codebase—logic, style, and standards—while allowing dynamic updates and fast retrieval. It's designed to be language-agnostic, leveraging the LLM's understanding of code across multiple languages.

Features

  • Language Agnostic: Works with any programming language the LLM understands
  • Knowledge Graph Storage: Efficiently stores code entities, relationships, patterns, and style conventions
  • LLM-Driven Analysis: Relies on the LLM to analyze code and provide insights
  • Token-Efficient Storage: Optimizes for minimal token usage while maximizing memory capacity
  • Incremental Updates: Updates knowledge when code changes without redundant storage
  • Fast Retrieval: Enables quick and accurate retrieval of relevant information

How It Works

SourceSage uses a novel approach where:

  1. The LLM analyzes code files (in any language)
  2. The LLM uses MCP tools to register entities, relationships, patterns, and style conventions
  3. SourceSage stores this knowledge in a token-efficient graph structure
  4. The LLM can later query this knowledge when needed

This approach leverages the LLM's inherent language understanding while focusing the MCP server on efficient memory management.

Installation

# Clone the repository
git clone https://github.com/yourusername/sourcesage.git
cd sourcesage

# Install the package
pip install -e .

Usage

Running the MCP Server

# Run the server
sourcesage

# Or run directly from the repository
python -m sourcesage.mcp_server

Connecting to Claude for Desktop

  1. Open Claude for Desktop
  2. Go to Settings > Developer > Edit Config
  3. Add the following to your claude_desktop_config.json:

If you've installed the package:

{
  "mcpServers": {
    "sourcesage": {
      "command": "sourcesage",
      "args": []
    }
  }
}

If you're running from a local directory without installing:

{
  "sourcesage": {
      "command": "uv", 
      "args": [
        "--directory",
        "/path/to/sourcesage",
        "run",
        "main.py"
      ]
    },
}
  1. Restart Claude for Desktop

Available Tools

SourceSage provides the following MCP tools:

  1. register_entity: Register a code entity in the knowledge graph

    Input:
      - name: Name of the entity (e.g., class name, function name)
      - entity_type: Type of entity (class, function, module, etc.)
      - summary: Brief description of the entity
      - signature: Entity signature (optional)
      - language: Programming language (optional)
      - observations: List of observations about the entity (optional)
      - metadata: Additional metadata (optional)
    Output: Confirmation message with entity ID
    
  2. register_relationship: Register a relationship between entities

    Input:
      - from_entity: Name of the source entity
      - to_entity: Name of the target entity
      - relationship_type: Type of relationship (calls, inherits, imports, etc.)
      - metadata: Additional metadata (optional)
    Output: Confirmation message with relationship ID
    
  3. register_pattern: Register a code pattern

    Input:
      - name: Name of the pattern
      - description: Description of the pattern
      - language: Programming language (optional)
      - example: Example code demonstrating the pattern (optional)
      - metadata: Additional metadata (optional)
    Output: Confirmation message with pattern ID
    
  4. register_style_convention: Register a coding style convention

    Input:
      - name: Name of the convention
      - description: Description of the convention
      - language: Programming language (optional)
      - examples: Example code snippets demonstrating the convention (optional)
      - metadata: Additional metadata (optional)
    Output: Confirmation message with convention ID
    
  5. add_entity_observation: Add an observation to an entity

    Input:
      - entity_name: Name of the entity
      - observation: Observation to add
    Output: Confirmation message
    
  6. query_entities: Query entities in the knowledge graph

    Input:
      - entity_type: Filter by entity type (optional)
      - language: Filter by programming language (optional)
      - name_pattern: Filter by name pattern (regex, optional)
      - limit: Maximum number of results to return (optional)
    Output: List of matching entities
    
  7. get_entity_details: Get detailed information about an entity

    Input:
      - entity_name: Name of the entity
    Output: Detailed information about the entity
    
  8. query_patterns: Query code patterns in the knowledge graph

    Input:
      - language: Filter by programming language (optional)
      - pattern_name: Filter by pattern name (optional)
    Output: List of matching patterns
    
  9. query_style_conventions: Query coding style conventions

    Input:
      - language: Filter by programming language (optional)
      - convention_name: Filter by convention name (optional)
    Output: List of matching style conventions
    
  10. get_knowledge_statistics: Get statistics about the knowledge graph

    Input: None
    Output: Statistics about the knowledge graph
    
  11. clear_knowledge: Clear all knowledge from the graph

    Input: None
    Output: Confirmation message
    

Example Workflow with Claude

  1. Analyze Code: Ask Claude to analyze your code files

    "Please analyze this Python file and register the key entities and relationships."
    
  2. Register Entities: Claude will use the register_entity tool to store code entities

    "I'll register the main class in this file."
    
  3. Register Relationships: Claude will use the register_relationship tool to store relationships

    "I'll register the inheritance relationship between these classes."
    
  4. Query Knowledge: Later, ask Claude about your codebase

    "What classes are defined in my codebase?"
    "Show me the details of the User class."
    "What's the relationship between the User and Profile classes?"
    
  5. Get Coding Patterns: Ask Claude about coding patterns

    "What design patterns are used in my codebase?"
    "Show me examples of the Factory pattern in my code."
    

How It's Different

Unlike traditional code analysis tools, SourceSage:

  1. Leverages LLM Understanding: Uses the LLM's ability to understand code semantics across languages
  2. Stores Semantic Knowledge: Focuses on meaning and relationships, not just syntax
  3. Is Language Agnostic: Works with any programming language the LLM understands
  4. Optimizes for Token Efficiency: Stores knowledge in a way that minimizes token usage
  5. Evolves with LLM Capabilities: As LLMs improve, so does code understanding

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

License

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

sourcesage FAQ

How does SourceSage handle multiple programming languages?
SourceSage is language-agnostic and leverages the LLM's understanding to analyze and store code from any programming language supported by the model.
How does SourceSage optimize token usage?
It uses a knowledge graph to store code entities and relationships efficiently, minimizing redundant token storage and maximizing memory capacity.
Can SourceSage update its stored knowledge when code changes?
Yes, it supports incremental updates to refresh only the changed parts of the codebase without redundant reprocessing.
What kind of insights can SourceSage provide?
It offers LLM-driven analysis of code logic, style conventions, and relationships between code entities to support better understanding and tooling.
Is SourceSage suitable for real-time developer workflows?
Yes, it is designed for fast retrieval and dynamic updates, making it ideal for integration with real-time coding environments.
Does SourceSage require specific LLM providers?
No, it is provider-agnostic and can work with OpenAI, Anthropic Claude, and Google Gemini models.
How does SourceSage store code as a knowledge graph?
It maps code entities, their relationships, and patterns into a graph structure optimized for efficient querying and memory.
Can SourceSage be integrated with existing MCP clients?
Yes, it is designed as an MCP server to seamlessly integrate with MCP clients for enhanced code context and memory.