Bug fix

plasma-umass · Mar 20, 2024 · 9d8d6cc · 9d8d6cc
1 parent 5f765d7
commit 9d8d6cc
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/chatdbg/assistant/lite_assistant.py b/src/chatdbg/assistant/lite_assistant.py
@@ -172,11 +172,11 @@ def run(
                             }
                             responses.append(response)
                             self._print_message(response, 4, append_message)
+                        self._conversation.append(choice.message)
+                        self._conversation.extend(responses)
                     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":
                     break
                 else: