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

scale_gp_beta/resources/spans.py

@@ -4,13 +4,13 @@ from __future__ import annotations
 
 from typing import Dict, Union, Iterable
 from datetime import datetime
-from typing_extensions import Literal, overload
+from typing_extensions import Literal
 
 import httpx
 
-from ..types import span_list_params, span_batch_params, span_create_params, span_update_params
+from ..types import 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 .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (
@@ -19,9 +19,9 @@ from .._response import (
     async_to_raw_response_wrapper,
     async_to_streamed_response_wrapper,
 )
-from ..pagination import SyncCursorPage, AsyncCursorPage, SyncAPIListPage, AsyncAPIListPage
 from ..types.span import Span
-from .._base_client import AsyncPaginator, make_request_options
+from .._base_client import make_request_options
+from ..types.span_batch_response import SpanBatchResponse
 
 __all__ = ["SpansResource", "AsyncSpansResource"]
 
@@ -46,7 +46,6 @@ class SpansResource(SyncAPIResource):
         """
         return SpansResourceWithStreamingResponse(self)
 
-    @overload
     def create(
         self,
         *,
@@ -54,7 +53,10 @@ class SpansResource(SyncAPIResource):
         start_timestamp: Union[str, datetime],
         trace_id: str,
         id: str | NotGiven = NOT_GIVEN,
+        application_interaction_id: str | NotGiven = NOT_GIVEN,
+        application_variant_id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        group_id: str | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
         metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -104,77 +106,11 @@ class SpansResource(SyncAPIResource):
 
           id: The id of the span
 
-          parent_id: Reference to a parent span_id
-
-          extra_headers: Send extra headers
-
-          extra_query: Add additional query parameters to the request
+          application_interaction_id: The optional application interaction ID this span belongs to
 
-          extra_body: Add additional JSON properties to the request
+          application_variant_id: The optional application variant ID this span belongs to
 
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        ...
-
-    @overload
-    def create(
-        self,
-        *,
-        application_interaction_id: str,
-        application_variant_id: str,
-        name: str,
-        start_timestamp: Union[str, datetime],
-        trace_id: str,
-        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,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Span:
-        """
-        Create Span
-
-        Args:
-          trace_id: id for grouping traces together, uuid is recommended
-
-          id: The id of the span
+          group_id: Reference to a group_id
 
           parent_id: Reference to a parent span_id
 
@@ -186,63 +122,6 @@ class SpansResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        ...
-
-    @required_args(
-        ["name", "start_timestamp", "trace_id"],
-        ["application_interaction_id", "application_variant_id", "name", "start_timestamp", "trace_id"],
-    )
-    def create(
-        self,
-        *,
-        name: str,
-        start_timestamp: Union[str, datetime],
-        trace_id: str,
-        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.
-        # 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,
-    ) -> Span:
         return self._post(
             "/v5/spans",
             body=maybe_transform(
@@ -251,15 +130,16 @@ class SpansResource(SyncAPIResource):
                     "start_timestamp": start_timestamp,
                     "trace_id": trace_id,
                     "id": id,
+                    "application_interaction_id": application_interaction_id,
+                    "application_variant_id": application_variant_id,
                     "end_timestamp": end_timestamp,
+                    "group_id": group_id,
                     "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,
                 },
                 span_create_params.SpanCreateParams,
             ),
@@ -350,65 +230,6 @@ class SpansResource(SyncAPIResource):
             cast_to=Span,
         )
 
-    def list(
-        self,
-        *,
-        ending_before: str | NotGiven = NOT_GIVEN,
-        from_ts: int | NotGiven = NOT_GIVEN,
-        limit: int | 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: 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,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncCursorPage[Span]:
-        """
-        List Spans
-
-        Args:
-          from_ts: The starting (oldest) timestamp window in seconds.
-
-          to_ts: The ending (most recent) timestamp in seconds.
-
-          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",
-            page=SyncCursorPage[Span],
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "ending_before": ending_before,
-                        "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,
-                    },
-                    span_list_params.SpanListParams,
-                ),
-            ),
-            model=Span,
-        )
-
     def batch(
         self,
         *,
@@ -419,7 +240,7 @@ class SpansResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> SyncAPIListPage[Span]:
+    ) -> SpanBatchResponse:
         """
         Create Spans in Batch
 
@@ -432,15 +253,13 @@ class SpansResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return self._post(
             "/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",
+            cast_to=SpanBatchResponse,
         )
 
 
@@ -464,7 +283,6 @@ class AsyncSpansResource(AsyncAPIResource):
         """
         return AsyncSpansResourceWithStreamingResponse(self)
 
-    @overload
     async def create(
         self,
         *,
@@ -472,7 +290,10 @@ class AsyncSpansResource(AsyncAPIResource):
         start_timestamp: Union[str, datetime],
         trace_id: str,
         id: str | NotGiven = NOT_GIVEN,
+        application_interaction_id: str | NotGiven = NOT_GIVEN,
+        application_variant_id: str | NotGiven = NOT_GIVEN,
         end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        group_id: str | NotGiven = NOT_GIVEN,
         input: Dict[str, object] | NotGiven = NOT_GIVEN,
         metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
@@ -522,77 +343,11 @@ class AsyncSpansResource(AsyncAPIResource):
 
           id: The id of the span
 
-          parent_id: Reference to a parent span_id
-
-          extra_headers: Send extra headers
+          application_interaction_id: The optional application interaction ID this span belongs to
 
-          extra_query: Add additional query parameters to the request
-
-          extra_body: Add additional JSON properties to the request
+          application_variant_id: The optional application variant ID this span belongs to
 
-          timeout: Override the client-level default timeout for this request, in seconds
-        """
-        ...
-
-    @overload
-    async def create(
-        self,
-        *,
-        application_interaction_id: str,
-        application_variant_id: str,
-        name: str,
-        start_timestamp: Union[str, datetime],
-        trace_id: str,
-        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,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> Span:
-        """
-        Create Span
-
-        Args:
-          trace_id: id for grouping traces together, uuid is recommended
-
-          id: The id of the span
+          group_id: Reference to a group_id
 
           parent_id: Reference to a parent span_id
 
@@ -604,63 +359,6 @@ class AsyncSpansResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        ...
-
-    @required_args(
-        ["name", "start_timestamp", "trace_id"],
-        ["application_interaction_id", "application_variant_id", "name", "start_timestamp", "trace_id"],
-    )
-    async def create(
-        self,
-        *,
-        name: str,
-        start_timestamp: Union[str, datetime],
-        trace_id: str,
-        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.
-        # 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,
-    ) -> Span:
         return await self._post(
             "/v5/spans",
             body=await async_maybe_transform(
@@ -669,15 +367,16 @@ class AsyncSpansResource(AsyncAPIResource):
                     "start_timestamp": start_timestamp,
                     "trace_id": trace_id,
                     "id": id,
+                    "application_interaction_id": application_interaction_id,
+                    "application_variant_id": application_variant_id,
                     "end_timestamp": end_timestamp,
+                    "group_id": group_id,
                     "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,
                 },
                 span_create_params.SpanCreateParams,
             ),
@@ -768,66 +467,7 @@ class AsyncSpansResource(AsyncAPIResource):
             cast_to=Span,
         )
 
-    def list(
-        self,
-        *,
-        ending_before: str | NotGiven = NOT_GIVEN,
-        from_ts: int | NotGiven = NOT_GIVEN,
-        limit: int | 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: 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,
-        extra_query: Query | None = None,
-        extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[Span, AsyncCursorPage[Span]]:
-        """
-        List Spans
-
-        Args:
-          from_ts: The starting (oldest) timestamp window in seconds.
-
-          to_ts: The ending (most recent) timestamp in seconds.
-
-          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",
-            page=AsyncCursorPage[Span],
-            options=make_request_options(
-                extra_headers=extra_headers,
-                extra_query=extra_query,
-                extra_body=extra_body,
-                timeout=timeout,
-                query=maybe_transform(
-                    {
-                        "ending_before": ending_before,
-                        "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,
-                    },
-                    span_list_params.SpanListParams,
-                ),
-            ),
-            model=Span,
-        )
-
-    def batch(
+    async def batch(
         self,
         *,
         items: Iterable[span_batch_params.Item],
@@ -837,7 +477,7 @@ class AsyncSpansResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> AsyncPaginator[Span, AsyncAPIListPage[Span]]:
+    ) -> SpanBatchResponse:
         """
         Create Spans in Batch
 
@@ -850,15 +490,13 @@ class AsyncSpansResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        return self._get_api_list(
+        return await self._post(
             "/v5/spans/batch",
-            page=AsyncAPIListPage[Span],
-            body=maybe_transform({"items": items}, span_batch_params.SpanBatchParams),
+            body=await async_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",
+            cast_to=SpanBatchResponse,
         )
 
 
@@ -875,9 +513,6 @@ class SpansResourceWithRawResponse:
         self.update = to_raw_response_wrapper(
             spans.update,
         )
-        self.list = to_raw_response_wrapper(
-            spans.list,
-        )
         self.batch = to_raw_response_wrapper(
             spans.batch,
         )
@@ -896,9 +531,6 @@ class AsyncSpansResourceWithRawResponse:
         self.update = async_to_raw_response_wrapper(
             spans.update,
         )
-        self.list = async_to_raw_response_wrapper(
-            spans.list,
-        )
         self.batch = async_to_raw_response_wrapper(
             spans.batch,
         )
@@ -917,9 +549,6 @@ class SpansResourceWithStreamingResponse:
         self.update = to_streamed_response_wrapper(
             spans.update,
         )
-        self.list = to_streamed_response_wrapper(
-            spans.list,
-        )
         self.batch = to_streamed_response_wrapper(
             spans.batch,
         )
@@ -938,9 +567,6 @@ class AsyncSpansResourceWithStreamingResponse:
         self.update = async_to_streamed_response_wrapper(
             spans.update,
         )
-        self.list = async_to_streamed_response_wrapper(
-            spans.list,
-        )
         self.batch = async_to_streamed_response_wrapper(
             spans.batch,
         )

scale_gp_beta/types/__init__.py

@@ -18,7 +18,6 @@ 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
@@ -29,6 +28,7 @@ from .span_update_params import SpanUpdateParams as SpanUpdateParams
 from .dataset_list_params import DatasetListParams as DatasetListParams
 from .model_create_params import ModelCreateParams as ModelCreateParams
 from .model_update_params import ModelUpdateParams as ModelUpdateParams
+from .span_batch_response import SpanBatchResponse as SpanBatchResponse
 from .file_delete_response import FileDeleteResponse as FileDeleteResponse
 from .inference_model_list import InferenceModelList as InferenceModelList
 from .dataset_create_params import DatasetCreateParams as DatasetCreateParams
@@ -54,3 +54,4 @@ from .dataset_item_delete_response import DatasetItemDeleteResponse as DatasetIt
 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/model_definition.py

@@ -10,13 +10,13 @@ __all__ = ["ModelDefinition"]
 
 
 class ModelDefinition(BaseModel):
-    api_model_name: str = FieldInfo(alias="model_name")
+    name: str = FieldInfo(alias="model_name")
     """model name, for example `gpt-4o`"""
 
-    api_model_type: Literal["generic", "completion", "chat_completion"] = FieldInfo(alias="model_type")
+    type: Literal["generic", "completion", "chat_completion"] = FieldInfo(alias="model_type")
     """model type, for example `chat_completion`"""
 
-    api_model_vendor: Literal[
+    vendor: Literal[
         "openai",
         "cohere",
         "vertex_ai",

scale_gp_beta/types/dataset_item_batch_create_response.py

@@ -0,0 +1,15 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from typing_extensions import Literal
+
+from .._models import BaseModel
+from .dataset_item import DatasetItem
+
+__all__ = ["DatasetItemBatchCreateResponse"]
+
+
+class DatasetItemBatchCreateResponse(BaseModel):
+    items: List[DatasetItem]
+
+    object: Optional[Literal["list"]] = None