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.12.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.12-py3-none-any.whl (40.3 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: mcp_rag-0.3.12.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.12.tar.gz
Algorithm Hash digest
SHA256 e81a6b94acc4c445be3215b9ef42700110e1ddea748cc32ced59ab1715466063
MD5 0492cca7c6f8229d8fb4b0a4ccb16d8f
BLAKE2b-256 7e0edb654f9d541673ce039f30fd58dba078516b60f426668f0fe4494b53a6df

See more details on using hashes here.

File details

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

File metadata

  • Download URL: mcp_rag-0.3.12-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.12-py3-none-any.whl
Algorithm Hash digest
SHA256 e5dd2f1ba06175b2fdbde6c8be3b55fc3694ddc02c1b3c5b1b06cdf585b39e52
MD5 55f80e15b9024ccec6eddd2585d3325e
BLAKE2b-256 a7b16ab5970972bbbf329ebe067050c36f77f8f56284c2b829270f3e77e06d6e

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