Skip to main content

Add your description here

Project description

MCP RAG 工具集

基于模型上下文协议(MCP)的智能知识库系统,提供文档处理、知识问答和向量库管理功能。

支持使用豆包与OpenAI

✨ 主要特性

  • 🧠 智能知识库:基于向量检索的 RAG 系统,支持语义搜索和智能问答
  • 📄 多格式文档处理:支持超过 25 种文档格式,包括 PDF、DOCX、PPTX、XLSX、图片、邮件等
  • 🌐 直观 Web 界面:Bento 风格布局,分类展示所有工具功能
  • 🤖 多模型支持:兼容 OpenAI、豆包、Ollama 等主流 AI 模型
  • 🔍 高级过滤搜索:支持按文件类型、内容结构等条件进行精确检索
  • 📊 统计分析:提供知识库统计、嵌入缓存分析等数据洞察
  • ⚡ 本地化处理:支持本地模型推理,保护数据隐私
  • 🔧 向量库管理:提供缓存清理、数据库优化等维护功能

安装

# 安装工具
uv tool install mcp_rag

# 升级工具
uv tool install mcp_rag --upgrade

# 卸载工具
uv tool uninstall mcp_rag

使用

启动 MCP 服务器

mcp_rag server

启动 Web 界面

mcp_rag web

Web 界面提供直观的 Bento 布局,支持以下工具分类:

  • 📥 添加内容:添加文本和文档到知识库
  • ❓ 智能问答:基于知识库进行问答和检索
  • 📊 数据统计:查看知识库和系统统计信息
  • ⚙️ 向量库管理:优化和维护向量数据库

配置

在项目根目录创建 .env 文件进行配置:

# OpenAI 配置
OPENAI_API_KEY=
OPENAI_API_BASE=https://api.openai.com/v1
OPENAI_MODEL=gpt-4o-mini
OPENAI_TEMPERATURE=0
OPENAI_EMBEDDING_MODEL=text-embedding-3-large

# 豆包 配置
# OPENAI_API_KEY=
# OPENAI_API_BASE=https://ark.cn-beijing.volces.com/api/v3
# OPENAI_MODEL=doubao-1-5-pro-32k-250115
# OPENAI_TEMPERATURE=0
# OPENAI_EMBEDDING_MODEL=doubao-embedding-text-240715

mcp客户端配置(豆包为例)

{
    "mcpServers": {
        "rag": {
            "command": "uv",
            "args": [
                "run",
                "mcp-rag",
                "serve"
            ],
            "env": {
                "PYTHONUNBUFFERED": "1",

                "OPENAI_API_KEY": "key",
                "OPENAI_API_BASE": "https://ark.cn-beijing.volces.com/api/v3",
                "OPENAI_MODEL": "doubao-1-5-pro-32k-250115",
                "OPENAI_TEMPERATURE": "0",

                "OPENAI_EMBEDDING_MODEL": "doubao-embedding-text-240715",
            }
        }
    }
}

可用工具

添加内容

  • learn_text(text, source_name) - 添加文本到知识库
  • learn_document(file_path) - 处理并添加文档到知识库

智能问答

  • ask_rag(query) - 基于知识库回答问题
  • ask_rag_filtered(query, file_type, min_tables, min_titles, processing_method) - 带过滤条件的智能检索

支持格式

支持超过 25 种文档格式,包括 PDF、DOCX、PPTX、XLSX、图片、邮件等。

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

mcp_rag-0.3.11.tar.gz (35.6 kB view details)

Uploaded Source

Built Distribution

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

mcp_rag-0.3.11-py3-none-any.whl (40.3 kB view details)

Uploaded Python 3

File details

Details for the file mcp_rag-0.3.11.tar.gz.

File metadata

  • Download URL: mcp_rag-0.3.11.tar.gz
  • Upload date:
  • Size: 35.6 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.5

File hashes

Hashes for mcp_rag-0.3.11.tar.gz
Algorithm Hash digest
SHA256 a9b5048d0ab77363430107589af9bbe52e6a30710a17d7c24a1b0a10a40c6f26
MD5 ab854cfcb5ec3c2482453410c67ff290
BLAKE2b-256 d97047552b44adf8ffacc53545cc9c9ae459bfbb1c0671e3687f9cf1d701021f

See more details on using hashes here.

File details

Details for the file mcp_rag-0.3.11-py3-none-any.whl.

File metadata

  • Download URL: mcp_rag-0.3.11-py3-none-any.whl
  • Upload date:
  • Size: 40.3 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.5

File hashes

Hashes for mcp_rag-0.3.11-py3-none-any.whl
Algorithm Hash digest
SHA256 8ee00e0f4c036afd1428e18489f948a7ce1687722f855e3e8fcc6185aefc4ff2
MD5 f34c3d34e664dacfbcaae8b6d3a52d2b
BLAKE2b-256 8790d3dd66091d023577b2520f71d85e50d8c37062deb9c6e1a2f7f072f9ab61

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