update internal repo to use latest stainless python sdk

Author: LukeMainwaring

.stats.yml

@@ -1,2 +1,2 @@
-configured_endpoints: 38
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/cleanlab%2Fagility-545a995322d24ba70cd2e52703fe1e6422a02fe73623439a2815f1d55142edff.yml
+configured_endpoints: 36
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/cleanlab%2Fagility-36e7c2cdfead9cc1628d9e75a9be97a5580130287e8610c5bceac14770743512.yml

api.md

@@ -3,66 +3,60 @@
 Types:
 
 ```python
-from agility.types import Assistant, AssistantWithConfig, AssistantListResponse
+from agility.types import Assistant, AssistantWithConfig
 ```
 
 Methods:
 
 - <code title="post /api/assistants/">client.assistants.<a href="./src/agility/resources/assistants/assistants.py">create</a>(\*\*<a href="src/agility/types/assistant_create_params.py">params</a>) -> <a href="./src/agility/types/assistant.py">Assistant</a></code>
 - <code title="get /api/assistants/{assistant_id}">client.assistants.<a href="./src/agility/resources/assistants/assistants.py">retrieve</a>(assistant_id) -> <a href="./src/agility/types/assistant_with_config.py">AssistantWithConfig</a></code>
 - <code title="put /api/assistants/{assistant_id}">client.assistants.<a href="./src/agility/resources/assistants/assistants.py">update</a>(assistant_id, \*\*<a href="src/agility/types/assistant_update_params.py">params</a>) -> <a href="./src/agility/types/assistant_with_config.py">AssistantWithConfig</a></code>
-- <code title="get /api/assistants/">client.assistants.<a href="./src/agility/resources/assistants/assistants.py">list</a>(\*\*<a href="src/agility/types/assistant_list_params.py">params</a>) -> <a href="./src/agility/types/assistant_list_response.py">AssistantListResponse</a></code>
+- <code title="get /api/assistants/">client.assistants.<a href="./src/agility/resources/assistants/assistants.py">list</a>(\*\*<a href="src/agility/types/assistant_list_params.py">params</a>) -> <a href="./src/agility/types/assistant_with_config.py">SyncMyOffsetPage[AssistantWithConfig]</a></code>
 - <code title="delete /api/assistants/{assistant_id}">client.assistants.<a href="./src/agility/resources/assistants/assistants.py">delete</a>(assistant_id) -> None</code>
 
 ## AccessKeys
 
 Types:
 
 ```python
-from agility.types.assistants import AccessKey, AccessKeyListResponse
+from agility.types.assistants import AccessKey
 ```
 
 Methods:
 
 - <code title="post /api/assistants/{assistant_id}/access_keys/">client.assistants.access_keys.<a href="./src/agility/resources/assistants/access_keys.py">create</a>(assistant_id, \*\*<a href="src/agility/types/assistants/access_key_create_params.py">params</a>) -> <a href="./src/agility/types/assistants/access_key.py">AccessKey</a></code>
-- <code title="get /api/assistants/{assistant_id}/access_keys/{access_key_id}">client.assistants.access_keys.<a href="./src/agility/resources/assistants/access_keys.py">retrieve</a>(access_key_id, \*, assistant_id) -> <a href="./src/agility/types/assistants/access_key.py">AccessKey</a></code>
-- <code title="get /api/assistants/{assistant_id}/access_keys/">client.assistants.access_keys.<a href="./src/agility/resources/assistants/access_keys.py">list</a>(assistant_id, \*\*<a href="src/agility/types/assistants/access_key_list_params.py">params</a>) -> <a href="./src/agility/types/assistants/access_key_list_response.py">AccessKeyListResponse</a></code>
+- <code title="get /api/assistants/{assistant_id}/access_keys/">client.assistants.access_keys.<a href="./src/agility/resources/assistants/access_keys.py">list</a>(assistant_id, \*\*<a href="src/agility/types/assistants/access_key_list_params.py">params</a>) -> <a href="./src/agility/types/assistants/access_key.py">SyncMyOffsetPage[AccessKey]</a></code>
 
 # KnowledgeBases
 
 Types:
 
 ```python
-from agility.types import KnowledgeBaseWithConfig, KnowledgeBaseListResponse
+from agility.types import KnowledgeBaseWithConfig
 ```
 
 Methods:
 
 - <code title="post /api/knowledge_bases/">client.knowledge_bases.<a href="./src/agility/resources/knowledge_bases/knowledge_bases.py">create</a>(\*\*<a href="src/agility/types/knowledge_base_create_params.py">params</a>) -> <a href="./src/agility/types/knowledge_base_with_config.py">KnowledgeBaseWithConfig</a></code>
 - <code title="get /api/knowledge_bases/{knowledge_base_id}">client.knowledge_bases.<a href="./src/agility/resources/knowledge_bases/knowledge_bases.py">retrieve</a>(knowledge_base_id) -> <a href="./src/agility/types/knowledge_base_with_config.py">KnowledgeBaseWithConfig</a></code>
 - <code title="put /api/knowledge_bases/{knowledge_base_id}">client.knowledge_bases.<a href="./src/agility/resources/knowledge_bases/knowledge_bases.py">update</a>(knowledge_base_id, \*\*<a href="src/agility/types/knowledge_base_update_params.py">params</a>) -> <a href="./src/agility/types/knowledge_base_with_config.py">KnowledgeBaseWithConfig</a></code>
-- <code title="get /api/knowledge_bases/">client.knowledge_bases.<a href="./src/agility/resources/knowledge_bases/knowledge_bases.py">list</a>(\*\*<a href="src/agility/types/knowledge_base_list_params.py">params</a>) -> <a href="./src/agility/types/knowledge_base_list_response.py">KnowledgeBaseListResponse</a></code>
+- <code title="get /api/knowledge_bases/">client.knowledge_bases.<a href="./src/agility/resources/knowledge_bases/knowledge_bases.py">list</a>(\*\*<a href="src/agility/types/knowledge_base_list_params.py">params</a>) -> <a href="./src/agility/types/knowledge_base_with_config.py">SyncMyOffsetPage[KnowledgeBaseWithConfig]</a></code>
 - <code title="delete /api/knowledge_bases/{knowledge_base_id}">client.knowledge_bases.<a href="./src/agility/resources/knowledge_bases/knowledge_bases.py">delete</a>(knowledge_base_id) -> None</code>
 
 ## Sources
 
 Types:
 
 ```python
-from agility.types.knowledge_bases import (
-    Source,
-    SourceStatusResponse,
-    SourceListResponse,
-    SourceSyncResponse,
-)
+from agility.types.knowledge_bases import Source, SourceStatusResponse, SourceSyncResponse
 ```
 
 Methods:
 
 - <code title="post /api/knowledge_bases/{knowledge_base_id}/sources/">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">create</a>(knowledge_base_id, \*\*<a href="src/agility/types/knowledge_bases/source_create_params.py">params</a>) -> <a href="./src/agility/types/knowledge_bases/source.py">Source</a></code>
 - <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">retrieve</a>(source_id, \*, knowledge_base_id) -> <a href="./src/agility/types/knowledge_bases/source.py">Source</a></code>
 - <code title="put /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">update</a>(source_id, \*, knowledge_base_id, \*\*<a href="src/agility/types/knowledge_bases/source_update_params.py">params</a>) -> <a href="./src/agility/types/knowledge_bases/source.py">Source</a></code>
-- <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">list</a>(knowledge_base_id, \*\*<a href="src/agility/types/knowledge_bases/source_list_params.py">params</a>) -> <a href="./src/agility/types/knowledge_bases/source_list_response.py">SourceListResponse</a></code>
+- <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">list</a>(knowledge_base_id, \*\*<a href="src/agility/types/knowledge_bases/source_list_params.py">params</a>) -> <a href="./src/agility/types/knowledge_bases/source.py">SyncMyOffsetPage[Source]</a></code>
 - <code title="delete /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">delete</a>(source_id, \*, knowledge_base_id) -> None</code>
 - <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}/status">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">status</a>(source_id, \*, knowledge_base_id) -> <a href="./src/agility/types/knowledge_bases/source_status_response.py">SourceStatusResponse</a></code>
 - <code title="post /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}/sync">client.knowledge_bases.sources.<a href="./src/agility/resources/knowledge_bases/sources/sources.py">sync</a>(source_id, \*, knowledge_base_id) -> <a href="./src/agility/types/knowledge_bases/source_sync_response.py">object</a></code>
@@ -72,25 +66,13 @@ Methods:
 Types:
 
 ```python
-from agility.types.knowledge_bases.sources import Document, DocumentListResponse
+from agility.types.knowledge_bases.sources import Document
 ```
 
 Methods:
 
 - <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}/documents/{document_id}">client.knowledge_bases.sources.documents.<a href="./src/agility/resources/knowledge_bases/sources/documents.py">retrieve</a>(document_id, \*, knowledge_base_id, source_id) -> <a href="./src/agility/types/knowledge_bases/sources/document.py">Document</a></code>
-- <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}/documents/">client.knowledge_bases.sources.documents.<a href="./src/agility/resources/knowledge_bases/sources/documents.py">list</a>(source_id, \*, knowledge_base_id, \*\*<a href="src/agility/types/knowledge_bases/sources/document_list_params.py">params</a>) -> <a href="./src/agility/types/knowledge_bases/sources/document_list_response.py">DocumentListResponse</a></code>
-
-# Health
-
-Types:
-
-```python
-from agility.types import HealthCheckResponse
-```
-
-Methods:
-
-- <code title="get /api/health/">client.health.<a href="./src/agility/resources/health.py">check</a>() -> <a href="./src/agility/types/health_check_response.py">HealthCheckResponse</a></code>
+- <code title="get /api/knowledge_bases/{knowledge_base_id}/sources/{source_id}/documents/">client.knowledge_bases.sources.documents.<a href="./src/agility/resources/knowledge_bases/sources/documents.py">list</a>(source_id, \*, knowledge_base_id, \*\*<a href="src/agility/types/knowledge_bases/sources/document_list_params.py">params</a>) -> <a href="./src/agility/types/knowledge_bases/sources/document.py">SyncMyOffsetPage[Document]</a></code>
 
 # Users
 
@@ -122,29 +104,29 @@ Methods:
 Types:
 
 ```python
-from agility.types import Thread, ThreadListResponse
+from agility.types import Thread
 ```
 
 Methods:
 
 - <code title="post /api/threads/">client.threads.<a href="./src/agility/resources/threads/threads.py">create</a>() -> <a href="./src/agility/types/thread.py">Thread</a></code>
 - <code title="get /api/threads/{thread_id}">client.threads.<a href="./src/agility/resources/threads/threads.py">retrieve</a>(thread_id) -> <a href="./src/agility/types/thread.py">Thread</a></code>
-- <code title="get /api/threads/">client.threads.<a href="./src/agility/resources/threads/threads.py">list</a>(\*\*<a href="src/agility/types/thread_list_params.py">params</a>) -> <a href="./src/agility/types/thread_list_response.py">ThreadListResponse</a></code>
+- <code title="get /api/threads/">client.threads.<a href="./src/agility/resources/threads/threads.py">list</a>(\*\*<a href="src/agility/types/thread_list_params.py">params</a>) -> <a href="./src/agility/types/thread.py">SyncMyOffsetPage[Thread]</a></code>
 - <code title="delete /api/threads/{thread_id}">client.threads.<a href="./src/agility/resources/threads/threads.py">delete</a>(thread_id) -> None</code>
 
 ## Messages
 
 Types:
 
 ```python
-from agility.types.threads import Message, MessageListResponse
+from agility.types.threads import Message
 ```
 
 Methods:
 
 - <code title="post /api/threads/{thread_id}/messages/">client.threads.messages.<a href="./src/agility/resources/threads/messages.py">create</a>(thread_id, \*\*<a href="src/agility/types/threads/message_create_params.py">params</a>) -> <a href="./src/agility/types/threads/message.py">Message</a></code>
 - <code title="get /api/threads/{thread_id}/messages/{message_id}">client.threads.messages.<a href="./src/agility/resources/threads/messages.py">retrieve</a>(message_id, \*, thread_id) -> <a href="./src/agility/types/threads/message.py">Message</a></code>
-- <code title="get /api/threads/{thread_id}/messages/">client.threads.messages.<a href="./src/agility/resources/threads/messages.py">list</a>(thread_id, \*\*<a href="src/agility/types/threads/message_list_params.py">params</a>) -> <a href="./src/agility/types/threads/message_list_response.py">MessageListResponse</a></code>
+- <code title="get /api/threads/{thread_id}/messages/">client.threads.messages.<a href="./src/agility/resources/threads/messages.py">list</a>(thread_id, \*\*<a href="src/agility/types/threads/message_list_params.py">params</a>) -> <a href="./src/agility/types/threads/message.py">SyncMyOffsetPage[Message]</a></code>
 - <code title="delete /api/threads/{thread_id}/messages/{message_id}">client.threads.messages.<a href="./src/agility/resources/threads/messages.py">delete</a>(message_id, \*, thread_id) -> None</code>
 
 ## Runs

pyproject.toml

@@ -63,11 +63,11 @@ format = { chain = [
   "format:ruff",
   "format:docs",
   "fix:ruff",
+  # run formatting again to fix any inconsistencies when imports are stripped
+  "format:ruff",
 ]}
-"format:black" = "black ."
 "format:docs" = "python scripts/utils/ruffen-docs.py README.md api.md"
 "format:ruff" = "ruff format"
-"format:isort" = "isort ."
 
 "lint" = { chain = [
   "check:ruff",
@@ -125,10 +125,6 @@ path = "README.md"
 pattern = '\[(.+?)\]\(((?!https?://)\S+?)\)'
 replacement = '[\1](https://github.com/stainless-sdks/agility-python/tree/main/\g<2>)'
 
-[tool.black]
-line-length = 120
-target-version = ["py37"]
-
 [tool.pytest.ini_options]
 testpaths = ["tests"]
 addopts = "--tb=short"

requirements-dev.lock

@@ -16,8 +16,6 @@ anyio==4.4.0
     # via httpx
 argcomplete==3.1.2
     # via nox
-attrs==23.1.0
-    # via pytest
 certifi==2023.7.22
     # via httpcore
     # via httpx
@@ -28,8 +26,9 @@ distlib==0.3.7
     # via virtualenv
 distro==1.8.0
     # via agility
-exceptiongroup==1.1.3
+exceptiongroup==1.2.2
     # via anyio
+    # via pytest
 filelock==3.12.4
     # via virtualenv
 h11==0.14.0
@@ -49,7 +48,7 @@ markdown-it-py==3.0.0
     # via rich
 mdurl==0.1.2
     # via markdown-it-py
-mypy==1.11.2
+mypy==1.13.0
 mypy-extensions==1.0.0
     # via mypy
 nodeenv==1.8.0
@@ -60,27 +59,25 @@ packaging==23.2
     # via pytest
 platformdirs==3.11.0
     # via virtualenv
-pluggy==1.3.0
-    # via pytest
-py==1.11.0
+pluggy==1.5.0
     # via pytest
-pydantic==2.7.1
+pydantic==2.9.2
     # via agility
-pydantic-core==2.18.2
+pydantic-core==2.23.4
     # via pydantic
 pygments==2.18.0
     # via rich
 pyright==1.1.380
-pytest==7.1.1
+pytest==8.3.3
     # via pytest-asyncio
-pytest-asyncio==0.21.1
+pytest-asyncio==0.24.0
 python-dateutil==2.8.2
     # via time-machine
 pytz==2023.3.post1
     # via dirty-equals
 respx==0.20.2
 rich==13.7.1
-ruff==0.6.5
+ruff==0.6.9
 setuptools==68.2.2
     # via nodeenv
 six==1.16.0
@@ -90,10 +87,10 @@ sniffio==1.3.0
     # via anyio
     # via httpx
 time-machine==2.9.0
-tomli==2.0.1
+tomli==2.0.2
     # via mypy
     # via pytest
-typing-extensions==4.8.0
+typing-extensions==4.12.2
     # via agility
     # via anyio
     # via mypy

requirements.lock

@@ -19,7 +19,7 @@ certifi==2023.7.22
     # via httpx
 distro==1.8.0
     # via agility
-exceptiongroup==1.1.3
+exceptiongroup==1.2.2
     # via anyio
 h11==0.14.0
     # via httpcore
@@ -30,15 +30,15 @@ httpx==0.25.2
 idna==3.4
     # via anyio
     # via httpx
-pydantic==2.7.1
+pydantic==2.9.2
     # via agility
-pydantic-core==2.18.2
+pydantic-core==2.23.4
     # via pydantic
 sniffio==1.3.0
     # via agility
     # via anyio
     # via httpx
-typing-extensions==4.8.0
+typing-extensions==4.12.2
     # via agility
     # via anyio
     # via pydantic

src/agility/__init__.py

@@ -3,7 +3,18 @@
 from . import types
 from ._types import NOT_GIVEN, NoneType, NotGiven, Transport, ProxiesTypes
 from ._utils import file_from_path
-from ._client import Client, Stream, Agility, Timeout, Transport, AsyncClient, AsyncStream, AsyncAgility, RequestOptions
+from ._client import (
+    ENVIRONMENTS,
+    Client,
+    Stream,
+    Agility,
+    Timeout,
+    Transport,
+    AsyncClient,
+    AsyncStream,
+    AsyncAgility,
+    RequestOptions,
+)
 from ._models import BaseModel
 from ._version import __title__, __version__
 from ._response import APIResponse as APIResponse, AsyncAPIResponse as AsyncAPIResponse
@@ -58,6 +69,7 @@
     "AsyncStream",
     "Agility",
     "AsyncAgility",
+    "ENVIRONMENTS",
     "file_from_path",
     "BaseModel",
     "DEFAULT_TIMEOUT",

src/agility/_base_client.py

@@ -1575,7 +1575,7 @@ async def _request(
         except Exception as err:
             log.debug("Encountered Exception", exc_info=True)
 
-            if retries_taken > 0:
+            if remaining_retries > 0:
                 return await self._retry_request(
                     input_options,
                     cast_to,