oagi 0.6.0__py3-none-any.whl → 0.6.2__py3-none-any.whl

oagi/__init__.py

@@ -5,9 +5,8 @@
 #  This file is part of the official API project.
 #  Licensed under the MIT License.
 # -----------------------------------------------------------------------------
+import importlib
 
-from oagi.async_pyautogui_action_handler import AsyncPyautoguiActionHandler
-from oagi.async_screenshot_maker import AsyncScreenshotMaker
 from oagi.async_single_step import async_single_step
 from oagi.client import AsyncClient, SyncClient
 from oagi.exceptions import (
@@ -22,9 +21,6 @@ from oagi.exceptions import (
     ServerError,
     ValidationError,
 )
-from oagi.pil_image import PILImage
-from oagi.pyautogui_action_handler import PyautoguiActionHandler, PyautoguiConfig
-from oagi.screenshot_maker import ScreenshotMaker
 from oagi.single_step import single_step
 from oagi.task import AsyncShortTask, AsyncTask, ShortTask, Task
 from oagi.types import (
@@ -34,6 +30,27 @@ from oagi.types import (
 )
 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",
+}
+
+
+def __getattr__(name: str):
+    """Lazy import for pyautogui-dependent modules."""
+    if name in _LAZY_IMPORTS:
+        module_name = _LAZY_IMPORTS[name]
+        module = importlib.import_module(module_name)
+        return getattr(module, name)
+    raise AttributeError(f"module '{__name__}' has no attribute '{name}'")
+
+
 __all__ = [
     # Core sync classes
     "Task",
@@ -46,15 +63,6 @@ __all__ = [
     # Functions
     "single_step",
     "async_single_step",
-    # Image classes
-    "PILImage",
-    # Handler classes
-    "PyautoguiActionHandler",
-    "PyautoguiConfig",
-    "ScreenshotMaker",
-    # Async handler classes
-    "AsyncPyautoguiActionHandler",
-    "AsyncScreenshotMaker",
     # Async protocols
     "AsyncActionHandler",
     "AsyncImageProvider",
@@ -75,4 +83,14 @@ __all__ = [
     "ServerError",
     "RequestTimeoutError",
     "ValidationError",
+    # Lazy imports
+    # Image classes
+    "PILImage",
+    # Handler classes
+    "PyautoguiActionHandler",
+    "PyautoguiConfig",
+    "ScreenshotMaker",
+    # Async handler classes
+    "AsyncPyautoguiActionHandler",
+    "AsyncScreenshotMaker",
 ]

oagi/async_single_step.py

@@ -8,7 +8,6 @@
 
 from pathlib import Path
 
-from .pil_image import PILImage
 from .task import AsyncTask
 from .types import Image, Step
 
@@ -64,12 +63,13 @@ async def async_single_step(
         ...     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)):
-        # Convert file path to PILImage
         screenshot = PILImage.from_file(str(screenshot))
     elif isinstance(screenshot, bytes):
-        # Convert bytes to PILImage
         screenshot = PILImage.from_bytes(screenshot)
 
     # Create a temporary task instance

oagi/pil_image.py

@@ -9,7 +9,6 @@
 import io
 from typing import Optional
 
-import pyautogui
 from PIL import Image as PILImageLib
 
 from .types.models.image_config import ImageConfig
@@ -39,6 +38,8 @@ class PILImage:
     @classmethod
     def from_screenshot(cls, config: ImageConfig | None = None) -> "PILImage":
         """Create PILImage from screenshot."""
+        import pyautogui  # noqa: PLC0415, avoid no DISPLAY issue in headless environment
+
         screenshot = pyautogui.screenshot()
         return cls(screenshot, config)
 

oagi/single_step.py

@@ -8,7 +8,6 @@
 
 from pathlib import Path
 
-from .pil_image import PILImage
 from .task import Task
 from .types import Image, Step
 
@@ -62,6 +61,9 @@ def single_step(
         ...     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

{oagi-0.6.0.dist-info → oagi-0.6.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: oagi
-Version: 0.6.0
+Version: 0.6.2
 Summary: Official API of OpenAGI Foundation
 Project-URL: Homepage, https://github.com/agiopen-org/oagi
 Author-email: OpenAGI Foundation <contact@agiopen.org>

{oagi-0.6.0.dist-info → oagi-0.6.2.dist-info}/RECORD

@@ -1,13 +1,13 @@
-oagi/__init__.py,sha256=XM1BgoAFQveyvevQo1g9C2jrWs3rQaN44HPJZaqpVss,2095
+oagi/__init__.py,sha256=Mr_V_lXIZa7n7jrxBMxXAWIaGVi7MblDfcrRsX4IrZI,2605
 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=VPCfpd3GIzgNtYSG-IiKNflB0qFXQ8sZvlIGEvwcy0w,2965
+oagi/async_single_step.py,sha256=FxOSoZKKegHx0by41a4qrJDPoYZV0qZKtdTNMU8Uqz4,2955
 oagi/exceptions.py,sha256=VMwVS8ouE9nHhBpN3AZMYt5_U2kGcihWaTnBhoQLquo,1662
 oagi/logging.py,sha256=CWe89mA5MKTipIvfrqSYkv2CAFNBSwHMDQMDkG_g64g,1350
-oagi/pil_image.py,sha256=Zp7YNwyE_AT25ZEFsWKbzMxbO8JOQsJ1Espph5ye8k8,3804
+oagi/pil_image.py,sha256=c-DmvjZLNU3fzZgF0qiEPEXN1UYFpHrxxr4vuxWLaCo,3878
 oagi/pyautogui_action_handler.py,sha256=8IFbU4p907L4b3TV3Eeh0-c-pYL2lYw-_qf1r8TtPTw,9811
 oagi/screenshot_maker.py,sha256=sVuW7jn-K4FmLhmYI-akdNI-UVcTeBzh9P1_qJhoq1s,1282
-oagi/single_step.py,sha256=-OKFyxpFwlflDnayJ6f-mjTxmnFgfF2u__vWlvwF-jI,3072
+oagi/single_step.py,sha256=62Zip4Uql6E-ZIX6vAlrBoveKqrnABzGqHdLCzju4ag,3138
 oagi/client/__init__.py,sha256=F9DShPUdb6vZYmN1fpM1VYzp4MWqUao_e_R1KYmM4Q4,410
 oagi/client/async_.py,sha256=okeFWLqqaVSZF5AD4WZ2_4jVPeZ_SZefe9yrmkF_lX0,4940
 oagi/client/base.py,sha256=qc2Le28MNpRFvjlO0U0z1rjzqCh9-6YZmO5uOoZO_u8,6538
@@ -29,7 +29,7 @@ oagi/types/models/action.py,sha256=hh6mRRSSWgrW4jpZo71zGMCOcZpV5_COu4148uG6G48,9
 oagi/types/models/client.py,sha256=a6aFwoGG5tBNg3Qr_Dd_TYsg6CpKvA-4tU6b6dCQkoU,983
 oagi/types/models/image_config.py,sha256=tl6abVg_-IAPLwpaWprgknXu7wRWriMg-AEVyUX73v0,1567
 oagi/types/models/step.py,sha256=RSI4H_2rrUBq_xyCoWKaq7JHdJWNobtQppaKC1l0aWU,471
-oagi-0.6.0.dist-info/METADATA,sha256=Y_abrzYAE5P7Sam9syDhSEZpcG6TdRYsstsnnmw3g3c,4799
-oagi-0.6.0.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-oagi-0.6.0.dist-info/licenses/LICENSE,sha256=sy5DLA2M29jFT4UfWsuBF9BAr3FnRkYtnAu6oDZiIf8,1075
-oagi-0.6.0.dist-info/RECORD,,
+oagi-0.6.2.dist-info/METADATA,sha256=VGC76x5orIWmHBHuuPZghKdCMJvP1W6q4U-4BINjlJc,4799
+oagi-0.6.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+oagi-0.6.2.dist-info/licenses/LICENSE,sha256=sy5DLA2M29jFT4UfWsuBF9BAr3FnRkYtnAu6oDZiIf8,1075
+oagi-0.6.2.dist-info/RECORD,,