oagi 0.6.1__tar.gz → 0.6.2__tar.gz

{oagi-0.6.1 → oagi-0.6.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: oagi
-Version: 0.6.1
+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.1 → oagi-0.6.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "oagi"
-version = "0.6.1"
+version = "0.6.2"
 description = "Official API of OpenAGI Foundation"
 readme = "README.md"
 license = { file = "LICENSE" }

{oagi-0.6.1 → oagi-0.6.2}/src/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-0.6.1 → oagi-0.6.2}/tests/test_pil_image.py

@@ -224,7 +224,7 @@ class TestPILImageFactoryMethods:
         assert isinstance(result, PILImage)
         assert result.image is mock_image
 
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_from_screenshot(self, mock_screenshot):
         mock_image = MagicMock()
         mock_screenshot.return_value = mock_image

{oagi-0.6.1 → oagi-0.6.2}/tests/test_screenshot_maker.py

@@ -27,7 +27,7 @@ def mock_screenshot_image():
 
 
 class TestScreenshotMaker:
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_maker_takes_screenshot(
         self, mock_screenshot, mock_screenshot_image
     ):
@@ -42,7 +42,7 @@ class TestScreenshotMaker:
         assert isinstance(result, PILImage)
         assert result.image is mock_resized_image
 
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_maker_stores_last_screenshot(self, mock_screenshot):
         def create_mock_image():
             mock = MagicMock()
@@ -66,7 +66,7 @@ class TestScreenshotMaker:
         assert maker.last_image() is second
         assert maker.last_image() is not first
 
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_maker_last_image_creates_if_none(
         self, mock_screenshot, mock_screenshot_image
     ):
@@ -79,7 +79,7 @@ class TestScreenshotMaker:
         mock_screenshot.assert_called_once()
         assert isinstance(result, PILImage)
 
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_image_returns_png_bytes(self, mock_screenshot):
         pil_image = PILImageLib.new("RGB", (10, 10), color="red")
         mock_screenshot.return_value = pil_image
@@ -99,7 +99,7 @@ class TestScreenshotMaker:
             (None, 600, (1920, 600)),  # Uses original width
         ],
     )
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_maker_resize_dimensions(
         self, mock_screenshot, mock_screenshot_image, width, height, expected_size
     ):
@@ -123,7 +123,7 @@ class TestScreenshotMaker:
             ("PNG", b"\x89PNG\r\n\x1a\n"),
         ],
     )
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_maker_format_output(
         self, mock_screenshot, format_name, expected_signature
     ):
@@ -140,7 +140,7 @@ class TestScreenshotMaker:
         result_image = PILImageLib.open(BytesIO(image_bytes))
         assert result_image.format == format_name
 
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_screenshot_maker_default_resize_1260x700(
         self, mock_screenshot, mock_screenshot_image
     ):
@@ -156,7 +156,7 @@ class TestScreenshotMaker:
         assert isinstance(result, PILImage)
         assert result.image is mock_resized_image
 
-    @patch("oagi.pil_image.pyautogui.screenshot")
+    @patch("pyautogui.screenshot")
     def test_resize_happens_before_format_conversion(self, mock_screenshot):
         original_image = PILImageLib.new("RGB", (2000, 1000), color="green")
         mock_screenshot.return_value = original_image

{oagi-0.6.1 → oagi-0.6.2}/uv.lock

@@ -138,7 +138,7 @@ sdist = { url = "https://files.pythonhosted.org/packages/28/fa/b2ba8229b9381e8f6
 
 [[package]]
 name = "oagi"
-version = "0.6.1"
+version = "0.6.2"
 source = { editable = "." }
 dependencies = [
     { name = "httpx" },