Merge pull request #10 from jihe520/dev

更新正确获取文献引用
添加了建模手agent
为每个agent设置不同模型
修改文档
修改部分架构设计和算法逻辑
获得新的 bug

Author: jihe520

README.md

@@ -24,11 +24,15 @@
 ## ✨ 功能特性
 
 - 🔍 自动分析问题，数学建模，编写代码，纠正错误，撰写论文
-- 💻 本地代码解释器
+- 💻 Code Interperter
+    - loacl Interperter: 基于 jupyter , 代码保存为 notebook 方便再编辑
+    - 云端 code interperter: [E2B](https://e2b.dev/) 和 [daytona](https://app.daytona.io/)
 - 📝 生成一份编排好格式的论文
-- 🤝 muti-agents: ~~建模手~~，代码手(反思模块，本地代码解释器)，论文手
-- 🔄 muti-llms: 每个agent设置不同的模型
-- 💰 成本低 agentless(单次任务成本约 1 rmb)
+- 🤝 muti-agents: 建模手，代码手，论文手等
+- 🔄 muti-llms: 每个 agent 设置不同的、合适的模型
+- 🤖 支持所有模型: [litellm](https://docs.litellm.ai/docs/providers)
+- 💰 成本低：workflow agentless，不依赖 agent 框架
+- 🧩 自定义模板：prompt inject 为每个 subtask 单独设置需求
 
 ## 🚀 后期计划
 
@@ -38,14 +42,18 @@
 - [ ] 英文支持（美赛）
 - [ ] 集成 latex 模板
 - [ ] 接入视觉模型
-- [ ] 添加正确文献引用
+- [x] 添加正确文献引用
 - [x] 更多测试案例
 - [ ] docker 部署
-- [ ] 引入用户的交互（选择模型，重写等等）
+- [ ] human in loop: 引入用户的交互（选择模型，@agent重写，等等）
+- [ ] feedback: evaluate the result and modify
 - [x] codeinterpreter 接入云端 如 e2b 等供应商..
 - [ ] 多语言: R 语言, matlab
-- [ ] 绘图 napki,draw.io
+- [ ] 绘图 napki,draw.io,plantuml,svg, mermaid.js
 - [ ] 添加 benchmark
+- [ ] web search tool
+- [ ] RAG 知识库
+- [ ] A2A hand off: 代码手多次反思错误，hand off 更聪明模型 agent
 
 ## 视频demo
 
@@ -55,6 +63,10 @@
 > 项目处于实验探索迭代demo阶段，有许多需要改进优化改进地方，我(项目作者)很忙，有时间会优化更新
 > 欢迎贡献
 
+
+案例参考 [demo](./demo/) 文件夹下
+**如果你有跑出来好的案例可以提交 PR 在该目录下**
+
 ## 📖 使用教程
 
 > 确保电脑中安装好 Python, Nodejs, **Redis** 环境
@@ -63,19 +75,13 @@
 
 
 
-1. 配置模型
+1. 配置环境变量
 
 复制`/backend/.env.dev.example`到`/backend/.env.dev`(删除`.example` 后缀)
-填写配置模型和 APIKEY
-推荐模型能力较强的、参数量大的模型。
 
-```bash
-# support all model, check out https://docs.litellm.ai/docs/ 
-API_KEY=
-# gpt-4.1,deepseek/deepseek-chat
-MODEL=
-# 确保安装 Redis
-```
+**配置环境变量**
+
+推荐模型能力较强的、参数量大的模型。
 
 复制`/fronted/.env.example`到`/fronted/.env`(删除`.example` 后缀)
 
@@ -91,6 +97,8 @@ git clone https://github.com/jihe520/MathModelAgent.git # 克隆项目
 
 启动后端
 
+*启动 redis*
+
 ```bash
 cd backend # 切换到 backend 目录下
 pip install uv # 推荐使用 uv 管理 python 项目
@@ -114,10 +122,18 @@ pnpm i #确保电脑安装了 pnpm
 pnpm run dev
 ```
 
+[教程](./docs/md/tutorial.md)
 
 运行的结果和产生在`backend/project/work_dir/xxx/*`目录下
 - notebook.ipynb: 保存运行过程中产生的代码
-- res.md: 保存最后运行产生的结果为 markdown 格式，使用 markdown 转 word(研究下 pandoc)
+- res.md: 保存最后运行产生的结果为 markdown 格式
+
+有没有自动部署的脚本 ？
+[mmaAutoSetupRun](https://github.com/Fitia-UCAS/mmaAutoSetupRun)
+
+需要自定义自定义提示词模板 template ？
+Prompt Inject : [prompt](./backend/app/config/md_template.toml)
+
 
 ## 🤝 贡献和开发
 
@@ -134,7 +150,8 @@ clone 项目后，下载 **Todo Tree** 插件，可以查看代码中所有具
 
 ## 📄 版权License
 
-个人免费使用，请勿商业用途，商业用途联系我（作者）
+个人免费使用，请勿商业用途。
+[License](./docs/md/License.md)
 
 ## 🙏 Reference
 
@@ -147,10 +164,18 @@ Thanks to the following projects:
 
 ## 其他
 
+### Sponsor
+
+[Buy Me a Coffee](./docs/sponser.md)
+
 感谢赞助
 [danmo-tyc](https://github.com/danmo-tyc)
 
+### GROUP
+
 有问题可以进群问
 [QQ 群：699970403](http://qm.qq.com/cgi-bin/qm/qr?_wv=1027&k=rFKquDTSxKcWpEhRgpJD-dPhTtqLwJ9r&authKey=xYKvCFG5My4uYZTbIIoV5MIPQedW7hYzf0%2Fbs4EUZ100UegQWcQ8xEEgTczHsyU6&noverify=0&group_code=699970403)
 
-<img src="./docs/qq.jpg" height="400px">
+<div align="center">
+    <img src="./docs/qq.jpg" height="400px">
+</div>

backend/.env.dev.example

@@ -1,8 +1,27 @@
 ENV=dev
 # support all model, check out https://docs.litellm.ai/docs/ 
-API_KEY=
-# gpt-4.1,deepseek/deepseek-chat
-MODEL=
+# e.g. gpt-4.1,deepseek/deepseek-chat
+# 为每个 agent 配置合适的模型
+COORDINATOR_API_KEY=
+COORDINATOR_MODEL=
+# COORDINATOR_BASE_URL= 如果需要中转自定义等 
+
+# 推荐 thinking model
+MODELER_API_KEY=
+MODELER_MODEL=
+# MODELER_BASE_URL=
+
+CODER_API_KEY=
+CODER_MODEL=
+# CODER_BASE_URL=
+
+WRITER_API_KEY=
+WRITER_MODEL=
+# WRITER_BASE_URL=
+
+DEFAULT_API_KEY=
+DEFAULT_MODEL=
+# DEFAULT_BASE_URL= 
 
 # 模型最大问答次数
 MAX_CHAT_TURNS=60
@@ -12,6 +31,8 @@ MAX_RETRIES=5
 # 不需要填，默认调用本地 Python
 # E2B_API_KEY=
 SERVER_HOST=http://localhost:8000
+# 使用 email 注册账号从 https://openalex.org/ 文献
+OPENALEX_EMAIL=
 
 LOG_LEVEL=DEBUG
 DEBUG=true

backend/app/config/md_template.toml

@@ -1,22 +1,17 @@
 firstPage = """
 # 标题： 基于全文取一个标题
 例子：基于哈里斯鹰算法的农业种植优化模型研究
-
 摘要
 第一段：背景
-例子：随着农业现代化的推进，农业种植问题逐渐成为乡村发展中的关键课题之一
 大约100字
 
-第二段：针对问题一，根据问题一回答，使用到什么模型，如何去构建该模型，做了什么工作，得到什么结果，一些具体的数值
-大约170字
-
-每个问题都需要单独撰写一段，与上面相同要求
+每个问题都需要按照下面类似的单独撰写成一段'\n'，不要分点叙述。
+针对问题几，根据问题几回答，使用到什么模型，如何去构建该模型，做了什么工作，得到什么结果，一些具体的数值
+每段大约170字
 
 最后一段：总结敏感性分析使用和效果，结论。
 大约150字
 
-注意：每段都是按段落编写，不要分点叙述。
-
 关键词： 用到的模型等，关键词大约4/5个，没有标号
 例子：HHO  MOHHO  遗传算法  Mealpy  蒙特卡洛模拟  规划问题
 
@@ -25,22 +20,17 @@ firstPage = """
 {问题}
 {模型的建立与求解}
 """
-
 RepeatQues = """
 # 一、问题重述
 ## 1.1 问题背景
-
-根据题目总结问题背景
+查询文献
+根据题目总结问题背景（查询文献）
 格式：一段话
 大约160字
 ## 1.2 问题重述
 参考模板如下
 本文基于以上信息建立数学模型来解决以下问题。
-- 问题1：考虑农作物的预期销售量、种植成本、亩产量及销售价格与2023年持平的情况下，提出2024至2030年乡村农作物的最优种植方案。
-- 并且对于预计销售量的结果在两种情况下进行计算：（1）超出部分滞销浪费；（2）超出部分以2023年销售价格的50%售出。
-- 问题2：在考虑农作物预期销售量、亩产量、种植成本、销售价格以及气候、市场变化等不确定因素的情况下，给出2024至2030年的最优种植方案，并充分评估种植风险。
-- 问题3：进一步考虑不同农作物之间的可替代性和互补性，销售量、价格和成本之间的相关性，通过数据模拟，求解最优种植策略，并与问题2的结果进行比较。
-- 如果还有问题
+分点叙述
 
 **参考下面的题目要求和问题，按照上面模板和要求完成 **问题重述中的问题背景和问题重述**
 
@@ -70,7 +60,6 @@ modelAssumption = """
 写大概3/4条，下面例子：
 (1) 数据有效性：假设所用数据真实可靠
 (2) 市场稳定性：假设农业种植中没有自然灾害、政府政策等因素影响。
-(3) 农民自主性：假设2023年农作物的销量为农民农作物全年产量。
 
 参考下面的模型的建立与求解 和 题目 ，按照上面模板和要求完成 **模型的假设**
 
@@ -97,12 +86,10 @@ symbol = """
 
 {模型的建立与求解}
 """
-
 eda = """
 ## 4.2 描述性统计
 大约200字
 """
-
 ques1 = """模板和要求如下
 # 五、模型的建立与求解
 ## 5.1 问题一模型的建立与求解
@@ -122,7 +109,6 @@ ques2 = """参考模板
 模型的求解过程
 大约600字
 """
-
 ques3 = """参考模板
 ## 5.3 问题三模型的建立与求解
 ### 5.3.1 模型的建立
@@ -132,7 +118,6 @@ ques3 = """参考模板
 模型的求解过程
 大约600字
 """
-
 ques4 = """参考模板
 ## 5.4 问题四模型的建立与求解
 ### 5.4.1 模型的建立
@@ -142,7 +127,6 @@ ques4 = """参考模板
 模型的求解过程
 大约600字
 """
-
 ques5 = """参考模板
 ## 5.5 问题五模型的建立与求解
 ### 5.5.1 模型的建立
@@ -152,7 +136,6 @@ ques5 = """参考模板
 模型的求解过程
 大约600字
 """
-
 ques6 = """参考模板
 ## 5.6 问题六模型的建立与求解
 ### 5.6.1 模型的建立
@@ -162,28 +145,15 @@ ques6 = """参考模板
 模型的求解过程
 大约600字
 """
-
 sensitivity_analysis = """参考模板
 # 六、模型的分析与检验
 ## 6.1 灵敏度分析
 """
-
 judge = """参考模板和要求
 # 七、模型的评价、改进与推广
 ## 7.1 模型的优点
 ## 7.2 模型的缺点
 ## 7.3 模型的改进与推广
 优点数量要多于缺点，缺点大约2/3个
 大约200字
-"""
-
-reference = """
-# 参考文献
-[1] 作者. (年份). 题目. 期刊, 卷(期), 页码.
-[2] 作者. (年份). 题目. 期刊, 卷(期), 页码.
-[3] 作者. (年份). 题目. 期刊, 卷(期), 页码.
-
-例子
-[1] 刘培杰,李军英.体验式营销模式对农业旅游经济发展的影响[J].山西农经,2024,(16):67-69.DOI:10.16675/j.cnki.cn14-1065/f.2024.16.019.
-[2] ..
 """

backend/app/config/setting.py

@@ -17,9 +17,27 @@ def parse_cors(value: str) -> list[str]:
 
 class Settings(BaseSettings):
     ENV: str
-    API_KEY: str
-    MODEL: str
-    BASE_URL: Optional[str] = None
+
+    COORDINATOR_API_KEY: str
+    COORDINATOR_MODEL: str
+    COORDINATOR_BASE_URL: Optional[str] = None
+
+    MODELER_API_KEY: str
+    MODELER_MODEL: str
+    MODELER_BASE_URL: Optional[str] = None
+
+    CODER_API_KEY: str
+    CODER_MODEL: str
+    CODER_BASE_URL: Optional[str] = None
+
+    WRITER_API_KEY: str
+    WRITER_MODEL: str
+    WRITER_BASE_URL: Optional[str] = None
+
+    DEFAULT_API_KEY: str
+    DEFAULT_MODEL: str
+    DEFAULT_BASE_URL: Optional[str] = None
+
     MAX_CHAT_TURNS: int
     MAX_RETRIES: int
     E2B_API_KEY: Optional[str] = None
@@ -29,6 +47,7 @@ class Settings(BaseSettings):
     REDIS_MAX_CONNECTIONS: int
     CORS_ALLOW_ORIGINS: Annotated[list[str] | str, BeforeValidator(parse_cors)]
     SERVER_HOST: str = "http://localhost:8000"  # 默认值
+    OPENALEX_EMAIL: Optional[str] = None
 
     model_config = SettingsConfigDict(
         env_file=".env.dev",

backend/app/core/agents/__init__.py

@@ -0,0 +1,11 @@
+from .coder_agent import CoderAgent
+from .writer_agent import WriterAgent
+from .coordinator_agent import CoordinatorAgent
+from .modeler_agent import ModelerAgent
+
+__all__ = [
+    "CoderAgent",
+    "WriterAgent",
+    "CoordinatorAgent",
+    "ModelerAgent",
+]