Update final version no debug

didiforgithub · Apr 11, 2024 · 23ed1b6 · 23ed1b6
1 parent ef8d266
commit 23ed1b6
Show file tree

Hide file tree

Showing 7 changed files with 12 additions and 6,326 deletions.
diff --git a/MathAI-README.md b/MathAI-README.md
@@ -0,0 +1,9 @@
+# 配置
+1. 首先按照README.md文件，完成MetaGPT项目配置
+2. 进入工作目录，在`math_ai/codebase/engine/llm.py` 中，配置`apikey`与`base_url`
+3. 进入工作目录，进入`run.py`填入JSON文件路径，运行 `python run.py`
+
+# JSON 题目文件
+JSON 题目文件将由赛事组提供的LaTex格式赛题生成，如需复现，我们将在赛后提供处理好的JSON格式题目，或提供数据处理的程序。
+
+在提交的代码方案中，我们使用历史赛题生成的JSON数据进行测试。
diff --git a/math_ai/codebase/gate_controller.py b/math_ai/codebase/gate_controller.py
@@ -46,7 +46,7 @@ def run(self, problem: Dict) -> Dict:
         }
         """
         prob = problem['desc']
-        response = json.loads(self.llm.llm_response(prob))
+        response = json.loads(self.llm.llm_response(prob, json_mode=True))
         problem['if_muti'] = 'simple' if 'simple' in response['if_muti'] else 'muti'
         problem['strategy'] = response['type']
         return problem

diff --git a/math_ai/dataset/2021.json b/math_ai/dataset/2021.json
diff --git a/math_ai/dataset/math/level_5.jsonl b/math_ai/dataset/math/level_5.jsonl
diff --git a/math_ai/dataset/math/test.json b/math_ai/dataset/math/test.json
diff --git a/math_ai/codebase/run.py → run.py b/math_ai/codebase/run.py → run.py
@@ -39,5 +39,5 @@ async def main(question_path: str):
     print(solutions)
 
 if __name__ == '__main__':
-    question_path = '/Users/mac/Github_project/MathAI/math_ai/dataset/dataset.json'
+    question_path = '/Users/mac/Github_project/MathAI/math_ai/dataset/2021.json'
     asyncio.run(main(question_path))
diff --git a/test.py b/test.py