scale-gp-beta 0.1.0a29__py3-none-any.whl → 0.1.0a31__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, 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, SequenceNotStr
+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: SequenceNotStr[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: SequenceNotStr[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

@@ -17,7 +17,7 @@ from ..types import (
     span_update_params,
     span_upsert_batch_params,
 )
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven, SequenceNotStr
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -255,18 +255,17 @@ 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,
-        excluded_span_ids: List[str] | NotGiven = NOT_GIVEN,
-        excluded_trace_ids: List[str] | NotGiven = NOT_GIVEN,
+        application_variant_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
+        excluded_span_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
+        excluded_trace_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         group_id: str | NotGiven = NOT_GIVEN,
-        names: List[str] | NotGiven = NOT_GIVEN,
+        names: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         parents_only: bool | NotGiven = NOT_GIVEN,
-        search_text: str | NotGiven = NOT_GIVEN,
-        span_id: str | NotGiven = NOT_GIVEN,
-        span_ids: List[str] | NotGiven = NOT_GIVEN,
+        search_texts: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
+        span_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
-        trace_id: str | NotGiven = NOT_GIVEN,
-        trace_ids: List[str] | NotGiven = NOT_GIVEN,
+        trace_ids: SequenceNotStr[str] | 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.
@@ -283,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
@@ -295,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
@@ -320,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,
                 },
@@ -607,18 +603,17 @@ 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,
-        excluded_span_ids: List[str] | NotGiven = NOT_GIVEN,
-        excluded_trace_ids: List[str] | NotGiven = NOT_GIVEN,
+        application_variant_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
+        excluded_span_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
+        excluded_trace_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         group_id: str | NotGiven = NOT_GIVEN,
-        names: List[str] | NotGiven = NOT_GIVEN,
+        names: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         parents_only: bool | NotGiven = NOT_GIVEN,
-        search_text: str | NotGiven = NOT_GIVEN,
-        span_id: str | NotGiven = NOT_GIVEN,
-        span_ids: List[str] | NotGiven = NOT_GIVEN,
+        search_texts: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
+        span_ids: SequenceNotStr[str] | NotGiven = NOT_GIVEN,
         statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
-        trace_id: str | NotGiven = NOT_GIVEN,
-        trace_ids: List[str] | NotGiven = NOT_GIVEN,
+        trace_ids: SequenceNotStr[str] | 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.
@@ -635,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
@@ -647,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
@@ -672,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,10 +2,14 @@
 
 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
@@ -44,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
@@ -52,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
@@ -64,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_V1:
+    evaluation.Evaluation.update_forward_refs()  # type: ignore
+    container.Container.update_forward_refs()  # type: ignore
+else:
+    evaluation.Evaluation.model_rebuild(_parent_namespace_depth=0)
+    container.Container.model_rebuild(_parent_namespace_depth=0)