Merge pull request #16 from jihe520/codex/修复拼写错误和代码问题

Fix interpreter restart and improve tests

Author: jihe520

README.md

@@ -24,12 +24,12 @@
 ## ✨ 功能特性
 
 - 🔍 自动分析问题，数学建模，编写代码，纠正错误，撰写论文
-- 💻 Code Interperter
-    - loacl Interperter: 基于 jupyter , 代码保存为 notebook 方便再编辑
-    - 云端 code interperter: [E2B](https://e2b.dev/) 和 [daytona](https://app.daytona.io/)
+- 💻 Code Interpreter
+    - local Interpreter: 基于 jupyter , 代码保存为 notebook 方便再编辑
+    - 云端 code interpreter: [E2B](https://e2b.dev/) 和 [daytona](https://app.daytona.io/)
 - 📝 生成一份编排好格式的论文
-- 🤝 muti-agents: 建模手，代码手，论文手等
-- 🔄 muti-llms: 每个 agent 设置不同的、合适的模型
+- 🤝 multi-agents: 建模手，代码手，论文手等
+- 🔄 multi-llms: 每个 agent 设置不同的、合适的模型
 - 🤖 支持所有模型: [litellm](https://docs.litellm.ai/docs/providers)
 - 💰 成本低：workflow agentless，不依赖 agent 框架
 - 🧩 自定义模板：prompt inject 为每个 subtask 单独设置需求

backend/app/tests/test_cite.py

@@ -0,0 +1,15 @@
+import unittest
+
+from app.utils.common_utils import split_footnotes
+
+
+class TestCommonUtils(unittest.TestCase):
+    def test_split_footnotes(self):
+        text = "Example[^1]\n\n[^1]: Footnote content"
+        main, notes = split_footnotes(text)
+        self.assertEqual(main, "Example")
+        self.assertEqual(notes, [("1", "Footnote content")])
+
+
+if __name__ == "__main__":
+    unittest.main()

backend/app/tests/test_common_utils.py

@@ -1,5 +1,13 @@
+import os
+import asyncio
 import unittest
 
+
+from dotenv import load_dotenv
+
+from app.tools.e2b_interpreter import E2BCodeInterpreter
+from app.utils.common_utils import create_work_dir
+
 try:
     from dotenv import load_dotenv
 except ModuleNotFoundError:  # Fallback if python-dotenv is not installed
@@ -11,21 +19,26 @@ def load_dotenv(*args, **kwargs):
 except ModuleNotFoundError:
     E2BCodeInterpreter = None
 from app.utils.common_utils import create_task_id, create_work_dir
-from app.tools.notebook_serializer import NotebookSerializer
+
 
 
 class TestE2BCodeInterpreter(unittest.TestCase):
     def setUp(self):
         load_dotenv()
+
         if E2BCodeInterpreter is None:
             self.skipTest("e2b_code_interpreter not available")
         _, dirs = create_work_dir("20250312-104132-d3625cab")
         notebook = NotebookSerializer(dirs["jupyter"])
+
         self.code_interpreter = E2BCodeInterpreter(
-            dirs, "20250312-104132-d3625cab", notebook
+            self.task_id, self.work_dir, notebook
         )
 
     def test_execute_code(self):
+        if not os.getenv("E2B_API_KEY"):
+            self.skipTest("E2B_API_KEY not set")
+
         code = """
 import matplotlib.pyplot as plt
 import numpy as np
@@ -48,6 +61,8 @@ def test_execute_code(self):
 plt.legend()
 
 # 显示图像
-plt.show()    
+plt.show()
 """
-        self.code_interpreter.execute_code(code)
+        asyncio.run(self.code_interpreter.initialize())
+        asyncio.run(self.code_interpreter.execute_code(code))
+

backend/app/tests/test_e2b.py

@@ -219,9 +219,14 @@ def send_interrupt_signal(self):
         self.interrupt_signal = True
 
     def restart_jupyter_kernel(self):
-        self.kernel_client.shutdown()
-        self.kernel_manager, self.kernel_client = (
-            jupyter_client.manager.start_new_kernel(kernel_name="python3")
+        """Restart the Jupyter kernel and recreate the work directory."""
+        self.kc.shutdown()
+        self.km, self.kc = jupyter_client.manager.start_new_kernel(
+            kernel_name="python3"
         )
         self.interrupt_signal = False
         self._create_work_dir()
+
+    def _create_work_dir(self):
+        """Ensure the working directory exists after a restart."""
+        os.makedirs(self.work_dir, exist_ok=True)