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

amigo_sdk/__init__.py

@@ -1,8 +1,4 @@
-__version__ = "0.9.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-25T20:27:06+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')
 
@@ -9453,11 +9458,17 @@ class AmigoLibPydanticBaseModelStrippedNonemptyString1(RootModel[str]):
     root: str = Field(..., min_length=1)
 
 
-class AmigoLibPydanticBaseModelStrippedNonemptyString2(RootModel[str]):
-    root: str = Field(..., max_length=256, min_length=1)
+class AmigoLibPydanticBaseModelStrippedNonemptyString2(
+    AmigoLibPydanticBaseModelStrippedNonemptyString1
+):
+    pass
 
 
 class AmigoLibPydanticBaseModelStrippedNonemptyString3(RootModel[str]):
+    root: str = Field(..., max_length=256, min_length=1)
+
+
+class AmigoLibPydanticBaseModelStrippedNonemptyString4(RootModel[str]):
     root: str = Field(..., max_length=3, min_length=1)
 
 
@@ -9785,7 +9796,7 @@ class MetricUpdateMetricRequest(BaseModel):
 
 
 class OrganizationCreateAgentRequest(BaseModel):
-    agent_name: AmigoLibPydanticBaseModelStrippedNonemptyString2 = Field(
+    agent_name: AmigoLibPydanticBaseModelStrippedNonemptyString3 = Field(
         ..., description='The name of the new agent.'
     )
 
@@ -10472,7 +10483,7 @@ class OrganizationCreateOrganizationResponse(BaseModel):
 
 
 class OrganizationCreateServiceHierarchicalStateMachineRequest(BaseModel):
-    state_machine_name: AmigoLibPydanticBaseModelStrippedNonemptyString2 = Field(
+    state_machine_name: AmigoLibPydanticBaseModelStrippedNonemptyString3 = Field(
         ..., description='The name of the new state machine.'
     )
 
@@ -11082,11 +11093,6 @@ class ToolModifyToolRequest(BaseModel):
     )
 
 
-class ToolPublishToolVersionRequestBranch(Enum):
-    main = 'main'
-    team = 'team'
-
-
 class ToolPublishToolVersionRequestBumpType(Enum):
     major = 'major'
     minor = 'minor'
@@ -11094,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(
         ...,
@@ -11642,7 +11648,7 @@ class GetConversationMessagesParametersQuery(BaseModel):
         [], description='The IDs of the messages to retrieve.', title='Id'
     )
     message_type: Optional[List[MessageType]] = Field(
-        ['agent-message', 'external-event', 'user-message'],
+        ['external-event', 'user-message', 'agent-message'],
         description='The type of messages to retrieve.',
         title='Message Type',
     )
@@ -13786,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(
@@ -14205,7 +14211,7 @@ class OrganizationCreateOrganizationRequest(BaseModel):
             title='Onboarding Instructions',
         )
     )
-    azure_devops_team_name: AmigoLibPydanticBaseModelStrippedNonemptyString1 = Field(
+    azure_devops_team_name: AmigoLibPydanticBaseModelStrippedNonemptyString2 = Field(
         ...,
         description="The name of the Azure DevOps team to create (if it doesn't exist) for this organization. All orgs with the same Azure DevOps team name will share access to the same Azure DevOps resources, namely\nthe source code for tools.",
     )
@@ -14353,7 +14359,7 @@ class OrganizationModifyOrganizationRequest(BaseModel):
 
 
 class RoleCreateRoleRequest(BaseModel):
-    role_name: AmigoLibPydanticBaseModelStrippedNonemptyString2 = Field(
+    role_name: AmigoLibPydanticBaseModelStrippedNonemptyString3 = Field(
         ...,
         description='The name of the role to create. The role must have a max length of 256 characters.',
     )
@@ -15206,7 +15212,7 @@ class MetricSearchMetricsResponse(BaseModel):
 
 
 class OrganizationCreateAgentVersionRequest(BaseModel):
-    initials: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString3] = Field(
+    initials: Optional[AmigoLibPydanticBaseModelStrippedNonemptyString4] = Field(
         None, description="The agent's initials."
     )
     identity: Optional[IdentityInput] = 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.9.0.dist-info → amigo_sdk-0.11.0.dist-info}/METADATA

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

@@ -0,0 +1,18 @@
+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=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=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.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,,

amigo_sdk-0.9.0.dist-info/RECORD

@@ -1,17 +0,0 @@
-amigo_sdk/__init__.py,sha256=kCOODDDBTcYl4OwHYOLi80txgJizsTwkNb49c88xU0M,153
-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/sdk_client.py,sha256=Kr9M9o66pOLu0T2VDvqdYMmPZzgKJyTELu7BSPgGrYQ,6152
-amigo_sdk/generated/model.py,sha256=jYhncFPqXUiwOSmV6OJT82TpkmjG5LZH_gFRFr7mTEs,427161
-amigo_sdk/resources/conversation.py,sha256=pGM2vtUsem8ClVfzZne1qqKZdM4o7aWaRdAYXXKEMFw,14784
-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.9.0.dist-info/METADATA,sha256=KOQJZyfLcA-ZbG8w22B24yP4XUvUDiUbeUct_4gWMyE,5982
-amigo_sdk-0.9.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-amigo_sdk-0.9.0.dist-info/entry_points.txt,sha256=ivKZ8S9W6SH796zUDHeM-qHodrwmkmUItophi-jJWK0,82
-amigo_sdk-0.9.0.dist-info/licenses/LICENSE,sha256=tx3FiTVbGxwBUOxQbNh05AAQlC2jd5hGvNpIkSfVbCo,1062
-amigo_sdk-0.9.0.dist-info/RECORD,,