From 5269aac4c0b2601c202d75a08c346cc91f4412f4 Mon Sep 17 00:00:00 2001 From: Rohit Vinnakota <148245014+rohitvinnakota-codecov@users.noreply.github.com> Date: Fri, 20 Sep 2024 14:00:59 -0400 Subject: [PATCH] Updates from editor --- src/seer/automation/agent/agent.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/src/seer/automation/agent/agent.py b/src/seer/automation/agent/agent.py index 52be31dc8..374133dcd 100644 --- a/src/seer/automation/agent/agent.py +++ b/src/seer/automation/agent/agent.py @@ -63,6 +63,7 @@ def get_completion(self): def run_iteration(self, context: Optional[AutofixContext] = None): logger.debug(f"----[{self.name}] Running Iteration {self.iterations}----") + message, usage = self.get_completion() self.memory.append(message) @@ -126,6 +127,7 @@ def run(self, prompt: str, context: Optional[AutofixContext] = None): raise MaxIterationsReachedException( f"Agent {self.name} reached maximum iterations without finishing." ) + return self.get_last_message_content() def reset_iterations(self):