Merge remote-tracking branch 'stainless/main'

Author: axl1313

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 42
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/cleanlab%2Fagility-f22c2d6c16f00b8d93c5db45c5586c80f1daebec0abec5ed155a17159ef44a61.yml
-openapi_spec_hash: 6d4b1288f7ca75879573547b316bf701
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/cleanlab%2Fagility-bb543bebe38d4cc889a3fa1ebc212458cd4321233d904357be98f8b22db82960.yml
+openapi_spec_hash: 7a30a005e382a8db9fafa55903c3a977
 config_hash: 6d2156cfe279456cf3c35ba5c66be1c1

pyproject.toml

@@ -147,6 +147,7 @@ exclude = [
 ]
 
 reportImplicitOverride = true
+reportOverlappingOverload = false
 
 reportImportCycles = false
 reportPrivateUsage = false

src/agility/_base_client.py

@@ -409,7 +409,8 @@ def _build_headers(self, options: FinalRequestOptions, *, retries_taken: int = 0
 
         idempotency_header = self._idempotency_header
         if idempotency_header and options.method.lower() != "get" and idempotency_header not in headers:
-            headers[idempotency_header] = options.idempotency_key or self._idempotency_key()
+            options.idempotency_key = options.idempotency_key or self._idempotency_key()
+            headers[idempotency_header] = options.idempotency_key
 
         # Don't set these headers if they were already set or removed by the caller. We check
         # `custom_headers`, which can contain `Omit()`, instead of `headers` to account for the removal case.
@@ -943,6 +944,10 @@ def _request(
         request = self._build_request(options, retries_taken=retries_taken)
         self._prepare_request(request)
 
+        if options.idempotency_key:
+            # ensure the idempotency key is reused between requests
+            input_options.idempotency_key = options.idempotency_key
+
         kwargs: HttpxSendArgs = {}
         if self.custom_auth is not None:
             kwargs["auth"] = self.custom_auth
@@ -1475,6 +1480,10 @@ async def _request(
         request = self._build_request(options, retries_taken=retries_taken)
         await self._prepare_request(request)
 
+        if options.idempotency_key:
+            # ensure the idempotency key is reused between requests
+            input_options.idempotency_key = options.idempotency_key
+
         kwargs: HttpxSendArgs = {}
         if self.custom_auth is not None:
             kwargs["auth"] = self.custom_auth

src/agility/types/threads/message.py

@@ -24,7 +24,7 @@
 
 
 class MetadataScoresContextSufficiencyLog(BaseModel):
-    explanation: str
+    explanation: Optional[str] = None
 
 
 class MetadataScoresContextSufficiency(BaseModel):
@@ -36,7 +36,7 @@ class MetadataScoresContextSufficiency(BaseModel):
 
 
 class MetadataScoresQueryEaseLog(BaseModel):
-    explanation: str
+    explanation: Optional[str] = None
 
 
 class MetadataScoresQueryEase(BaseModel):
@@ -48,7 +48,7 @@ class MetadataScoresQueryEase(BaseModel):
 
 
 class MetadataScoresResponseGroundednessLog(BaseModel):
-    explanation: str
+    explanation: Optional[str] = None
 
 
 class MetadataScoresResponseGroundedness(BaseModel):
@@ -60,7 +60,7 @@ class MetadataScoresResponseGroundedness(BaseModel):
 
 
 class MetadataScoresResponseHelpfulnessLog(BaseModel):
-    explanation: str
+    explanation: Optional[str] = None
 
 
 class MetadataScoresResponseHelpfulness(BaseModel):
@@ -72,7 +72,7 @@ class MetadataScoresResponseHelpfulness(BaseModel):
 
 
 class MetadataScoresTrustworthinessLog(BaseModel):
-    explanation: str
+    explanation: Optional[str] = None
 
 
 class MetadataScoresTrustworthiness(BaseModel):

src/agility/types/threads/message_create_params.py

@@ -31,7 +31,7 @@ class MessageCreateParams(TypedDict, total=False):
 
 
 class MetadataScoresContextSufficiencyLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class MetadataScoresContextSufficiency(TypedDict, total=False):
@@ -43,7 +43,7 @@ class MetadataScoresContextSufficiency(TypedDict, total=False):
 
 
 class MetadataScoresQueryEaseLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class MetadataScoresQueryEase(TypedDict, total=False):
@@ -55,7 +55,7 @@ class MetadataScoresQueryEase(TypedDict, total=False):
 
 
 class MetadataScoresResponseGroundednessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class MetadataScoresResponseGroundedness(TypedDict, total=False):
@@ -67,7 +67,7 @@ class MetadataScoresResponseGroundedness(TypedDict, total=False):
 
 
 class MetadataScoresResponseHelpfulnessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class MetadataScoresResponseHelpfulness(TypedDict, total=False):
@@ -79,7 +79,7 @@ class MetadataScoresResponseHelpfulness(TypedDict, total=False):
 
 
 class MetadataScoresTrustworthinessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class MetadataScoresTrustworthiness(TypedDict, total=False):

src/agility/types/threads/run_create_params.py

@@ -51,7 +51,7 @@ class RunCreateParams(TypedDict, total=False):
 
 
 class AdditionalMessageMetadataScoresContextSufficiencyLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresContextSufficiency(TypedDict, total=False):
@@ -63,7 +63,7 @@ class AdditionalMessageMetadataScoresContextSufficiency(TypedDict, total=False):
 
 
 class AdditionalMessageMetadataScoresQueryEaseLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresQueryEase(TypedDict, total=False):
@@ -75,7 +75,7 @@ class AdditionalMessageMetadataScoresQueryEase(TypedDict, total=False):
 
 
 class AdditionalMessageMetadataScoresResponseGroundednessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresResponseGroundedness(TypedDict, total=False):
@@ -87,7 +87,7 @@ class AdditionalMessageMetadataScoresResponseGroundedness(TypedDict, total=False
 
 
 class AdditionalMessageMetadataScoresResponseHelpfulnessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresResponseHelpfulness(TypedDict, total=False):
@@ -99,7 +99,7 @@ class AdditionalMessageMetadataScoresResponseHelpfulness(TypedDict, total=False)
 
 
 class AdditionalMessageMetadataScoresTrustworthinessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresTrustworthiness(TypedDict, total=False):

src/agility/types/threads/run_stream_params.py

@@ -51,7 +51,7 @@ class RunStreamParams(TypedDict, total=False):
 
 
 class AdditionalMessageMetadataScoresContextSufficiencyLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresContextSufficiency(TypedDict, total=False):
@@ -63,7 +63,7 @@ class AdditionalMessageMetadataScoresContextSufficiency(TypedDict, total=False):
 
 
 class AdditionalMessageMetadataScoresQueryEaseLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresQueryEase(TypedDict, total=False):
@@ -75,7 +75,7 @@ class AdditionalMessageMetadataScoresQueryEase(TypedDict, total=False):
 
 
 class AdditionalMessageMetadataScoresResponseGroundednessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresResponseGroundedness(TypedDict, total=False):
@@ -87,7 +87,7 @@ class AdditionalMessageMetadataScoresResponseGroundedness(TypedDict, total=False
 
 
 class AdditionalMessageMetadataScoresResponseHelpfulnessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresResponseHelpfulness(TypedDict, total=False):
@@ -99,7 +99,7 @@ class AdditionalMessageMetadataScoresResponseHelpfulness(TypedDict, total=False)
 
 
 class AdditionalMessageMetadataScoresTrustworthinessLog(TypedDict, total=False):
-    explanation: Required[str]
+    explanation: Optional[str]
 
 
 class AdditionalMessageMetadataScoresTrustworthiness(TypedDict, total=False):