llamagen-python 0.1.3__tar.gz → 0.1.5__tar.gz

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/.gitignore

@@ -6,6 +6,7 @@
 /.ruff_cache/
 /.mypy_cache/
 /.venv/
+/.env
 __pycache__/
 .ipynb_checkpoints/
 *.py[cod]

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llamagen-python
-Version: 0.1.3
+Version: 0.1.5
 Summary: Official Python SDK for LlamaGen Comic API and Animation API
 Project-URL: Homepage, https://llamagen.ai/comic-api
 Project-URL: Documentation, https://llamagen.ai/comic-api/docs

llamagen_python-0.1.5/examples/quickstart.py

@@ -0,0 +1,40 @@
+import os
+from pathlib import Path
+
+from llamagen import LlamaGenAPIError, LlamaGenConnectionError, LlamaGenClient, LlamaGenTimeoutError
+
+
+def load_dotenv(path: Path) -> None:
+    if not path.exists():
+        return
+
+    for raw_line in path.read_text(encoding="utf-8").splitlines():
+        line = raw_line.strip()
+        if not line or line.startswith("#") or "=" not in line:
+            continue
+        key, value = line.split("=", 1)
+        key = key.strip()
+        value = value.strip().strip('"').strip("'")
+        os.environ.setdefault(key, value)
+
+
+load_dotenv(Path(__file__).resolve().parents[1] / ".env")
+client = LlamaGenClient(api_key=os.environ["LLAMAGEN_API_KEY"], max_retries=5)
+
+try:
+    created = client.comic.create(
+        {
+            "prompt": "A 4-panel comic about Leo finding a glowing key in a quiet library.",
+            "style": "manga",
+            "fixPanelNum": 4,
+        }
+    )
+    print("created:", created["id"], created.get("status"))
+    done = client.comic.wait_for_completion(created["id"])
+    print("done:", done["status"])
+except LlamaGenAPIError as error:
+    print("LlamaGen API error:", error.status, error.data)
+except LlamaGenConnectionError as error:
+    print("LlamaGen connection error:", error)
+except LlamaGenTimeoutError as error:
+    print("LlamaGen timeout:", error)

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "llamagen-python"
-version = "0.1.3"
+version = "0.1.5"
 description = "Official Python SDK for LlamaGen Comic API and Animation API"
 readme = "README.md"
 requires-python = ">=3.9"

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/src/llamagen/__init__.py

@@ -19,4 +19,4 @@ __all__ = [
     "verify_webhook_signature",
 ]
 
-__version__ = "0.1.3"
+__version__ = "0.1.5"

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/src/llamagen/_http.py

@@ -10,7 +10,7 @@ from typing import Any, Dict, Mapping, Optional, Protocol, Tuple
 
 from ._errors import LlamaGenAPIError, LlamaGenConnectionError, LlamaGenTimeoutError
 
-DEFAULT_USER_AGENT = "llamagen-python/0.1.3"
+DEFAULT_USER_AGENT = "llamagen-python/0.1.5"
 
 
 class Transport(Protocol):
@@ -26,6 +26,10 @@ class Transport(Protocol):
 
 
 class UrllibTransport:
+    def __init__(self) -> None:
+        self.context = ssl.create_default_context()
+        self.context.minimum_version = ssl.TLSVersion.TLSv1_2
+
     def request(
         self,
         method: str,
@@ -36,7 +40,7 @@ class UrllibTransport:
     ) -> Tuple[int, Mapping[str, str], bytes]:
         request = urllib.request.Request(url, data=body, method=method, headers=dict(headers))
         try:
-            with urllib.request.urlopen(request, timeout=timeout) as response:
+            with urllib.request.urlopen(request, timeout=timeout, context=self.context) as response:
                 return response.status, dict(response.headers.items()), response.read()
         except urllib.error.HTTPError as error:
             return error.code, dict(error.headers.items()), error.read()

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/src/llamagen/resources/animations.py

@@ -3,7 +3,7 @@ from __future__ import annotations
 import time
 from typing import Mapping, Optional, Sequence
 
-from .._errors import LlamaGenTimeoutError
+from .._errors import LlamaGenConnectionError, LlamaGenTimeoutError
 from .._http import HTTPClient
 from .._types import AnimationArtworkResponse
 
@@ -34,7 +34,14 @@ class AnimationsResource:
         start = time.monotonic()
 
         while True:
-            result = self.get(artwork_id)
+            try:
+                result = self.get(artwork_id)
+            except (LlamaGenConnectionError, LlamaGenTimeoutError):
+                if (time.monotonic() - start) * 1000 >= timeout_ms:
+                    raise
+                time.sleep(interval_ms / 1000)
+                continue
+
             if str(result.get("status", "")).upper() in done:
                 return result
 

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/src/llamagen/resources/comics.py

@@ -9,7 +9,7 @@ from pathlib import Path
 from typing import Any, BinaryIO, Dict, Iterable, List, Mapping, Optional, Sequence, Union
 from urllib.parse import urlencode
 
-from .._errors import LlamaGenTimeoutError
+from .._errors import LlamaGenConnectionError, LlamaGenTimeoutError
 from .._http import HTTPClient
 from .._types import ComicArtworkResponse, ComicUploadResponse, ComicUsage, SUPPORTED_COMIC_SIZES
 
@@ -99,7 +99,14 @@ class ComicsResource:
         start = time.monotonic()
 
         while True:
-            result = self.get(artwork_id)
+            try:
+                result = self.get(artwork_id)
+            except (LlamaGenConnectionError, LlamaGenTimeoutError):
+                if (time.monotonic() - start) * 1000 >= timeout_ms:
+                    raise
+                time.sleep(interval_ms / 1000)
+                continue
+
             if str(result.get("status", "")).upper() in done:
                 return result
 

{llamagen_python-0.1.3 → llamagen_python-0.1.5}/tests/test_client.py

@@ -44,6 +44,27 @@ class FailingTransport:
         raise self.errors.pop(0)
 
 
+class MixedTransport:
+    def __init__(self, events):
+        self.events = list(events)
+        self.calls = []
+
+    def request(
+        self,
+        method: str,
+        url: str,
+        headers: Mapping[str, str],
+        body: Optional[bytes],
+        timeout: float,
+    ) -> Tuple[int, Mapping[str, str], bytes]:
+        self.calls.append({"method": method, "url": url, "headers": dict(headers), "body": body})
+        event = self.events.pop(0)
+        if isinstance(event, BaseException):
+            raise event
+        status, payload = event
+        return status, {"content-type": "application/json"}, json.dumps(payload).encode("utf-8")
+
+
 class ClientTests(unittest.TestCase):
     def test_create_comic_applies_defaults_and_auth(self):
         transport = FakeTransport([(200, {"id": "gen_1", "status": "PENDING"})])
@@ -57,7 +78,7 @@ class ClientTests(unittest.TestCase):
         self.assertEqual(call["url"], "https://api.llamagen.ai/v1/comics/generations")
         self.assertEqual(call["headers"]["Authorization"], "Bearer test-key")
         self.assertEqual(call["headers"]["Accept"], "application/json")
-        self.assertEqual(call["headers"]["User-Agent"], "llamagen-python/0.1.3")
+        self.assertEqual(call["headers"]["User-Agent"], "llamagen-python/0.1.5")
         body = json.loads(call["body"].decode("utf-8"))
         self.assertEqual(body["preset"], "neutral")
         self.assertEqual(body["size"], "1024x1024")
@@ -122,6 +143,20 @@ class ClientTests(unittest.TestCase):
 
         self.assertEqual(len(transport.calls), 3)
 
+    def test_wait_for_completion_continues_after_transient_connection_error(self):
+        transport = MixedTransport(
+            [
+                urllib.error.URLError(ssl.SSLEOFError("unexpected eof")),
+                (200, {"id": "gen_1", "status": "SUCCEEDED"}),
+            ]
+        )
+        client = LlamaGenClient(api_key="test-key", max_retries=0, retry_delay_ms=1, transport=transport)
+
+        result = client.comic.wait_for_completion("gen_1", interval_ms=1, timeout_ms=1000)
+
+        self.assertEqual(result["status"], "SUCCEEDED")
+        self.assertEqual(len(transport.calls), 2)
+
 
 if __name__ == "__main__":
     unittest.main()

llamagen_python-0.1.3/examples/quickstart.py

@@ -1,23 +0,0 @@
-import os
-
-from llamagen import LlamaGenAPIError, LlamaGenConnectionError, LlamaGenClient, LlamaGenTimeoutError
-
-
-client = LlamaGenClient(api_key=os.environ["LLAMAGEN_API_KEY"])
-
-try:
-    created = client.comic.create(
-        {
-            "prompt": "A 4-panel comic about Leo finding a glowing key in a quiet library.",
-            "style": "manga",
-            "fixPanelNum": 4,
-        }
-    )
-    done = client.comic.wait_for_completion(created["id"])
-    print(done["status"])
-except LlamaGenAPIError as error:
-    print("LlamaGen API error:", error.status, error.data)
-except LlamaGenConnectionError as error:
-    print("LlamaGen connection error:", error)
-except LlamaGenTimeoutError as error:
-    print("LlamaGen timeout:", error)