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

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

@@ -7,4 +7,5 @@
 /.mypy_cache/
 /.venv/
 __pycache__/
+.ipynb_checkpoints/
 *.py[cod]

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: llamagen-python
-Version: 0.1.1
+Version: 0.1.3
 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
@@ -155,6 +155,25 @@ event = construct_webhook_event(
 The helper verifies `X-Llama-Webhook-Timestamp` and
 `X-Llama-Webhook-Signature` with HMAC SHA-256 and a default 5-minute tolerance.
 
+## Errors
+
+```python
+from llamagen import LlamaGenAPIError, LlamaGenConnectionError, LlamaGenTimeoutError
+
+try:
+    usage = llamagen.comic.usage()
+except LlamaGenAPIError as error:
+    print(error.status, error.data)
+except LlamaGenConnectionError as error:
+    print("Network connection failed:", error)
+except LlamaGenTimeoutError as error:
+    print("Request timed out:", error)
+```
+
+- `LlamaGenAPIError`: the API returned a non-2xx response, such as `401`, `403`, or `429`.
+- `LlamaGenConnectionError`: the SDK could not connect to the API after retries.
+- `LlamaGenTimeoutError`: the request timed out after retries.
+
 ## Publishing
 
 Maintainers can follow the release guide in [`docs/RELEASE.md`](docs/RELEASE.md).

{llamagen_python-0.1.1 → llamagen_python-0.1.3}/README.md

@@ -130,6 +130,25 @@ event = construct_webhook_event(
 The helper verifies `X-Llama-Webhook-Timestamp` and
 `X-Llama-Webhook-Signature` with HMAC SHA-256 and a default 5-minute tolerance.
 
+## Errors
+
+```python
+from llamagen import LlamaGenAPIError, LlamaGenConnectionError, LlamaGenTimeoutError
+
+try:
+    usage = llamagen.comic.usage()
+except LlamaGenAPIError as error:
+    print(error.status, error.data)
+except LlamaGenConnectionError as error:
+    print("Network connection failed:", error)
+except LlamaGenTimeoutError as error:
+    print("Request timed out:", error)
+```
+
+- `LlamaGenAPIError`: the API returned a non-2xx response, such as `401`, `403`, or `429`.
+- `LlamaGenConnectionError`: the SDK could not connect to the API after retries.
+- `LlamaGenTimeoutError`: the request timed out after retries.
+
 ## Publishing
 
 Maintainers can follow the release guide in [`docs/RELEASE.md`](docs/RELEASE.md).

llamagen_python-0.1.3/examples/quickstart.py

@@ -0,0 +1,23 @@
+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)

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

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

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

@@ -1,11 +1,17 @@
 from ._client import LlamaGenClient
-from ._errors import LlamaGenAPIError, LlamaGenTimeoutError, LlamaGenWebhookSignatureError
+from ._errors import (
+    LlamaGenAPIError,
+    LlamaGenConnectionError,
+    LlamaGenTimeoutError,
+    LlamaGenWebhookSignatureError,
+)
 from ._types import SUPPORTED_COMIC_SIZES
 from ._webhooks import construct_webhook_event, verify_webhook_signature
 
 __all__ = [
     "LlamaGenAPIError",
     "LlamaGenClient",
+    "LlamaGenConnectionError",
     "LlamaGenTimeoutError",
     "LlamaGenWebhookSignatureError",
     "SUPPORTED_COMIC_SIZES",
@@ -13,4 +19,4 @@ __all__ = [
     "verify_webhook_signature",
 ]
 
-__version__ = "0.1.1"
+__version__ = "0.1.3"

{llamagen_python-0.1.1 → llamagen_python-0.1.3}/src/llamagen/_errors.py

@@ -16,5 +16,9 @@ class LlamaGenTimeoutError(TimeoutError):
     """Raised when a request or polling operation times out."""
 
 
+class LlamaGenConnectionError(ConnectionError):
+    """Raised when the SDK cannot connect to the LlamaGen API."""
+
+
 class LlamaGenWebhookSignatureError(ValueError):
     """Raised when webhook signature verification fails."""

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

@@ -2,12 +2,15 @@ from __future__ import annotations
 
 import json
 import socket
+import ssl
 import time
 import urllib.error
 import urllib.request
 from typing import Any, Dict, Mapping, Optional, Protocol, Tuple
 
-from ._errors import LlamaGenAPIError, LlamaGenTimeoutError
+from ._errors import LlamaGenAPIError, LlamaGenConnectionError, LlamaGenTimeoutError
+
+DEFAULT_USER_AGENT = "llamagen-python/0.1.3"
 
 
 class Transport(Protocol):
@@ -69,7 +72,11 @@ class HTTPClient:
         if json_body is not None and body is not None:
             raise ValueError("Pass either json_body or body, not both.")
 
-        prepared_headers: Dict[str, str] = {"Authorization": f"Bearer {self.api_key}"}
+        prepared_headers: Dict[str, str] = {
+            "Accept": "application/json",
+            "Authorization": f"Bearer {self.api_key}",
+            "User-Agent": DEFAULT_USER_AGENT,
+        }
         if json_body is not None:
             body = json.dumps(json_body, separators=(",", ":")).encode("utf-8")
             prepared_headers["Content-Type"] = "application/json"
@@ -90,7 +97,19 @@ class HTTPClient:
                 )
             except Exception as error:
                 if _is_timeout_error(error):
+                    if attempt < self.max_retries:
+                        attempt += 1
+                        time.sleep(self.retry_delay * attempt)
+                        continue
                     raise LlamaGenTimeoutError(f"Request timed out after {int(self.timeout * 1000)}ms") from error
+                if _is_connection_error(error):
+                    if attempt < self.max_retries:
+                        attempt += 1
+                        time.sleep(self.retry_delay * attempt)
+                        continue
+                    raise LlamaGenConnectionError(
+                        f"Could not connect to LlamaGen API after {attempt + 1} attempts: {error}"
+                    ) from error
                 raise
 
             data = _decode_response(response_body, response_headers)
@@ -135,3 +154,11 @@ def _is_timeout_error(error: BaseException) -> bool:
     if isinstance(error, urllib.error.URLError):
         return isinstance(error.reason, (TimeoutError, socket.timeout))
     return False
+
+
+def _is_connection_error(error: BaseException) -> bool:
+    if isinstance(error, (ConnectionError, ssl.SSLError)):
+        return True
+    if isinstance(error, urllib.error.URLError):
+        return not _is_timeout_error(error)
+    return False

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

@@ -1,10 +1,12 @@
 from __future__ import annotations
 
 import json
+import ssl
 import unittest
+import urllib.error
 from typing import Mapping, Optional, Tuple
 
-from llamagen import LlamaGenAPIError, LlamaGenClient
+from llamagen import LlamaGenAPIError, LlamaGenClient, LlamaGenConnectionError
 
 
 class FakeTransport:
@@ -25,6 +27,23 @@ class FakeTransport:
         return status, {"content-type": "application/json"}, json.dumps(payload).encode("utf-8")
 
 
+class FailingTransport:
+    def __init__(self, errors):
+        self.errors = list(errors)
+        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})
+        raise self.errors.pop(0)
+
+
 class ClientTests(unittest.TestCase):
     def test_create_comic_applies_defaults_and_auth(self):
         transport = FakeTransport([(200, {"id": "gen_1", "status": "PENDING"})])
@@ -37,6 +56,8 @@ class ClientTests(unittest.TestCase):
         self.assertEqual(call["method"], "POST")
         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")
         body = json.loads(call["body"].decode("utf-8"))
         self.assertEqual(body["preset"], "neutral")
         self.assertEqual(body["size"], "1024x1024")
@@ -86,6 +107,21 @@ class ClientTests(unittest.TestCase):
         body = json.loads(transport.calls[0]["body"].decode("utf-8"))
         self.assertEqual(body["action"], "regeneratePanel")
 
+    def test_retries_ssl_url_errors_and_raises_connection_error(self):
+        transport = FailingTransport(
+            [
+                urllib.error.URLError(ssl.SSLEOFError("unexpected eof")),
+                urllib.error.URLError(ssl.SSLEOFError("unexpected eof")),
+                urllib.error.URLError(ssl.SSLEOFError("unexpected eof")),
+            ]
+        )
+        client = LlamaGenClient(api_key="test-key", retry_delay_ms=1, transport=transport)
+
+        with self.assertRaises(LlamaGenConnectionError):
+            client.comic.usage()
+
+        self.assertEqual(len(transport.calls), 3)
+
 
 if __name__ == "__main__":
     unittest.main()

llamagen_python-0.1.1/examples/quickstart.py

@@ -1,17 +0,0 @@
-import os
-
-from llamagen import LlamaGenClient
-
-
-client = LlamaGenClient(api_key=os.environ["LLAMAGEN_API_KEY"])
-
-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"])