Improved rendering, fixed screenshots

Author: KillianLucas

interpreter_1/interpreter.py

@@ -262,7 +262,6 @@ async def async_respond(self):
                     elif isinstance(chunk, BetaRawContentBlockDeltaEvent):
                         if chunk.delta.type == "text_delta":
                             md.feed(chunk.delta.text)
-                            yield {"type": "chunk", "chunk": chunk.delta.text}
                             await asyncio.sleep(0)
                             if current_block and current_block.type == "text":
                                 current_block.text += chunk.delta.text
@@ -291,7 +290,6 @@ async def async_respond(self):
                                 delattr(current_block, "partial_json")
                             else:
                                 md.feed("\n")
-                                yield {"type": "chunk", "chunk": "\n"}
                                 await asyncio.sleep(0)
 
                             for attr in [
@@ -305,6 +303,8 @@ async def async_respond(self):
                             response_content.append(current_block)
                             current_block = None
 
+                edit.close()
+
                 response = BetaMessage(
                     id=str(uuid.uuid4()),
                     content=response_content,
@@ -406,8 +406,6 @@ async def async_respond(self):
                 tool_result_content: list[BetaToolResultBlockParam] = []
                 for content_block in cast(list[BetaContentBlock], response.content):
                     if content_block.type == "tool_use":
-                        edit.close()
-
                         if user_approval in ["y", "a"]:
                             result = await tool_collection.run(
                                 name=content_block.name,

interpreter_1/ui/tool.py

@@ -313,8 +313,9 @@ def flush(self):
         pass  # No need to flush since we render when we get a complete command
 
     def close(self):
-        if self.json_obj and self.json_obj.get("action") == "screenshot":
-            SchemaRenderer.print_separator("┴", newline=True)
+        if self.json_obj and self.json_obj.get("name") == "computer":
+            if set(self.json_obj.keys()) == {"name", "action"}:
+                SchemaRenderer.print_separator("┴", newline=True)
 
 
 class InsertRenderer(ContentRenderer):

pyproject.toml

@@ -13,7 +13,7 @@ readme = "README.md"
 [tool.poetry.dependencies]
 
 # Required dependencies
-python = ">=3.8,<4"
+python = ">=3.9,<4"
 litellm = "^1.52.3"
 anthropic = "^0.39.0"
 pygments = "^2.18.0"
@@ -23,6 +23,7 @@ prompt-toolkit = "^3.0.48"
 pyte = "^0.8.2"
 screeninfo = "^0.8.1"
 readchar = "^4.2.1"
+pillow = "^11.0.0"
 
 [build-system]
 requires = ["poetry-core>=1.0.0"]