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.3.tar.gz (15.9 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.3-py3-none-any.whl (16.9 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: agent_rag_mcp-1.0.3.tar.gz
  • Upload date:
  • Size: 15.9 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.3.tar.gz
Algorithm Hash digest
SHA256 bcf3b6407f511eafa851d6d84940661ddb00dbefe9a528231832c42201f16e7a
MD5 8cbc1782b1196bfe41e1e068c97782c8
BLAKE2b-256 d11f08775362efa090ef332903523759c1f2d698f6692671413718be7a8719d9

See more details on using hashes here.

File details

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

File metadata

  • Download URL: agent_rag_mcp-1.0.3-py3-none-any.whl
  • Upload date:
  • Size: 16.9 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.3-py3-none-any.whl
Algorithm Hash digest
SHA256 c32d137cba78729d07a0335553cc7e52b889606896d2d1caef514a21cf8b4def
MD5 c41bf5d44be309cc741e5f462620ed48
BLAKE2b-256 522a76a426a80bfd032d64d1feebefde508eaee869f16c75dfb6c1322a8322a4

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