Improve tool output handling

OpenInterpreter · Dec 10, 2024 · 19b9a9d · 19b9a9d
1 parent d437f1d
commit 19b9a9d
Showing 1 changed file with 19 additions and 13 deletions.
diff --git a/interpreter/interpreter.py b/interpreter/interpreter.py
@@ -896,20 +896,15 @@ async def async_respond(self, user_input=None):
                             output = result.error
                         else:
                             output = result.output
-                        self.messages.append(
-                            {
-                                "role": "tool",
-                                "content": output,
-                                "tool_call_id": tool_call.id,
-                            }
-                        )
+
+                        tool_output = ""
+
+                        if output:
+                            tool_output += output
+
                         if result.base64_image:
-                            self.messages.append(
-                                {
-                                    "role": "tool",
-                                    "content": "The user will reply with the tool's image output.",
-                                    "tool_call_id": tool_call.id,
-                                }
+                            tool_output += (
+                                "\nThe user will reply with the tool's image output."
                             )
                             user_content_to_add.append(
                                 {
@@ -919,6 +914,17 @@ async def async_respond(self, user_input=None):
                                     },
                                 }
                             )
+
+                        if tool_output == "":
+                            tool_output = "No output from tool."
+
+                        self.messages.append(
+                            {
+                                "role": "tool",
+                                "content": tool_output.strip(),
+                                "tool_call_id": tool_call.id,
+                            }
+                        )
                     else:
                         text_content = (
                             "This was the output of the tool call. What does it mean/what's next?\n"