fix: add VisionAgent and AndroidVisionAgent aliases for ComputerAgent and AndroidAgent to ensure backward compatibility

Author: philipph-askui

src/askui/__init__.py

@@ -7,7 +7,7 @@
 
 os.environ["FASTMCP_EXPERIMENTAL_ENABLE_NEW_OPENAPI_PARSER"] = "true"
 
-from .agent import ComputerAgent
+from .agent import ComputerAgent, VisionAgent
 from .agent_base import Agent
 from .agent_settings import AgentSettings
 from .locators import Locator
@@ -47,7 +47,7 @@
 from .utils.source_utils import InputSource
 
 try:
-    from .android_agent import AndroidAgent
+    from .android_agent import AndroidAgent, AndroidVisionAgent
 
     _ANDROID_AGENT_AVAILABLE = True
 except ImportError:
@@ -66,6 +66,7 @@
 __all__ = [
     "Agent",
     "ComputerAgent",
+    "VisionAgent",
     "AgentSettings",
     "ActSettings",
     "Base64ImageSourceParam",
@@ -104,7 +105,7 @@
 ]
 
 if _ANDROID_AGENT_AVAILABLE:
-    __all__ += ["AndroidAgent"]
+    __all__ += ["AndroidAgent", "AndroidVisionAgent"]
 
 if _WEB_AGENTS_AVAILABLE:
     __all__ += ["WebVisionAgent", "WebTestingAgent"]

src/askui/agent.py

@@ -1,4 +1,5 @@
 import logging
+import warnings
 from typing import Annotated, Literal, Optional
 
 from pydantic import ConfigDict, Field, validate_call
@@ -501,3 +502,13 @@ def cli(
             "ComputerAgent received instruction to execute '%s' on cli", command
         )
         self.tools.os.run_command(command)
+
+
+class VisionAgent(ComputerAgent):
+    def __init__(self, *args, **kwargs) -> None:  # type: ignore
+        warnings.warn(
+            "VisionAgent is deprecated, use ComputerAgent instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        super().__init__(*args, **kwargs)

src/askui/android_agent.py

@@ -1,4 +1,5 @@
 import logging
+import warnings
 from typing import Annotated, overload
 
 from pydantic import ConfigDict, Field, validate_call
@@ -352,3 +353,13 @@ def set_device_by_serial_number(
             f"set_device_by_serial_number(device_sn='{device_sn}')",
         )
         self.os.set_device_by_serial_number(device_sn)
+
+
+class AndroidVisionAgent(AndroidAgent):
+    def __init__(self, *args, **kwargs) -> None:  # type: ignore
+        warnings.warn(
+            "AndroidVisionAgent is deprecated, use AndroidAgent instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        super().__init__(*args, **kwargs)