chore: fix linting and typing issues

Author: adi-wan-askui

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

@@ -150,7 +150,7 @@ async def async_on_message(
 
         on_message = syncify(async_on_message)
 
-        mcp_client = await self._mcp_client_manager_manager.get_mcp_client_manager(  # type: ignore
+        mcp_client = await self._mcp_client_manager_manager.get_mcp_client_manager(
             self._workspace_id
         )
 
@@ -159,7 +159,7 @@ def _run_agent_inner() -> None:
                 mcp_client=mcp_client,
                 include=set(self._assistant.tools),
             )
-            # TODO Remove this after having extracted tools into Android MCP
+            # Remove this after having extracted tools into Android MCP
             if self._run.assistant_id == ANDROID_AGENT.id:
                 tools.append_tool(*_get_android_tools())
             custom_agent = CustomAgent()

src/askui/models/shared/tools.py

@@ -1,3 +1,4 @@
+import json
 import types
 from abc import ABC, abstractmethod
 from datetime import timedelta
@@ -154,7 +155,7 @@ async def __aexit__(
 
 def _replace_refs(tool_name: str, input_schema: InputSchema) -> InputSchema:
     try:
-        return jsonref.replace_refs(
+        return jsonref.replace_refs(  # type: ignore[no-any-return]
             input_schema,
             lazy_load=False,
             proxies=False,