fix crash KeyError: 'user_feedback_qa'

This commit is contained in:
LeonOstrez
2024-06-28 12:47:19 +01:00
parent 53c29dee7a
commit 86ef2d426b

View File

@@ -57,10 +57,12 @@ class CodeMonkey(BaseAgent):
user_feedback = None
user_feedback_qa = None
llm = self.get_llm()
if iterations:
instructions = iterations[-1]["description"]
user_feedback = iterations[-1]["user_feedback"]
user_feedback_qa = iterations[-1]["user_feedback_qa"]
last_iteration = iterations[-1]
instructions = last_iteration.get("description")
user_feedback = last_iteration.get("user_feedback")
user_feedback_qa = last_iteration.get("user_feedback_qa")
else:
instructions = self.current_state.current_task["instructions"]