scale-gp-beta 0.1.0a12__py3-none-any.whl → 0.1.0a14__py3-none-any.whl

scale_gp_beta/resources/spans.py

@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Optional
+from typing import Dict, Union, Iterable
 from datetime import datetime
-from typing_extensions import overload
+from typing_extensions import Literal, overload
 
 import httpx
 
-from ..types import span_list_params, span_create_params, span_update_params
+from ..types import span_list_params, span_batch_params, span_create_params, span_update_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -19,7 +19,7 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..pagination import SyncCursorPage, AsyncCursorPage
+from ..pagination import SyncCursorPage, AsyncCursorPage, SyncAPIListPage, AsyncAPIListPage
 from ..types.span import Span
 from .._base_client import AsyncPaginator, make_request_options
 
@@ -53,11 +53,42 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        type: Literal[
+            "TEXT_INPUT",
+            "TEXT_OUTPUT",
+            "COMPLETION_INPUT",
+            "COMPLETION",
+            "KB_RETRIEVAL",
+            "KB_INPUT",
+            "RERANKING",
+            "EXTERNAL_ENDPOINT",
+            "PROMPT_ENGINEERING",
+            "DOCUMENT_INPUT",
+            "MAP_REDUCE",
+            "DOCUMENT_SEARCH",
+            "DOCUMENT_PROMPT",
+            "CUSTOM",
+            "INPUT_GUARDRAIL",
+            "OUTPUT_GUARDRAIL",
+            "CODE_EXECUTION",
+            "DATA_MANIPULATION",
+            "EVALUATION",
+            "FILE_RETRIEVAL",
+            "KB_ADD_CHUNK",
+            "KB_MANAGEMENT",
+            "TRACER",
+            "AGENT_TRACER",
+            "AGENT_WORKFLOW",
+            "STANDALONE",
+        ]
+        | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -71,6 +102,8 @@ class SpansResource(SyncAPIResource):
         Args:
           trace_id: id for grouping traces together, uuid is recommended
 
+          id: The id of the span
+
           parent_id: Reference to a parent span_id
 
           extra_headers: Send extra headers
@@ -92,11 +125,42 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        type: Literal[
+            "TEXT_INPUT",
+            "TEXT_OUTPUT",
+            "COMPLETION_INPUT",
+            "COMPLETION",
+            "KB_RETRIEVAL",
+            "KB_INPUT",
+            "RERANKING",
+            "EXTERNAL_ENDPOINT",
+            "PROMPT_ENGINEERING",
+            "DOCUMENT_INPUT",
+            "MAP_REDUCE",
+            "DOCUMENT_SEARCH",
+            "DOCUMENT_PROMPT",
+            "CUSTOM",
+            "INPUT_GUARDRAIL",
+            "OUTPUT_GUARDRAIL",
+            "CODE_EXECUTION",
+            "DATA_MANIPULATION",
+            "EVALUATION",
+            "FILE_RETRIEVAL",
+            "KB_ADD_CHUNK",
+            "KB_MANAGEMENT",
+            "TRACER",
+            "AGENT_TRACER",
+            "AGENT_WORKFLOW",
+            "STANDALONE",
+        ]
+        | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -110,6 +174,8 @@ class SpansResource(SyncAPIResource):
         Args:
           trace_id: id for grouping traces together, uuid is recommended
 
+          id: The id of the span
+
           parent_id: Reference to a parent span_id
 
           extra_headers: Send extra headers
@@ -132,11 +198,42 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        type: Literal[
+            "TEXT_INPUT",
+            "TEXT_OUTPUT",
+            "COMPLETION_INPUT",
+            "COMPLETION",
+            "KB_RETRIEVAL",
+            "KB_INPUT",
+            "RERANKING",
+            "EXTERNAL_ENDPOINT",
+            "PROMPT_ENGINEERING",
+            "DOCUMENT_INPUT",
+            "MAP_REDUCE",
+            "DOCUMENT_SEARCH",
+            "DOCUMENT_PROMPT",
+            "CUSTOM",
+            "INPUT_GUARDRAIL",
+            "OUTPUT_GUARDRAIL",
+            "CODE_EXECUTION",
+            "DATA_MANIPULATION",
+            "EVALUATION",
+            "FILE_RETRIEVAL",
+            "KB_ADD_CHUNK",
+            "KB_MANAGEMENT",
+            "TRACER",
+            "AGENT_TRACER",
+            "AGENT_WORKFLOW",
+            "STANDALONE",
+        ]
+        | NotGiven = NOT_GIVEN,
         application_interaction_id: str | NotGiven = NOT_GIVEN,
         application_variant_id: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -153,11 +250,14 @@ class SpansResource(SyncAPIResource):
                     "name": name,
                     "start_timestamp": start_timestamp,
                     "trace_id": trace_id,
-                    "data": data,
+                    "id": id,
                     "end_timestamp": end_timestamp,
                     "input": input,
+                    "metadata": metadata,
                     "output": output,
                     "parent_id": parent_id,
+                    "status": status,
+                    "type": type,
                     "application_interaction_id": application_interaction_id,
                     "application_variant_id": application_variant_id,
                 },
@@ -206,10 +306,11 @@ class SpansResource(SyncAPIResource):
         self,
         span_id: str,
         *,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         name: str | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -235,10 +336,11 @@ class SpansResource(SyncAPIResource):
             f"/v5/spans/{span_id}",
             body=maybe_transform(
                 {
-                    "data": data,
                     "end_timestamp": end_timestamp,
+                    "metadata": metadata,
                     "name": name,
                     "output": output,
+                    "status": status,
                 },
                 span_update_params.SpanUpdateParams,
             ),
@@ -251,13 +353,14 @@ class SpansResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         from_ts: int | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        parents_only: Optional[bool] | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        parents_only: bool | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
         to_ts: int | NotGiven = NOT_GIVEN,
-        trace_id: Optional[str] | NotGiven = NOT_GIVEN,
+        trace_id: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -295,6 +398,7 @@ class SpansResource(SyncAPIResource):
                         "from_ts": from_ts,
                         "limit": limit,
                         "parents_only": parents_only,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "to_ts": to_ts,
                         "trace_id": trace_id,
@@ -305,6 +409,40 @@ class SpansResource(SyncAPIResource):
             model=Span,
         )
 
+    def batch(
+        self,
+        *,
+        items: Iterable[span_batch_params.Item],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> SyncAPIListPage[Span]:
+        """
+        Create Spans in Batch
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._get_api_list(
+            "/v5/spans/batch",
+            page=SyncAPIListPage[Span],
+            body=maybe_transform({"items": items}, span_batch_params.SpanBatchParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            model=Span,
+            method="post",
+        )
+
 
 class AsyncSpansResource(AsyncAPIResource):
     @cached_property
@@ -333,11 +471,42 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        type: Literal[
+            "TEXT_INPUT",
+            "TEXT_OUTPUT",
+            "COMPLETION_INPUT",
+            "COMPLETION",
+            "KB_RETRIEVAL",
+            "KB_INPUT",
+            "RERANKING",
+            "EXTERNAL_ENDPOINT",
+            "PROMPT_ENGINEERING",
+            "DOCUMENT_INPUT",
+            "MAP_REDUCE",
+            "DOCUMENT_SEARCH",
+            "DOCUMENT_PROMPT",
+            "CUSTOM",
+            "INPUT_GUARDRAIL",
+            "OUTPUT_GUARDRAIL",
+            "CODE_EXECUTION",
+            "DATA_MANIPULATION",
+            "EVALUATION",
+            "FILE_RETRIEVAL",
+            "KB_ADD_CHUNK",
+            "KB_MANAGEMENT",
+            "TRACER",
+            "AGENT_TRACER",
+            "AGENT_WORKFLOW",
+            "STANDALONE",
+        ]
+        | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -351,6 +520,8 @@ class AsyncSpansResource(AsyncAPIResource):
         Args:
           trace_id: id for grouping traces together, uuid is recommended
 
+          id: The id of the span
+
           parent_id: Reference to a parent span_id
 
           extra_headers: Send extra headers
@@ -372,11 +543,42 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        type: Literal[
+            "TEXT_INPUT",
+            "TEXT_OUTPUT",
+            "COMPLETION_INPUT",
+            "COMPLETION",
+            "KB_RETRIEVAL",
+            "KB_INPUT",
+            "RERANKING",
+            "EXTERNAL_ENDPOINT",
+            "PROMPT_ENGINEERING",
+            "DOCUMENT_INPUT",
+            "MAP_REDUCE",
+            "DOCUMENT_SEARCH",
+            "DOCUMENT_PROMPT",
+            "CUSTOM",
+            "INPUT_GUARDRAIL",
+            "OUTPUT_GUARDRAIL",
+            "CODE_EXECUTION",
+            "DATA_MANIPULATION",
+            "EVALUATION",
+            "FILE_RETRIEVAL",
+            "KB_ADD_CHUNK",
+            "KB_MANAGEMENT",
+            "TRACER",
+            "AGENT_TRACER",
+            "AGENT_WORKFLOW",
+            "STANDALONE",
+        ]
+        | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -390,6 +592,8 @@ class AsyncSpansResource(AsyncAPIResource):
         Args:
           trace_id: id for grouping traces together, uuid is recommended
 
+          id: The id of the span
+
           parent_id: Reference to a parent span_id
 
           extra_headers: Send extra headers
@@ -412,11 +616,42 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        type: Literal[
+            "TEXT_INPUT",
+            "TEXT_OUTPUT",
+            "COMPLETION_INPUT",
+            "COMPLETION",
+            "KB_RETRIEVAL",
+            "KB_INPUT",
+            "RERANKING",
+            "EXTERNAL_ENDPOINT",
+            "PROMPT_ENGINEERING",
+            "DOCUMENT_INPUT",
+            "MAP_REDUCE",
+            "DOCUMENT_SEARCH",
+            "DOCUMENT_PROMPT",
+            "CUSTOM",
+            "INPUT_GUARDRAIL",
+            "OUTPUT_GUARDRAIL",
+            "CODE_EXECUTION",
+            "DATA_MANIPULATION",
+            "EVALUATION",
+            "FILE_RETRIEVAL",
+            "KB_ADD_CHUNK",
+            "KB_MANAGEMENT",
+            "TRACER",
+            "AGENT_TRACER",
+            "AGENT_WORKFLOW",
+            "STANDALONE",
+        ]
+        | NotGiven = NOT_GIVEN,
         application_interaction_id: str | NotGiven = NOT_GIVEN,
         application_variant_id: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -433,11 +668,14 @@ class AsyncSpansResource(AsyncAPIResource):
                     "name": name,
                     "start_timestamp": start_timestamp,
                     "trace_id": trace_id,
-                    "data": data,
+                    "id": id,
                     "end_timestamp": end_timestamp,
                     "input": input,
+                    "metadata": metadata,
                     "output": output,
                     "parent_id": parent_id,
+                    "status": status,
+                    "type": type,
                     "application_interaction_id": application_interaction_id,
                     "application_variant_id": application_variant_id,
                 },
@@ -486,10 +724,11 @@ class AsyncSpansResource(AsyncAPIResource):
         self,
         span_id: str,
         *,
-        data: Dict[str, object] | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         name: str | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -515,10 +754,11 @@ class AsyncSpansResource(AsyncAPIResource):
             f"/v5/spans/{span_id}",
             body=await async_maybe_transform(
                 {
-                    "data": data,
                     "end_timestamp": end_timestamp,
+                    "metadata": metadata,
                     "name": name,
                     "output": output,
+                    "status": status,
                 },
                 span_update_params.SpanUpdateParams,
             ),
@@ -531,13 +771,14 @@ class AsyncSpansResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         from_ts: int | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        parents_only: Optional[bool] | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        parents_only: bool | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
         to_ts: int | NotGiven = NOT_GIVEN,
-        trace_id: Optional[str] | NotGiven = NOT_GIVEN,
+        trace_id: str | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
@@ -575,6 +816,7 @@ class AsyncSpansResource(AsyncAPIResource):
                         "from_ts": from_ts,
                         "limit": limit,
                         "parents_only": parents_only,
+                        "sort_order": sort_order,
                         "starting_after": starting_after,
                         "to_ts": to_ts,
                         "trace_id": trace_id,
@@ -585,6 +827,40 @@ class AsyncSpansResource(AsyncAPIResource):
             model=Span,
         )
 
+    def batch(
+        self,
+        *,
+        items: Iterable[span_batch_params.Item],
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> AsyncPaginator[Span, AsyncAPIListPage[Span]]:
+        """
+        Create Spans in Batch
+
+        Args:
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._get_api_list(
+            "/v5/spans/batch",
+            page=AsyncAPIListPage[Span],
+            body=maybe_transform({"items": items}, span_batch_params.SpanBatchParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            model=Span,
+            method="post",
+        )
+
 
 class SpansResourceWithRawResponse:
     def __init__(self, spans: SpansResource) -> None:
@@ -602,6 +878,9 @@ class SpansResourceWithRawResponse:
         self.list = to_raw_response_wrapper(
             spans.list,
         )
+        self.batch = to_raw_response_wrapper(
+            spans.batch,
+        )
 
 
 class AsyncSpansResourceWithRawResponse:
@@ -620,6 +899,9 @@ class AsyncSpansResourceWithRawResponse:
         self.list = async_to_raw_response_wrapper(
             spans.list,
         )
+        self.batch = async_to_raw_response_wrapper(
+            spans.batch,
+        )
 
 
 class SpansResourceWithStreamingResponse:
@@ -638,6 +920,9 @@ class SpansResourceWithStreamingResponse:
         self.list = to_streamed_response_wrapper(
             spans.list,
         )
+        self.batch = to_streamed_response_wrapper(
+            spans.batch,
+        )
 
 
 class AsyncSpansResourceWithStreamingResponse:
@@ -656,3 +941,6 @@ class AsyncSpansResourceWithStreamingResponse:
         self.list = async_to_streamed_response_wrapper(
             spans.list,
         )
+        self.batch = async_to_streamed_response_wrapper(
+            spans.batch,
+        )

scale_gp_beta/types/__init__.py

@@ -5,16 +5,22 @@ from __future__ import annotations
 from .file import File as File
 from .span import Span as Span
 from .dataset import Dataset as Dataset
+from .component import Component as Component
+from .container import Container as Container
 from .file_list import FileList as FileList
 from .completion import Completion as Completion
 from .evaluation import Evaluation as Evaluation
 from .dataset_item import DatasetItem as DatasetItem
+from .item_locator import ItemLocator as ItemLocator
+from .component_param import ComponentParam as ComponentParam
+from .container_param import ContainerParam as ContainerParam
 from .evaluation_item import EvaluationItem as EvaluationItem
 from .evaluation_task import EvaluationTask as EvaluationTask
 from .inference_model import InferenceModel as InferenceModel
 from .file_list_params import FileListParams as FileListParams
 from .span_list_params import SpanListParams as SpanListParams
 from .model_list_params import ModelListParams as ModelListParams
+from .span_batch_params import SpanBatchParams as SpanBatchParams
 from .file_create_params import FileCreateParams as FileCreateParams
 from .file_update_params import FileUpdateParams as FileUpdateParams
 from .inference_response import InferenceResponse as InferenceResponse
@@ -28,6 +34,7 @@ from .inference_model_list import InferenceModelList as InferenceModelList
 from .dataset_create_params import DatasetCreateParams as DatasetCreateParams
 from .dataset_update_params import DatasetUpdateParams as DatasetUpdateParams
 from .evaluation_task_param import EvaluationTaskParam as EvaluationTaskParam
+from .item_locator_template import ItemLocatorTemplate as ItemLocatorTemplate
 from .model_delete_response import ModelDeleteResponse as ModelDeleteResponse
 from .evaluation_list_params import EvaluationListParams as EvaluationListParams
 from .dataset_delete_response import DatasetDeleteResponse as DatasetDeleteResponse
@@ -36,14 +43,14 @@ from .inference_create_params import InferenceCreateParams as InferenceCreatePar
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams
 from .dataset_item_list_params import DatasetItemListParams as DatasetItemListParams
 from .evaluation_create_params import EvaluationCreateParams as EvaluationCreateParams
+from .evaluation_update_params import EvaluationUpdateParams as EvaluationUpdateParams
 from .inference_response_chunk import InferenceResponseChunk as InferenceResponseChunk
 from .inference_create_response import InferenceCreateResponse as InferenceCreateResponse
 from .dataset_item_update_params import DatasetItemUpdateParams as DatasetItemUpdateParams
+from .evaluation_delete_response import EvaluationDeleteResponse as EvaluationDeleteResponse
 from .evaluation_retrieve_params import EvaluationRetrieveParams as EvaluationRetrieveParams
-from .evaluation_archive_response import EvaluationArchiveResponse as EvaluationArchiveResponse
 from .evaluation_item_list_params import EvaluationItemListParams as EvaluationItemListParams
 from .dataset_item_delete_response import DatasetItemDeleteResponse as DatasetItemDeleteResponse
 from .dataset_item_retrieve_params import DatasetItemRetrieveParams as DatasetItemRetrieveParams
 from .evaluation_item_retrieve_params import EvaluationItemRetrieveParams as EvaluationItemRetrieveParams
 from .dataset_item_batch_create_params import DatasetItemBatchCreateParams as DatasetItemBatchCreateParams
-from .dataset_item_batch_create_response import DatasetItemBatchCreateResponse as DatasetItemBatchCreateResponse

scale_gp_beta/types/chat/__init__.py

@@ -3,6 +3,9 @@
 from __future__ import annotations
 
 from .chat_completion import ChatCompletion as ChatCompletion
+from .model_definition import ModelDefinition as ModelDefinition
 from .chat_completion_chunk import ChatCompletionChunk as ChatCompletionChunk
 from .completion_create_params import CompletionCreateParams as CompletionCreateParams
+from .completion_models_params import CompletionModelsParams as CompletionModelsParams
 from .completion_create_response import CompletionCreateResponse as CompletionCreateResponse
+from .completion_models_response import CompletionModelsResponse as CompletionModelsResponse

scale_gp_beta/types/chat/completion_models_params.py

@@ -0,0 +1,31 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing_extensions import Literal, TypedDict
+
+__all__ = ["CompletionModelsParams"]
+
+
+class CompletionModelsParams(TypedDict, total=False):
+    ending_before: str
+
+    limit: int
+
+    model_vendor: Literal[
+        "openai",
+        "cohere",
+        "vertex_ai",
+        "anthropic",
+        "azure",
+        "gemini",
+        "launch",
+        "llmengine",
+        "model_zoo",
+        "bedrock",
+        "xai",
+    ]
+
+    sort_order: Literal["asc", "desc"]
+
+    starting_after: str

scale_gp_beta/types/{dataset_item_batch_create_response.py → chat/completion_models_response.py}

@@ -3,13 +3,13 @@
 from typing import List, Optional
 from typing_extensions import Literal
 
-from .._models import BaseModel
-from .dataset_item import DatasetItem
+from ..._models import BaseModel
+from .model_definition import ModelDefinition
 
-__all__ = ["DatasetItemBatchCreateResponse"]
+__all__ = ["CompletionModelsResponse"]
 
 
-class DatasetItemBatchCreateResponse(BaseModel):
-    items: List[DatasetItem]
+class CompletionModelsResponse(BaseModel):
+    items: List[ModelDefinition]
 
     object: Optional[Literal["list"]] = None