delete arg only_code.

This commit is contained in:
刘棒棒 2024-01-18 20:57:43 +08:00
parent f3612f8123
commit d78db8994c

View file

@ -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):