Bug fix

plasma-umass · Mar 20, 2024 · 5f765d7 · 5f765d7
1 parent dfa632c
commit 5f765d7
Showing 1 changed file with 13 additions and 12 deletions.
diff --git a/src/chatdbg/assistant/lite_assistant.py b/src/chatdbg/assistant/lite_assistant.py
@@ -161,19 +161,20 @@ def run(
 
                 if choice.finish_reason == "tool_calls":
                     responses = []
-                    for tool_call in choice.message.tool_calls:
-                        try:
+                    try:
+                        for tool_call in choice.message.tool_calls:
                             function_response = self._make_call(tool_call)
-                        except Exception as e:
-                            function_response = f"Error: {e}"
-                        response = {
-                            "tool_call_id": tool_call.id,
-                            "role": "tool",
-                            "name": tool_call.function.name,
-                            "content": function_response,
-                        }
-                        responses.append(response)
-                        self._print_message(response, 4, append_message)
+                            response = {
+                                "tool_call_id": tool_call.id,
+                                "role": "tool",
+                                "name": tool_call.function.name,
+                                "content": function_response,
+                            }
+                            responses.append(response)
+                            self._print_message(response, 4, append_message)
+                    except Exception as e:
+                        # Warning: potential infinite loop.
+                        append_warning(f"error processing tool calls.")
                     self._conversation.append(choice.message)
                     self._conversation.extend(responses)
                 elif choice.finish_reason == "stop":