From d1aef99ef4f3beff7265fb5e76faef952164a937 Mon Sep 17 00:00:00 2001 From: Mandlin Sarah Date: Thu, 15 Aug 2024 20:08:46 -0700 Subject: [PATCH] Add exception logging to improve debugging in BlockAgent --- rnd/autogpt_server/autogpt_server/blocks/agent.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/rnd/autogpt_server/autogpt_server/blocks/agent.py b/rnd/autogpt_server/autogpt_server/blocks/agent.py index 5f6f32140214..a919668cf69f 100644 --- a/rnd/autogpt_server/autogpt_server/blocks/agent.py +++ b/rnd/autogpt_server/autogpt_server/blocks/agent.py @@ -33,7 +33,7 @@ class BlockAgentSettings(AgentSettings): class OutputComponent(CommandProvider): - def get_commands(self) -> Iterator[Command]: + def get_commands() -> Iterator[Command]: yield self.output @command( @@ -74,7 +74,6 @@ def __init__( ): delattr(self, attr_name) - class AutoGPTAgentBlock(Block): class Input(BlockSchema): task: str = SchemaField( @@ -151,6 +150,7 @@ def get_result(agent: BlockAgent) -> str: result = asyncio.run(agent.execute(proposal)) return str(result) except Exception as e: + logger.error(f"Error during agent action execution: {e}") # Detailed exception logging error = e raise error or Exception("Failed to get result") @@ -188,3 +188,4 @@ def run(self, input_data: Input) -> BlockOutput: result = self.get_result(agent) yield "result", result +