Skip to main content

Integrate RAG into AI Agents via MCP Server. Supports multiple Vector database technologies.

Project description

Vector Database MCP Server

PyPI - Version PyPI - Downloads GitHub Repo stars GitHub forks GitHub contributors PyPI - License GitHub

GitHub last commit (by committer) GitHub pull requests GitHub closed pull requests GitHub issues

GitHub top language GitHub language count GitHub repo size GitHub repo file count (file type) PyPI - Wheel PyPI - Implementation

Version: 0.1.2

This is an MCP Server implementation which allows for a standardized collection management system across vector database technologies.

This was heavily inspired by the RAG implementation of Microsoft's Autogen V1 framework, however, this was changed to an MCP server model instead.

AI Agents can:

  • Create collections with documents stored on the local filesystem or URLs
  • Add documents to a collection
  • Utilize collection for retrieval augmented generation (RAG)
  • Delete collection

Supports:

  • ChromaDB - 95% Tested
  • PGVector - 90% Tested
  • Couchbase - 80% Tested
  • Qdrant - 80% Tested
  • MongoDB - 80% Tested

This repository is actively maintained - Contributions and bug reports are welcome!

Automated tests are planned

Usage:

Using as an MCP Server:

Creating Collection

AI Prompt:

Create a collection called zapdos with the documents in this directory:
/home/user/Documents/Chroma

AI Response:

The collection named "zapdos" has been successfully created or retrieved from the vector database
using the documents in the directory /home/user/Documents/Chroma.
Let me know if you'd like to perform any further actions, such as querying the collection or adding more documents!

Retrieving from Collection

AI Prompt:

Create a collection called zapdos with the documents in this directory:
/home/user/Documents/Chroma

AI Response:

The collection named "zapdos" has been successfully created or retrieved from the vector database
using the documents in the directory /home/user/Documents/Chroma.
Let me know if you'd like to perform any further actions, such as querying the collection or adding more documents!

Deleting Collection

AI Prompt:

Delete the collection called memory

AI Response:

The collection named "memory" has been successfully deleted.
Let me know if you'd like to create a new collection or perform any other actions!
Example:

Use in CLI

Short Flag Long Flag Description
-h --help See Usage
-h --host Host of Vector Database
-p --port Port of Vector Database
-d --path Path of local Vector Database
-t --transport Transport Type (https/stdio)
vector-mcp

Use with AI

Deploy MCP Server as a Service

docker pull knucklessg1/vector-mcp:latest

Modify the compose.yml

services:
  vector-mcp-mcp:
    image: knucklessg1/vector-mcp:latest
    volumes:
      - development:/root/Development
    environment:
      - HOST=0.0.0.0
      - PORT=8001
    ports:
      - 8001:8001

Configure mcp.json

{
  "mcpServers": {
    "vector_mcp": {
      "command": "uv",
      "args": [
        "run",
        "--with",
        "vector-mcp",
        "vector-mcp"
      ],
      "env": {
        "DATABASE_TYPE": "chromadb",                   // Optional
        "COLLECTION_NAME": "memory",                   // Optional
        "DOCUMENT_DIRECTORY": "/home/user/Documents/"  // Optional
      },
      "timeout": 300000
    }
  }
}
Installation Instructions:

Install Python Package

python -m pip install vector-mcp

PGVector dependencies

python -m pip install vector-mcp[pgvector]

All

python -m pip install vector-mcp[all]
Repository Owners:

GitHub followers GitHub User's stars

Special shoutouts to Microsoft Autogen V1 ♥️

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

vector_mcp-0.1.2.tar.gz (48.1 kB view details)

Uploaded Source

Built Distribution

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

vector_mcp-0.1.2-py3-none-any.whl (67.3 kB view details)

Uploaded Python 3

File details

Details for the file vector_mcp-0.1.2.tar.gz.

File metadata

  • Download URL: vector_mcp-0.1.2.tar.gz
  • Upload date:
  • Size: 48.1 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.2

File hashes

Hashes for vector_mcp-0.1.2.tar.gz
Algorithm Hash digest
SHA256 7ec7004249f4f8ddacbebec6eabfa8c9d860874130cc9405416b7970fe4b1400
MD5 a03e26c7cb0203832e3ea398d22d2d58
BLAKE2b-256 f9ed7291421d361655bdb725bb19e64fec87a317d6982b84c3b32f27599b2692

See more details on using hashes here.

File details

Details for the file vector_mcp-0.1.2-py3-none-any.whl.

File metadata

  • Download URL: vector_mcp-0.1.2-py3-none-any.whl
  • Upload date:
  • Size: 67.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.11.2

File hashes

Hashes for vector_mcp-0.1.2-py3-none-any.whl
Algorithm Hash digest
SHA256 fac5f1998ed51b24a19f7e249b038a2de3fa01d8235c179e7835db336ebfb8bc
MD5 43e7eabb023303f6919e79f73aa863bc
BLAKE2b-256 3bb52ca11573b31e05460da94b014b5a76bf877b8bcc4281bb6cf179134d7b54

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