oagi-core 0.9.0__py3-none-any.whl → 0.9.1__py3-none-any.whl

oagi/__init__.py

@@ -7,7 +7,6 @@
 # -----------------------------------------------------------------------------
 import importlib
 
-from oagi.async_single_step import async_single_step
 from oagi.client import AsyncClient, SyncClient
 from oagi.exceptions import (
     APIError,
@@ -21,8 +20,7 @@ from oagi.exceptions import (
     ServerError,
     ValidationError,
 )
-from oagi.single_step import single_step
-from oagi.task import AsyncShortTask, AsyncTask, ShortTask, Task
+from oagi.task import Actor, AsyncActor, AsyncShortTask, AsyncTask, ShortTask, Task
 from oagi.types import (
     AsyncActionHandler,
     AsyncImageProvider,
@@ -33,12 +31,12 @@ from oagi.types.models import ErrorDetail, ErrorResponse, LLMResponse
 # Lazy imports for pyautogui-dependent modules
 # These will only be imported when actually accessed
 _LAZY_IMPORTS = {
-    "AsyncPyautoguiActionHandler": "oagi.async_pyautogui_action_handler",
-    "AsyncScreenshotMaker": "oagi.async_screenshot_maker",
-    "PILImage": "oagi.pil_image",
-    "PyautoguiActionHandler": "oagi.pyautogui_action_handler",
-    "PyautoguiConfig": "oagi.pyautogui_action_handler",
-    "ScreenshotMaker": "oagi.screenshot_maker",
+    "AsyncPyautoguiActionHandler": "oagi.handler.async_pyautogui_action_handler",
+    "AsyncScreenshotMaker": "oagi.handler.async_screenshot_maker",
+    "PILImage": "oagi.handler.pil_image",
+    "PyautoguiActionHandler": "oagi.handler.pyautogui_action_handler",
+    "PyautoguiConfig": "oagi.handler.pyautogui_action_handler",
+    "ScreenshotMaker": "oagi.handler.screenshot_maker",
     # Agent modules (to avoid circular imports)
     "TaskerAgent": "oagi.agent.tasker",
     # Server modules (optional - requires server dependencies)
@@ -59,18 +57,17 @@ def __getattr__(name: str):
 
 __all__ = [
     # Core sync classes
-    "Task",
-    "ShortTask",
+    "Actor",
+    "AsyncActor",
+    "Task",  # Deprecated: Use Actor instead
+    "ShortTask",  # Deprecated
     "SyncClient",
     # Core async classes
-    "AsyncTask",
-    "AsyncShortTask",
+    "AsyncTask",  # Deprecated: Use AsyncActor instead
+    "AsyncShortTask",  # Deprecated
     "AsyncClient",
     # Agent classes
     "TaskerAgent",
-    # Functions
-    "single_step",
-    "async_single_step",
     # Async protocols
     "AsyncActionHandler",
     "AsyncImageProvider",

oagi/agent/default.py

@@ -8,10 +8,11 @@
 
 import logging
 
-from .. import AsyncTask
+from .. import AsyncActor
 from ..types import (
     AsyncActionHandler,
     AsyncImageProvider,
+    AsyncStepObserver,
 )
 
 logger = logging.getLogger(__name__)
@@ -27,12 +28,14 @@ class AsyncDefaultAgent:
         model: str = "lux-v1",
         max_steps: int = 30,
         temperature: float | None = None,
+        step_observer: AsyncStepObserver | None = None,
     ):
         self.api_key = api_key
         self.base_url = base_url
         self.model = model
         self.max_steps = max_steps
         self.temperature = temperature
+        self.step_observer = step_observer
 
     async def execute(
         self,
@@ -40,11 +43,11 @@ class AsyncDefaultAgent:
         action_handler: AsyncActionHandler,
         image_provider: AsyncImageProvider,
     ) -> bool:
-        async with AsyncTask(
+        async with AsyncActor(
             api_key=self.api_key, base_url=self.base_url, model=self.model
-        ) as self.task:
+        ) as self.actor:
             logger.info(f"Starting async task execution: {instruction}")
-            await self.task.init_task(instruction, max_steps=self.max_steps)
+            await self.actor.init_task(instruction, max_steps=self.max_steps)
 
             for i in range(self.max_steps):
                 logger.debug(f"Executing step {i + 1}/{self.max_steps}")
@@ -53,15 +56,28 @@ class AsyncDefaultAgent:
                 image = await image_provider()
 
                 # Get next step from OAGI
-                step = await self.task.step(image, temperature=self.temperature)
+                step = await self.actor.step(image, temperature=self.temperature)
 
                 # Log reasoning
                 if step.reason:
-                    logger.debug(f"Step {i + 1} reasoning: {step.reason}")
+                    logger.info(f"Step {i + 1}: {step.reason}")
+
+                # Notify observer if present
+                if self.step_observer and step.actions:
+                    await self.step_observer.on_step(i + 1, step.reason, step.actions)
 
                 # Execute actions if any
                 if step.actions:
-                    logger.debug(f"Executing {len(step.actions)} actions")
+                    logger.info(f"Actions ({len(step.actions)}):")
+                    for action in step.actions:
+                        count_suffix = (
+                            f" x{action.count}"
+                            if action.count and action.count > 1
+                            else ""
+                        )
+                        logger.info(
+                            f"  [{action.type.value}] {action.argument}{count_suffix}"
+                        )
                     await action_handler(step.actions)
 
                 # Check if task is complete

oagi/agent/factories.py

@@ -6,6 +6,7 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 from oagi.agent.tasker import TaskerAgent
+from oagi.types import AsyncStepObserver
 
 from .default import AsyncDefaultAgent
 from .protocol import AsyncAgent
@@ -19,6 +20,7 @@ def create_default_agent(
     model: str = "lux-v1",
     max_steps: int = 20,
     temperature: float = 0.1,
+    step_observer: AsyncStepObserver | None = None,
 ) -> AsyncAgent:
     return AsyncDefaultAgent(
         api_key=api_key,
@@ -26,6 +28,7 @@ def create_default_agent(
         model=model,
         max_steps=max_steps,
         temperature=temperature,
+        step_observer=step_observer,
     )
 
 
@@ -37,6 +40,7 @@ def create_planner_agent(
     max_steps: int = 30,
     temperature: float = 0.0,
     reflection_interval: int = 20,
+    step_observer: AsyncStepObserver | None = None,
 ) -> AsyncAgent:
     tasker = TaskerAgent(
         api_key=api_key,
@@ -45,6 +49,7 @@ def create_planner_agent(
         max_steps=max_steps,
         temperature=temperature,
         reflection_interval=reflection_interval,
+        step_observer=step_observer,
     )
     # tasker.set_task()
     return tasker

oagi/agent/tasker/taskee_agent.py

@@ -10,8 +10,8 @@ import logging
 from datetime import datetime
 from typing import Any
 
-from oagi import AsyncTask
-from oagi.types import AsyncActionHandler, AsyncImageProvider
+from oagi import AsyncActor
+from oagi.types import AsyncActionHandler, AsyncImageProvider, AsyncStepObserver
 
 from ..protocol import AsyncAgent
 from .memory import PlannerMemory
@@ -42,6 +42,7 @@ class TaskeeAgent(AsyncAgent):
         planner: Planner | None = None,
         external_memory: PlannerMemory | None = None,
         todo_index: int | None = None,
+        step_observer: AsyncStepObserver | None = None,
     ):
         """Initialize the taskee agent.
 
@@ -55,6 +56,7 @@ class TaskeeAgent(AsyncAgent):
             planner: Planner for planning and reflection
             external_memory: External memory from parent agent
             todo_index: Index of the todo being executed
+            step_observer: Optional observer for step tracking
         """
         self.api_key = api_key
         self.base_url = base_url
@@ -65,9 +67,10 @@ class TaskeeAgent(AsyncAgent):
         self.planner = planner or Planner()
         self.external_memory = external_memory
         self.todo_index = todo_index
+        self.step_observer = step_observer
 
         # Internal state
-        self.task: AsyncTask | None = None
+        self.actor: AsyncActor | None = None
         self.current_todo: str = ""
         self.current_instruction: str = ""
         self.actions: list[Action] = []
@@ -135,10 +138,10 @@ class TaskeeAgent(AsyncAgent):
             )
             return False
         finally:
-            # Clean up task
-            if self.task:
-                await self.task.close()
-                self.task = None
+            # Clean up actor
+            if self.actor:
+                await self.actor.close()
+                self.actor = None
 
     async def _initial_plan(self, image_provider: AsyncImageProvider) -> None:
         """Generate initial plan for the todo.
@@ -199,17 +202,17 @@ class TaskeeAgent(AsyncAgent):
         logger.info(f"Executing subtask with max {max_steps} steps")
 
         # Use async with for automatic resource management
-        async with AsyncTask(
+        async with AsyncActor(
             api_key=self.api_key,
             base_url=self.base_url,
             model=self.model,
             temperature=self.temperature,
-        ) as task:
+        ) as actor:
             # Store reference for potential cleanup in execute's finally block
-            self.task = task
+            self.actor = actor
 
-            # Initialize task with current instruction
-            await task.init_task(self.current_instruction)
+            # Initialize actor with current instruction
+            await actor.init_task(self.current_instruction)
 
             steps_taken = 0
             for step_num in range(max_steps):
@@ -218,7 +221,7 @@ class TaskeeAgent(AsyncAgent):
 
                 # Get next step from OAGI
                 try:
-                    step = await task.step(screenshot, instruction=None)
+                    step = await actor.step(screenshot, instruction=None)
                 except Exception as e:
                     logger.error(f"Error getting step from OAGI: {e}")
                     self._record_action(
@@ -228,8 +231,24 @@ class TaskeeAgent(AsyncAgent):
                     )
                     break
 
+                # Log reasoning
+                if step.reason:
+                    logger.info(f"Step {self.total_actions + 1}: {step.reason}")
+
                 # Record OAGI actions
                 if step.actions:
+                    # Log actions with details
+                    logger.info(f"Actions ({len(step.actions)}):")
+                    for action in step.actions:
+                        count_suffix = (
+                            f" x{action.count}"
+                            if action.count and action.count > 1
+                            else ""
+                        )
+                        logger.info(
+                            f"  [{action.type.value}] {action.argument}{count_suffix}"
+                        )
+
                     for action in step.actions:
                         self._record_action(
                             action_type=action.type.lower(),
@@ -237,6 +256,12 @@ class TaskeeAgent(AsyncAgent):
                             reasoning=step.reason,
                         )
 
+                    # Notify observer if present
+                    if self.step_observer:
+                        await self.step_observer.on_step(
+                            self.total_actions + 1, step.reason, step.actions
+                        )
+
                     # Execute actions
                     await action_handler(step.actions)
                     self.total_actions += len(step.actions)
@@ -255,9 +280,9 @@ class TaskeeAgent(AsyncAgent):
                     logger.info("Reflection interval reached")
                     break
 
-            # Task will be automatically closed by async with context manager
+            # Actor will be automatically closed by async with context manager
             # Clear reference after context manager closes it
-            self.task = None
+            self.actor = None
             return steps_taken
 
     async def _reflect_and_decide(self, image_provider: AsyncImageProvider) -> bool:

oagi/agent/tasker/tasker_agent.py

@@ -9,7 +9,7 @@
 import logging
 from typing import Any
 
-from oagi.types import AsyncActionHandler, AsyncImageProvider
+from oagi.types import AsyncActionHandler, AsyncImageProvider, AsyncStepObserver
 
 from ..protocol import AsyncAgent
 from .memory import PlannerMemory
@@ -39,6 +39,7 @@ class TaskerAgent(AsyncAgent):
         temperature: float = 0.0,
         reflection_interval: int = 20,
         planner: Planner | None = None,
+        step_observer: AsyncStepObserver | None = None,
     ):
         """Initialize the tasker agent.
 
@@ -50,6 +51,7 @@ class TaskerAgent(AsyncAgent):
             temperature: Sampling temperature
             reflection_interval: Actions before reflection
             planner: Planner for planning and reflection
+            step_observer: Optional observer for step tracking
         """
         self.api_key = api_key
         self.base_url = base_url
@@ -58,6 +60,7 @@ class TaskerAgent(AsyncAgent):
         self.temperature = temperature
         self.reflection_interval = reflection_interval
         self.planner = planner or Planner()
+        self.step_observer = step_observer
 
         # Memory for tracking workflow
         self.memory = PlannerMemory()
@@ -174,6 +177,7 @@ class TaskerAgent(AsyncAgent):
             planner=self.planner,
             external_memory=self.memory,  # Share memory with child
             todo_index=todo_index,  # Pass the todo index
+            step_observer=self.step_observer,  # Pass step observer
         )
 
         self.current_todo_index = todo_index

oagi/cli/agent.py

@@ -10,9 +10,14 @@ import argparse
 import asyncio
 import os
 import sys
+import time
+import traceback
 
 from oagi.exceptions import check_optional_dependency
 
+from .display import display_step_table
+from .tracking import StepTracker
+
 
 def add_agent_parser(subparsers: argparse._SubParsersAction) -> None:
     agent_parser = subparsers.add_parser("agent", help="Agent execution commands")
@@ -79,7 +84,10 @@ def run_agent(args: argparse.Namespace) -> None:
     temperature = args.temperature if args.temperature is not None else 0.0
     mode = args.mode or "actor"
 
-    # Create agent
+    # Create step tracker
+    step_tracker = StepTracker()
+
+    # Create agent with step tracker
     agent = create_agent(
         mode=mode,
         api_key=api_key,
@@ -87,6 +95,7 @@ def run_agent(args: argparse.Namespace) -> None:
         model=model,
         max_steps=max_steps,
         temperature=temperature,
+        step_observer=step_tracker,
     )
 
     # Create handlers
@@ -99,7 +108,10 @@ def run_agent(args: argparse.Namespace) -> None:
     )
     print("-" * 60)
 
-    # Run agent
+    start_time = time.time()
+    success = False
+    interrupted = False
+
     try:
         success = asyncio.run(
             agent.execute(
@@ -108,18 +120,22 @@ def run_agent(args: argparse.Namespace) -> None:
                 image_provider=image_provider,
             )
         )
-
-        print("-" * 60)
-        if success:
-            print("Task completed successfully!")
-            sys.exit(0)
-        else:
-            print("Task failed or reached max steps without completion.")
-            sys.exit(1)
-
     except KeyboardInterrupt:
-        print("\nAgent execution interrupted.")
-        sys.exit(130)
+        print("\nAgent execution interrupted by user (Ctrl+C)")
+        interrupted = True
     except Exception as e:
-        print(f"Error during agent execution: {e}", file=sys.stderr)
-        sys.exit(1)
+        print(f"\nError during agent execution: {e}", file=sys.stderr)
+        traceback.print_exc()
+    finally:
+        duration = time.time() - start_time
+
+        if step_tracker.steps:
+            print("\n" + "=" * 60)
+            display_step_table(step_tracker.steps, success, duration)
+        else:
+            print("\nNo steps were executed.")
+
+        if interrupted:
+            sys.exit(130)
+        elif not success:
+            sys.exit(1)

oagi/cli/display.py

@@ -0,0 +1,56 @@
+# -----------------------------------------------------------------------------
+#  Copyright (c) OpenAGI Foundation
+#  All rights reserved.
+#
+#  This file is part of the official API project.
+#  Licensed under the MIT License.
+# -----------------------------------------------------------------------------
+
+from rich.console import Console
+from rich.table import Table
+
+from .tracking import StepData
+
+
+def display_step_table(
+    steps: list[StepData], success: bool, duration: float | None = None
+):
+    console = Console()
+
+    table = Table(title="Agent Execution Summary", show_lines=True)
+    table.add_column("Step", justify="center", style="cyan", width=6)
+    table.add_column("Reasoning", style="white")
+    table.add_column("Actions", style="yellow", width=35)
+    table.add_column("Status", justify="center", width=8)
+
+    for step in steps:
+        reason = step.reasoning or "N/A"
+
+        actions_display = []
+        for action in step.actions[:3]:
+            arg = action.argument[:20] if action.argument else ""
+            actions_display.append(f"{action.type.value}({arg})")
+
+        actions_str = ", ".join(actions_display)
+        if len(step.actions) > 3:
+            actions_str += f" (+{len(step.actions) - 3} more)"
+
+        status_display = "✓" if step.status == "complete" else "→"
+
+        table.add_row(
+            str(step.step_num),
+            reason,
+            actions_str,
+            status_display,
+        )
+
+    console.print(table)
+
+    status_text = "Success" if success else "Failed/Interrupted"
+    console.print(
+        f"\nTotal Steps: {len(steps)} | Status: {status_text}",
+        style="bold",
+    )
+
+    if duration:
+        console.print(f"Duration: {duration:.2f}s")

oagi/cli/tracking.py

@@ -0,0 +1,45 @@
+# -----------------------------------------------------------------------------
+#  Copyright (c) OpenAGI Foundation
+#  All rights reserved.
+#
+#  This file is part of the official API project.
+#  Licensed under the MIT License.
+# -----------------------------------------------------------------------------
+
+from dataclasses import dataclass
+from datetime import datetime
+
+from oagi.types import Action
+
+
+@dataclass
+class StepData:
+    step_num: int
+    timestamp: datetime
+    reasoning: str | None
+    actions: list[Action]
+    action_count: int
+    status: str
+
+
+class StepTracker:
+    """Tracks agent step execution by implementing AsyncStepObserver protocol."""
+
+    def __init__(self):
+        self.steps: list[StepData] = []
+
+    async def on_step(
+        self,
+        step_num: int,
+        reasoning: str | None,
+        actions: list[Action],
+    ) -> None:
+        step_data = StepData(
+            step_num=step_num,
+            timestamp=datetime.now(),
+            reasoning=reasoning,
+            actions=actions,
+            action_count=len(actions),
+            status="running",
+        )
+        self.steps.append(step_data)

oagi/handler/__init__.py

@@ -0,0 +1,24 @@
+# -----------------------------------------------------------------------------
+#  Copyright (c) OpenAGI Foundation
+#  All rights reserved.
+#
+#  This file is part of the official API project.
+#  Licensed under the MIT License.
+# -----------------------------------------------------------------------------
+from oagi.handler.async_pyautogui_action_handler import AsyncPyautoguiActionHandler
+from oagi.handler.async_screenshot_maker import AsyncScreenshotMaker
+from oagi.handler.pil_image import PILImage
+from oagi.handler.pyautogui_action_handler import (
+    PyautoguiActionHandler,
+    PyautoguiConfig,
+)
+from oagi.handler.screenshot_maker import ScreenshotMaker
+
+__all__ = [
+    "PILImage",
+    "PyautoguiActionHandler",
+    "PyautoguiConfig",
+    "AsyncPyautoguiActionHandler",
+    "ScreenshotMaker",
+    "AsyncScreenshotMaker",
+]

oagi/{async_pyautogui_action_handler.py → handler/async_pyautogui_action_handler.py}

@@ -8,8 +8,8 @@
 
 import asyncio
 
+from ..types import Action
 from .pyautogui_action_handler import PyautoguiActionHandler, PyautoguiConfig
-from .types import Action
 
 
 class AsyncPyautoguiActionHandler:

oagi/{async_screenshot_maker.py → handler/async_screenshot_maker.py}

@@ -8,8 +8,8 @@
 
 import asyncio
 
+from ..types import Image, ImageConfig
 from .screenshot_maker import ScreenshotMaker
-from .types import Image, ImageConfig
 
 
 class AsyncScreenshotMaker:

oagi/{pil_image.py → handler/pil_image.py}

@@ -8,8 +8,8 @@
 
 import io
 
-from .exceptions import check_optional_dependency
-from .types.models.image_config import ImageConfig
+from ..exceptions import check_optional_dependency
+from ..types.models.image_config import ImageConfig
 
 check_optional_dependency("PIL", "PILImage", "desktop")
 from PIL import Image as PILImageLib  # noqa: E402

oagi/{pyautogui_action_handler.py → handler/pyautogui_action_handler.py}

@@ -11,8 +11,8 @@ import time
 
 from pydantic import BaseModel, Field
 
-from .exceptions import check_optional_dependency
-from .types import Action, ActionType
+from ..exceptions import check_optional_dependency
+from ..types import Action, ActionType
 
 check_optional_dependency("pyautogui", "PyautoguiActionHandler", "desktop")
 import pyautogui  # noqa: E402

oagi/{screenshot_maker.py → handler/screenshot_maker.py}

@@ -8,9 +8,9 @@
 
 from typing import Optional
 
+from ..types import Image
+from ..types.models.image_config import ImageConfig
 from .pil_image import PILImage
-from .types import Image
-from .types.models.image_config import ImageConfig
 
 
 class ScreenshotMaker:

oagi/logging.py

@@ -44,4 +44,12 @@ def get_logger(name: str) -> logging.Logger:
     # Always update level in case environment variable changed
     oagi_root.setLevel(level)
 
+    # Suppress verbose httpx logs unless DEBUG level is enabled
+    # httpx logs every HTTP request at INFO level by default
+    httpx_logger = logging.getLogger("httpx")
+    if level == logging.DEBUG:
+        httpx_logger.setLevel(logging.DEBUG)
+    else:
+        httpx_logger.setLevel(logging.WARNING)
+
     return logger

oagi/task/__init__.py

@@ -6,9 +6,16 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 
-from .async_ import AsyncTask
+from .async_ import AsyncActor, AsyncTask
 from .async_short import AsyncShortTask
 from .short import ShortTask
-from .sync import Task
+from .sync import Actor, Task
 
-__all__ = ["Task", "AsyncTask", "ShortTask", "AsyncShortTask"]
+__all__ = [
+    "Actor",
+    "AsyncActor",
+    "Task",  # Deprecated: Use Actor instead
+    "AsyncTask",  # Deprecated: Use AsyncActor instead
+    "ShortTask",  # Deprecated
+    "AsyncShortTask",  # Deprecated
+]

oagi/task/async_.py

@@ -6,6 +6,8 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 
+import warnings
+
 from ..client import AsyncClient
 from ..logging import get_logger
 from ..types import Image, Step
@@ -14,7 +16,7 @@ from .base import BaseTask
 logger = get_logger("async_task")
 
 
-class AsyncTask(BaseTask):
+class AsyncActor(BaseTask):
     """Async base class for task automation with the OAGI API."""
 
     def __init__(
@@ -95,3 +97,26 @@ class AsyncTask(BaseTask):
 
     async def __aexit__(self, exc_type, exc_val, exc_tb):
         await self.close()
+
+
+class AsyncTask(AsyncActor):
+    """Deprecated: Use AsyncActor instead.
+
+    This class is deprecated and will be removed in a future version.
+    Please use AsyncActor instead.
+    """
+
+    def __init__(
+        self,
+        api_key: str | None = None,
+        base_url: str | None = None,
+        model: str = "vision-model-v1",
+        temperature: float | None = None,
+    ):
+        warnings.warn(
+            "AsyncTask is deprecated and will be removed in a future version. "
+            "Please use AsyncActor instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        super().__init__(api_key, base_url, model, temperature)

oagi/task/async_short.py

@@ -6,16 +6,22 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 
+import warnings
+
 from ..logging import get_logger
 from ..types import AsyncActionHandler, AsyncImageProvider
-from .async_ import AsyncTask
+from .async_ import AsyncActor
 from .base import BaseAutoMode
 
 logger = get_logger("async_short_task")
 
 
-class AsyncShortTask(AsyncTask, BaseAutoMode):
-    """Async task implementation with automatic mode for short-duration tasks."""
+class AsyncShortTask(AsyncActor, BaseAutoMode):
+    """Deprecated: This class is deprecated and will be removed in a future version.
+
+    Async task implementation with automatic mode for short-duration tasks.
+    Please use AsyncActor directly with custom automation logic instead.
+    """
 
     def __init__(
         self,
@@ -24,6 +30,12 @@ class AsyncShortTask(AsyncTask, BaseAutoMode):
         model: str = "vision-model-v1",
         temperature: float | None = None,
     ):
+        warnings.warn(
+            "AsyncShortTask is deprecated and will be removed in a future version. "
+            "Please use AsyncActor with custom automation logic instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
         super().__init__(
             api_key=api_key, base_url=base_url, model=model, temperature=temperature
         )

oagi/task/base.py

@@ -45,7 +45,9 @@ class BaseTask:
             task_desc: Task description
             max_steps: Maximum number of steps
         """
+        self.task_id = uuid4().hex
         self.task_description = task_desc
+        self.message_history = []
         logger.info(f"Task initialized: '{task_desc}' (max_steps: {max_steps})")
 
     def _validate_step_preconditions(self):

oagi/task/short.py

@@ -6,16 +6,22 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 
+import warnings
+
 from ..logging import get_logger
 from ..types import ActionHandler, ImageProvider
 from .base import BaseAutoMode
-from .sync import Task
+from .sync import Actor
 
 logger = get_logger("short_task")
 
 
-class ShortTask(Task, BaseAutoMode):
-    """Task implementation with automatic mode for short-duration tasks."""
+class ShortTask(Actor, BaseAutoMode):
+    """Deprecated: This class is deprecated and will be removed in a future version.
+
+    Task implementation with automatic mode for short-duration tasks.
+    Please use Actor directly with custom automation logic instead.
+    """
 
     def __init__(
         self,
@@ -24,6 +30,12 @@ class ShortTask(Task, BaseAutoMode):
         model: str = "vision-model-v1",
         temperature: float | None = None,
     ):
+        warnings.warn(
+            "ShortTask is deprecated and will be removed in a future version. "
+            "Please use Actor with custom automation logic instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
         super().__init__(
             api_key=api_key, base_url=base_url, model=model, temperature=temperature
         )

oagi/task/sync.py

@@ -6,6 +6,8 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 
+import warnings
+
 from ..client import SyncClient
 from ..logging import get_logger
 from ..types import Image, Step
@@ -14,7 +16,7 @@ from .base import BaseTask
 logger = get_logger("task")
 
 
-class Task(BaseTask):
+class Actor(BaseTask):
     """Base class for task automation with the OAGI API."""
 
     def __init__(
@@ -95,3 +97,26 @@ class Task(BaseTask):
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         self.close()
+
+
+class Task(Actor):
+    """Deprecated: Use Actor instead.
+
+    This class is deprecated and will be removed in a future version.
+    Please use Actor instead.
+    """
+
+    def __init__(
+        self,
+        api_key: str | None = None,
+        base_url: str | None = None,
+        model: str = "vision-model-v1",
+        temperature: float | None = None,
+    ):
+        warnings.warn(
+            "Task is deprecated and will be removed in a future version. "
+            "Please use Actor instead.",
+            DeprecationWarning,
+            stacklevel=2,
+        )
+        super().__init__(api_key, base_url, model, temperature)

oagi/types/__init__.py

@@ -12,6 +12,7 @@ from .async_image_provider import AsyncImageProvider
 from .image import Image
 from .image_provider import ImageProvider
 from .models import Action, ActionType, ImageConfig, Step
+from .step_observer import AsyncStepObserver
 from .url_image import URLImage
 
 __all__ = [
@@ -24,5 +25,6 @@ __all__ = [
     "AsyncActionHandler",
     "ImageProvider",
     "AsyncImageProvider",
+    "AsyncStepObserver",
     "URLImage",
 ]

oagi/types/step_observer.py

@@ -0,0 +1,34 @@
+# -----------------------------------------------------------------------------
+#  Copyright (c) OpenAGI Foundation
+#  All rights reserved.
+#
+#  This file is part of the official API project.
+#  Licensed under the MIT License.
+# -----------------------------------------------------------------------------
+
+from typing import Protocol
+
+from .models import Action
+
+
+class AsyncStepObserver(Protocol):
+    """Protocol for observing agent step execution.
+
+    Observers receive step information (reasoning and actions) as agents
+    execute tasks, enabling tracking, logging, or other side effects.
+    """
+
+    async def on_step(
+        self,
+        step_num: int,
+        reasoning: str | None,
+        actions: list[Action],
+    ) -> None:
+        """Called when an agent executes a step.
+
+        Args:
+            step_num: The step number (1-indexed)
+            reasoning: The reasoning/thinking for this step (if available)
+            actions: The list of actions being executed in this step
+        """
+        ...

{oagi_core-0.9.0.dist-info → oagi_core-0.9.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: oagi-core
-Version: 0.9.0
+Version: 0.9.1
 Summary: Official API of OpenAGI Foundation
 Project-URL: Homepage, https://github.com/agiopen-org/oagi
 Author-email: OpenAGI Foundation <contact@agiopen.org>
@@ -28,6 +28,7 @@ License: MIT License
 Requires-Python: >=3.10
 Requires-Dist: httpx>=0.28.0
 Requires-Dist: pydantic>=2.0.0
+Requires-Dist: rich>=13.0.0
 Provides-Extra: desktop
 Requires-Dist: pillow>=11.3.0; extra == 'desktop'
 Requires-Dist: pyautogui>=0.9.54; extra == 'desktop'
@@ -75,22 +76,6 @@ export OAGI_API_KEY="your-api-key"
 export OAGI_BASE_URL="https://api.oagi.com"  # or your server URL
 ```
 
-### Single-Step Analysis
-
-Analyze a screenshot and get recommended actions:
-
-```python
-from oagi import single_step
-
-step = single_step(
-    task_description="Click the submit button",
-    screenshot="screenshot.png"  # or bytes, or Image object
-)
-
-print(f"Actions: {step.actions}")
-print(f"Complete: {step.is_complete}")
-```
-
 ### Automated Task Execution
 
 Run tasks automatically with screenshot capture and action execution:
@@ -142,9 +127,6 @@ config = ImageConfig(
     height=700
 )
 compressed = image.transform(config)
-
-# Use with single_step
-step = single_step("Click button", screenshot=compressed)
 ```
 
 ### Async Support
@@ -153,16 +135,9 @@ Use async client for non-blocking operations and better concurrency:
 
 ```python
 import asyncio
-from oagi import async_single_step, AsyncShortTask
+from oagi import AsyncShortTask
 
 async def main():
-    # Single-step async analysis
-    step = await async_single_step(
-        "Find the search bar",
-        screenshot="screenshot.png"
-    )
-    print(f"Found {len(step.actions)} actions")
-    
     # Async task automation
     task = AsyncShortTask()
     async with task:
@@ -176,7 +151,6 @@ asyncio.run(main())
 
 See the [`examples/`](examples/) directory for more usage patterns:
 - `google_weather.py` - Basic task execution with `ShortTask`
-- `single_step.py` - Basic single-step inference
 - `screenshot_with_config.py` - Image compression and optimization
 - `execute_task_auto.py` - Automated task execution
 - `socketio_server_basic.py` - Socket.IO server example

{oagi_core-0.9.0.dist-info → oagi_core-0.9.1.dist-info}/RECORD

@@ -1,33 +1,34 @@
-oagi/__init__.py,sha256=2yFfGqF5Ve2ZmQHYNHvlTQJIENUV2j5AV_fROsUrbyU,2998
-oagi/async_pyautogui_action_handler.py,sha256=F-lKyePCONWI03WnSxpX_QwxONbvnfdQu51wTod6mdw,1614
-oagi/async_screenshot_maker.py,sha256=pI-dbLcYOzcO1ffgTmozAdbYJQNBPKA7hmqj1RxEmIY,1688
-oagi/async_single_step.py,sha256=FxOSoZKKegHx0by41a4qrJDPoYZV0qZKtdTNMU8Uqz4,2955
+oagi/__init__.py,sha256=cSqJ61OyscGJLHBCn53cS4PrCKC4DQl5xRDNkGR0TXo,3041
 oagi/exceptions.py,sha256=Rco37GQTPYUfc2vRO3hozxPF_s8mKFDpFvBg2UKWo3Y,3066
-oagi/logging.py,sha256=YaG-UQWjqXJTfk6MsxcX_xfPVeVCq2ryIH1zwgR2_fQ,1455
-oagi/pil_image.py,sha256=xs6nDRyjTCaoeIMb3GV5d4U6dbG5M5jr2TdMc57eJMo,4078
-oagi/pyautogui_action_handler.py,sha256=N-4XcCV8f92ViuwXnpgk3ILGrlhR_8L1CKkGq_ub5gQ,9952
-oagi/screenshot_maker.py,sha256=sVuW7jn-K4FmLhmYI-akdNI-UVcTeBzh9P1_qJhoq1s,1282
-oagi/single_step.py,sha256=62Zip4Uql6E-ZIX6vAlrBoveKqrnABzGqHdLCzju4ag,3138
+oagi/logging.py,sha256=YT3KCMFj5fzO98R9xlDDgfSotUuz1xRD6OZeYM2rKoo,1760
 oagi/agent/__init__.py,sha256=JU9zuWuDzpzitsVJB4z5ddvx8RMm5nbP-bjUCL1Sfvo,834
-oagi/agent/default.py,sha256=XJXbltDOak3Wer92MXiocLgD9_y0OWJ88isWNOFhaLk,2397
-oagi/agent/factories.py,sha256=JsOqzUWY2_MlL16XznLR8MrWUqgOQTmAyDOvtsA1zvY,1386
+oagi/agent/default.py,sha256=FaaNTssDt5rcuEUppznATqNOTTEJFThcxbBpPChzwnU,3108
+oagi/agent/factories.py,sha256=eE0hcXyJX4dk2nvRRlEvQPJqkBraeIaT7nXiED2lH_Q,1605
 oagi/agent/protocol.py,sha256=IQJGiMN4yZIacrh5e9JQsoM9TyHb8wJRQR4LAk8dSA0,1615
 oagi/agent/registry.py,sha256=4oG65E_bV47Xl6F-HX9KaVoV0pcoC1uRRDU4RT_m3uU,4841
 oagi/agent/tasker/__init__.py,sha256=faOC5ONY8ZKr4CjofC6HYg1WKWc1UiaGB9VHy8W280M,800
 oagi/agent/tasker/memory.py,sha256=JsJjUMpnJoKW4VFzd8FI4M-FhnEihTecL61KVgO_YBI,6051
 oagi/agent/tasker/models.py,sha256=VzvHB5hLv6qyYcyNiojVIEDlTzeGE4Quswk4EVIbzoI,2180
 oagi/agent/tasker/planner.py,sha256=j0Zhk6kw0LFtxkpjFhmeaTBxZfKLf3qqCNpGAuMuODw,13650
-oagi/agent/tasker/taskee_agent.py,sha256=L_5JvWQLVeqY35Xclv7YBOb0gHnGMZ4tJdbskGLc-_o,12966
-oagi/agent/tasker/tasker_agent.py,sha256=PzfsBZvMCH1QMRXDMPQtezw4rX_PfOj7sXSsDAH0U6g,10705
+oagi/agent/tasker/taskee_agent.py,sha256=9t_uIilH2KCnkv013mIfk2VPPNKz6wJtTyjEJLvZOzQ,14093
+oagi/agent/tasker/tasker_agent.py,sha256=d4khAfd1_Ra1KP7Y2un8BwUBeGh-Frwp2xWc7prI6F4,10954
 oagi/cli/__init__.py,sha256=aDnJViTseShpo5fdGPTj-ELysZhmdvB6Z8mEj2D-_N4,359
-oagi/cli/agent.py,sha256=U7Yjmfi7aGt--iisuylg0C2-17bVB3tTBq2sXYU0Gtg,4020
+oagi/cli/agent.py,sha256=q0dLyjrzT3PkZreaXSqpHN_rH5fAB_jHyLonPy-V6hk,4453
+oagi/cli/display.py,sha256=rkAxuHa40ZtKdmvwARev1rgyfsNyVvQ-J6RdjOZIPwc,1729
 oagi/cli/main.py,sha256=faHns0HaQCGyylDn2YZLpjQESuEiMYjoQVoMkt8FsH4,2292
 oagi/cli/server.py,sha256=Z1ic8r55yaeQBFRCsMNZStC1jRiJdnDGqe9On9LmFzQ,3031
+oagi/cli/tracking.py,sha256=jPH6QDUUwnfZ8bjQU6deofBmBflTEOOCINwinQJz9OI,1147
 oagi/cli/utils.py,sha256=QvDyoP0nU9aq4ORjnvJJAGRECmjh9WAmilKuk6zOGM8,2615
 oagi/client/__init__.py,sha256=F9DShPUdb6vZYmN1fpM1VYzp4MWqUao_e_R1KYmM4Q4,410
 oagi/client/async_.py,sha256=t-GPHcz6xbHx_RPFv1V_hwZ1_f-O9ONH-Ahr0w-Nz8M,11046
 oagi/client/base.py,sha256=TlZk4LaYFlOQFmTTgTDzZ1XRAixqNO0pCwgJ7VhZOSU,17101
 oagi/client/sync.py,sha256=QKd6nTUXtyn1Am8YlFcpsoLh1KuHhQgbMemIkb7r39g,10882
+oagi/handler/__init__.py,sha256=Ha11L42K33K3L9S4lQ10UC0DnD5g6egtQUsJpS_tKgg,835
+oagi/handler/async_pyautogui_action_handler.py,sha256=hQzseR1yBD0QMpgsEVNsUmuApGVAIIyGYD06BXd82Dc,1615
+oagi/handler/async_screenshot_maker.py,sha256=8QCtUV59ozpOpvkqhUMb8QDI2qje2gsoFT1qB60tfJM,1689
+oagi/handler/pil_image.py,sha256=yUcAoGBL-aZ0PCjSaAmQsDwtyzjldXHqXQp_OYRk6e4,4080
+oagi/handler/pyautogui_action_handler.py,sha256=rhQAes81-hR5KoFiB4Ph6Cw5r04vYiqvZx64eA31pBw,9954
+oagi/handler/screenshot_maker.py,sha256=j1jTW-awx3vAnb1N5_FIMBC0Z-rNVQbiBP-S6Gh5dlE,1284
 oagi/server/__init__.py,sha256=uZx8u3vJUb87kkNzwmmVrgAgbqRu0WxyMIQCLSx56kk,452
 oagi/server/agent_wrappers.py,sha256=4f6ZKvqy9TDA57QRHGjAQVhpHmPE5QNeewmmURg5Ajo,3288
 oagi/server/config.py,sha256=RstTWbPwWCfW9ZRJcns3uCbdhXxuUnerKjxR1pCGbqY,1602
@@ -35,26 +36,27 @@ oagi/server/main.py,sha256=jnTxk7Prc5CzlsUnkBNJp4MOoYN-7HN_Be_m1d3COa8,4829
 oagi/server/models.py,sha256=E0R80zIz-YLpwx4VQyADQEnM89ZY9bXmjuswfhCuqMc,2595
 oagi/server/session_store.py,sha256=UI14NiApAwvZWmMOG4SvPE2WkbGkNTHToZhTXQjN2_U,3624
 oagi/server/socketio_server.py,sha256=UeNm9bemYMZqtYWp3i09BLUyEWW6t8yTkl2pjpSeT_M,14359
-oagi/task/__init__.py,sha256=kvB8ENQ3iII9Uzqa9SSIxBv5JFKv44j2T6J-_Qc8A-4,494
-oagi/task/async_.py,sha256=QDXaw3wAI8Utbph_kB1a3Elv2aeZwPnv2R7wT16LDXo,3242
-oagi/task/async_short.py,sha256=h7Kchl4opUjyKQd_0Vd7DHWoBOTcYqN1lPdGeyTjoq8,2324
-oagi/task/base.py,sha256=gUpWWhgbkrN17pmZNISmyzxLedhWZ39wVt7fRhH6JDM,4378
-oagi/task/short.py,sha256=pjhoYN_a5KS_cGKfXbB575wdw-nx9ZdzJTplMr3sgqs,2148
-oagi/task/sync.py,sha256=_tMjw4Gjyk3wKCD_q7Pwvug0-DMD7DalXZocrREZogc,3135
-oagi/types/__init__.py,sha256=j-ShAARBbz-fGgkaQj5j_xyhFGM4XWnUTp-h3xhm9Bo,814
+oagi/task/__init__.py,sha256=g_8_7ZLDLKuCGzyrB42OzY3gSOjd_SxzkJW3_pf-PXs,662
+oagi/task/async_.py,sha256=IwoLy0SHLNI7k6I5Ja0WmWa8U-SDLogoM-qHFjbVISU,3924
+oagi/task/async_short.py,sha256=LjgWotSxyuTRpNNFsAhAP0Y0raze0GHyoWqthLaPoS8,2755
+oagi/task/base.py,sha256=Udp_Yl5GVUWR7RtayrQlwa8Yg_Nrsyal3mGEkaG8FmE,4447
+oagi/task/short.py,sha256=mgXao6AXuR8XvHwnf3KjzsCm1PZRLzKg1R-2JmS2yoM,2564
+oagi/task/sync.py,sha256=Ag171gJxcWkLgmna4X5Ipgf3qEDDj2JnZEQ9b-mH0mM,3787
+oagi/types/__init__.py,sha256=Vz6JArE8XvBWlES8CVLy-Nx97gooh1OSsltBL6iaFiM,884
 oagi/types/action_handler.py,sha256=NH8E-m5qpGqWcXzTSWfF7W0Xdp8SkzJsbhCmQ0B96cg,1075
 oagi/types/async_action_handler.py,sha256=k1AaqSkFcXlxwW8sn-w0WFHGsIqHFLbcOPrkknmSVug,1116
 oagi/types/async_image_provider.py,sha256=wnhRyPtTmuALt45Qore74-RCkP5yxU9sZGjvOzFqzOk,1170
 oagi/types/image.py,sha256=KgPCCTJ6D5vHIaGZdbTE7eQEa1WlT6G9tf59ZuUCV2U,537
 oagi/types/image_provider.py,sha256=oYFdOYznrK_VOR9egzOjw5wFM5w8EY2sY01pH0ANAgU,1112
+oagi/types/step_observer.py,sha256=KDw7yQxA_I6T2DqElspAOMa8rBJTYFWBNHfC-9NmasM,1025
 oagi/types/url_image.py,sha256=iOwtXj2uwY6dVtDP7uvQLPvK-aTxkdrzhw_R4C6GwBw,1334
 oagi/types/models/__init__.py,sha256=I86Z2moM8hCog_1K1FG_uATcBmWFv_UFetLAjzPzWAY,742
 oagi/types/models/action.py,sha256=hh6mRRSSWgrW4jpZo71zGMCOcZpV5_COu4148uG6G48,967
 oagi/types/models/client.py,sha256=fCN18DBq5XDjNyYB8w-2dFeQ_K9ywwdyh-rXa0GToU4,1357
 oagi/types/models/image_config.py,sha256=tl6abVg_-IAPLwpaWprgknXu7wRWriMg-AEVyUX73v0,1567
 oagi/types/models/step.py,sha256=RSI4H_2rrUBq_xyCoWKaq7JHdJWNobtQppaKC1l0aWU,471
-oagi_core-0.9.0.dist-info/METADATA,sha256=4GekQr0XrNc9sDArtojzQTS6SYXNJz9ax8jQtDW-hMY,8187
-oagi_core-0.9.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-oagi_core-0.9.0.dist-info/entry_points.txt,sha256=zzgsOSWX6aN3KUB0Z1it8DMxFFBJBqmZVqMVAJRjYuw,44
-oagi_core-0.9.0.dist-info/licenses/LICENSE,sha256=sy5DLA2M29jFT4UfWsuBF9BAr3FnRkYtnAu6oDZiIf8,1075
-oagi_core-0.9.0.dist-info/RECORD,,
+oagi_core-0.9.1.dist-info/METADATA,sha256=zJ3xNe4fGIUHAff90wp0s_d98QsjEunz848CWcoQd2w,7543
+oagi_core-0.9.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+oagi_core-0.9.1.dist-info/entry_points.txt,sha256=zzgsOSWX6aN3KUB0Z1it8DMxFFBJBqmZVqMVAJRjYuw,44
+oagi_core-0.9.1.dist-info/licenses/LICENSE,sha256=sy5DLA2M29jFT4UfWsuBF9BAr3FnRkYtnAu6oDZiIf8,1075
+oagi_core-0.9.1.dist-info/RECORD,,

oagi/async_single_step.py

@@ -1,85 +0,0 @@
-# -----------------------------------------------------------------------------
-#  Copyright (c) OpenAGI Foundation
-#  All rights reserved.
-#
-#  This file is part of the official API project.
-#  Licensed under the MIT License.
-# -----------------------------------------------------------------------------
-
-from pathlib import Path
-
-from .task import AsyncTask
-from .types import Image, Step
-
-
-async def async_single_step(
-    task_description: str,
-    screenshot: str | bytes | Path | Image,
-    instruction: str | None = None,
-    api_key: str | None = None,
-    base_url: str | None = None,
-    temperature: float | None = None,
-) -> Step:
-    """
-    Perform a single-step inference asynchronously without maintaining task state.
-
-    This is useful for one-off analyses where you don't need to maintain
-    a conversation or task context across multiple steps.
-
-    Args:
-        task_description: Description of the task to perform
-        screenshot: Screenshot as Image, bytes, or file path
-        instruction: Optional additional instruction for the task
-        api_key: OAGI API key (uses environment variable if not provided)
-        base_url: OAGI base URL (uses environment variable if not provided)
-        temperature: Sampling temperature (0.0-2.0) for LLM inference
-
-    Returns:
-        Step: Object containing reasoning, actions, and completion status
-
-    Example:
-        >>> # Using with bytes
-        >>> import asyncio
-        >>> async def main():
-        ...     with open("screenshot.png", "rb") as f:
-        ...         screenshot_bytes = f.read()
-        ...     step = await async_single_step(
-        ...         "Click the submit button",
-        ...         screenshot=screenshot_bytes
-        ...     )
-        ...     print(f"Actions: {step.actions}")
-        >>> asyncio.run(main())
-
-        >>> # Using with file path
-        >>> step = await async_single_step(
-        ...     "Find the search box",
-        ...     screenshot="screenshot.png"
-        ... )
-
-        >>> # Using with PILImage
-        >>> image = PILImage.from_file("screenshot.png")
-        >>> step = await async_single_step(
-        ...     "Click next page",
-        ...     screenshot=image
-        ... )
-    """
-    # Lazy import PILImage only when needed
-    from .pil_image import PILImage  # noqa: PLC0415
-
-    # Handle different screenshot input types
-    if isinstance(screenshot, (str, Path)):
-        screenshot = PILImage.from_file(str(screenshot))
-    elif isinstance(screenshot, bytes):
-        screenshot = PILImage.from_bytes(screenshot)
-
-    # Create a temporary task instance
-    task = AsyncTask(api_key=api_key, base_url=base_url, temperature=temperature)
-
-    try:
-        # Initialize task and perform single step
-        await task.init_task(task_description)
-        result = await task.step(screenshot, instruction=instruction)
-        return result
-    finally:
-        # Clean up resources
-        await task.close()

oagi/single_step.py

@@ -1,87 +0,0 @@
-# -----------------------------------------------------------------------------
-#  Copyright (c) OpenAGI Foundation
-#  All rights reserved.
-#
-#  This file is part of the official API project.
-#  Licensed under the MIT License.
-# -----------------------------------------------------------------------------
-
-from pathlib import Path
-
-from .task import Task
-from .types import Image, Step
-
-
-def single_step(
-    task_description: str,
-    screenshot: str | bytes | Path | Image,
-    instruction: str | None = None,
-    api_key: str | None = None,
-    base_url: str | None = None,
-    temperature: float | None = None,
-) -> Step:
-    """
-    Perform a single-step inference without maintaining task state.
-
-    This is useful for one-off analyses where you don't need to maintain
-    a conversation or task context across multiple steps.
-
-    Args:
-        task_description: Description of the task to perform
-        screenshot: Screenshot as Image, bytes, or file path
-        instruction: Optional additional instruction for the task
-        api_key: OAGI API key (uses environment variable if not provided)
-        base_url: OAGI base URL (uses environment variable if not provided)
-        temperature: Sampling temperature (0.0-2.0) for LLM inference
-
-    Returns:
-        Step: Object containing reasoning, actions, and completion status
-
-    Example:
-        >>> # Using with bytes
-        >>> with open("screenshot.png", "rb") as f:
-        ...     image_bytes = f.read()
-        >>> step = single_step(
-        ...     task_description="Click the submit button",
-        ...     screenshot=image_bytes
-        ... )
-
-        >>> # Using with file path
-        >>> step = single_step(
-        ...     task_description="Fill in the form",
-        ...     screenshot=Path("screenshot.png"),
-        ...     instruction="Use test@example.com for email"
-        ... )
-
-        >>> # Using with Image object
-        >>> from oagi.types import Image
-        >>> image = Image(...)
-        >>> step = single_step(
-        ...     task_description="Navigate to settings",
-        ...     screenshot=image
-        ... )
-    """
-    # Lazy import PILImage only when needed
-    from .pil_image import PILImage  # noqa: PLC0415
-
-    # Convert file paths to bytes using PILImage
-    if isinstance(screenshot, (str, Path)):
-        path = Path(screenshot) if isinstance(screenshot, str) else screenshot
-        if path.exists():
-            pil_image = PILImage.from_file(str(path))
-            screenshot_bytes = pil_image.read()
-        else:
-            raise FileNotFoundError(f"Screenshot file not found: {path}")
-    elif isinstance(screenshot, bytes):
-        screenshot_bytes = screenshot
-    elif isinstance(screenshot, Image):
-        screenshot_bytes = screenshot.read()
-    else:
-        raise ValueError(
-            f"screenshot must be Image, bytes, str, or Path, got {type(screenshot)}"
-        )
-
-    # Use Task to perform single step
-    with Task(api_key=api_key, base_url=base_url, temperature=temperature) as task:
-        task.init_task(task_description)
-        return task.step(screenshot_bytes, instruction=instruction)