oagi-core 0.10.1__py3-none-any.whl → 0.10.3__py3-none-any.whl

oagi/cli/display.py

@@ -29,7 +29,8 @@ def display_step_table(
         actions_display = []
         for action in step.actions[:3]:
             arg = action.argument[:20] if action.argument else ""
-            actions_display.append(f"{action.type.value}({arg})")
+            count_str = f" x{action.count}" if action.count and action.count > 1 else ""
+            actions_display.append(f"{action.type.value}({arg}){count_str}")
 
         actions_str = ", ".join(actions_display)
         if len(step.actions) > 3:

oagi/cli/server.py

@@ -25,7 +25,7 @@ def add_server_parser(subparsers: argparse._SubParsersAction) -> None:
     start_parser.add_argument(
         "--host",
         type=str,
-        help="Server host (default: 0.0.0.0, or OAGI_SERVER_HOST env var)",
+        help="Server host (default: 127.0.0.1, or OAGI_SERVER_HOST env var)",
     )
     start_parser.add_argument(
         "--port",

oagi/cli/utils.py

@@ -11,6 +11,7 @@ import os
 import sys
 from importlib.metadata import version as get_version
 
+from oagi.constants import DEFAULT_BASE_URL, MODEL_ACTOR
 from oagi.exceptions import check_optional_dependency
 
 
@@ -55,10 +56,10 @@ def display_version() -> None:
 def display_config() -> None:
     config_vars = {
         "OAGI_API_KEY": os.getenv("OAGI_API_KEY", ""),
-        "OAGI_BASE_URL": os.getenv("OAGI_BASE_URL", "https://api.agiopen.org"),
-        "OAGI_DEFAULT_MODEL": os.getenv("OAGI_DEFAULT_MODEL", "lux-actor-1"),
+        "OAGI_BASE_URL": os.getenv("OAGI_BASE_URL", DEFAULT_BASE_URL),
+        "OAGI_DEFAULT_MODEL": os.getenv("OAGI_DEFAULT_MODEL", MODEL_ACTOR),
         "OAGI_LOG_LEVEL": os.getenv("OAGI_LOG_LEVEL", "INFO"),
-        "OAGI_SERVER_HOST": os.getenv("OAGI_SERVER_HOST", "0.0.0.0"),
+        "OAGI_SERVER_HOST": os.getenv("OAGI_SERVER_HOST", "127.0.0.1"),
         "OAGI_SERVER_PORT": os.getenv("OAGI_SERVER_PORT", "8000"),
         "OAGI_MAX_STEPS": os.getenv("OAGI_MAX_STEPS", "30"),
     }

oagi/client/async_.py

@@ -10,6 +10,13 @@ from functools import wraps
 
 import httpx
 
+from ..constants import (
+    API_HEALTH_ENDPOINT,
+    API_V1_FILE_UPLOAD_ENDPOINT,
+    API_V1_GENERATE_ENDPOINT,
+    API_V2_MESSAGE_ENDPOINT,
+    HTTP_CLIENT_TIMEOUT,
+)
 from ..logging import get_logger
 from ..types import Image
 from ..types.models import GenerateResponse, LLMResponse, UploadFileResponse
@@ -41,7 +48,7 @@ class AsyncClient(BaseClient[httpx.AsyncClient]):
     def __init__(self, base_url: str | None = None, api_key: str | None = None):
         super().__init__(base_url, api_key)
         self.client = httpx.AsyncClient(base_url=self.base_url)
-        self.upload_client = httpx.AsyncClient(timeout=60)  # client for uploading image
+        self.upload_client = httpx.AsyncClient(timeout=HTTP_CLIENT_TIMEOUT)
         logger.info(f"AsyncClient initialized with base_url: {self.base_url}")
 
     async def __aenter__(self):
@@ -121,7 +128,10 @@ class AsyncClient(BaseClient[httpx.AsyncClient]):
         # Make request
         try:
             response = await self.client.post(
-                "/v2/message", json=payload, headers=headers, timeout=self.timeout
+                API_V2_MESSAGE_ENDPOINT,
+                json=payload,
+                headers=headers,
+                timeout=self.timeout,
             )
             return self._process_response(response)
         except (httpx.TimeoutException, httpx.NetworkError) as e:
@@ -136,7 +146,7 @@ class AsyncClient(BaseClient[httpx.AsyncClient]):
         """
         logger.debug("Making async health check request")
         try:
-            response = await self.client.get("/health")
+            response = await self.client.get(API_HEALTH_ENDPOINT)
             response.raise_for_status()
             result = response.json()
             logger.debug("Async health check successful")
@@ -158,12 +168,12 @@ class AsyncClient(BaseClient[httpx.AsyncClient]):
         Returns:
             UploadFileResponse: The response from /v1/file/upload with uuid and presigned S3 URL
         """
-        logger.debug("Making async API request to /v1/file/upload")
+        logger.debug(f"Making async API request to {API_V1_FILE_UPLOAD_ENDPOINT}")
 
         try:
             headers = self._build_headers(api_version)
             response = await self.client.get(
-                "/v1/file/upload", headers=headers, timeout=self.timeout
+                API_V1_FILE_UPLOAD_ENDPOINT, headers=headers, timeout=self.timeout
             )
             return self._process_upload_response(response)
         except (httpx.TimeoutException, httpx.NetworkError, httpx.HTTPStatusError) as e:
@@ -283,7 +293,10 @@ class AsyncClient(BaseClient[httpx.AsyncClient]):
         # Make request
         try:
             response = await self.client.post(
-                "/v1/generate", json=payload, headers=headers, timeout=self.timeout
+                API_V1_GENERATE_ENDPOINT,
+                json=payload,
+                headers=headers,
+                timeout=self.timeout,
             )
             return self._process_generate_response(response)
         except (httpx.TimeoutException, httpx.NetworkError) as e:

oagi/client/base.py

@@ -11,6 +11,7 @@ from typing import Any, Generic, TypeVar
 
 import httpx
 
+from ..constants import API_KEY_HELP_URL, DEFAULT_BASE_URL, HTTP_CLIENT_TIMEOUT
 from ..exceptions import (
     APIError,
     AuthenticationError,
@@ -41,20 +42,19 @@ class BaseClient(Generic[HttpClientT]):
 
     def __init__(self, base_url: str | None = None, api_key: str | None = None):
         # Get from environment if not provided
-        self.base_url = (
-            base_url or os.getenv("OAGI_BASE_URL") or "https://api.agiopen.org"
-        )
+        self.base_url = base_url or os.getenv("OAGI_BASE_URL") or DEFAULT_BASE_URL
         self.api_key = api_key or os.getenv("OAGI_API_KEY")
 
         # Validate required configuration
         if not self.api_key:
             raise ConfigurationError(
                 "OAGI API key must be provided either as 'api_key' parameter or "
-                "OAGI_API_KEY environment variable"
+                "OAGI_API_KEY environment variable. "
+                f"Get your API key at {API_KEY_HELP_URL}"
             )
 
         self.base_url = self.base_url.rstrip("/")
-        self.timeout = 60
+        self.timeout = HTTP_CLIENT_TIMEOUT
         self.client: HttpClientT  # Will be set by subclasses
 
         logger.info(f"Client initialized with base_url: {self.base_url}")
@@ -273,22 +273,20 @@ class BaseClient(Generic[HttpClientT]):
             NetworkError: If network error occurs
             APIError: If API returns error or invalid response
         """
+        response_data = self._parse_response_json(response)
+
+        # Check for error status codes first (follows _process_response pattern)
+        if response.status_code != 200:
+            self._handle_response_error(response, response_data)
+
         try:
-            response_data = response.json()
             upload_file_response = UploadFileResponse(**response_data)
             logger.debug("Calling /v1/file/upload successful")
             return upload_file_response
-        except ValueError:
-            logger.error(f"Non-JSON API response: {response.status_code}")
-            raise APIError(
-                f"Invalid response format (status {response.status_code})",
-                status_code=response.status_code,
-                response=response,
-            )
-        except KeyError as e:
-            logger.error(f"Invalid response: {response.status_code}")
+        except Exception as e:
+            logger.error(f"Invalid upload response: {response.status_code}")
             raise APIError(
-                f"Invalid presigned S3 URL response: missing field {e}",
+                f"Invalid presigned S3 URL response: {e}",
                 status_code=response.status_code,
                 response=response,
             )

oagi/client/sync.py

@@ -11,6 +11,13 @@ from functools import wraps
 import httpx
 from httpx import Response
 
+from ..constants import (
+    API_HEALTH_ENDPOINT,
+    API_V1_FILE_UPLOAD_ENDPOINT,
+    API_V1_GENERATE_ENDPOINT,
+    API_V2_MESSAGE_ENDPOINT,
+    HTTP_CLIENT_TIMEOUT,
+)
 from ..logging import get_logger
 from ..types import Image
 from ..types.models import GenerateResponse, LLMResponse, UploadFileResponse
@@ -46,7 +53,7 @@ class SyncClient(BaseClient[httpx.Client]):
     def __init__(self, base_url: str | None = None, api_key: str | None = None):
         super().__init__(base_url, api_key)
         self.client = httpx.Client(base_url=self.base_url)
-        self.upload_client = httpx.Client(timeout=60)  # client for uploading image
+        self.upload_client = httpx.Client(timeout=HTTP_CLIENT_TIMEOUT)
         logger.info(f"SyncClient initialized with base_url: {self.base_url}")
 
     def __enter__(self):
@@ -124,7 +131,10 @@ class SyncClient(BaseClient[httpx.Client]):
         # Make request
         try:
             response = self.client.post(
-                "/v2/message", json=payload, headers=headers, timeout=self.timeout
+                API_V2_MESSAGE_ENDPOINT,
+                json=payload,
+                headers=headers,
+                timeout=self.timeout,
             )
             return self._process_response(response)
         except (httpx.TimeoutException, httpx.NetworkError) as e:
@@ -139,7 +149,7 @@ class SyncClient(BaseClient[httpx.Client]):
         """
         logger.debug("Making health check request")
         try:
-            response = self.client.get("/health")
+            response = self.client.get(API_HEALTH_ENDPOINT)
             response.raise_for_status()
             result = response.json()
             logger.debug("Health check successful")
@@ -161,12 +171,12 @@ class SyncClient(BaseClient[httpx.Client]):
         Returns:
             UploadFileResponse: The response from /v1/file/upload with uuid and presigned S3 URL
         """
-        logger.debug("Making API request to /v1/file/upload")
+        logger.debug(f"Making API request to {API_V1_FILE_UPLOAD_ENDPOINT}")
 
         try:
             headers = self._build_headers(api_version)
             response = self.client.get(
-                "/v1/file/upload", headers=headers, timeout=self.timeout
+                API_V1_FILE_UPLOAD_ENDPOINT, headers=headers, timeout=self.timeout
             )
             return self._process_upload_response(response)
         except (httpx.TimeoutException, httpx.NetworkError, httpx.HTTPStatusError) as e:
@@ -286,7 +296,10 @@ class SyncClient(BaseClient[httpx.Client]):
         # Make request
         try:
             response = self.client.post(
-                "/v1/generate", json=payload, headers=headers, timeout=self.timeout
+                API_V1_GENERATE_ENDPOINT,
+                json=payload,
+                headers=headers,
+                timeout=self.timeout,
             )
             return self._process_generate_response(response)
         except (httpx.TimeoutException, httpx.NetworkError) as e:

oagi/constants.py

@@ -0,0 +1,43 @@
+# -----------------------------------------------------------------------------
+#  Copyright (c) OpenAGI Foundation
+#  All rights reserved.
+#
+#  This file is part of the official API project.
+#  Licensed under the MIT License.
+# -----------------------------------------------------------------------------
+
+# URLs & API Endpoints
+DEFAULT_BASE_URL = "https://api.agiopen.org"
+API_KEY_HELP_URL = "https://developer.agiopen.org/api-keys"
+API_V2_MESSAGE_ENDPOINT = "/v2/message"
+API_V1_FILE_UPLOAD_ENDPOINT = "/v1/file/upload"
+API_V1_GENERATE_ENDPOINT = "/v1/generate"
+API_HEALTH_ENDPOINT = "/health"
+
+# Model identifiers
+MODEL_ACTOR = "lux-actor-1"
+MODEL_THINKER = "lux-thinker-1"
+
+# Agent modes
+MODE_ACTOR = "actor"
+MODE_THINKER = "thinker"
+MODE_TASKER = "tasker"
+
+# Default max steps per model
+DEFAULT_MAX_STEPS = 20
+DEFAULT_MAX_STEPS_THINKER = 100
+DEFAULT_MAX_STEPS_TASKER = 60
+
+# Reflection intervals
+DEFAULT_REFLECTION_INTERVAL = 4
+DEFAULT_REFLECTION_INTERVAL_TASKER = 20
+
+# Timing & Delays
+DEFAULT_STEP_DELAY = 0.3
+
+# Temperature Defaults
+DEFAULT_TEMPERATURE = 0.5
+DEFAULT_TEMPERATURE_LOW = 0.1
+
+# Timeout Values
+HTTP_CLIENT_TIMEOUT = 60

oagi/handler/pyautogui_action_handler.py

@@ -6,14 +6,13 @@
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
 
-import re
 import sys
 import time
 
 from pydantic import BaseModel, Field
 
 from ..exceptions import check_optional_dependency
-from ..types import Action, ActionType
+from ..types import Action, ActionType, parse_coords, parse_drag_coords, parse_scroll
 
 check_optional_dependency("pyautogui", "PyautoguiActionHandler", "desktop")
 import pyautogui  # noqa: E402
@@ -65,7 +64,8 @@ class PyautoguiConfig(BaseModel):
         default=0.5, description="Duration for drag operations in seconds"
     )
     scroll_amount: int = Field(
-        default=30, description="Amount to scroll (positive for up, negative for down)"
+        default=2 if sys.platform == "darwin" else 100,
+        description="Amount to scroll (positive for up, negative for down)",
     )
     wait_duration: float = Field(
         default=1.0, description="Duration for wait actions in seconds"
@@ -136,36 +136,27 @@ class PyautoguiActionHandler:
 
     def _parse_coords(self, args_str: str) -> tuple[int, int]:
         """Extract x, y coordinates from argument string."""
-        match = re.match(r"(\d+),\s*(\d+)", args_str)
-        if not match:
+        coords = parse_coords(args_str)
+        if not coords:
             raise ValueError(f"Invalid coordinates format: {args_str}")
-        x, y = int(match.group(1)), int(match.group(2))
-        return self._denormalize_coords(x, y)
+        return self._denormalize_coords(coords[0], coords[1])
 
     def _parse_drag_coords(self, args_str: str) -> tuple[int, int, int, int]:
         """Extract x1, y1, x2, y2 coordinates from drag argument string."""
-        match = re.match(r"(\d+),\s*(\d+),\s*(\d+),\s*(\d+)", args_str)
-        if not match:
+        coords = parse_drag_coords(args_str)
+        if not coords:
             raise ValueError(f"Invalid drag coordinates format: {args_str}")
-        x1, y1, x2, y2 = (
-            int(match.group(1)),
-            int(match.group(2)),
-            int(match.group(3)),
-            int(match.group(4)),
-        )
-        x1, y1 = self._denormalize_coords(x1, y1)
-        x2, y2 = self._denormalize_coords(x2, y2)
+        x1, y1 = self._denormalize_coords(coords[0], coords[1])
+        x2, y2 = self._denormalize_coords(coords[2], coords[3])
         return x1, y1, x2, y2
 
     def _parse_scroll(self, args_str: str) -> tuple[int, int, str]:
         """Extract x, y, direction from scroll argument string."""
-        match = re.match(r"(\d+),\s*(\d+),\s*(\w+)", args_str)
-        if not match:
+        result = parse_scroll(args_str)
+        if not result:
             raise ValueError(f"Invalid scroll format: {args_str}")
-        x, y = int(match.group(1)), int(match.group(2))
-        x, y = self._denormalize_coords(x, y)
-        direction = match.group(3).lower()
-        return x, y, direction
+        x, y = self._denormalize_coords(result[0], result[1])
+        return x, y, result[2]
 
     def _normalize_key(self, key: str) -> str:
         """Normalize key names for consistency."""

oagi/server/config.py

@@ -8,6 +8,7 @@
 
 from pydantic import Field
 
+from ..constants import DEFAULT_MAX_STEPS, MODEL_ACTOR
 from ..exceptions import check_optional_dependency
 
 check_optional_dependency("pydantic_settings", "Server features", "server")
@@ -20,7 +21,7 @@ class ServerConfig(BaseSettings):
     oagi_base_url: str = Field(default="https://api.agiopen.org", alias="OAGI_BASE_URL")
 
     # Server settings
-    server_host: str = Field(default="0.0.0.0", alias="OAGI_SERVER_HOST")
+    server_host: str = Field(default="127.0.0.1", alias="OAGI_SERVER_HOST")
     server_port: int = Field(default=8000, alias="OAGI_SERVER_PORT")
     cors_allowed_origins: str = Field(default="*", alias="OAGI_CORS_ORIGINS")
 
@@ -28,11 +29,13 @@ class ServerConfig(BaseSettings):
     session_timeout_seconds: float = Field(default=10.0)
 
     # Model settings
-    default_model: str = Field(default="lux-actor-1", alias="OAGI_DEFAULT_MODEL")
+    default_model: str = Field(default=MODEL_ACTOR, alias="OAGI_DEFAULT_MODEL")
     default_temperature: float = Field(default=0.5, ge=0.0, le=2.0)
 
     # Agent settings
-    max_steps: int = Field(default=20, alias="OAGI_MAX_STEPS", ge=1, le=100)
+    max_steps: int = Field(
+        default=DEFAULT_MAX_STEPS, alias="OAGI_MAX_STEPS", ge=1, le=200
+    )
 
     # Socket.IO settings
     socketio_path: str = Field(default="/socket.io")

oagi/server/models.py

@@ -10,13 +10,15 @@ from typing import Literal
 
 from pydantic import BaseModel, Field
 
+from ..constants import DEFAULT_TEMPERATURE_LOW, MODE_ACTOR, MODEL_ACTOR
+
 
 # Client-to-server events
 class InitEventData(BaseModel):
     instruction: str = Field(...)
-    mode: str | None = Field(default="actor")
-    model: str | None = Field(default="lux-actor-1")
-    temperature: float | None = Field(default=0.1, ge=0.0, le=2.0)
+    mode: str | None = Field(default=MODE_ACTOR)
+    model: str | None = Field(default=MODEL_ACTOR)
+    temperature: float | None = Field(default=DEFAULT_TEMPERATURE_LOW, ge=0.0, le=2.0)
 
 
 # Server-to-client events

oagi/server/session_store.py

@@ -11,14 +11,16 @@ from datetime import datetime
 from typing import Any
 from uuid import uuid4
 
+from ..constants import MODE_ACTOR, MODEL_ACTOR
+
 
 class Session:
     def __init__(
         self,
         session_id: str,
         instruction: str,
-        mode: str = "actor",
-        model: str = "lux-actor-1",
+        mode: str = MODE_ACTOR,
+        model: str = MODEL_ACTOR,
         temperature: float = 0.0,
     ):
         self.session_id: str = session_id
@@ -53,8 +55,8 @@ class SessionStore:
     def create_session(
         self,
         instruction: str,
-        mode: str = "actor",
-        model: str = "lux-actor-1",
+        mode: str = MODE_ACTOR,
+        model: str = MODEL_ACTOR,
         temperature: float = 0.0,
         session_id: str | None = None,
     ) -> str:

oagi/server/socketio_server.py

@@ -15,8 +15,15 @@ from pydantic import ValidationError
 
 from ..agent import AsyncDefaultAgent, create_agent
 from ..client import AsyncClient
+from ..constants import MODE_ACTOR
 from ..exceptions import check_optional_dependency
-from ..types.models.action import Action, ActionType
+from ..types.models.action import (
+    Action,
+    ActionType,
+    parse_coords,
+    parse_drag_coords,
+    parse_scroll,
+)
 from .agent_wrappers import SocketIOActionHandler, SocketIOImageProvider
 from .config import ServerConfig
 from .models import (
@@ -74,7 +81,7 @@ class SessionNamespace(socketio.AsyncNamespace):
             session = Session(
                 session_id=session_id,
                 instruction="",
-                mode="actor",  # Default mode
+                mode=MODE_ACTOR,
                 model=self.config.default_model,
                 temperature=self.config.default_temperature,
             )
@@ -275,31 +282,29 @@ class SessionNamespace(socketio.AsyncNamespace):
                 | ActionType.LEFT_TRIPLE
                 | ActionType.RIGHT_SINGLE
             ):
-                coords = arg.split(",")
-                if len(coords) >= 2:
-                    x, y = int(coords[0]), int(coords[1])
-                else:
+                coords = parse_coords(arg)
+                if not coords:
                     logger.warning(f"Invalid action coordinates: {arg}")
                     return None
 
                 return await self.call(
                     action.type.value,
-                    ClickEventData(**common, x=x, y=y).model_dump(),
+                    ClickEventData(**common, x=coords[0], y=coords[1]).model_dump(),
                     to=session.socket_id,
                     timeout=self.config.socketio_timeout,
                 )
 
             case ActionType.DRAG:
-                coords = arg.split(",")
-                if len(coords) >= 4:
-                    x1, y1, x2, y2 = (int(coords[i]) for i in range(4))
-                else:
+                coords = parse_drag_coords(arg)
+                if not coords:
                     logger.warning(f"Invalid drag coordinates: {arg}")
                     return None
 
                 return await self.call(
                     "drag",
-                    DragEventData(**common, x1=x1, y1=y1, x2=x2, y2=y2).model_dump(),
+                    DragEventData(
+                        **common, x1=coords[0], y1=coords[1], x2=coords[2], y2=coords[3]
+                    ).model_dump(),
                     to=session.socket_id,
                     timeout=self.config.socketio_timeout,
                 )
@@ -326,11 +331,8 @@ class SessionNamespace(socketio.AsyncNamespace):
                 )
 
             case ActionType.SCROLL:
-                parts = arg.split(",")
-                if len(parts) >= 3:
-                    x, y = int(parts[0]), int(parts[1])
-                    direction = parts[2].strip().lower()
-                else:
+                result = parse_scroll(arg)
+                if not result:
                     logger.warning(f"Invalid scroll coordinates: {arg}")
                     return None
 
@@ -340,9 +342,9 @@ class SessionNamespace(socketio.AsyncNamespace):
                     "scroll",
                     ScrollEventData(
                         **common,
-                        x=x,
-                        y=y,
-                        direction=direction,
+                        x=result[0],
+                        y=result[1],
+                        direction=result[2],
                         count=count,  # type: ignore
                     ).model_dump(),
                     to=session.socket_id,

oagi/task/async_.py

@@ -9,6 +9,7 @@
 import warnings
 
 from ..client import AsyncClient
+from ..constants import DEFAULT_MAX_STEPS, MODEL_ACTOR
 from ..types import URL, Image, Step
 from .base import BaseActor
 
@@ -20,7 +21,7 @@ class AsyncActor(BaseActor):
         self,
         api_key: str | None = None,
         base_url: str | None = None,
-        model: str = "lux-actor-1",
+        model: str = MODEL_ACTOR,
         temperature: float | None = None,
     ):
         super().__init__(api_key, base_url, model, temperature)
@@ -31,7 +32,7 @@ class AsyncActor(BaseActor):
     async def init_task(
         self,
         task_desc: str,
-        max_steps: int = 20,
+        max_steps: int = DEFAULT_MAX_STEPS,
     ):
         """Initialize a new task with the given description.
 
@@ -89,7 +90,7 @@ class AsyncTask(AsyncActor):
         self,
         api_key: str | None = None,
         base_url: str | None = None,
-        model: str = "lux-actor-1",
+        model: str = MODEL_ACTOR,
         temperature: float | None = None,
     ):
         warnings.warn(

oagi/task/async_short.py

@@ -8,6 +8,7 @@
 
 import warnings
 
+from ..constants import DEFAULT_MAX_STEPS, MODEL_ACTOR
 from ..logging import get_logger
 from ..types import AsyncActionHandler, AsyncImageProvider
 from .async_ import AsyncActor
@@ -27,7 +28,7 @@ class AsyncShortTask(AsyncActor, BaseAutoMode):
         self,
         api_key: str | None = None,
         base_url: str | None = None,
-        model: str = "lux-actor-1",
+        model: str = MODEL_ACTOR,
         temperature: float | None = None,
     ):
         warnings.warn(
@@ -43,7 +44,7 @@ class AsyncShortTask(AsyncActor, BaseAutoMode):
     async def auto_mode(
         self,
         task_desc: str,
-        max_steps: int = 20,
+        max_steps: int = DEFAULT_MAX_STEPS,
         executor: AsyncActionHandler = None,
         image_provider: AsyncImageProvider = None,
         temperature: float | None = None,

oagi/task/base.py

@@ -8,6 +8,7 @@
 
 from uuid import uuid4
 
+from ..constants import DEFAULT_MAX_STEPS
 from ..logging import get_logger
 from ..types import URL, Image, Step
 from ..types.models import LLMResponse
@@ -30,7 +31,7 @@ class BaseActor:
         self.model = model
         self.temperature = temperature
         self.message_history: list = []  # OpenAI-compatible message history
-        self.max_steps: int = 20  # Maximum steps allowed
+        self.max_steps: int = DEFAULT_MAX_STEPS
         self.current_step: int = 0  # Current step counter
         # Client will be set by subclasses
         self.api_key: str | None = None

oagi/task/short.py

@@ -8,6 +8,7 @@
 
 import warnings
 
+from ..constants import DEFAULT_MAX_STEPS, MODEL_ACTOR
 from ..logging import get_logger
 from ..types import ActionHandler, ImageProvider
 from .base import BaseAutoMode
@@ -27,7 +28,7 @@ class ShortTask(Actor, BaseAutoMode):
         self,
         api_key: str | None = None,
         base_url: str | None = None,
-        model: str = "lux-actor-1",
+        model: str = MODEL_ACTOR,
         temperature: float | None = None,
     ):
         warnings.warn(
@@ -43,7 +44,7 @@ class ShortTask(Actor, BaseAutoMode):
     def auto_mode(
         self,
         task_desc: str,
-        max_steps: int = 20,
+        max_steps: int = DEFAULT_MAX_STEPS,
         executor: ActionHandler = None,
         image_provider: ImageProvider = None,
         temperature: float | None = None,