computer-use-ootb-internal 0.0.145__py3-none-any.whl → 0.0.147__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

@@ -382,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")
 
@@ -389,7 +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.5)
+                time.sleep(0.25)
                 x0, y0 = pyautogui.position()
                 # x0, y0 = self.scale_coordinates(ScalingSource.COMPUTER, x0, y0)
                 x1 = coordinate[0]+self.offset_x
@@ -397,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")