look for DONE in the bigger chunk of the response

This commit is contained in:
Zvonimir Sabljic
2024-07-28 19:16:40 -07:00
parent b9cd3a8fe8
commit 7970dc8772

View File

@@ -47,7 +47,7 @@ class TaskReviewer(BaseAgent):
)
llm_response: str = await llm(convo, temperature=0.7)
if "done" in llm_response.strip().lower()[-7:]:
if "done" in llm_response.strip().lower()[-20:]:
return AgentResponse.done(self)
else:
return AgentResponse.task_review_feedback(self, llm_response)