computer-use-ootb-internal 0.0.144__py3-none-any.whl → 0.0.146__py3-none-any.whl

computer_use_ootb_internal/computer_use_demo/animation/test_animation.py

@@ -1,40 +1,40 @@
-"""
-Test script to verify cursor animation is working
-"""
-import asyncio
-import sys
-import time
-from pathlib import Path
-from computer_use_ootb_internal.computer_use_demo.tools.computer import ComputerTool
-
-async def test_animations():
-    
-    # Initialize the computer tool
-    computer = ComputerTool()
-    
-    # Test mouse move animation
-    print("Testing mouse move animation...")
-    await computer(action="mouse_move_windll", coordinate=(500, 500))
-    print("Waiting 2 seconds...")
-    await asyncio.sleep(2)
-    
-    # Test click animation
-    print("Testing click animation...")
-    await computer(action="left_click_windll", coordinate=(700, 300))
-    print("Waiting 2 seconds...")
-    await asyncio.sleep(2)
-    
-    # Test another move
-    print("Testing move and click sequence...")
-    await computer(action="mouse_move_windll", coordinate=(300, 300))
-    await asyncio.sleep(1)
-    await computer(action="left_click_windll", coordinate=(300, 300))
-    
-    # Wait for animations to complete
-    print("Waiting for animations to complete...")
-    await asyncio.sleep(3)
-    
-    print("Test completed")
-
-if __name__ == "__main__":
+"""
+Test script to verify cursor animation is working
+"""
+import asyncio
+import sys
+import time
+from pathlib import Path
+from computer_use_ootb_internal.computer_use_demo.tools.computer import ComputerTool
+
+async def test_animations():
+    
+    # Initialize the computer tool
+    computer = ComputerTool()
+    
+    # Test mouse move animation
+    print("Testing mouse move animation...")
+    await computer(action="mouse_move_windll", coordinate=(500, 500))
+    print("Waiting 2 seconds...")
+    await asyncio.sleep(2)
+    
+    # Test click animation
+    print("Testing click animation...")
+    await computer(action="left_click_windll", coordinate=(700, 300))
+    print("Waiting 2 seconds...")
+    await asyncio.sleep(2)
+    
+    # Test another move
+    print("Testing move and click sequence...")
+    await computer(action="mouse_move_windll", coordinate=(300, 300))
+    await asyncio.sleep(1)
+    await computer(action="left_click_windll", coordinate=(300, 300))
+    
+    # Wait for animations to complete
+    print("Waiting for animations to complete...")
+    await asyncio.sleep(3)
+    
+    print("Test completed")
+
+if __name__ == "__main__":
     asyncio.run(test_animations()) 

computer_use_ootb_internal/computer_use_demo/tools/computer.py

@@ -374,6 +374,7 @@ class ComputerTool(BaseAnthropicTool):
         if action in ("left_click_windll", "mouse_move_windll", "right_click_windll", "key_down_windll", "key_up_windll"):
             if action == "left_click_windll":
                 if coordinate is None:
+                    time.sleep(0.5)
                     x, y = pyautogui.position()
                     show_click(x, y)
                     self.marbot_auto_gui.click()
@@ -381,6 +382,8 @@ class ComputerTool(BaseAnthropicTool):
                     x = coordinate[0]+self.offset_x
                     y = coordinate[1]+self.offset_y
                     show_click(x, y)
+                    time.sleep(0.25)
+                    self.marbot_auto_gui.click(x=x, y=y)
                     self.marbot_auto_gui.click(x=x, y=y)
                 return ToolResult(output=f"Left click", action_base_type="click")
 
@@ -388,6 +391,7 @@ class ComputerTool(BaseAnthropicTool):
                 if coordinate is None:
                     raise ToolError(output=f"coordinate is required for {action}", action_base_type="error")
                 
+                time.sleep(0.25)
                 x0, y0 = pyautogui.position()
                 # x0, y0 = self.scale_coordinates(ScalingSource.COMPUTER, x0, y0)
                 x1 = coordinate[0]+self.offset_x
@@ -395,6 +399,7 @@ class ComputerTool(BaseAnthropicTool):
 
                 show_move_to(x0, y0, x1, y1)
                 self.marbot_auto_gui.moveTo(x=x1, y=y1)
+                time.sleep(0.5)
             
                 return ToolResult(output=f"Mouse move", action_base_type="move")
             
@@ -402,9 +407,11 @@ class ComputerTool(BaseAnthropicTool):
             #     self.marbot_auto_gui.rightClick(x=coordinate[0], y=coordinate[1])
             elif action == "key_down_windll":
                 self.marbot_auto_gui.keyDown(text)
+                time.sleep(0.5)
                 return ToolResult(output=f"Key down '{text}'", type="hidden", action_base_type="key")
             
             elif action == "key_up_windll":
+                time.sleep(0.5)
                 self.marbot_auto_gui.keyUp(text)
                 return ToolResult(output=f"Key up '{text}'", type="hidden", action_base_type="key")