feat: Change the operation of transmitting file content during the QA process to transmitting file names instead.

metagpt/actions/debug_error.py

@@ -5,7 +5,6 @@
 @Author  : alexanderwu
 @File    : debug_error.py
 """
-import re
 
 from metagpt.actions.action import Action
 from metagpt.logs import logger
@@ -36,18 +35,17 @@ class DebugError(Action):
     #     fixed_code = await self._aask(prompt)
     #     return fixed_code
 
-    async def run(self, context):
-        if "PASS" in context:
+    async def run(self, *args, **kwargs) -> str:
+        if "PASS" in self.context.output:
             return "", "the original code works fine, no need to debug"
 
-        file_name = re.search("## File To Rewrite:\s*(.+\\.py)", context).group(1)
-
+        file_name = self.context.code_filename
         logger.info(f"Debug and rewrite {file_name}")
 
-        prompt = PROMPT_TEMPLATE.format(context=context)
+        prompt = PROMPT_TEMPLATE.format(context=self.context.output)
 
         rsp = await self._aask(prompt)
 
         code = CodeParser.parse_code(block="", text=rsp)
 
-        return file_name, code
+        return code

metagpt/actions/run_code.py

@@ -98,24 +98,22 @@ class RunCode(Action):
             stdout, stderr = process.communicate()
         return stdout.decode("utf-8"), stderr.decode("utf-8")
 
-    async def run(
-        self, code, mode="script", code_file_name="", test_code="", test_file_name="", command=[], **kwargs
-    ) -> str:
-        logger.info(f"Running {' '.join(command)}")
-        if mode == "script":
-            outs, errs = await self.run_script(command=command, **kwargs)
-        elif mode == "text":
-            outs, errs = await self.run_text(code=code)
+    async def run(self, *args, **kwargs) -> str:
+        logger.info(f"Running {' '.join(self.context.command)}")
+        if self.context.mode == "script":
+            outs, errs = await self.run_script(command=self.context.command, **kwargs)
+        elif self.context.mode == "text":
+            outs, errs = await self.run_text(code=self.context.code)
 
         logger.info(f"{outs=}")
         logger.info(f"{errs=}")
 
         context = CONTEXT.format(
-            code=code,
-            code_file_name=code_file_name,
-            test_code=test_code,
-            test_file_name=test_file_name,
-            command=" ".join(command),
+            code=self.context.code,
+            code_file_name=self.context.code_filename,
+            test_code=self.context.test_code,
+            test_file_name=self.context.test_filename,
+            command=" ".join(self.context.command),
             outs=outs[:500],  # outs might be long but they are not important, truncate them to avoid token overflow
             errs=errs[:10000],  # truncate errors to avoid token overflow
         )

metagpt/actions/write_test.py

@@ -6,7 +6,9 @@
 @File    : environment.py
 """
 from metagpt.actions.action import Action
+from metagpt.config import CONFIG
 from metagpt.logs import logger
+from metagpt.schema import TestingContext
 from metagpt.utils.common import CodeParser
 
 PROMPT_TEMPLATE = """
@@ -47,12 +49,12 @@ class WriteTest(Action):
             code = code_rsp
         return code
 
-    async def run(self, code_to_test, test_file_name, source_file_path, workspace):
+    async def run(self, *args, **kwargs) -> TestingContext:
         prompt = PROMPT_TEMPLATE.format(
-            code_to_test=code_to_test,
-            test_file_name=test_file_name,
-            source_file_path=source_file_path,
-            workspace=workspace,
+            code_to_test=self.context.code_doc.content,
+            test_file_name=self.context.test_doc.filename,
+            source_file_path=self.context.code_doc.root_relative_path,
+            workspace=CONFIG.git_repo.workdir,
         )
-        code = await self.write_code(prompt)
-        return code
+        self.context.test_doc.content = await self.write_code(prompt)
+        return self.context