scale-gp-beta 0.1.0a11__py3-none-any.whl → 0.1.0a13__py3-none-any.whl

scale_gp_beta/resources/evaluations.py

@@ -2,12 +2,17 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Iterable, Optional
+from typing import Dict, List, Iterable
 from typing_extensions import Literal, overload
 
 import httpx
 
-from ..types import evaluation_list_params, evaluation_create_params, evaluation_retrieve_params
+from ..types import (
+    evaluation_list_params,
+    evaluation_create_params,
+    evaluation_update_params,
+    evaluation_retrieve_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
@@ -22,7 +27,7 @@ from ..pagination import SyncCursorPage, AsyncCursorPage
 from .._base_client import AsyncPaginator, make_request_options
 from ..types.evaluation import Evaluation
 from ..types.evaluation_task_param import EvaluationTaskParam
-from ..types.evaluation_archive_response import EvaluationArchiveResponse
+from ..types.evaluation_delete_response import EvaluationDeleteResponse
 
 __all__ = ["EvaluationsResource", "AsyncEvaluationsResource"]
 
@@ -246,13 +251,61 @@ class EvaluationsResource(SyncAPIResource):
             cast_to=Evaluation,
         )
 
+    def update(
+        self,
+        evaluation_id: str,
+        *,
+        description: str | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        tags: List[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,
+    ) -> Evaluation:
+        """
+        Update Evaluation
+
+        Args:
+          tags: The tags associated with the entity
+
+          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
+        """
+        if not evaluation_id:
+            raise ValueError(f"Expected a non-empty value for `evaluation_id` but received {evaluation_id!r}")
+        return self._patch(
+            f"/v5/evaluations/{evaluation_id}",
+            body=maybe_transform(
+                {
+                    "description": description,
+                    "name": name,
+                    "tags": tags,
+                },
+                evaluation_update_params.EvaluationUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Evaluation,
+        )
+
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         views: List[Literal["tasks"]] | 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.
@@ -286,7 +339,9 @@ class EvaluationsResource(SyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "name": name,
                         "starting_after": starting_after,
+                        "tags": tags,
                         "views": views,
                     },
                     evaluation_list_params.EvaluationListParams,
@@ -295,7 +350,7 @@ class EvaluationsResource(SyncAPIResource):
             model=Evaluation,
         )
 
-    def archive(
+    def delete(
         self,
         evaluation_id: str,
         *,
@@ -305,7 +360,7 @@ class EvaluationsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> EvaluationArchiveResponse:
+    ) -> EvaluationDeleteResponse:
         """
         Archive Evaluation
 
@@ -325,7 +380,7 @@ class EvaluationsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EvaluationArchiveResponse,
+            cast_to=EvaluationDeleteResponse,
         )
 
 
@@ -548,13 +603,61 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             cast_to=Evaluation,
         )
 
+    async def update(
+        self,
+        evaluation_id: str,
+        *,
+        description: str | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        tags: List[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,
+    ) -> Evaluation:
+        """
+        Update Evaluation
+
+        Args:
+          tags: The tags associated with the entity
+
+          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
+        """
+        if not evaluation_id:
+            raise ValueError(f"Expected a non-empty value for `evaluation_id` but received {evaluation_id!r}")
+        return await self._patch(
+            f"/v5/evaluations/{evaluation_id}",
+            body=await async_maybe_transform(
+                {
+                    "description": description,
+                    "name": name,
+                    "tags": tags,
+                },
+                evaluation_update_params.EvaluationUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Evaluation,
+        )
+
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         include_archived: bool | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
+        tags: List[str] | NotGiven = NOT_GIVEN,
         views: List[Literal["tasks"]] | 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.
@@ -588,7 +691,9 @@ class AsyncEvaluationsResource(AsyncAPIResource):
                         "ending_before": ending_before,
                         "include_archived": include_archived,
                         "limit": limit,
+                        "name": name,
                         "starting_after": starting_after,
+                        "tags": tags,
                         "views": views,
                     },
                     evaluation_list_params.EvaluationListParams,
@@ -597,7 +702,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             model=Evaluation,
         )
 
-    async def archive(
+    async def delete(
         self,
         evaluation_id: str,
         *,
@@ -607,7 +712,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
-    ) -> EvaluationArchiveResponse:
+    ) -> EvaluationDeleteResponse:
         """
         Archive Evaluation
 
@@ -627,7 +732,7 @@ class AsyncEvaluationsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=EvaluationArchiveResponse,
+            cast_to=EvaluationDeleteResponse,
         )
 
 
@@ -641,11 +746,14 @@ class EvaluationsResourceWithRawResponse:
         self.retrieve = to_raw_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = to_raw_response_wrapper(
+            evaluations.update,
+        )
         self.list = to_raw_response_wrapper(
             evaluations.list,
         )
-        self.archive = to_raw_response_wrapper(
-            evaluations.archive,
+        self.delete = to_raw_response_wrapper(
+            evaluations.delete,
         )
 
 
@@ -659,11 +767,14 @@ class AsyncEvaluationsResourceWithRawResponse:
         self.retrieve = async_to_raw_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = async_to_raw_response_wrapper(
+            evaluations.update,
+        )
         self.list = async_to_raw_response_wrapper(
             evaluations.list,
         )
-        self.archive = async_to_raw_response_wrapper(
-            evaluations.archive,
+        self.delete = async_to_raw_response_wrapper(
+            evaluations.delete,
         )
 
 
@@ -677,11 +788,14 @@ class EvaluationsResourceWithStreamingResponse:
         self.retrieve = to_streamed_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = to_streamed_response_wrapper(
+            evaluations.update,
+        )
         self.list = to_streamed_response_wrapper(
             evaluations.list,
         )
-        self.archive = to_streamed_response_wrapper(
-            evaluations.archive,
+        self.delete = to_streamed_response_wrapper(
+            evaluations.delete,
         )
 
 
@@ -695,9 +809,12 @@ class AsyncEvaluationsResourceWithStreamingResponse:
         self.retrieve = async_to_streamed_response_wrapper(
             evaluations.retrieve,
         )
+        self.update = async_to_streamed_response_wrapper(
+            evaluations.update,
+        )
         self.list = async_to_streamed_response_wrapper(
             evaluations.list,
         )
-        self.archive = async_to_streamed_response_wrapper(
-            evaluations.archive,
+        self.delete = async_to_streamed_response_wrapper(
+            evaluations.delete,
         )

scale_gp_beta/resources/files/files.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Mapping, Optional, cast
+from typing import Dict, Mapping, cast
 
 import httpx
 
@@ -167,9 +167,9 @@ class FilesResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        starting_after: 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,
@@ -377,9 +377,9 @@ class AsyncFilesResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        starting_after: 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,

scale_gp_beta/resources/models.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-from typing import Dict, Optional
+from typing import Dict
 from typing_extensions import Literal, overload
 
 import httpx
@@ -303,26 +303,24 @@ class ModelsResource(SyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        model_vendor: Optional[
-            Literal[
-                "openai",
-                "cohere",
-                "vertex_ai",
-                "anthropic",
-                "azure",
-                "gemini",
-                "launch",
-                "llmengine",
-                "model_zoo",
-                "bedrock",
-                "xai",
-            ]
+        model_vendor: Literal[
+            "openai",
+            "cohere",
+            "vertex_ai",
+            "anthropic",
+            "azure",
+            "gemini",
+            "launch",
+            "llmengine",
+            "model_zoo",
+            "bedrock",
+            "xai",
         ]
         | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        starting_after: 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,
@@ -675,26 +673,24 @@ class AsyncModelsResource(AsyncAPIResource):
     def list(
         self,
         *,
-        ending_before: Optional[str] | NotGiven = NOT_GIVEN,
+        ending_before: str | NotGiven = NOT_GIVEN,
         limit: int | NotGiven = NOT_GIVEN,
-        model_vendor: Optional[
-            Literal[
-                "openai",
-                "cohere",
-                "vertex_ai",
-                "anthropic",
-                "azure",
-                "gemini",
-                "launch",
-                "llmengine",
-                "model_zoo",
-                "bedrock",
-                "xai",
-            ]
+        model_vendor: Literal[
+            "openai",
+            "cohere",
+            "vertex_ai",
+            "anthropic",
+            "azure",
+            "gemini",
+            "launch",
+            "llmengine",
+            "model_zoo",
+            "bedrock",
+            "xai",
         ]
         | NotGiven = NOT_GIVEN,
-        name: Optional[str] | NotGiven = NOT_GIVEN,
-        starting_after: Optional[str] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        starting_after: 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,

scale_gp_beta/resources/spans.py

@@ -2,9 +2,9 @@
 
 from __future__ import annotations
 
-from typing import Dict, Union, Optional
+from typing import Dict, Union
 from datetime import datetime
-from typing_extensions import overload
+from typing_extensions import Literal, overload
 
 import httpx
 
@@ -53,11 +53,13 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | 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: 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,
@@ -92,11 +94,13 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | 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: 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,
@@ -132,11 +136,13 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | 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: str | 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 +159,13 @@ class SpansResource(SyncAPIResource):
                     "name": name,
                     "start_timestamp": start_timestamp,
                     "trace_id": trace_id,
-                    "data": data,
                     "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 +214,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 +244,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 +261,13 @@ 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,
+        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,
@@ -333,11 +343,13 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | 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: 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,
@@ -372,11 +384,13 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | 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: 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,
@@ -412,11 +426,13 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        data: Dict[str, object] | 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: str | 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 +449,13 @@ class AsyncSpansResource(AsyncAPIResource):
                     "name": name,
                     "start_timestamp": start_timestamp,
                     "trace_id": trace_id,
-                    "data": data,
                     "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 +504,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 +534,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 +551,13 @@ 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,
+        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,

scale_gp_beta/types/__init__.py

@@ -5,10 +5,15 @@ 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
@@ -28,6 +33,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,11 +42,12 @@ 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