apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-05-15 11:02:36 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

resolve CR in MR17.

Browse source
This commit is contained in:
刘棒棒 2023-12-26 14:08:10 +08:00
parent 7806013dce
commit be47f6171d
1 changed files with 9 additions and 14 deletions
Download patch file Download diff file Expand all files Collapse all files

23
metagpt/roles/ml_engineer.py
View file

@ -168,22 +168,16 @@ class MLEngineer(Role):
\n{json.dumps(list(UDFS_YAML.keys()), indent=2, ensure_ascii=False)}")
# set task_type to `udf`
self.plan.current_task.task_type = 'udf'
tool_context, code = await WriteCodeWithTools(schema_path=UDFS_YAML).run(
context=context,
plan=self.plan,
column_info=self.data_desc.get("column_info", ""),
)
debug_context = tool_context
cause_by = WriteCodeWithTools
schema_path = UDFS_YAML
else:
schema_path = PROJECT_ROOT / "metagpt/tools/functions/schemas"
tool_context, code = await WriteCodeWithTools(schema_path=schema_path).run(
context=context,
plan=self.plan,
column_info=self.data_desc.get("column_info", ""),
)
debug_context = tool_context
cause_by = WriteCodeWithTools
tool_context, code = await WriteCodeWithTools(schema_path=schema_path).run(
context=context,
plan=self.plan,
column_info=self.data_desc.get("column_info", ""),
)
debug_context = tool_context
cause_by = WriteCodeWithTools
self.working_memory.add(
Message(content=code, role="assistant", cause_by=cause_by)
)
@ -301,6 +295,7 @@ class MLEngineer(Role):
# tool_code = await make_tools.run(code_prompt)
tool_code = await make_tools.run(code)
_, success = await self.execute_code.run(tool_code)
make_tool_retries += 1
if make_tool_current_retry > make_tool_retries:
logger.error(f"We have tried the maximum number of attempts {make_tool_retries}\
and still have not created tools for task_id {self.plan.current_task_id} successfully,\