Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
27 changes: 21 additions & 6 deletions src/askui/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,6 @@
__version__ = "0.10.0"

from .agent import VisionAgent
from .android_agent import AndroidVisionAgent
from .locators import Locator
from .models import (
ActModel,
Expand Down Expand Up @@ -38,13 +37,25 @@
from .retry import ConfigurableRetry, Retry
from .tools import ModifierKey, PcKey
from .utils.image_utils import ImageSource, Img
from .web_agent import WebVisionAgent
from .web_testing_agent import WebTestingAgent

try:
from .android_agent import AndroidVisionAgent

_ANDROID_AGENT_AVAILABLE = True
except ImportError:
_ANDROID_AGENT_AVAILABLE = False

try:
from .web_agent import WebVisionAgent
from .web_testing_agent import WebTestingAgent

_WEB_AGENTS_AVAILABLE = True
except ImportError:
_WEB_AGENTS_AVAILABLE = False

__all__ = [
"ActModel",
"ActSettings",
"AndroidVisionAgent",
"Base64ImageSourceParam",
"CacheControlEphemeralParam",
"CitationCharLocationParam",
Expand Down Expand Up @@ -81,6 +92,10 @@
"ToolUseBlockParam",
"UrlImageSourceParam",
"VisionAgent",
"WebTestingAgent",
"WebVisionAgent",
]

if _ANDROID_AGENT_AVAILABLE:
__all__ += ["AndroidVisionAgent"]

if _WEB_AGENTS_AVAILABLE:
__all__ += ["WebVisionAgent", "WebTestingAgent"]