add computer connect tool for chat

Author: mlikasam-askui

src/askui/chat/api/mcp_servers/computer.py

@@ -2,6 +2,7 @@
 
 from askui.tools.askui.askui_controller import AskUiControllerClient
 from askui.tools.computer import (
+    ComputerConnectTool,
     ComputerDisconnectTool,
     ComputerGetMousePositionTool,
     ComputerKeyboardPressedTool,
@@ -13,7 +14,6 @@
     ComputerMouseReleaseTool,
     ComputerMouseScrollTool,
     ComputerMoveMouseTool,
-    ComputerReconnectTool,
     ComputerRetrieveActiveDisplayTool,
     ComputerScreenshotTool,
     ComputerSetActiveDisplayTool,
@@ -41,7 +41,7 @@
     ComputerScreenshotTool(COMPUTER_AGENT_OS_FACADE),
     ComputerSetActiveDisplayTool(COMPUTER_AGENT_OS_FACADE),
     ComputerTypeTool(COMPUTER_AGENT_OS_FACADE),
-    ComputerReconnectTool(COMPUTER_AGENT_OS_FACADE),
+    ComputerConnectTool(COMPUTER_AGENT_OS_FACADE),
     ComputerDisconnectTool(COMPUTER_AGENT_OS_FACADE),
 ]
 

src/askui/chat/migrations/shared/assistants/seeds.py

@@ -25,7 +25,7 @@
     ),
     tools=[
         "computer_disconnect",
-        "computer_reconnect",
+        "computer_connect",
         "computer_mouse_click",
         "computer_get_mouse_position",
         "computer_keyboard_pressed",

src/askui/tools/computer/__init__.py

@@ -1,3 +1,4 @@
+from .connect_tool import ComputerConnectTool
 from .disconnect_tool import ComputerDisconnectTool
 from .get_mouse_position_tool import ComputerGetMousePositionTool
 from .keyboard_pressed_tool import ComputerKeyboardPressedTool
@@ -9,14 +10,13 @@
 from .mouse_release_tool import ComputerMouseReleaseTool
 from .mouse_scroll_tool import ComputerMouseScrollTool
 from .move_mouse_tool import ComputerMoveMouseTool
-from .reconnect_tool import ComputerReconnectTool
 from .retrieve_active_display_tool import ComputerRetrieveActiveDisplayTool
 from .screenshot_tool import ComputerScreenshotTool
 from .set_active_display_tool import ComputerSetActiveDisplayTool
 from .type_tool import ComputerTypeTool
 
 __all__ = [
-    "ComputerReconnectTool",
+    "ComputerConnectTool",
     "ComputerDisconnectTool",
     "ComputerGetMousePositionTool",
     "ComputerKeyboardPressedTool",

src/askui/tools/computer/connect_tool.py

@@ -0,0 +1,27 @@
+from askui.models.shared import ComputerBaseTool
+from askui.tools.computer_agent_os_facade import ComputerAgentOsFacade
+
+
+class ComputerConnectTool(ComputerBaseTool):
+    """Computer Connect Tool"""
+
+    def __init__(self, agent_os: ComputerAgentOsFacade | None = None) -> None:
+        super().__init__(
+            name="computer_connect",
+            description=(
+                "Connect to the agent OS controller to enable computer control. "
+                "Useful for establishing an initial connection or reconnecting "
+                "after controller errors. If already connected, this will "
+                "disconnect and reconnect, which may cause previous configuration "
+                "to be lost (e.g., selected display must be reconfigured)."
+                "Must be used once before any other computer tools are used."
+            ),
+            agent_os=agent_os,
+        )
+
+    def __call__(self) -> str:
+        try:
+            self.agent_os.disconnect()
+        finally:
+            self.agent_os.connect()
+            return "Agent OS controller was connected."  # noqa: B012