scale-gp-beta 0.1.0a16__tar.gz → 0.1.0a17__tar.gz

scale_gp_beta-0.1.0a17/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.1.0-alpha.17"
+}

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.1.0-alpha.17 (2025-06-11)
+
+Full Changelog: [v0.1.0-alpha.16...v0.1.0-alpha.17](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.16...v0.1.0-alpha.17)
+
+### Features
+
+* **api:** Added Span upsert batch and search endpoints ([f109f69](https://github.com/scaleapi/sgp-python-beta/commit/f109f6919886874122f541bf3bc8bb7d690648f1))
+
 ## 0.1.0-alpha.16 (2025-06-10)
 
 Full Changelog: [v0.1.0-alpha.15...v0.1.0-alpha.16](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.15...v0.1.0-alpha.16)

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: scale-gp-beta
-Version: 0.1.0a16
+Version: 0.1.0a17
 Summary: The official Python library for the Scale GP API
 Project-URL: Homepage, https://github.com/scaleapi/sgp-python-beta
 Project-URL: Repository, https://github.com/scaleapi/sgp-python-beta

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/api.md

@@ -157,7 +157,7 @@ Methods:
 Types:
 
 ```python
-from scale_gp_beta.types import Span, SpanBatchResponse
+from scale_gp_beta.types import Span, SpanBatchResponse, SpanSearchResponse, SpanUpsertBatchResponse
 ```
 
 Methods:
@@ -166,3 +166,5 @@ Methods:
 - <code title="get /v5/spans/{span_id}">client.spans.<a href="./src/scale_gp_beta/resources/spans.py">retrieve</a>(span_id) -> <a href="./src/scale_gp_beta/types/span.py">Span</a></code>
 - <code title="patch /v5/spans/{span_id}">client.spans.<a href="./src/scale_gp_beta/resources/spans.py">update</a>(span_id, \*\*<a href="src/scale_gp_beta/types/span_update_params.py">params</a>) -> <a href="./src/scale_gp_beta/types/span.py">Span</a></code>
 - <code title="post /v5/spans/batch">client.spans.<a href="./src/scale_gp_beta/resources/spans.py">batch</a>(\*\*<a href="src/scale_gp_beta/types/span_batch_params.py">params</a>) -> <a href="./src/scale_gp_beta/types/span_batch_response.py">SpanBatchResponse</a></code>
+- <code title="post /v5/spans/search">client.spans.<a href="./src/scale_gp_beta/resources/spans.py">search</a>(\*\*<a href="src/scale_gp_beta/types/span_search_params.py">params</a>) -> <a href="./src/scale_gp_beta/types/span_search_response.py">SpanSearchResponse</a></code>
+- <code title="put /v5/spans/batch">client.spans.<a href="./src/scale_gp_beta/resources/spans.py">upsert_batch</a>(\*\*<a href="src/scale_gp_beta/types/span_upsert_batch_params.py">params</a>) -> <a href="./src/scale_gp_beta/types/span_upsert_batch_response.py">SpanUpsertBatchResponse</a></code>

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "scale-gp-beta"
-version = "0.1.0-alpha.16"
+version = "0.1.0-alpha.17"
 description = "The official Python library for the Scale GP API"
 dynamic = ["readme"]
 license = "Apache-2.0"

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/src/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.16"  # x-release-please-version
+__version__ = "0.1.0-alpha.17"  # x-release-please-version

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/src/scale_gp_beta/resources/spans.py

@@ -8,7 +8,13 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..types import span_batch_params, span_create_params, span_update_params
+from ..types import (
+    span_batch_params,
+    span_create_params,
+    span_search_params,
+    span_update_params,
+    span_upsert_batch_params,
+)
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -22,6 +28,8 @@ from .._response import (
 from ..types.span import Span
 from .._base_client import make_request_options
 from ..types.span_batch_response import SpanBatchResponse
+from ..types.span_search_response import SpanSearchResponse
+from ..types.span_upsert_batch_response import SpanUpsertBatchResponse
 
 __all__ = ["SpansResource", "AsyncSpansResource"]
 
@@ -262,6 +270,155 @@ class SpansResource(SyncAPIResource):
             cast_to=SpanBatchResponse,
         )
 
+    def search(
+        self,
+        *,
+        ending_before: str | NotGiven = NOT_GIVEN,
+        from_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        limit: int | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
+        to_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        group_id: str | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        parents_only: bool | NotGiven = NOT_GIVEN,
+        span_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        trace_id: str | 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,
+    ) -> SpanSearchResponse:
+        """
+        Search and list spans
+
+        Args:
+          from_ts: The starting (oldest) timestamp in ISO format.
+
+          to_ts: The ending (most recent) timestamp in ISO format.
+
+          extra_metadata: Filter on custom metadata key-value pairs
+
+          group_id: Filter by group ID
+
+          name: Filter by trace/span name
+
+          parents_only: Only fetch spans that are the top-level (ie. have no parent_id)
+
+          span_id: Filter by span ID
+
+          status: Filter on span status
+
+          trace_id: Which trace_id to filter on
+
+          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._post(
+            "/v5/spans/search",
+            body=maybe_transform(
+                {
+                    "extra_metadata": extra_metadata,
+                    "group_id": group_id,
+                    "name": name,
+                    "parents_only": parents_only,
+                    "span_id": span_id,
+                    "status": status,
+                    "trace_id": trace_id,
+                    "type": type,
+                },
+                span_search_params.SpanSearchParams,
+            ),
+            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,
+                        "sort_order": sort_order,
+                        "starting_after": starting_after,
+                        "to_ts": to_ts,
+                    },
+                    span_search_params.SpanSearchParams,
+                ),
+            ),
+            cast_to=SpanSearchResponse,
+        )
+
+    def upsert_batch(
+        self,
+        *,
+        items: Iterable[span_upsert_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,
+    ) -> SpanUpsertBatchResponse:
+        """
+        Upsert 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._put(
+            "/v5/spans/batch",
+            body=maybe_transform({"items": items}, span_upsert_batch_params.SpanUpsertBatchParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SpanUpsertBatchResponse,
+        )
+
 
 class AsyncSpansResource(AsyncAPIResource):
     @cached_property
@@ -499,6 +656,155 @@ class AsyncSpansResource(AsyncAPIResource):
             cast_to=SpanBatchResponse,
         )
 
+    async def search(
+        self,
+        *,
+        ending_before: str | NotGiven = NOT_GIVEN,
+        from_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        limit: int | NotGiven = NOT_GIVEN,
+        sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
+        starting_after: str | NotGiven = NOT_GIVEN,
+        to_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        group_id: str | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        parents_only: bool | NotGiven = NOT_GIVEN,
+        span_id: str | NotGiven = NOT_GIVEN,
+        status: Literal["SUCCESS", "ERROR"] | NotGiven = NOT_GIVEN,
+        trace_id: str | 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,
+    ) -> SpanSearchResponse:
+        """
+        Search and list spans
+
+        Args:
+          from_ts: The starting (oldest) timestamp in ISO format.
+
+          to_ts: The ending (most recent) timestamp in ISO format.
+
+          extra_metadata: Filter on custom metadata key-value pairs
+
+          group_id: Filter by group ID
+
+          name: Filter by trace/span name
+
+          parents_only: Only fetch spans that are the top-level (ie. have no parent_id)
+
+          span_id: Filter by span ID
+
+          status: Filter on span status
+
+          trace_id: Which trace_id to filter on
+
+          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 await self._post(
+            "/v5/spans/search",
+            body=await async_maybe_transform(
+                {
+                    "extra_metadata": extra_metadata,
+                    "group_id": group_id,
+                    "name": name,
+                    "parents_only": parents_only,
+                    "span_id": span_id,
+                    "status": status,
+                    "trace_id": trace_id,
+                    "type": type,
+                },
+                span_search_params.SpanSearchParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=await async_maybe_transform(
+                    {
+                        "ending_before": ending_before,
+                        "from_ts": from_ts,
+                        "limit": limit,
+                        "sort_order": sort_order,
+                        "starting_after": starting_after,
+                        "to_ts": to_ts,
+                    },
+                    span_search_params.SpanSearchParams,
+                ),
+            ),
+            cast_to=SpanSearchResponse,
+        )
+
+    async def upsert_batch(
+        self,
+        *,
+        items: Iterable[span_upsert_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,
+    ) -> SpanUpsertBatchResponse:
+        """
+        Upsert 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 await self._put(
+            "/v5/spans/batch",
+            body=await async_maybe_transform({"items": items}, span_upsert_batch_params.SpanUpsertBatchParams),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=SpanUpsertBatchResponse,
+        )
+
 
 class SpansResourceWithRawResponse:
     def __init__(self, spans: SpansResource) -> None:
@@ -516,6 +822,12 @@ class SpansResourceWithRawResponse:
         self.batch = to_raw_response_wrapper(
             spans.batch,
         )
+        self.search = to_raw_response_wrapper(
+            spans.search,
+        )
+        self.upsert_batch = to_raw_response_wrapper(
+            spans.upsert_batch,
+        )
 
 
 class AsyncSpansResourceWithRawResponse:
@@ -534,6 +846,12 @@ class AsyncSpansResourceWithRawResponse:
         self.batch = async_to_raw_response_wrapper(
             spans.batch,
         )
+        self.search = async_to_raw_response_wrapper(
+            spans.search,
+        )
+        self.upsert_batch = async_to_raw_response_wrapper(
+            spans.upsert_batch,
+        )
 
 
 class SpansResourceWithStreamingResponse:
@@ -552,6 +870,12 @@ class SpansResourceWithStreamingResponse:
         self.batch = to_streamed_response_wrapper(
             spans.batch,
         )
+        self.search = to_streamed_response_wrapper(
+            spans.search,
+        )
+        self.upsert_batch = to_streamed_response_wrapper(
+            spans.upsert_batch,
+        )
 
 
 class AsyncSpansResourceWithStreamingResponse:
@@ -570,3 +894,9 @@ class AsyncSpansResourceWithStreamingResponse:
         self.batch = async_to_streamed_response_wrapper(
             spans.batch,
         )
+        self.search = async_to_streamed_response_wrapper(
+            spans.search,
+        )
+        self.upsert_batch = async_to_streamed_response_wrapper(
+            spans.upsert_batch,
+        )

{scale_gp_beta-0.1.0a16 → scale_gp_beta-0.1.0a17}/src/scale_gp_beta/types/__init__.py

@@ -24,6 +24,7 @@ from .file_create_params import FileCreateParams as FileCreateParams
 from .file_update_params import FileUpdateParams as FileUpdateParams
 from .inference_response import InferenceResponse as InferenceResponse
 from .span_create_params import SpanCreateParams as SpanCreateParams
+from .span_search_params import SpanSearchParams as SpanSearchParams
 from .span_update_params import SpanUpdateParams as SpanUpdateParams
 from .dataset_list_params import DatasetListParams as DatasetListParams
 from .model_create_params import ModelCreateParams as ModelCreateParams
@@ -31,6 +32,7 @@ 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 .span_search_response import SpanSearchResponse as SpanSearchResponse
 from .dataset_create_params import DatasetCreateParams as DatasetCreateParams
 from .dataset_update_params import DatasetUpdateParams as DatasetUpdateParams
 from .evaluation_task_param import EvaluationTaskParam as EvaluationTaskParam
@@ -45,10 +47,12 @@ from .dataset_item_list_params import DatasetItemListParams as DatasetItemListPa
 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 .span_upsert_batch_params import SpanUpsertBatchParams as SpanUpsertBatchParams
 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 .span_upsert_batch_response import SpanUpsertBatchResponse as SpanUpsertBatchResponse
 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

scale_gp_beta-0.1.0a17/src/scale_gp_beta/types/span_search_params.py

@@ -0,0 +1,77 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union
+from datetime import datetime
+from typing_extensions import Literal, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["SpanSearchParams"]
+
+
+class SpanSearchParams(TypedDict, total=False):
+    ending_before: str
+
+    from_ts: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
+    """The starting (oldest) timestamp in ISO format."""
+
+    limit: int
+
+    sort_order: Literal["asc", "desc"]
+
+    starting_after: str
+
+    to_ts: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
+    """The ending (most recent) timestamp in ISO format."""
+
+    extra_metadata: Dict[str, object]
+    """Filter on custom metadata key-value pairs"""
+
+    group_id: str
+    """Filter by group ID"""
+
+    name: str
+    """Filter by trace/span name"""
+
+    parents_only: bool
+    """Only fetch spans that are the top-level (ie. have no parent_id)"""
+
+    span_id: str
+    """Filter by span ID"""
+
+    status: Literal["SUCCESS", "ERROR"]
+    """Filter on span status"""
+
+    trace_id: str
+    """Which trace_id to filter on"""
+
+    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",
+    ]

scale_gp_beta-0.1.0a17/src/scale_gp_beta/types/span_search_response.py

@@ -0,0 +1,27 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+from typing_extensions import Literal
+
+from .span import Span
+from .._models import BaseModel
+
+__all__ = ["SpanSearchResponse"]
+
+
+class SpanSearchResponse(BaseModel):
+    has_more: bool
+    """Whether there are more items left to be fetched."""
+
+    items: List[Span]
+
+    total: int
+    """The total of items that match the query.
+
+    This is greater than or equal to the number of items returned.
+    """
+
+    limit: Optional[int] = None
+    """The maximum number of items to return."""
+
+    object: Optional[Literal["list"]] = None

scale_gp_beta-0.1.0a17/src/scale_gp_beta/types/span_upsert_batch_params.py

@@ -0,0 +1,78 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union, Iterable
+from datetime import datetime
+from typing_extensions import Literal, Required, Annotated, TypedDict
+
+from .._utils import PropertyInfo
+
+__all__ = ["SpanUpsertBatchParams", "Item"]
+
+
+class SpanUpsertBatchParams(TypedDict, total=False):
+    items: Required[Iterable[Item]]
+
+
+class Item(TypedDict, total=False):
+    name: Required[str]
+
+    start_timestamp: Required[Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]]
+
+    trace_id: Required[str]
+    """id for grouping traces together, uuid is recommended"""
+
+    id: str
+    """The id of the span"""
+
+    application_interaction_id: str
+    """The optional application interaction ID this span belongs to"""
+
+    application_variant_id: str
+    """The optional application variant ID this span belongs to"""
+
+    end_timestamp: Annotated[Union[str, datetime], PropertyInfo(format="iso8601")]
+
+    group_id: str
+    """Reference to a group_id"""
+
+    input: Dict[str, object]
+
+    metadata: Dict[str, object]
+
+    output: Dict[str, object]
+
+    parent_id: str
+    """Reference to a parent span_id"""
+
+    status: Literal["SUCCESS", "ERROR"]
+
+    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",
+    ]

scale_gp_beta-0.1.0a17/src/scale_gp_beta/types/span_upsert_batch_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 .span import Span
+from .._models import BaseModel
+
+__all__ = ["SpanUpsertBatchResponse"]
+
+
+class SpanUpsertBatchResponse(BaseModel):
+    items: List[Span]
+
+    object: Optional[Literal["list"]] = None