apunkt/MetaGPT
1
0
Fork 0
mirror of https://github.com/FoundationAgents/MetaGPT.git synced 2026-06-02 14:45:17 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

update JUDGE_PROMPT_TEMPLATE in ml_engineer_simple.py

Browse source
This commit is contained in:
mannaandpoem 2023-12-19 18:50:49 +08:00
parent 5db0063342
commit 0f3c0c21e5
1 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

10
metagpt/roles/ml_engineer_simple.py
View file

@ -32,10 +32,10 @@ JUDGE_PROMPT_TEMPLATE = """
# State
Output "Ture" or "False". Judging from the code perspective, whether the user's needs have been completely fulfilled.
=====
# Finally output State, Thought and Next Action separately in one sentence
# Output State("Ture" or "False") firstly, then output Thought and Next Steps for the code requirements based on the context respectively in one sentence
State:
Thought:
Next Action:
Next Steps:
"""
@ -132,10 +132,10 @@ class MLEngineerSimple(Role):
)
context_msg = [Message(content=context, role="user")]
return context_msg + self.get_working_memories()
return context_msg + self.get_working_memories(6)
def get_working_memories(self, num=6) -> List[Message]:
return self.working_memory.get(num) # 默认为6
def get_working_memories(self, num=0) -> List[Message]:
return self.working_memory.get(num) # 默认为6
if __name__ == "__main__":