fix: fix using time.sleep in async code

Author: adi-wan-askui

src/askui/agent_base.py

@@ -136,8 +136,8 @@ def act(
                 be used for achieving the `goal`.
             on_message (OnMessageCb | None, optional): Callback for new messages. If
                 it returns `None`, stops and does not add the message.
-            tools (list[Tool] | ToolCollection | None, optional): The tools for the agent.
-                Defaults to default tools depending on the selected model.
+            tools (list[Tool] | ToolCollection | None, optional): The tools for the
+                agent. Defaults to default tools depending on the selected model.
             settings (AgentSettings | None, optional): The settings for the agent.
                 Defaults to a default settings depending on the selected model.
 

src/askui/chat/api/runs/runner/runner.py

@@ -4,6 +4,7 @@
 from pathlib import Path
 from typing import TYPE_CHECKING, Literal, Sequence
 
+import anyio
 from anyio.abc import ObjectStream
 from asyncer import asyncify, syncify
 from fastmcp import Client
@@ -113,7 +114,7 @@ async def _run_human_agent(self, send_stream: ObjectStream[Events]) -> None:
         )
         self._agent_os.start_listening()
         screenshot = self._agent_os.screenshot()
-        time.sleep(0.1)
+        await anyio.sleep(0.1)
         recorded_events: list[InputEvent] = []
         while True:
             updated_run = self._retrieve_run()
@@ -159,7 +160,7 @@ async def _run_human_agent(self, send_stream: ObjectStream[Events]) -> None:
                         )
                     )
             screenshot = self._agent_os.screenshot()
-            time.sleep(0.1)
+            await anyio.sleep(0.1)
         self._agent_os.stop_listening()
         if len(recorded_events) == 0:
             text = "Nevermind, I didn't do anything."

src/askui/chat/api/runs/service.py

@@ -64,7 +64,7 @@ async def event_generator() -> AsyncGenerator[Events, None]:
                 # Start the runner in a background task
                 async def run_runner() -> None:
                     try:
-                        await runner.run(send_stream)
+                        await runner.run(send_stream)  # type: ignore[arg-type]
                     finally:
                         await send_stream.aclose()
 

src/askui/models/ui_tars_ep/ui_tars_api.py

@@ -21,9 +21,10 @@
 from askui.models.shared.agent_message_param import MessageParam
 from askui.models.shared.agent_on_message_cb import OnMessageCb
 from askui.models.shared.settings import ActSettings
-from askui.models.shared.tools import ToolCollection
+from askui.models.shared.tools import Tool, ToolCollection
 from askui.models.types.response_schemas import ResponseSchema
 from askui.reporting import Reporter
+from askui.tools.computer import Computer20241022Tool
 from askui.utils.image_utils import ImageSource, image_to_base64
 from askui.utils.pdf_utils import PdfSource
 from askui.utils.source_utils import Source
@@ -228,11 +229,11 @@ def act(
             raise ValueError(error_msg)  # noqa: TRY004
 
         # Find the computer tool
-        computer_tool = None
+        computer_tool: Computer20241022Tool | None = None
         if tools:
             for tool in tools:
                 if tool.name == "computer":
-                    computer_tool = tool
+                    computer_tool: Computer20241022Tool = tool
                     break
 
         if computer_tool is None:
@@ -260,7 +261,7 @@ def act(
         self.execute_act(self.act_history, computer_tool)
 
     def add_screenshot_to_history(
-        self, message_history: list[dict[str, Any]], computer_tool: Tool
+        self, message_history: list[dict[str, Any]], computer_tool: Computer20241022Tool
     ) -> None:
         screenshot = computer_tool(action="screenshot")
         message_history.append(

src/askui/tools/mcp/servers/sse.py

@@ -1,6 +1,8 @@
+from typing import Any
+
 from fastmcp import FastMCP
 
-mcp = FastMCP("Test StdIO MCP App", port=8001)
+mcp: FastMCP[Any] = FastMCP("Test StdIO MCP App", port=8001)
 
 
 @mcp.tool

src/askui/tools/mcp/servers/stdio.py

@@ -1,6 +1,8 @@
+from typing import Any
+
 from fastmcp import FastMCP
 
-mcp = FastMCP("Test StdIO MCP App")
+mcp: FastMCP[Any] = FastMCP("Test StdIO MCP App")
 
 
 @mcp.tool

tests/integration/test_custom_models.py

@@ -22,7 +22,7 @@
 from askui.models.shared.agent_message_param import MessageParam
 from askui.models.shared.agent_on_message_cb import OnMessageCb
 from askui.models.shared.settings import ActSettings
-from askui.models.shared.tools import Tool
+from askui.models.shared.tools import ToolCollection
 from askui.tools.toolbox import AgentToolbox
 from askui.utils.image_utils import ImageSource
 from askui.utils.source_utils import Source
@@ -41,7 +41,7 @@ def act(
         messages: list[MessageParam],
         model_choice: str,
         on_message: OnMessageCb | None = None,
-        tools: list[Tool] | None = None,
+        tools: ToolCollection | None = None,
         settings: ActSettings | None = None,
     ) -> None:
         self.goals.append([message.model_dump(mode="json") for message in messages])
@@ -230,7 +230,7 @@ def act(
                 messages: list[MessageParam],
                 model_choice: str,
                 on_message: OnMessageCb | None = None,
-                tools: list[Tool] | None = None,
+                tools: ToolCollection | None = None,
                 settings: ActSettings | None = None,
             ) -> None:
                 pass