Mathematical Proof Agent v2.0

交互式数学证明助手，支持 OpenAI GPT 和 Google Gemini 两类大模型。

功能特点

• 双模型支持：可选择使用 OpenAI GPT 系列或 Google Gemini 系列模型
• 自定义 API Key：选择模型类型后需要输入对应的 API key
• 全流程统一模型：选择某一类模型后，整个证明流程都使用该模型

支持的模型

OpenAI GPT 系列

• GPT-4o (推荐)
• GPT-4o-mini (快速)
• GPT-4 Turbo
• GPT-3.5 Turbo
• o1-preview
• o1-mini
• 自定义模型名称

Google Gemini 系列

• Gemini 1.5 Pro (推荐)
• Gemini 1.5 Flash (快速)
• Gemini Pro
• Gemini 1.0 Pro
• 自定义模型名称

安装

pip install -r requirements.txt

使用方法

方式一：Web界面（推荐）

启动Web服务器：

python start_web.py

然后在浏览器中访问 http://localhost:5000

Web界面特点：

• 🎨 类似ChatGPT的现代化聊天界面
• 💬 通过自然对话完成证明流程
• 🔄 实时流式响应
• ⚙️ 图形化模型配置

方式二：命令行界面

python main.py

运行流程

1. 选择模型类型

   • 选择 OpenAI GPT 或 Google Gemini

2. 输入 API Key

   • OpenAI API Key (通常以 sk- 开头)
   • Google API Key (可在 Google AI Studio 获取)

3. 选择具体模型

   • 从预设列表中选择，或输入自定义模型名称

4. 选择运行模式

   • 完整交互模式：四阶段全部交互
   • 快速模式：使用示例问题
   • 直接输入模式：跳过问题形式化

5. 四阶段证明流程

   • 问题形式化：明确假设和结论
   • 知识检索：查找相关参考资料
   • 任务规划：制定证明步骤
   • 交互式证明：生成并完善证明

文件结构

.
├── main.py              # 主入口文件
├── model_config.py      # 模型配置管理模块
├── formulator.py        # 问题形式化模块
├── planner.py           # 任务规划模块
├── prover.py            # 交互式证明模块
├── search_agent.py      # RAG 检索代理模块
├── checker.py           # 证明检查模块
├── navigator.py         # 流程导航模块
├── tool.py              # 工具类
├── rag.py               # RAG 向量检索
├── ragsetup.py          # RAG 数据库构建
├── storage.py           # 数据存储
├── state.py             # 状态定义
├── logger.py            # 日志记录
└── requirements.txt     # 依赖列表

配置说明

API Key 获取

OpenAI API Key:

1. 访问 https://platform.openai.com/
2. 登录或注册账号
3. 在 API Keys 页面创建新的 API Key

Google API Key:

1. 访问 https://makersuite.google.com/app/apikey
2. 登录 Google 账号
3. 创建新的 API Key

注意事项

1. API Key 在会话期间保存在内存中，不会持久化存储
2. 选择模型后，整个证明流程都会使用该模型
3. 如需切换模型，需要重新运行程序
4. 建议根据任务复杂度选择合适的模型

示例

█████████████████████████████████████████████████████████████
   Mathematical Proof Agent v2.0
   交互式数学证明助手
█████████████████████████████████████████████████████████████

============================================================
模型配置
============================================================

【第一步】请选择模型类型：
----------------------------------------
  [1] OpenAI GPT 系列
  [2] Google Gemini 系列
----------------------------------------
请输入选项 (1 或 2): 1

✓ 已选择: OpenAI GPT 系列

【第二步】请输入 API Key：
----------------------------------------
提示: OpenAI API Key 格式通常为 sk-...
请输入 OpenAI API Key: sk-xxx...

✓ API Key 已设置

【第三步】请选择具体模型：
----------------------------------------
  [1] GPT-4o (推荐)
  [2] GPT-4o-mini (快速)
  ...
----------------------------------------
请输入选项 (默认为1): 1

✓ 已选择模型: gpt-4o

============================================================
配置完成！
  模型类型: OpenAI GPT 系列
  模型名称: gpt-4o
  API Key: sk-xxx...xxx
============================================================

License

MIT License