From 45700d77ab3e93c6e684131ebe7ed2a50b17c92c Mon Sep 17 00:00:00 2001 From: martin legrand Date: Fri, 25 Apr 2025 21:54:36 +0200 Subject: [PATCH] fix : was showing code after execution --- sources/agents/agent.py | 9 +++++++++ sources/agents/code_agent.py | 2 +- sources/schemas.py | 7 +++---- 3 files changed, 13 insertions(+), 5 deletions(-) diff --git a/sources/agents/agent.py b/sources/agents/agent.py index 0e69648..4e10c59 100644 --- a/sources/agents/agent.py +++ b/sources/agents/agent.py @@ -213,6 +213,14 @@ class Agent(): post_lines.append(f"block:{block_idx}") block_idx += 1 return "\n".join(post_lines) + + def show_block(self, block: str) -> None: + """ + Show the block in a pretty way. + """ + pretty_print('▂'*64, color="status") + pretty_print(block, color="code") + pretty_print('▂'*64, color="status") def execute_modules(self, answer: str) -> Tuple[bool, str]: """ @@ -231,6 +239,7 @@ class Agent(): if blocks != None: for block in blocks: + self.show_block(block) output = tool.execute([block]) feedback = tool.interpreter_feedback(output) # tool interpreter feedback success = not tool.execution_failure_check(output) diff --git a/sources/agents/code_agent.py b/sources/agents/code_agent.py index dd133ab..8a7f8b4 100644 --- a/sources/agents/code_agent.py +++ b/sources/agents/code_agent.py @@ -56,6 +56,7 @@ class CoderAgent(Agent): self.last_answer = answer await asyncio.sleep(0) break + self.show_answer() animate_thinking("Executing code...", color="status") self.status_message = "Executing code..." exec_success, _ = self.execute_modules(answer) @@ -67,7 +68,6 @@ class CoderAgent(Agent): pretty_print("Execution failure", color="failure") pretty_print("Correcting code...", color="status") self.status_message = "Correcting code..." - self.show_answer() attempt += 1 self.status_message = "Ready" if attempt == max_attempts: diff --git a/sources/schemas.py b/sources/schemas.py index ad72830..29410ba 100644 --- a/sources/schemas.py +++ b/sources/schemas.py @@ -68,9 +68,8 @@ class executorResult: "success": self.success, "tool_type": self.tool_type } - + def show(self): pretty_print('▂'*64, color="status") - pretty_print(self.block, color="code" if self.success else "failure") - pretty_print('▂'*64, color="status") - pretty_print(self.feedback, color="success" if self.success else "failure") \ No newline at end of file + pretty_print(self.feedback, color="success" if self.success else "failure") + pretty_print('▂'*64, color="status") \ No newline at end of file