update internal repo to use stainless python sdk with latest deployment

Author: LukeMainwaring

.stats.yml

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

api.md

@@ -143,3 +143,43 @@ Methods:
 - <code title="get /api/threads/{thread_id}/runs/{run_id}">client.threads.runs.<a href="./src/agility/resources/threads/runs.py">retrieve</a>(run_id, \*, thread_id) -> <a href="./src/agility/types/threads/run.py">Run</a></code>
 - <code title="delete /api/threads/{thread_id}/runs/{run_id}">client.threads.runs.<a href="./src/agility/resources/threads/runs.py">delete</a>(run_id, \*, thread_id) -> None</code>
 - <code title="post /api/threads/{thread_id}/runs/stream">client.threads.runs.<a href="./src/agility/resources/threads/runs.py">stream</a>(thread_id, \*\*<a href="src/agility/types/threads/run_stream_params.py">params</a>) -> <a href="./src/agility/types/threads/run_stream_response.py">object</a></code>
+
+# Integrations
+
+Types:
+
+```python
+from agility.types import (
+    GcSv0Integration,
+    Integration,
+    S3V0Integration,
+    IntegrationCreateResponse,
+    IntegrationRetrieveResponse,
+    IntegrationListResponse,
+)
+```
+
+Methods:
+
+- <code title="post /api/integrations/">client.integrations.<a href="./src/agility/resources/integrations/integrations.py">create</a>(\*\*<a href="src/agility/types/integration_create_params.py">params</a>) -> <a href="./src/agility/types/integration_create_response.py">IntegrationCreateResponse</a></code>
+- <code title="get /api/integrations/{integration_id}">client.integrations.<a href="./src/agility/resources/integrations/integrations.py">retrieve</a>(integration_id) -> <a href="./src/agility/types/integration_retrieve_response.py">IntegrationRetrieveResponse</a></code>
+- <code title="get /api/integrations/">client.integrations.<a href="./src/agility/resources/integrations/integrations.py">list</a>(\*\*<a href="src/agility/types/integration_list_params.py">params</a>) -> <a href="./src/agility/types/integration_list_response.py">SyncMyOffsetPage[IntegrationListResponse]</a></code>
+- <code title="delete /api/integrations/{integration_id}">client.integrations.<a href="./src/agility/resources/integrations/integrations.py">delete</a>(integration_id) -> None</code>
+
+## Available
+
+Types:
+
+```python
+from agility.types.integrations import IntegrationTypeDef, AvailableListResponse
+```
+
+Methods:
+
+- <code title="get /api/integrations/available">client.integrations.available.<a href="./src/agility/resources/integrations/available.py">list</a>() -> <a href="./src/agility/types/integrations/available_list_response.py">AvailableListResponse</a></code>
+
+## Rbac
+
+Methods:
+
+- <code title="post /api/integrations/rbac/{integration_id}/verify">client.integrations.rbac.<a href="./src/agility/resources/integrations/rbac.py">verify</a>(integration_id) -> <a href="./src/agility/types/integration.py">Integration</a></code>

pyproject.toml

@@ -16,11 +16,10 @@ dependencies = [
     "sniffio",
     "cached-property; python_version < '3.8'",
 ]
-requires-python = ">= 3.7"
+requires-python = ">= 3.8"
 classifiers = [
   "Typing :: Typed",
   "Intended Audience :: Developers",
-  "Programming Language :: Python :: 3.7",
   "Programming Language :: Python :: 3.8",
   "Programming Language :: Python :: 3.9",
   "Programming Language :: Python :: 3.10",
@@ -139,7 +138,7 @@ filterwarnings = [
 # there are a couple of flags that are still disabled by
 # default in strict mode as they are experimental and niche.
 typeCheckingMode = "strict"
-pythonVersion = "3.7"
+pythonVersion = "3.8"
 
 exclude = [
     "_dev",

src/agility/_client.py

@@ -58,6 +58,7 @@ class Agility(SyncAPIClient):
     knowledge_bases: resources.KnowledgeBasesResource
     users: resources.UsersResource
     threads: resources.ThreadsResource
+    integrations: resources.IntegrationsResource
     with_raw_response: AgilityWithRawResponse
     with_streaming_response: AgilityWithStreamedResponse
 
@@ -158,6 +159,7 @@ def __init__(
         self.knowledge_bases = resources.KnowledgeBasesResource(self)
         self.users = resources.UsersResource(self)
         self.threads = resources.ThreadsResource(self)
+        self.integrations = resources.IntegrationsResource(self)
         self.with_raw_response = AgilityWithRawResponse(self)
         self.with_streaming_response = AgilityWithStreamedResponse(self)
 
@@ -301,6 +303,7 @@ class AsyncAgility(AsyncAPIClient):
     knowledge_bases: resources.AsyncKnowledgeBasesResource
     users: resources.AsyncUsersResource
     threads: resources.AsyncThreadsResource
+    integrations: resources.AsyncIntegrationsResource
     with_raw_response: AsyncAgilityWithRawResponse
     with_streaming_response: AsyncAgilityWithStreamedResponse
 
@@ -401,6 +404,7 @@ def __init__(
         self.knowledge_bases = resources.AsyncKnowledgeBasesResource(self)
         self.users = resources.AsyncUsersResource(self)
         self.threads = resources.AsyncThreadsResource(self)
+        self.integrations = resources.AsyncIntegrationsResource(self)
         self.with_raw_response = AsyncAgilityWithRawResponse(self)
         self.with_streaming_response = AsyncAgilityWithStreamedResponse(self)
 
@@ -545,6 +549,7 @@ def __init__(self, client: Agility) -> None:
         self.knowledge_bases = resources.KnowledgeBasesResourceWithRawResponse(client.knowledge_bases)
         self.users = resources.UsersResourceWithRawResponse(client.users)
         self.threads = resources.ThreadsResourceWithRawResponse(client.threads)
+        self.integrations = resources.IntegrationsResourceWithRawResponse(client.integrations)
 
 
 class AsyncAgilityWithRawResponse:
@@ -553,6 +558,7 @@ def __init__(self, client: AsyncAgility) -> None:
         self.knowledge_bases = resources.AsyncKnowledgeBasesResourceWithRawResponse(client.knowledge_bases)
         self.users = resources.AsyncUsersResourceWithRawResponse(client.users)
         self.threads = resources.AsyncThreadsResourceWithRawResponse(client.threads)
+        self.integrations = resources.AsyncIntegrationsResourceWithRawResponse(client.integrations)
 
 
 class AgilityWithStreamedResponse:
@@ -561,6 +567,7 @@ def __init__(self, client: Agility) -> None:
         self.knowledge_bases = resources.KnowledgeBasesResourceWithStreamingResponse(client.knowledge_bases)
         self.users = resources.UsersResourceWithStreamingResponse(client.users)
         self.threads = resources.ThreadsResourceWithStreamingResponse(client.threads)
+        self.integrations = resources.IntegrationsResourceWithStreamingResponse(client.integrations)
 
 
 class AsyncAgilityWithStreamedResponse:
@@ -569,6 +576,7 @@ def __init__(self, client: AsyncAgility) -> None:
         self.knowledge_bases = resources.AsyncKnowledgeBasesResourceWithStreamingResponse(client.knowledge_bases)
         self.users = resources.AsyncUsersResourceWithStreamingResponse(client.users)
         self.threads = resources.AsyncThreadsResourceWithStreamingResponse(client.threads)
+        self.integrations = resources.AsyncIntegrationsResourceWithStreamingResponse(client.integrations)
 
 
 Client = Agility

src/agility/_compat.py

@@ -2,7 +2,7 @@
 
 from typing import TYPE_CHECKING, Any, Union, Generic, TypeVar, Callable, cast, overload
 from datetime import date, datetime
-from typing_extensions import Self
+from typing_extensions import Self, Literal
 
 import pydantic
 from pydantic.fields import FieldInfo
@@ -137,9 +137,11 @@ def model_dump(
     exclude_unset: bool = False,
     exclude_defaults: bool = False,
     warnings: bool = True,
+    mode: Literal["json", "python"] = "python",
 ) -> dict[str, Any]:
-    if PYDANTIC_V2:
+    if PYDANTIC_V2 or hasattr(model, "model_dump"):
         return model.model_dump(
+            mode=mode,
             exclude=exclude,
             exclude_unset=exclude_unset,
             exclude_defaults=exclude_defaults,

src/agility/_models.py

@@ -37,6 +37,7 @@
     PropertyInfo,
     is_list,
     is_given,
+    json_safe,
     lru_cache,
     is_mapping,
     parse_date,
@@ -279,8 +280,8 @@ def model_dump(
             Returns:
                 A dictionary representation of the model.
             """
-            if mode != "python":
-                raise ValueError("mode is only supported in Pydantic v2")
+            if mode not in {"json", "python"}:
+                raise ValueError("mode must be either 'json' or 'python'")
             if round_trip != False:
                 raise ValueError("round_trip is only supported in Pydantic v2")
             if warnings != True:
@@ -289,7 +290,7 @@ def model_dump(
                 raise ValueError("context is only supported in Pydantic v2")
             if serialize_as_any != False:
                 raise ValueError("serialize_as_any is only supported in Pydantic v2")
-            return super().dict(  # pyright: ignore[reportDeprecated]
+            dumped = super().dict(  # pyright: ignore[reportDeprecated]
                 include=include,
                 exclude=exclude,
                 by_alias=by_alias,
@@ -298,6 +299,8 @@ def model_dump(
                 exclude_none=exclude_none,
             )
 
+            return cast(dict[str, Any], json_safe(dumped)) if mode == "json" else dumped
+
         @override
         def model_dump_json(
             self,

src/agility/_utils/__init__.py

@@ -6,6 +6,7 @@
     is_list as is_list,
     is_given as is_given,
     is_tuple as is_tuple,
+    json_safe as json_safe,
     lru_cache as lru_cache,
     is_mapping as is_mapping,
     is_tuple_t as is_tuple_t,

src/agility/_utils/_transform.py

@@ -173,6 +173,11 @@ def _transform_recursive(
         # Iterable[T]
         or (is_iterable_type(stripped_type) and is_iterable(data) and not isinstance(data, str))
     ):
+        # dicts are technically iterable, but it is an iterable on the keys of the dict and is not usually
+        # intended as an iterable, so we don't transform it.
+        if isinstance(data, dict):
+            return cast(object, data)
+
         inner_type = extract_type_arg(stripped_type, 0)
         return [_transform_recursive(d, annotation=annotation, inner_type=inner_type) for d in data]
 
@@ -186,7 +191,7 @@ def _transform_recursive(
         return data
 
     if isinstance(data, pydantic.BaseModel):
-        return model_dump(data, exclude_unset=True)
+        return model_dump(data, exclude_unset=True, mode="json")
 
     annotated_type = _get_annotated_type(annotation)
     if annotated_type is None:
@@ -324,7 +329,7 @@ async def _async_transform_recursive(
         return data
 
     if isinstance(data, pydantic.BaseModel):
-        return model_dump(data, exclude_unset=True)
+        return model_dump(data, exclude_unset=True, mode="json")
 
     annotated_type = _get_annotated_type(annotation)
     if annotated_type is None:

src/agility/_utils/_utils.py

@@ -16,6 +16,7 @@
     overload,
 )
 from pathlib import Path
+from datetime import date, datetime
 from typing_extensions import TypeGuard
 
 import sniffio
@@ -395,3 +396,19 @@ def lru_cache(*, maxsize: int | None = 128) -> Callable[[CallableT], CallableT]:
         maxsize=maxsize,
     )
     return cast(Any, wrapper)  # type: ignore[no-any-return]
+
+
+def json_safe(data: object) -> object:
+    """Translates a mapping / sequence recursively in the same fashion
+    as `pydantic` v2's `model_dump(mode="json")`.
+    """
+    if is_mapping(data):
+        return {json_safe(key): json_safe(value) for key, value in data.items()}
+
+    if is_iterable(data) and not isinstance(data, (str, bytes, bytearray)):
+        return [json_safe(item) for item in data]
+
+    if isinstance(data, (datetime, date)):
+        return data.isoformat()
+
+    return data

src/agility/resources/__init__.py

@@ -24,6 +24,14 @@
     AssistantsResourceWithStreamingResponse,
     AsyncAssistantsResourceWithStreamingResponse,
 )
+from .integrations import (
+    IntegrationsResource,
+    AsyncIntegrationsResource,
+    IntegrationsResourceWithRawResponse,
+    AsyncIntegrationsResourceWithRawResponse,
+    IntegrationsResourceWithStreamingResponse,
+    AsyncIntegrationsResourceWithStreamingResponse,
+)
 from .knowledge_bases import (
     KnowledgeBasesResource,
     AsyncKnowledgeBasesResource,
@@ -58,4 +66,10 @@
     "AsyncThreadsResourceWithRawResponse",
     "ThreadsResourceWithStreamingResponse",
     "AsyncThreadsResourceWithStreamingResponse",
+    "IntegrationsResource",
+    "AsyncIntegrationsResource",
+    "IntegrationsResourceWithRawResponse",
+    "AsyncIntegrationsResourceWithRawResponse",
+    "IntegrationsResourceWithStreamingResponse",
+    "AsyncIntegrationsResourceWithStreamingResponse",
 ]