amigo_sdk 0.10.0__py3-none-any.whl → 0.12.0__py3-none-any.whl

amigo_sdk/http_client.py

@@ -1,6 +1,7 @@
 import asyncio
 import datetime as dt
 import random
+import threading
 import time
 from collections.abc import AsyncIterator, Iterator
 from dataclasses import dataclass
@@ -333,7 +334,7 @@ class AmigoHttpClient:
         self,
         method: str,
         path: str,
-        abort_flag: list[bool] | None = None,
+        abort_event: threading.Event | None = None,
         **kwargs,
     ) -> Iterator[str]:
         kwargs.setdefault("headers", {})
@@ -343,23 +344,23 @@ class AmigoHttpClient:
 
         def _yield_from_response(resp: httpx.Response) -> Iterator[str]:
             _raise_status_with_body_sync(resp)
-            if abort_flag and abort_flag[0]:
+            if abort_event and abort_event.is_set():
                 return
             for line in resp.iter_lines():
-                if abort_flag and abort_flag[0]:
+                if abort_event and abort_event.is_set():
                     return
                 line_stripped = (line or "").strip()
                 if not line_stripped:
                     continue
                 yield line_stripped
 
-        if abort_flag and abort_flag[0]:
+        if abort_event and abort_event.is_set():
             return iter(())
         with self._client.stream(method, path, **kwargs) as resp:
             if resp.status_code == 401:
                 self._token = None
                 headers["Authorization"] = f"Bearer {self._ensure_token()}"
-                if abort_flag and abort_flag[0]:
+                if abort_event and abort_event.is_set():
                     return iter(())
                 with self._client.stream(method, path, **kwargs) as retry_resp:
                     for ln in _yield_from_response(retry_resp):

amigo_sdk/models.py

@@ -0,0 +1 @@
+from .generated.model import *  # noqa: F403

amigo_sdk/resources/conversation.py

@@ -1,4 +1,5 @@
 import asyncio
+import threading
 from collections.abc import AsyncGenerator, Iterator
 from datetime import datetime
 from typing import Any, Literal
@@ -219,7 +220,7 @@ class ConversationResource:
         self,
         body: ConversationCreateConversationRequest,
         params: CreateConversationParametersQuery,
-        abort_flag: list[bool] | None = None,
+        abort_event: threading.Event | None = None,
     ) -> Iterator[ConversationCreateConversationResponse]:
         def _iter():
             for line in self._http.stream_lines(
@@ -228,7 +229,7 @@ class ConversationResource:
                 params=params.model_dump(mode="json", exclude_none=True),
                 json=body.model_dump(mode="json", exclude_none=True),
                 headers={"Accept": "application/x-ndjson"},
-                abort_flag=abort_flag,
+                abort_event=abort_event,
             ):
                 yield ConversationCreateConversationResponse.model_validate_json(line)
 
@@ -238,7 +239,7 @@ class ConversationResource:
         self,
         conversation_id: str,
         params: InteractWithConversationParametersQuery,
-        abort_flag: list[bool] | None = None,
+        abort_event: threading.Event | None = None,
         *,
         text_message: str | None = None,
         audio_bytes: bytes | None = None,
@@ -248,7 +249,7 @@ class ConversationResource:
             request_kwargs: dict[str, Any] = {
                 "params": params.model_dump(mode="json", exclude_none=True),
                 "headers": {"Accept": "application/x-ndjson"},
-                "abort_flag": abort_flag,
+                "abort_event": abort_event,
             }
             req_format = getattr(params, "request_format", None)
             if req_format == Format.text:

{amigo_sdk-0.10.0.dist-info → amigo_sdk-0.12.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amigo_sdk
-Version: 0.10.0
+Version: 0.12.0
 Summary: Amigo AI Python SDK
 Author: Amigo AI
 License-File: LICENSE
@@ -50,7 +50,7 @@ This SDK auto-generates its types from the latest [Amigo OpenAPI schema](https:/
 
 ```python
 from amigo_sdk import AmigoClient
-from amigo_sdk.generated.model import GetConversationsParametersQuery
+from amigo_sdk.models import GetConversationsParametersQuery
 
 # Initialize and use the client synchronously
 with AmigoClient(
@@ -65,9 +65,38 @@ with AmigoClient(
     print("Conversations:", conversations)
 ```
 
+## Quick Start (async)
+
+```python
+import asyncio
+
+from amigo_sdk import AsyncAmigoClient
+from amigo_sdk.models import GetConversationsParametersQuery
+
+
+async def main():
+    async with AsyncAmigoClient(
+        api_key="your-api-key",
+        api_key_id="your-api-key-id",
+        user_id="user-id",
+        organization_id="your-organization-id",
+    ) as client:
+        conversations = await client.conversation.get_conversations(
+            GetConversationsParametersQuery(limit=10, sort_by=["-created_at"])
+        )
+        print("Conversations:", conversations)
+
+
+asyncio.run(main())
+```
+
 ## Examples
 
-For more SDK usage examples see checkout the [examples/](examples/README.md) folder.
+For more SDK usage examples see the [examples overview](examples/README.md). Direct links:
+
+- **Conversation (sync)**: [examples/conversation/conversation.py](examples/conversation/conversation.py)
+- **Conversation (async)**: [examples/conversation/conversation_async.py](examples/conversation/conversation_async.py)
+- **User management (sync)**: [examples/user/user-management.py](examples/user/user-management.py)
 
 ## Configuration
 
@@ -133,6 +162,43 @@ The SDK provides access to the following resources:
 - **Conversation**: Manage conversations
 - **User**: Manage users
 
+## Generated types
+
+The SDK ships with Pydantic models generated from the latest OpenAPI schema.
+
+- **Importing types**: Import directly from `amigo_sdk.models`
+
+  ```python
+  from amigo_sdk.models import (
+      GetConversationsParametersQuery,
+      ConversationCreateConversationRequest,
+      GetUsersParametersQuery,
+  )
+  ```
+
+- **Using types when calling SDK functions**: Pass request/query models to resource methods.
+
+  ```python
+  from amigo_sdk import AmigoClient
+  from amigo_sdk.models import GetConversationsParametersQuery
+
+  with AmigoClient() as client:
+     conversations = client.conversation.get_conversations(
+         GetConversationsParametersQuery(limit=20, sort_by=["-created_at"])
+     )
+  ```
+
+- **Parsing returned objects**: Responses are Pydantic models. Access fields directly or convert to dict/JSON.
+
+  ```python
+  # Access fields
+  first = conversations.conversations[0]
+  print(first.id, first.created_at)
+
+  # Convert to plain dict for logging/serialization
+  print(first.model_dump(mode="json"))
+  ```
+
 ## Error Handling
 
 The SDK provides typed error handling:
@@ -162,6 +228,20 @@ except Exception as error:
     print("Unexpected error:", error)
 ```
 
+## Retries
+
+The HTTP client includes sensible, configurable retries:
+
+- **Defaults**:
+
+  - max attempts: 3
+  - backoff base: 0.25s (exponential with full jitter)
+  - max delay per attempt: 30s
+  - retry on status: {408, 429, 500, 502, 503, 504}
+  - retry on methods: {"GET"}
+  - special-case: POST is retried on 429 when `Retry-After` is present
+  - 401 triggers a one-time token refresh and immediate retry
+
 ## Development
 
 For detailed development setup, testing, and contribution guidelines, see [CONTRIBUTING.md](CONTRIBUTING.md).

{amigo_sdk-0.10.0.dist-info → amigo_sdk-0.12.0.dist-info}/RECORD

@@ -1,17 +1,18 @@
-amigo_sdk/__init__.py,sha256=zTslmbvamdtcfPsHWs1-RY6Zm9GqYxBfd9IfmADx-Co,154
+amigo_sdk/__init__.py,sha256=IEpHGle8VT0uoku9xjW1s95C5r-iuRhIXGH8muORDlY,139
 amigo_sdk/_retry_utils.py,sha256=kFjw9Wqye6MB5-B4rjLxsbSNcfYBIztcollIoncd1hY,2142
 amigo_sdk/auth.py,sha256=WaM9PcEcnaC6CzNsgRKueHkdSAxNbRylzpR_3Q6guQ0,1765
 amigo_sdk/config.py,sha256=0eZIo-hcJ8ODftKAr-mwB-FGJxGO5PT5T4dRpyWPqAg,1491
 amigo_sdk/errors.py,sha256=RkRyF5eAASd8fIOS6YvL9rLDvLAYWqHfpHSCR7jqvl4,4840
-amigo_sdk/http_client.py,sha256=z8h8FKHRxGzULRz_C60mL5PfYSAv8e_RHUndVo0vHrM,13452
+amigo_sdk/http_client.py,sha256=v25UoUbXcMeHTnfJMcrl8RSSwCVkKUL1Jv-0HoXP1B4,13507
+amigo_sdk/models.py,sha256=V-G6iL43_ZNOPDcatCJCSszGWGz-nzp_RSyGNm-rBAc,45
 amigo_sdk/sdk_client.py,sha256=Kr9M9o66pOLu0T2VDvqdYMmPZzgKJyTELu7BSPgGrYQ,6152
-amigo_sdk/generated/model.py,sha256=JHa-HjkOKDQoeoXIhqSU_Bh9b3DVfV7faofcbETr43M,427284
-amigo_sdk/resources/conversation.py,sha256=pGM2vtUsem8ClVfzZne1qqKZdM4o7aWaRdAYXXKEMFw,14784
+amigo_sdk/generated/model.py,sha256=_AwOzuy1mAzX8yNzK35ojD8OYKLUE768A8uDTJECKMY,421774
+amigo_sdk/resources/conversation.py,sha256=5PkJOvLKqnriSS9K9hKw2VRPxRLTuABEbCyPy1fz1r0,14817
 amigo_sdk/resources/organization.py,sha256=yX4UlOHNegRzFW4gCJrCxjiLCAGnGegasjviR1yad_Q,1211
 amigo_sdk/resources/service.py,sha256=SiwEHXCQk4r1b_tGv47M08VuB7RALDHJQzWlpuD937g,1571
 amigo_sdk/resources/user.py,sha256=i4t5aVzBI37KwAtLKSDWTMwf4D4KQdSDoUiblFe1u7o,3529
-amigo_sdk-0.10.0.dist-info/METADATA,sha256=cpQjUGl6f3XwRN6q6jud4Wnp_0TSzZZvADEWbFuWn18,5983
-amigo_sdk-0.10.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-amigo_sdk-0.10.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.10.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.10.0.dist-info/RECORD,,
+amigo_sdk-0.12.0.dist-info/METADATA,sha256=cVAo-SpuzcIHoBeomVa7KFCAKiWIxZThdcQbWj7lDqw,8350
+amigo_sdk-0.12.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+amigo_sdk-0.12.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.12.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.12.0.dist-info/RECORD,,