From 05504851758fccf4ac59936329b2333bc6a48806 Mon Sep 17 00:00:00 2001 From: codegen-bot Date: Wed, 12 Mar 2025 23:07:54 +0000 Subject: [PATCH] Fix IndexError in CodeAgent.run method when message.content is empty --- src/codegen/agents/code_agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/codegen/agents/code_agent.py b/src/codegen/agents/code_agent.py index 99c01a874..e27f485da 100644 --- a/src/codegen/agents/code_agent.py +++ b/src/codegen/agents/code_agent.py @@ -120,7 +120,7 @@ def run(self, prompt: str, thread_id: Optional[str] = None) -> str: if isinstance(message, tuple): print(message) else: - if isinstance(message, AIMessage) and isinstance(message.content, list) and "text" in message.content[0]: + if isinstance(message, AIMessage) and isinstance(message.content, list) and len(message.content) > 0 and "text" in message.content[0]: AIMessage(message.content[0]["text"]).pretty_print() else: message.pretty_print()