From d78db8994c6cb6c05f40c30891987358dcafd242 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=88=98=E6=A3=92=E6=A3=92?= <liubangbang@fuzhi.ai>
Date: Thu, 18 Jan 2024 20:57:43 +0800
Subject: [PATCH] delete arg only_code.

---
 metagpt/actions/write_analysis_code.py | 10 ++--------
 1 file changed, 2 insertions(+), 8 deletions(-)

diff --git a/metagpt/actions/write_analysis_code.py b/metagpt/actions/write_analysis_code.py
index bceb100b1..9104fdf82 100644
--- a/metagpt/actions/write_analysis_code.py
+++ b/metagpt/actions/write_analysis_code.py
@@ -85,17 +85,11 @@ class WriteCodeByGenerate(BaseWriteAnalysisCode):
         plan: Plan = None,
         system_msg: str = None,
         **kwargs,
-    ) -> str:
+    ) -> dict:
         # context.append(Message(content=self.REUSE_CODE_INSTRUCTION, role="user"))
         prompt = self.process_msg(context, system_msg)
-        is_only_code = kwargs.pop("only_code", False)
-
         code_content = await self.llm.aask_code(prompt, **kwargs)
-        if is_only_code:
-            return code_content["code"]
-        else:
-            return code_content
-
+        return code_content
 
 
 class WriteCodeWithTools(BaseWriteAnalysisCode):