Skip to content

Commit daa65ef

Browse files
refactor: clean up import statements and formatting
- Reformatted import statements in `agent.py`, `agent_os.py`, and `askui_controller.py` for improved readability. - Removed unnecessary blank lines to maintain consistent formatting across files.
1 parent 9ae1b02 commit daa65ef

File tree

3 files changed

+44
-24
lines changed

3 files changed

+44
-24
lines changed

src/askui/agent.py

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,12 @@
1010
from askui.agent_base import AgentBase
1111
from askui.container import telemetry
1212
from askui.locators.locators import Locator
13-
from askui.models.shared.settings import (COMPUTER_USE_20241022_BETA_FLAG,
14-
COMPUTER_USE_20250124_BETA_FLAG,
15-
ActSettings, MessageSettings)
13+
from askui.models.shared.settings import (
14+
COMPUTER_USE_20241022_BETA_FLAG,
15+
COMPUTER_USE_20250124_BETA_FLAG,
16+
ActSettings,
17+
MessageSettings,
18+
)
1619
from askui.models.shared.tools import Tool
1720
from askui.tools.computer import Computer20241022Tool, Computer20250124Tool
1821
from askui.tools.exception_tool import ExceptionTool
@@ -399,7 +402,6 @@ def _get_default_settings_for_act(self, model_choice: str) -> ActSettings:
399402

400403
@override
401404
def _get_default_tools_for_act(self, model_choice: str) -> list[Tool]:
402-
403405
self._tools.append(ScreenSwitchTool(agent_os=self.tools.os))
404406

405407
match model_choice:

src/askui/tools/agent_os.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,8 +5,9 @@
55
from pydantic import BaseModel, Field
66

77
if TYPE_CHECKING:
8-
from askui.tools.askui.askui_ui_controller_grpc.generated.AgentOS_Send_Request_2501 import \
9-
RenderObjectStyle # noqa: E501
8+
from askui.tools.askui.askui_ui_controller_grpc.generated.AgentOS_Send_Request_2501 import (
9+
RenderObjectStyle,
10+
) # noqa: E501
1011

1112

1213
ModifierKey = Literal[
@@ -159,6 +160,7 @@ class SizeInPixels(BaseModel):
159160
width: int
160161
height: int
161162

163+
162164
class DisplayInformation(BaseModel):
163165
"""Contains information about a single display."""
164166

@@ -171,6 +173,7 @@ class GetDisplayInformationResponse(BaseModel):
171173

172174
displays: list[DisplayInformation]
173175

176+
174177
class Coordinate(BaseModel):
175178
x: int
176179
y: int

src/askui/tools/askui/askui_controller.py

Lines changed: 33 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -16,28 +16,43 @@
1616
from askui.container import telemetry
1717
from askui.logger import logger
1818
from askui.reporting import Reporter
19-
from askui.tools.agent_os import (AgentOs, Coordinate,
20-
GetDisplayInformationResponse, ModifierKey,
21-
PcKey)
22-
from askui.tools.askui.askui_ui_controller_grpc.generated import \
23-
Controller_V1_pb2 as controller_v1_pbs
24-
from askui.tools.askui.askui_ui_controller_grpc.generated import \
25-
Controller_V1_pb2_grpc as controller_v1
26-
from askui.tools.askui.askui_ui_controller_grpc.generated.AgentOS_Send_Request_2501 import \
27-
RenderObjectStyle # noqa: E501
28-
from askui.tools.askui.askui_ui_controller_grpc.generated.AgentOS_Send_Response_2501 import \
29-
AskuiAgentosSendResponseSchema # noqa: E501
19+
from askui.tools.agent_os import (
20+
AgentOs,
21+
Coordinate,
22+
GetDisplayInformationResponse,
23+
ModifierKey,
24+
PcKey,
25+
)
26+
from askui.tools.askui.askui_ui_controller_grpc.generated import (
27+
Controller_V1_pb2 as controller_v1_pbs,
28+
)
29+
from askui.tools.askui.askui_ui_controller_grpc.generated import (
30+
Controller_V1_pb2_grpc as controller_v1,
31+
)
32+
from askui.tools.askui.askui_ui_controller_grpc.generated.AgentOS_Send_Request_2501 import (
33+
RenderObjectStyle,
34+
) # noqa: E501
35+
from askui.tools.askui.askui_ui_controller_grpc.generated.AgentOS_Send_Response_2501 import (
36+
AskuiAgentosSendResponseSchema,
37+
) # noqa: E501
3038
from askui.tools.askui.command_helpers import (
31-
create_clear_render_objects_command, create_delete_render_object_command,
32-
create_get_mouse_position_command, create_image_command,
33-
create_line_command, create_quad_command,
34-
create_set_mouse_position_command, create_text_command,
35-
create_update_render_object_command)
39+
create_clear_render_objects_command,
40+
create_delete_render_object_command,
41+
create_get_mouse_position_command,
42+
create_image_command,
43+
create_line_command,
44+
create_quad_command,
45+
create_set_mouse_position_command,
46+
create_text_command,
47+
create_update_render_object_command,
48+
)
3649
from askui.utils.image_utils import draw_point_on_image
3750

3851
from ..utils import process_exists, wait_for_port
39-
from .exceptions import (AskUiControllerOperationFailedError,
40-
AskUiControllerOperationTimeoutError)
52+
from .exceptions import (
53+
AskUiControllerOperationFailedError,
54+
AskUiControllerOperationTimeoutError,
55+
)
4156

4257

4358
class RemoteDeviceController(BaseModel):

0 commit comments

Comments
 (0)