Skip to main content

RAG MCP server: ChromaDB + sentence-transformers, exposes ingest/search/list/delete tools.

Project description

Agent_rag

Agent_rag is a RAG (Retrieval-Augmented Generation) MCP (Model Context Protocol) Server. It uses ChromaDB and sentence-transformers locally to provide a vector store for your intelligent agents.

This server exposes several tools for the orchestrator (Agent_head) or any other MCP client to call:

  • rag_ingest: Ingest documents, directories, or raw text into a collection.
  • rag_search: Semantic search against the ingested knowledge base.
  • rag_list_collections: List all active collections.
  • rag_delete_collection: Delete a specific collection.

Features

  • Local Embeddings: Uses sentence-transformers (default: all-MiniLM-L6-v2) locally, meaning no API keys or external services are required for embedding generation.
  • FastMCP Built-in: Asynchronous and thread-safe tool execution using FastMCP.
  • Easy Configuration: Configurable via config.yaml to set your desired chunk size, collection names, and embedding models.

Installation & Usage

Agent_rag is packaged and distributed via standard Python mechanisms. You can run it effortlessly using uv without needing to clone the repository or manually manage virtual environments.

Running with uvx

You can run the MCP server directly. uvx will automatically download and run the latest version of the agent-rag-mcp CLI:

uvx agent-rag-mcp

Transport Modes By default, the server runs in stdio transport mode (designed to be spawned as a subprocess by MCP clients like Agent_head).

To run it over HTTP using Server-Sent Events (SSE):

uvx agent-rag-mcp --transport sse --port 8002 --host 0.0.0.0

Specifying a Test Registry (If using TestPyPI)

If you published the package to: TestPyPI instead of the main PyPI, run it via

uvx --extra-index-url https://test.pypi.org/simple/ --index-strategy unsafe-best-match agent-rag-mcp@latest

Integrating with Agent_head

To connect this RAG server to your Agent_head orchestrator, add the following configuration to your Agent_head/config.yaml:

memory:
  enabled: true
  backend: "rag"

  # Configure this if backend is set to "rag"
  rag_server:
    command: "uvx"
    args: ["agent-rag-mcp"] # Or ["--from", "/path/to/local/Agent_rag", "agent-rag-mcp"] for local development
    collection: "agent_memory"

Local Development

If you are developing this package locally:

  1. Install dependencies:
    uv sync
    
  2. Run locally:
    uv run agent-rag-mcp
    
  3. Build the package:
    uv build
    

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

agent_rag_mcp-1.0.4.tar.gz (16.5 kB view details)

Uploaded Source

Built Distribution

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

agent_rag_mcp-1.0.4-py3-none-any.whl (17.6 kB view details)

Uploaded Python 3

File details

Details for the file agent_rag_mcp-1.0.4.tar.gz.

File metadata

  • Download URL: agent_rag_mcp-1.0.4.tar.gz
  • Upload date:
  • Size: 16.5 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.10.4 {"installer":{"name":"uv","version":"0.10.4","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":null,"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for agent_rag_mcp-1.0.4.tar.gz
Algorithm Hash digest
SHA256 08dfda0352f09a9d619976ce71d6ba797ee0ddcf5d6b4c95979c650ce270fdb0
MD5 22d611f5132aef9fd8626d79e47ace86
BLAKE2b-256 915fb6d5c639175d4a8c2eb15600bc0b47065d27b784c212d61fac6a79b8de34

See more details on using hashes here.

File details

Details for the file agent_rag_mcp-1.0.4-py3-none-any.whl.

File metadata

  • Download URL: agent_rag_mcp-1.0.4-py3-none-any.whl
  • Upload date:
  • Size: 17.6 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.10.4 {"installer":{"name":"uv","version":"0.10.4","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":null,"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for agent_rag_mcp-1.0.4-py3-none-any.whl
Algorithm Hash digest
SHA256 3bf69084387270c01678f8834e2886e370205b55574cab9fef9ce51f9f235885
MD5 48624b3888935b74924f172085c68a66
BLAKE2b-256 b17cb86e032a2839e727cc00d9ce7557a9ae7ba0ab7843e0c8f1456930205879

See more details on using hashes here.

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