python-sdk/src/askui/__init__.py at 3493c4ac7c3d8f2aa967ec55d5e2d61bd1122bed · askui/python-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
"""AskUI Python SDK"""

__version__ = "0.33.0"

import logging
import os

os.environ["FASTMCP_EXPERIMENTAL_ENABLE_NEW_OPENAPI_PARSER"] = "true"

from .agent_base import Agent
from .agent_settings import AgentSettings
from .callbacks import ConversationCallback
from .computer_agent import ComputerAgent, VisionAgent
from .locators import Locator
from .models import (
    Base64ImageSourceParam,
    CacheControlEphemeralParam,
    CitationCharLocationParam,
    CitationContentBlockLocationParam,
    CitationPageLocationParam,
    ContentBlockParam,
    ImageBlockParam,
    MessageParam,
    OnMessageCb,
    OnMessageCbParam,
    Point,
    PointList,
    TextBlockParam,
    TextCitationParam,
    ToolResultBlockParam,
    ToolUseBlockParam,
    UrlImageSourceParam,
)
from .models.exceptions import AutomationError
from .models.shared.settings import (
    DEFAULT_GET_RESOLUTION,
    DEFAULT_LOCATE_RESOLUTION,
    ActSettings,
    GetSettings,
    LocateSettings,
    MessageSettings,
    Resolution,
)
from .models.shared.tools import Tool
from .models.types.response_schemas import ResponseSchema, ResponseSchemaBase
from .retry import ConfigurableRetry, Retry
from .tools import ModifierKey, PcKey
from .tools.askui import LocalAgentOsServer, RemoteAgentOsServer
from .utils.image_utils import ImageSource
from .utils.source_utils import InputSource

try:
    from .android_agent import AndroidAgent, AndroidVisionAgent

    _ANDROID_AGENT_AVAILABLE = True
except ImportError:
    _ANDROID_AGENT_AVAILABLE = False

if _ANDROID_AGENT_AVAILABLE:
    from .multi_device_agent import MultiDeviceAgent

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

    _WEB_AGENTS_AVAILABLE = True
except ImportError:
    _WEB_AGENTS_AVAILABLE = False

logging.getLogger(__name__).addHandler(logging.NullHandler())

__all__ = [
    "RemoteAgentOsServer",
    "LocalAgentOsServer",
    "Agent",
    "AutomationError",
    "ComputerAgent",
    "VisionAgent",
    "AgentSettings",
    "ActSettings",
    "Base64ImageSourceParam",
    "CacheControlEphemeralParam",
    "CitationCharLocationParam",
    "CitationContentBlockLocationParam",
    "CitationPageLocationParam",
    "ConfigurableRetry",
    "ContentBlockParam",
    "ConversationCallback",
    "DEFAULT_GET_RESOLUTION",
    "DEFAULT_LOCATE_RESOLUTION",
    "GetSettings",
    "ImageBlockParam",
    "ImageSource",
    "InputSource",
    "Locator",
    "LocateSettings",
    "MessageParam",
    "MessageSettings",
    "ModifierKey",
    "OnMessageCb",
    "OnMessageCbParam",
    "PcKey",
    "Point",
    "PointList",
    "Resolution",
    "ResponseSchema",
    "ResponseSchemaBase",
    "Retry",
    "TextBlockParam",
    "TextCitationParam",
    "Tool",
    "ToolResultBlockParam",
    "ToolUseBlockParam",
    "UrlImageSourceParam",
]

if _ANDROID_AGENT_AVAILABLE:
    __all__ += ["AndroidAgent", "AndroidVisionAgent", "MultiDeviceAgent"]

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