arXiv论文搜索聊天机器人

一个使用MiniMax的LLM API通过功能调用搜索和提取arXiv论文信息的AI聊天机器人。

  • by leiwingqueen
  • Jupyter Notebook
  • 本地部署
  • 更新于
在 GitHub 查看 leiwingqueen/mcp-course

使用说明

MCP课程-arXiv论文聊天机器人

一个AI聊天机器人,使用MiniMax的LLM API通过函数调用(工具定义)搜索和提取有关arXiv论文的信息。

特性

  • 搜索arXiv论文:搜索任何主题的学术论文
  • 提取纸张信息:检索特定论文的详细信息
  • 交互式CLI:在交互式命令行界面中与机器人聊天
  • 工具执行LLM可以自主调用工具来搜索论文和检索信息

先决条件

  • Python 3.10+
  • MiniMax API密钥

安装

  1. 克隆存储库:
git clone <repository-url>
cd mcp-course
  1. 创建并激活虚拟环境:
python -m venv venv
source venv/bin/activate  # On macOS/Linux
# or
venv\Scripts\activate  # On Windows
  1. 安装依赖项:
pip install -r requirements.txt
  1. 配置环境变量:

创建一个 .env 项目根目录中的文件,内容如下:

ANTHROPIC_BASE_URL=https://api.minimaxi.com/anthropic
ANTHROPIC_API_KEY=your-api-key-here

替换 your-api-key-here 使用您的MiniMax API密钥。

用法

运行聊天机器人:

python chatbot.py

查询示例

  • “搜索有关机器学习的论文”
  • “搜索关于大型语言模型的论文”
  • “跟我说说纸2310.12345”

类型 quit 退出聊天。

项目结构

mcp-course/
├── chatbot.py          # Main chatbot implementation
├── .env                # Environment variables (API keys)
├── CLAUDE.md          # Claude Code instructions
├── papers/             # Directory for storing paper data
│   └── <topic>/
│       └── papers_info.json
├── venv/               # Python virtual environment
└── chatbot.log         # Chat logs

工具

聊天机器人提供两个工具:

  1. 搜索路径:在arXiv上搜索某个主题的论文

    • topic:要搜索的主题
    • max_results:最大结果数(默认值:5)
  2. 提取信息:检索特定论文的信息

    • paper_id:arXiv纸张ID(例如“2310.12345”)

建筑

  • chatbot.py:核心实施包括:

    • search_papers() -搜索arXiv并将纸张信息保存为JSON
    • extract_info() -检索已保存的纸张信息
    • process_query() -带有工具执行的主查询处理循环
    • chat_loop() -交互式CLI循环
  • 工具系统:工具使用Anthropic SDK的工具模式格式定义,使LLM能够根据用户查询自主决定何时调用工具。

许可证

麻省理工学院