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

amigo_sdk/__init__.py

@@ -1,8 +1,4 @@
-__version__ = "0.10.0"
+__version__ = "0.11.0"
 from .sdk_client import AmigoClient, AsyncAmigoClient
 
-__all__ = [
-    "__version__",
-    "AmigoClient",
-    "AsyncAmigoClient",
-]
+__all__ = ["__version__", "AmigoClient", "AsyncAmigoClient"]

amigo_sdk/generated/model.py

@@ -1,6 +1,6 @@
 # generated by datamodel-codegen:
 #   filename:  <stdin>
-#   timestamp: 2025-08-26T14:45:34+00:00
+#   timestamp: 2025-08-27T15:09:46+00:00
 
 from __future__ import annotations
 
@@ -9078,6 +9078,11 @@ class ToolInstance(BaseModel):
     )
 
 
+class ToolVersionInstanceToolRepo(Enum):
+    main = 'main'
+    team = 'team'
+
+
 class TotalARRMetricInput(BaseModel):
     type: Literal['total-arr'] = Field(..., title='Type')
 
@@ -11088,11 +11093,6 @@ class ToolModifyToolRequest(BaseModel):
     )
 
 
-class ToolPublishToolVersionRequestBranch(Enum):
-    main = 'main'
-    team = 'team'
-
-
 class ToolPublishToolVersionRequestBumpType(Enum):
     major = 'major'
     minor = 'minor'
@@ -11100,10 +11100,10 @@ class ToolPublishToolVersionRequestBumpType(Enum):
 
 
 class ToolPublishToolVersionRequest(BaseModel):
-    branch: ToolPublishToolVersionRequestBranch = Field(
+    repo: ToolVersionInstanceToolRepo = Field(
         ...,
-        description="The branch of the tools repo to publish the tool version from. If `team`, then it's published from the release branch of the organization's\nAzure DevOps team. If `main`, then it's published from the main branch of the tools repo.",
-        title='Branch',
+        description="The repo to publish the tool version from. If `team`, then it's published from the repo of the organization's Azure DevOps team. If `main`, then it's published\nfrom the `tools-main` repo.",
+        title='Repo',
     )
     project_path: str = Field(
         ...,
@@ -13792,14 +13792,14 @@ class ToolVersionInstance(BaseModel):
     input_schema: Dict[str, Any] = Field(
         ..., description='The input schema of the tool.', title='Input Schema'
     )
-    tool_branch_name: str = Field(
+    tool_repo: ToolVersionInstanceToolRepo = Field(
         ...,
-        description='The name of the branch in the tool repository that this version corresponds to.',
-        title='Tool Branch Name',
+        description='Whether this tool is published from the main repository or the team-specific repository.',
+        title='Tool Repo',
     )
     tool_commit_hash: str = Field(
         ...,
-        description='The commit hash of the tool within the `tool_branch_name`.',
+        description='The commit hash of the tool within the repository.',
         title='Tool Commit Hash',
     )
     amigo_scaffold_commit_hash: str = Field(

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.11.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: amigo_sdk
-Version: 0.10.0
+Version: 0.11.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.11.0.dist-info}/RECORD

@@ -1,17 +1,18 @@
-amigo_sdk/__init__.py,sha256=zTslmbvamdtcfPsHWs1-RY6Zm9GqYxBfd9IfmADx-Co,154
+amigo_sdk/__init__.py,sha256=MNEge8Pwx6KfzTaKsnEQdwJ4t9xpa5JWOMZude9UL70,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=1iw7NL8cMOWN8KcYP_P3oai0JMWgwhjSWVEZAb2BWy4,427233
+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.11.0.dist-info/METADATA,sha256=blllxaFKFVLxLkoEmft0TnniC8_6Qg_yEi_Xwgc2uO4,8350
+amigo_sdk-0.11.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+amigo_sdk-0.11.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
+amigo_sdk-0.11.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
+amigo_sdk-0.11.0.dist-info/RECORD,,