scale-gp-beta 0.1.0a39__py3-none-any.whl → 0.1.0a40__py3-none-any.whl

scale_gp_beta/_streaming.py

@@ -54,11 +54,12 @@ class Stream(Generic[_T]):
         process_data = self._client._process_response_data
         iterator = self._iter_events()
 
-        for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
-
-        # As we might not fully consume the response stream, we need to close it explicitly
-        response.close()
+        try:
+            for sse in iterator:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+        finally:
+            # Ensure the response is closed even if the consumer doesn't read all data
+            response.close()
 
     def __enter__(self) -> Self:
         return self
@@ -117,11 +118,12 @@ class AsyncStream(Generic[_T]):
         process_data = self._client._process_response_data
         iterator = self._iter_events()
 
-        async for sse in iterator:
-            yield process_data(data=sse.json(), cast_to=cast_to, response=response)
-
-        # As we might not fully consume the response stream, we need to close it explicitly
-        await response.aclose()
+        try:
+            async for sse in iterator:
+                yield process_data(data=sse.json(), cast_to=cast_to, response=response)
+        finally:
+            # Ensure the response is closed even if the consumer doesn't read all data
+            await response.aclose()
 
     async def __aenter__(self) -> Self:
         return self

scale_gp_beta/_types.py

@@ -243,6 +243,9 @@ _T_co = TypeVar("_T_co", covariant=True)
 if TYPE_CHECKING:
     # This works because str.__contains__ does not accept object (either in typeshed or at runtime)
     # https://github.com/hauntsaninja/useful_types/blob/5e9710f3875107d068e7679fd7fec9cfab0eff3b/useful_types/__init__.py#L285
+    #
+    # Note: index() and count() methods are intentionally omitted to allow pyright to properly
+    # infer TypedDict types when dict literals are used in lists assigned to SequenceNotStr.
     class SequenceNotStr(Protocol[_T_co]):
         @overload
         def __getitem__(self, index: SupportsIndex, /) -> _T_co: ...
@@ -251,8 +254,6 @@ if TYPE_CHECKING:
         def __contains__(self, value: object, /) -> bool: ...
         def __len__(self) -> int: ...
         def __iter__(self) -> Iterator[_T_co]: ...
-        def index(self, value: Any, start: int = 0, stop: int = ..., /) -> int: ...
-        def count(self, value: Any, /) -> int: ...
         def __reversed__(self) -> Iterator[_T_co]: ...
 else:
     # just point this to a normal `Sequence` at runtime to avoid having to special case

scale_gp_beta/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "scale_gp_beta"
-__version__ = "0.1.0-alpha.39"  # x-release-please-version
+__version__ = "0.1.0-alpha.40"  # x-release-please-version

scale_gp_beta/resources/chat/completions.py

@@ -539,6 +539,7 @@ class CompletionsResource(SyncAPIResource):
             "fireworks_ai",
         ]
         | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -572,6 +573,7 @@ class CompletionsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "limit": limit,
                         "model_vendor": model_vendor,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
@@ -1094,6 +1096,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
             "fireworks_ai",
         ]
         | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -1127,6 +1130,7 @@ class AsyncCompletionsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "limit": limit,
                         "model_vendor": model_vendor,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },

scale_gp_beta/resources/credentials.py

@@ -201,6 +201,7 @@ class CredentialsResource(SyncAPIResource):
         ending_before: str | Omit = omit,
         limit: int | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -238,6 +239,7 @@ class CredentialsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "limit": limit,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
@@ -554,6 +556,7 @@ class AsyncCredentialsResource(AsyncAPIResource):
         ending_before: str | Omit = omit,
         limit: int | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -591,6 +594,7 @@ class AsyncCredentialsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "limit": limit,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },

scale_gp_beta/resources/dataset_items.py

@@ -146,6 +146,7 @@ class DatasetItemsResource(SyncAPIResource):
         ending_before: str | Omit = omit,
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         version: int | Omit = omit,
@@ -189,6 +190,7 @@ class DatasetItemsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "version": version,
@@ -396,6 +398,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
         ending_before: str | Omit = omit,
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         version: int | Omit = omit,
@@ -439,6 +442,7 @@ class AsyncDatasetItemsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "version": version,

scale_gp_beta/resources/datasets.py

@@ -238,6 +238,7 @@ class DatasetsResource(SyncAPIResource):
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
@@ -274,6 +275,7 @@ class DatasetsResource(SyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,
@@ -530,6 +532,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
@@ -566,6 +569,7 @@ class AsyncDatasetsResource(AsyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,

scale_gp_beta/resources/evaluation_items.py

@@ -91,6 +91,7 @@ class EvaluationItemsResource(SyncAPIResource):
         evaluation_id: str | Omit = omit,
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -126,6 +127,7 @@ class EvaluationItemsResource(SyncAPIResource):
                         "evaluation_id": evaluation_id,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
@@ -203,6 +205,7 @@ class AsyncEvaluationItemsResource(AsyncAPIResource):
         evaluation_id: str | Omit = omit,
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -238,6 +241,7 @@ class AsyncEvaluationItemsResource(AsyncAPIResource):
                         "evaluation_id": evaluation_id,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },

scale_gp_beta/resources/evaluations.py

@@ -370,6 +370,7 @@ class EvaluationsResource(SyncAPIResource):
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
@@ -407,6 +408,7 @@ class EvaluationsResource(SyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,
@@ -791,6 +793,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         include_archived: bool | Omit = omit,
         limit: int | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         tags: SequenceNotStr[str] | Omit = omit,
@@ -828,6 +831,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                         "include_archived": include_archived,
                         "limit": limit,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "tags": tags,

scale_gp_beta/resources/files/files.py

@@ -172,6 +172,7 @@ class FilesResource(SyncAPIResource):
         ending_before: str | Omit = omit,
         filename: str | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -208,6 +209,7 @@ class FilesResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "filename": filename,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
@@ -422,6 +424,7 @@ class AsyncFilesResource(AsyncAPIResource):
         ending_before: str | Omit = omit,
         filename: str | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -458,6 +461,7 @@ class AsyncFilesResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "filename": filename,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },

scale_gp_beta/resources/models.py

@@ -320,6 +320,7 @@ class ModelsResource(SyncAPIResource):
         ]
         | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -355,6 +356,7 @@ class ModelsResource(SyncAPIResource):
                         "limit": limit,
                         "model_vendor": model_vendor,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
@@ -692,6 +694,7 @@ class AsyncModelsResource(AsyncAPIResource):
         ]
         | Omit = omit,
         name: str | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -727,6 +730,7 @@ class AsyncModelsResource(AsyncAPIResource):
                         "limit": limit,
                         "model_vendor": model_vendor,
                         "name": name,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },

scale_gp_beta/resources/questions.py

@@ -339,6 +339,7 @@ class QuestionsResource(SyncAPIResource):
         *,
         ending_before: str | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -372,6 +373,7 @@ class QuestionsResource(SyncAPIResource):
                     {
                         "ending_before": ending_before,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },
@@ -696,6 +698,7 @@ class AsyncQuestionsResource(AsyncAPIResource):
         *,
         ending_before: str | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -729,6 +732,7 @@ class AsyncQuestionsResource(AsyncAPIResource):
                     {
                         "ending_before": ending_before,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                     },

scale_gp_beta/resources/spans.py

@@ -252,9 +252,13 @@ class SpansResource(SyncAPIResource):
         ending_before: str | Omit = omit,
         from_ts: Union[str, datetime] | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         to_ts: Union[str, datetime] | Omit = omit,
+        acp_types: SequenceNotStr[str] | Omit = omit,
+        agentex_agent_ids: SequenceNotStr[str] | Omit = omit,
+        agentex_agent_names: SequenceNotStr[str] | Omit = omit,
         application_variant_ids: SequenceNotStr[str] | Omit = omit,
         assessment_types: SequenceNotStr[str] | Omit = omit,
         excluded_span_ids: SequenceNotStr[str] | Omit = omit,
@@ -285,6 +289,12 @@ class SpansResource(SyncAPIResource):
 
           to_ts: The ending (most recent) timestamp in ISO format.
 
+          acp_types: Filter by ACP types
+
+          agentex_agent_ids: Filter by Agentex agent IDs
+
+          agentex_agent_names: Filter by Agentex agent names
+
           application_variant_ids: Filter by application variant IDs
 
           assessment_types: Filter spans by traces that have assessments of these types
@@ -326,6 +336,9 @@ class SpansResource(SyncAPIResource):
             page=SyncCursorPage[Span],
             body=maybe_transform(
                 {
+                    "acp_types": acp_types,
+                    "agentex_agent_ids": agentex_agent_ids,
+                    "agentex_agent_names": agentex_agent_names,
                     "application_variant_ids": application_variant_ids,
                     "assessment_types": assessment_types,
                     "excluded_span_ids": excluded_span_ids,
@@ -354,6 +367,7 @@ class SpansResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "from_ts": from_ts,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "to_ts": to_ts,
@@ -612,9 +626,13 @@ class AsyncSpansResource(AsyncAPIResource):
         ending_before: str | Omit = omit,
         from_ts: Union[str, datetime] | Omit = omit,
         limit: int | Omit = omit,
+        sort_by: str | Omit = omit,
         sort_order: Literal["asc", "desc"] | Omit = omit,
         starting_after: str | Omit = omit,
         to_ts: Union[str, datetime] | Omit = omit,
+        acp_types: SequenceNotStr[str] | Omit = omit,
+        agentex_agent_ids: SequenceNotStr[str] | Omit = omit,
+        agentex_agent_names: SequenceNotStr[str] | Omit = omit,
         application_variant_ids: SequenceNotStr[str] | Omit = omit,
         assessment_types: SequenceNotStr[str] | Omit = omit,
         excluded_span_ids: SequenceNotStr[str] | Omit = omit,
@@ -645,6 +663,12 @@ class AsyncSpansResource(AsyncAPIResource):
 
           to_ts: The ending (most recent) timestamp in ISO format.
 
+          acp_types: Filter by ACP types
+
+          agentex_agent_ids: Filter by Agentex agent IDs
+
+          agentex_agent_names: Filter by Agentex agent names
+
           application_variant_ids: Filter by application variant IDs
 
           assessment_types: Filter spans by traces that have assessments of these types
@@ -686,6 +710,9 @@ class AsyncSpansResource(AsyncAPIResource):
             page=AsyncCursorPage[Span],
             body=maybe_transform(
                 {
+                    "acp_types": acp_types,
+                    "agentex_agent_ids": agentex_agent_ids,
+                    "agentex_agent_names": agentex_agent_names,
                     "application_variant_ids": application_variant_ids,
                     "assessment_types": assessment_types,
                     "excluded_span_ids": excluded_span_ids,
@@ -714,6 +741,7 @@ class AsyncSpansResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "from_ts": from_ts,
                         "limit": limit,
+                        "sort_by": sort_by,
                         "sort_order": sort_order,
                         "starting_after": starting_after,
                         "to_ts": to_ts,

scale_gp_beta/types/chat/completion_models_params.py

@@ -27,6 +27,8 @@ class CompletionModelsParams(TypedDict, total=False):
         "fireworks_ai",
     ]
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/credential_list_params.py

@@ -15,6 +15,8 @@ class CredentialListParams(TypedDict, total=False):
     name: str
     """Filter credentials by name"""
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/dataset_item_list_params.py

@@ -20,6 +20,8 @@ class DatasetItemListParams(TypedDict, total=False):
 
     limit: int
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/dataset_list_params.py

@@ -18,6 +18,8 @@ class DatasetListParams(TypedDict, total=False):
 
     name: str
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/evaluation.py

@@ -44,6 +44,8 @@ class ProgressWorkflows(BaseModel):
 
 
 class Progress(BaseModel):
+    """Progress of the evaluation's underlying async job"""
+
     items: Optional[ProgressItems] = None
 
     workflows: Optional[ProgressWorkflows] = None

scale_gp_beta/types/evaluation_create_params.py

@@ -93,6 +93,8 @@ class EvaluationWithDatasetCreateRequest(TypedDict, total=False):
 
 
 class EvaluationWithDatasetCreateRequestDataset(TypedDict, total=False):
+    """Create a reusable dataset from items in the `data` field"""
+
     name: Required[str]
 
     description: str

scale_gp_beta/types/evaluation_item_list_params.py

@@ -16,6 +16,8 @@ class EvaluationItemListParams(TypedDict, total=False):
 
     limit: int
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/evaluation_list_params.py

@@ -19,6 +19,8 @@ class EvaluationListParams(TypedDict, total=False):
 
     name: str
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/evaluation_task.py

@@ -210,6 +210,8 @@ class ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplication
 
 
 class ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverrides(BaseModel):
+    """Execution override options for agentic applications"""
+
     concurrent: Optional[bool] = None
 
     initial_state: Optional[
@@ -259,7 +261,9 @@ class ApplicationVariantV1EvaluationTask(BaseModel):
 class AgentexOutputEvaluationTaskConfiguration(BaseModel):
     agentex_agent_id: str
 
-    input_column: str
+    input_column: Union[ItemLocator, object]
+
+    include_traces: Union[bool, ItemLocator, None] = None
 
 
 class AgentexOutputEvaluationTask(BaseModel):

scale_gp_beta/types/evaluation_task_param.py

@@ -206,6 +206,8 @@ class ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplication
 
 
 class ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverrides(TypedDict, total=False):
+    """Execution override options for agentic applications"""
+
     concurrent: bool
 
     initial_state: ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverridesInitialState
@@ -251,7 +253,9 @@ class ApplicationVariantV1EvaluationTask(TypedDict, total=False):
 class AgentexOutputEvaluationTaskConfiguration(TypedDict, total=False):
     agentex_agent_id: Required[str]
 
-    input_column: Required[str]
+    input_column: Required[Union[ItemLocator, object]]
+
+    include_traces: Union[bool, ItemLocator]
 
 
 class AgentexOutputEvaluationTask(TypedDict, total=False):

scale_gp_beta/types/file_import_from_cloud_response.py

@@ -10,6 +10,10 @@ __all__ = ["FileImportFromCloudResponse", "Result", "ResultFile", "ResultFile_Fa
 
 
 class ResultFile_FailedFile(BaseModel):
+    """
+    Minimal file representation for failed uploads containing only essential information.
+    """
+
     filename: str
     """The original filename from the request"""
 

scale_gp_beta/types/file_list_params.py

@@ -15,6 +15,8 @@ class FileListParams(TypedDict, total=False):
 
     limit: int
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/inference_create_params.py

@@ -20,6 +20,8 @@ class InferenceCreateParams(TypedDict, total=False):
 
 
 class InferenceConfiguration(TypedDict, total=False):
+    """Vendor specific configuration"""
+
     num_retries: int
 
     timeout_seconds: int

scale_gp_beta/types/model_list_params.py

@@ -29,6 +29,8 @@ class ModelListParams(TypedDict, total=False):
 
     name: str
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/question_list_params.py

@@ -12,6 +12,8 @@ class QuestionListParams(TypedDict, total=False):
 
     limit: int
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str

scale_gp_beta/types/span_assessment.py

@@ -15,6 +15,8 @@ __all__ = ["SpanAssessment"]
 
 
 class SpanAssessment(BaseModel):
+    """Response model for span assessment"""
+
     assessment_id: str
     """Unique identifier for the assessment"""
 

scale_gp_beta/types/span_search_params.py

@@ -22,6 +22,8 @@ class SpanSearchParams(TypedDict, total=False):
 
     limit: int
 
+    sort_by: str
+
     sort_order: Literal["asc", "desc"]
 
     starting_after: str
@@ -29,6 +31,15 @@ class SpanSearchParams(TypedDict, total=False):
     to_ts: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
     """The ending (most recent) timestamp in ISO format."""
 
+    acp_types: SequenceNotStr[str]
+    """Filter by ACP types"""
+
+    agentex_agent_ids: SequenceNotStr[str]
+    """Filter by Agentex agent IDs"""
+
+    agentex_agent_names: SequenceNotStr[str]
+    """Filter by Agentex agent names"""
+
     application_variant_ids: SequenceNotStr[str]
     """Filter by application variant IDs"""