scale-gp-beta 0.1.0a28__py3-none-any.whl → 0.1.0a30__py3-none-any.whl

scale_gp_beta/resources/responses.py

@@ -0,0 +1,314 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Any, Dict, List, Union, Iterable, cast
+from typing_extensions import Literal
+
+import httpx
+
+from ..types import response_create_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import maybe_transform, async_maybe_transform
+from .._compat import cached_property
+from .._resource import SyncAPIResource, AsyncAPIResource
+from .._response import (
+    to_raw_response_wrapper,
+    to_streamed_response_wrapper,
+    async_to_raw_response_wrapper,
+    async_to_streamed_response_wrapper,
+)
+from .._base_client import make_request_options
+from ..types.response_create_response import ResponseCreateResponse
+
+__all__ = ["ResponsesResource", "AsyncResponsesResource"]
+
+
+class ResponsesResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> ResponsesResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/scaleapi/sgp-python-beta#accessing-raw-response-data-eg-headers
+        """
+        return ResponsesResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> ResponsesResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/scaleapi/sgp-python-beta#with_streaming_response
+        """
+        return ResponsesResourceWithStreamingResponse(self)
+
+    def create(
+        self,
+        *,
+        input: Union[str, Iterable[response_create_params.InputUnionMember1]],
+        model: str,
+        include: List[str] | NotGiven = NOT_GIVEN,
+        instructions: str | NotGiven = NOT_GIVEN,
+        max_output_tokens: int | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
+        previous_response_id: str | NotGiven = NOT_GIVEN,
+        reasoning: Dict[str, object] | NotGiven = NOT_GIVEN,
+        store: bool | NotGiven = NOT_GIVEN,
+        stream: bool | NotGiven = NOT_GIVEN,
+        temperature: float | NotGiven = NOT_GIVEN,
+        text: Dict[str, object] | NotGiven = NOT_GIVEN,
+        tool_choice: Union[str, Dict[str, object]] | NotGiven = NOT_GIVEN,
+        tools: Iterable[Dict[str, object]] | NotGiven = NOT_GIVEN,
+        top_p: float | NotGiven = NOT_GIVEN,
+        truncation: Literal["auto", "disabled"] | 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,
+    ) -> ResponseCreateResponse:
+        """
+        Responses
+
+        Args:
+          model: model specified as `model_vendor/model`, for example `openai/gpt-4o`
+
+          include: Which fields to include in the response
+
+          instructions: Instructions for the response generation
+
+          max_output_tokens: Maximum number of output tokens
+
+          metadata: Metadata for the response
+
+          parallel_tool_calls: Whether to enable parallel tool calls
+
+          previous_response_id: ID of the previous response for chaining
+
+          reasoning: Reasoning configuration for the response
+
+          store: Whether to store the response
+
+          stream: Whether to stream the response
+
+          temperature: Sampling temperature for randomness control
+
+          text: Text configuration parameters
+
+          tool_choice: Tool choice configuration
+
+          tools: Tools available for the response
+
+          top_p: Top-p sampling parameter
+
+          truncation: Truncation configuration
+
+          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 cast(
+            ResponseCreateResponse,
+            self._post(
+                "/v5/responses",
+                body=maybe_transform(
+                    {
+                        "input": input,
+                        "model": model,
+                        "include": include,
+                        "instructions": instructions,
+                        "max_output_tokens": max_output_tokens,
+                        "metadata": metadata,
+                        "parallel_tool_calls": parallel_tool_calls,
+                        "previous_response_id": previous_response_id,
+                        "reasoning": reasoning,
+                        "store": store,
+                        "stream": stream,
+                        "temperature": temperature,
+                        "text": text,
+                        "tool_choice": tool_choice,
+                        "tools": tools,
+                        "top_p": top_p,
+                        "truncation": truncation,
+                    },
+                    response_create_params.ResponseCreateParams,
+                ),
+                options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                ),
+                cast_to=cast(
+                    Any, ResponseCreateResponse
+                ),  # Union types cannot be passed in as arguments in the type system
+            ),
+        )
+
+
+class AsyncResponsesResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncResponsesResourceWithRawResponse:
+        """
+        This property can be used as a prefix for any HTTP method call to return
+        the raw response object instead of the parsed content.
+
+        For more information, see https://www.github.com/scaleapi/sgp-python-beta#accessing-raw-response-data-eg-headers
+        """
+        return AsyncResponsesResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncResponsesResourceWithStreamingResponse:
+        """
+        An alternative to `.with_raw_response` that doesn't eagerly read the response body.
+
+        For more information, see https://www.github.com/scaleapi/sgp-python-beta#with_streaming_response
+        """
+        return AsyncResponsesResourceWithStreamingResponse(self)
+
+    async def create(
+        self,
+        *,
+        input: Union[str, Iterable[response_create_params.InputUnionMember1]],
+        model: str,
+        include: List[str] | NotGiven = NOT_GIVEN,
+        instructions: str | NotGiven = NOT_GIVEN,
+        max_output_tokens: int | NotGiven = NOT_GIVEN,
+        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parallel_tool_calls: bool | NotGiven = NOT_GIVEN,
+        previous_response_id: str | NotGiven = NOT_GIVEN,
+        reasoning: Dict[str, object] | NotGiven = NOT_GIVEN,
+        store: bool | NotGiven = NOT_GIVEN,
+        stream: bool | NotGiven = NOT_GIVEN,
+        temperature: float | NotGiven = NOT_GIVEN,
+        text: Dict[str, object] | NotGiven = NOT_GIVEN,
+        tool_choice: Union[str, Dict[str, object]] | NotGiven = NOT_GIVEN,
+        tools: Iterable[Dict[str, object]] | NotGiven = NOT_GIVEN,
+        top_p: float | NotGiven = NOT_GIVEN,
+        truncation: Literal["auto", "disabled"] | 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,
+    ) -> ResponseCreateResponse:
+        """
+        Responses
+
+        Args:
+          model: model specified as `model_vendor/model`, for example `openai/gpt-4o`
+
+          include: Which fields to include in the response
+
+          instructions: Instructions for the response generation
+
+          max_output_tokens: Maximum number of output tokens
+
+          metadata: Metadata for the response
+
+          parallel_tool_calls: Whether to enable parallel tool calls
+
+          previous_response_id: ID of the previous response for chaining
+
+          reasoning: Reasoning configuration for the response
+
+          store: Whether to store the response
+
+          stream: Whether to stream the response
+
+          temperature: Sampling temperature for randomness control
+
+          text: Text configuration parameters
+
+          tool_choice: Tool choice configuration
+
+          tools: Tools available for the response
+
+          top_p: Top-p sampling parameter
+
+          truncation: Truncation configuration
+
+          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 cast(
+            ResponseCreateResponse,
+            await self._post(
+                "/v5/responses",
+                body=await async_maybe_transform(
+                    {
+                        "input": input,
+                        "model": model,
+                        "include": include,
+                        "instructions": instructions,
+                        "max_output_tokens": max_output_tokens,
+                        "metadata": metadata,
+                        "parallel_tool_calls": parallel_tool_calls,
+                        "previous_response_id": previous_response_id,
+                        "reasoning": reasoning,
+                        "store": store,
+                        "stream": stream,
+                        "temperature": temperature,
+                        "text": text,
+                        "tool_choice": tool_choice,
+                        "tools": tools,
+                        "top_p": top_p,
+                        "truncation": truncation,
+                    },
+                    response_create_params.ResponseCreateParams,
+                ),
+                options=make_request_options(
+                    extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+                ),
+                cast_to=cast(
+                    Any, ResponseCreateResponse
+                ),  # Union types cannot be passed in as arguments in the type system
+            ),
+        )
+
+
+class ResponsesResourceWithRawResponse:
+    def __init__(self, responses: ResponsesResource) -> None:
+        self._responses = responses
+
+        self.create = to_raw_response_wrapper(
+            responses.create,
+        )
+
+
+class AsyncResponsesResourceWithRawResponse:
+    def __init__(self, responses: AsyncResponsesResource) -> None:
+        self._responses = responses
+
+        self.create = async_to_raw_response_wrapper(
+            responses.create,
+        )
+
+
+class ResponsesResourceWithStreamingResponse:
+    def __init__(self, responses: ResponsesResource) -> None:
+        self._responses = responses
+
+        self.create = to_streamed_response_wrapper(
+            responses.create,
+        )
+
+
+class AsyncResponsesResourceWithStreamingResponse:
+    def __init__(self, responses: AsyncResponsesResource) -> None:
+        self._responses = responses
+
+        self.create = async_to_streamed_response_wrapper(
+            responses.create,
+        )

scale_gp_beta/resources/spans.py

@@ -9,6 +9,8 @@ from typing_extensions import Literal
 import httpx
 
 from ..types import (
+    SpanType,
+    SpanStatus,
     span_batch_params,
     span_create_params,
     span_search_params,
@@ -28,6 +30,8 @@ from .._response import (
 from ..pagination import SyncCursorPage, AsyncCursorPage
 from ..types.span import Span
 from .._base_client import AsyncPaginator, make_request_options
+from ..types.span_type import SpanType
+from ..types.span_status import SpanStatus
 from ..types.span_batch_response import SpanBatchResponse
 from ..types.span_upsert_batch_response import SpanUpsertBatchResponse
 
@@ -69,35 +73,8 @@ class SpansResource(SyncAPIResource):
         metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
-        status: Literal["SUCCESS", "ERROR", "CANCELED"] | 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",
-            "CODE_EXECUTION",
-            "DATA_MANIPULATION",
-            "EVALUATION",
-            "FILE_RETRIEVAL",
-            "KB_ADD_CHUNK",
-            "KB_MANAGEMENT",
-            "GUARDRAIL",
-            "TRACER",
-            "AGENT_TRACER",
-            "AGENT_WORKFLOW",
-            "STANDALONE",
-        ]
-        | NotGiven = NOT_GIVEN,
+        status: SpanStatus | NotGiven = NOT_GIVEN,
+        type: SpanType | 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,
@@ -197,7 +174,7 @@ class SpansResource(SyncAPIResource):
         metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         name: str | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
-        status: Literal["SUCCESS", "ERROR", "CANCELED"] | NotGiven = NOT_GIVEN,
+        status: SpanStatus | 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,
@@ -278,48 +255,18 @@ class SpansResource(SyncAPIResource):
         sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
         starting_after: str | NotGiven = NOT_GIVEN,
         to_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        application_variant_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_span_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_trace_ids: List[str] | NotGiven = NOT_GIVEN,
         extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         group_id: str | NotGiven = NOT_GIVEN,
         names: List[str] | NotGiven = NOT_GIVEN,
         parents_only: bool | NotGiven = NOT_GIVEN,
-        search_text: str | NotGiven = NOT_GIVEN,
-        span_id: str | NotGiven = NOT_GIVEN,
+        search_texts: List[str] | NotGiven = NOT_GIVEN,
         span_ids: List[str] | NotGiven = NOT_GIVEN,
-        statuses: List[Literal["SUCCESS", "ERROR", "CANCELED"]] | NotGiven = NOT_GIVEN,
-        trace_id: str | NotGiven = NOT_GIVEN,
+        statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
         trace_ids: List[str] | NotGiven = NOT_GIVEN,
-        types: List[
-            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",
-                "CODE_EXECUTION",
-                "DATA_MANIPULATION",
-                "EVALUATION",
-                "FILE_RETRIEVAL",
-                "KB_ADD_CHUNK",
-                "KB_MANAGEMENT",
-                "GUARDRAIL",
-                "TRACER",
-                "AGENT_TRACER",
-                "AGENT_WORKFLOW",
-                "STANDALONE",
-            ]
-        ]
-        | NotGiven = NOT_GIVEN,
+        types: List[SpanType] | 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,
@@ -335,6 +282,8 @@ class SpansResource(SyncAPIResource):
 
           to_ts: The ending (most recent) timestamp in ISO format.
 
+          application_variant_ids: Filter by application variant IDs
+
           excluded_span_ids: List of span IDs to exclude from results
 
           excluded_trace_ids: List of trace IDs to exclude from results
@@ -347,16 +296,12 @@ class SpansResource(SyncAPIResource):
 
           parents_only: Only fetch spans that are the top-level (ie. have no parent_id)
 
-          search_text: Free text search across span input and output fields
-
-          span_id: Filter by span ID (deprecated: use span_ids instead)
+          search_texts: Free text search across span input and output fields
 
           span_ids: Filter by span IDs
 
           statuses: Filter on span status
 
-          trace_id: Filter by trace ID (deprecated: use trace_ids instead)
-
           trace_ids: Filter by trace IDs
 
           extra_headers: Send extra headers
@@ -372,17 +317,16 @@ class SpansResource(SyncAPIResource):
             page=SyncCursorPage[Span],
             body=maybe_transform(
                 {
+                    "application_variant_ids": application_variant_ids,
                     "excluded_span_ids": excluded_span_ids,
                     "excluded_trace_ids": excluded_trace_ids,
                     "extra_metadata": extra_metadata,
                     "group_id": group_id,
                     "names": names,
                     "parents_only": parents_only,
-                    "search_text": search_text,
-                    "span_id": span_id,
+                    "search_texts": search_texts,
                     "span_ids": span_ids,
                     "statuses": statuses,
-                    "trace_id": trace_id,
                     "trace_ids": trace_ids,
                     "types": types,
                 },
@@ -477,35 +421,8 @@ class AsyncSpansResource(AsyncAPIResource):
         metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
         parent_id: str | NotGiven = NOT_GIVEN,
-        status: Literal["SUCCESS", "ERROR", "CANCELED"] | 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",
-            "CODE_EXECUTION",
-            "DATA_MANIPULATION",
-            "EVALUATION",
-            "FILE_RETRIEVAL",
-            "KB_ADD_CHUNK",
-            "KB_MANAGEMENT",
-            "GUARDRAIL",
-            "TRACER",
-            "AGENT_TRACER",
-            "AGENT_WORKFLOW",
-            "STANDALONE",
-        ]
-        | NotGiven = NOT_GIVEN,
+        status: SpanStatus | NotGiven = NOT_GIVEN,
+        type: SpanType | 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,
@@ -605,7 +522,7 @@ class AsyncSpansResource(AsyncAPIResource):
         metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         name: str | NotGiven = NOT_GIVEN,
         output: Dict[str, object] | NotGiven = NOT_GIVEN,
-        status: Literal["SUCCESS", "ERROR", "CANCELED"] | NotGiven = NOT_GIVEN,
+        status: SpanStatus | 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,
@@ -686,48 +603,18 @@ class AsyncSpansResource(AsyncAPIResource):
         sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
         starting_after: str | NotGiven = NOT_GIVEN,
         to_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        application_variant_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_span_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_trace_ids: List[str] | NotGiven = NOT_GIVEN,
         extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         group_id: str | NotGiven = NOT_GIVEN,
         names: List[str] | NotGiven = NOT_GIVEN,
         parents_only: bool | NotGiven = NOT_GIVEN,
-        search_text: str | NotGiven = NOT_GIVEN,
-        span_id: str | NotGiven = NOT_GIVEN,
+        search_texts: List[str] | NotGiven = NOT_GIVEN,
         span_ids: List[str] | NotGiven = NOT_GIVEN,
-        statuses: List[Literal["SUCCESS", "ERROR", "CANCELED"]] | NotGiven = NOT_GIVEN,
-        trace_id: str | NotGiven = NOT_GIVEN,
+        statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
         trace_ids: List[str] | NotGiven = NOT_GIVEN,
-        types: List[
-            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",
-                "CODE_EXECUTION",
-                "DATA_MANIPULATION",
-                "EVALUATION",
-                "FILE_RETRIEVAL",
-                "KB_ADD_CHUNK",
-                "KB_MANAGEMENT",
-                "GUARDRAIL",
-                "TRACER",
-                "AGENT_TRACER",
-                "AGENT_WORKFLOW",
-                "STANDALONE",
-            ]
-        ]
-        | NotGiven = NOT_GIVEN,
+        types: List[SpanType] | 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,
@@ -743,6 +630,8 @@ class AsyncSpansResource(AsyncAPIResource):
 
           to_ts: The ending (most recent) timestamp in ISO format.
 
+          application_variant_ids: Filter by application variant IDs
+
           excluded_span_ids: List of span IDs to exclude from results
 
           excluded_trace_ids: List of trace IDs to exclude from results
@@ -755,16 +644,12 @@ class AsyncSpansResource(AsyncAPIResource):
 
           parents_only: Only fetch spans that are the top-level (ie. have no parent_id)
 
-          search_text: Free text search across span input and output fields
-
-          span_id: Filter by span ID (deprecated: use span_ids instead)
+          search_texts: Free text search across span input and output fields
 
           span_ids: Filter by span IDs
 
           statuses: Filter on span status
 
-          trace_id: Filter by trace ID (deprecated: use trace_ids instead)
-
           trace_ids: Filter by trace IDs
 
           extra_headers: Send extra headers
@@ -780,17 +665,16 @@ class AsyncSpansResource(AsyncAPIResource):
             page=AsyncCursorPage[Span],
             body=maybe_transform(
                 {
+                    "application_variant_ids": application_variant_ids,
                     "excluded_span_ids": excluded_span_ids,
                     "excluded_trace_ids": excluded_trace_ids,
                     "extra_metadata": extra_metadata,
                     "group_id": group_id,
                     "names": names,
                     "parents_only": parents_only,
-                    "search_text": search_text,
-                    "span_id": span_id,
+                    "search_texts": search_texts,
                     "span_ids": span_ids,
                     "statuses": statuses,
-                    "trace_id": trace_id,
                     "trace_ids": trace_ids,
                     "types": types,
                 },

scale_gp_beta/types/__init__.py

@@ -2,15 +2,21 @@
 
 from __future__ import annotations
 
+from . import container, evaluation
+from .. import _compat
 from .file import File as File
 from .span import Span as Span
+from .shared import Identity as Identity
 from .dataset import Dataset as Dataset
 from .question import Question as Question
+from .response import Response as Response
 from .component import Component as Component
 from .container import Container as Container
 from .file_list import FileList as FileList
+from .span_type import SpanType as SpanType
 from .completion import Completion as Completion
 from .evaluation import Evaluation as Evaluation
+from .span_status import SpanStatus as SpanStatus
 from .dataset_item import DatasetItem as DatasetItem
 from .item_locator import ItemLocator as ItemLocator
 from .question_list import QuestionList as QuestionList
@@ -42,6 +48,7 @@ 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 .question_create_params import QuestionCreateParams as QuestionCreateParams
+from .response_create_params import ResponseCreateParams as ResponseCreateParams
 from .dataset_delete_response import DatasetDeleteResponse as DatasetDeleteResponse
 from .dataset_retrieve_params import DatasetRetrieveParams as DatasetRetrieveParams
 from .inference_create_params import InferenceCreateParams as InferenceCreateParams
@@ -50,6 +57,7 @@ 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 .response_create_response import ResponseCreateResponse as ResponseCreateResponse
 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
@@ -62,3 +70,14 @@ from .dataset_item_retrieve_params import DatasetItemRetrieveParams as DatasetIt
 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
+
+# Rebuild cyclical models only after all modules are imported.
+# This ensures that, when building the deferred (due to cyclical references) model schema,
+# Pydantic can resolve the necessary references.
+# See: https://github.com/pydantic/pydantic/issues/11250 for more context.
+if _compat.PYDANTIC_V2:
+    evaluation.Evaluation.model_rebuild(_parent_namespace_depth=0)
+    container.Container.model_rebuild(_parent_namespace_depth=0)
+else:
+    evaluation.Evaluation.update_forward_refs()  # type: ignore
+    container.Container.update_forward_refs()  # type: ignore