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

scale_gp_beta/resources/models.py

@@ -9,11 +9,7 @@ import httpx
 
 from ..types import model_list_params, model_create_params, model_update_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
-from .._utils import (
-    required_args,
-    maybe_transform,
-    async_maybe_transform,
-)
+from .._utils import required_args, maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
 from .._response import (

scale_gp_beta/resources/spans.py

@@ -0,0 +1,658 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union, Optional
+from datetime import datetime
+from typing_extensions import overload
+
+import httpx
+
+from ..types import span_list_params, span_create_params, span_update_params
+from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._utils import required_args, maybe_transform, async_maybe_transform
+from .._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 ..pagination import SyncCursorPage, AsyncCursorPage
+from ..types.span import Span
+from .._base_client import AsyncPaginator, make_request_options
+
+__all__ = ["SpansResource", "AsyncSpansResource"]
+
+
+class SpansResource(SyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> SpansResourceWithRawResponse:
+        """
+        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 SpansResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> SpansResourceWithStreamingResponse:
+        """
+        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 SpansResourceWithStreamingResponse(self)
+
+    @overload
+    def create(
+        self,
+        *,
+        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,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parent_id: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Create Span
+
+        Args:
+          trace_id: id for grouping traces together, uuid is recommended
+
+          parent_id: Reference to a parent span_id
+
+          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
+        """
+        ...
+
+    @overload
+    def create(
+        self,
+        *,
+        application_interaction_id: str,
+        application_variant_id: str,
+        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,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parent_id: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Create Span
+
+        Args:
+          trace_id: id for grouping traces together, uuid is recommended
+
+          parent_id: Reference to a parent span_id
+
+          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
+        """
+        ...
+
+    @required_args(
+        ["name", "start_timestamp", "trace_id"],
+        ["application_interaction_id", "application_variant_id", "name", "start_timestamp", "trace_id"],
+    )
+    def create(
+        self,
+        *,
+        name: str,
+        start_timestamp: Union[str, datetime],
+        trace_id: str,
+        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parent_id: 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.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        return self._post(
+            "/v5/spans",
+            body=maybe_transform(
+                {
+                    "name": name,
+                    "start_timestamp": start_timestamp,
+                    "trace_id": trace_id,
+                    "data": data,
+                    "end_timestamp": end_timestamp,
+                    "input": input,
+                    "output": output,
+                    "parent_id": parent_id,
+                    "application_interaction_id": application_interaction_id,
+                    "application_variant_id": application_variant_id,
+                },
+                span_create_params.SpanCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Span,
+        )
+
+    def retrieve(
+        self,
+        span_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Get Span
+
+        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
+        """
+        if not span_id:
+            raise ValueError(f"Expected a non-empty value for `span_id` but received {span_id!r}")
+        return self._get(
+            f"/v5/spans/{span_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Span,
+        )
+
+    def update(
+        self,
+        span_id: str,
+        *,
+        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Update Span
+
+        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
+        """
+        if not span_id:
+            raise ValueError(f"Expected a non-empty value for `span_id` but received {span_id!r}")
+        return self._patch(
+            f"/v5/spans/{span_id}",
+            body=maybe_transform(
+                {
+                    "data": data,
+                    "end_timestamp": end_timestamp,
+                    "name": name,
+                    "output": output,
+                },
+                span_update_params.SpanUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Span,
+        )
+
+    def list(
+        self,
+        *,
+        ending_before: Optional[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,
+        to_ts: int | NotGiven = NOT_GIVEN,
+        trace_id: Optional[str] | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> SyncCursorPage[Span]:
+        """
+        List Spans
+
+        Args:
+          from_ts: The starting (oldest) timestamp window in seconds.
+
+          to_ts: The ending (most recent) timestamp in seconds.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._get_api_list(
+            "/v5/spans",
+            page=SyncCursorPage[Span],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "ending_before": ending_before,
+                        "from_ts": from_ts,
+                        "limit": limit,
+                        "parents_only": parents_only,
+                        "starting_after": starting_after,
+                        "to_ts": to_ts,
+                        "trace_id": trace_id,
+                    },
+                    span_list_params.SpanListParams,
+                ),
+            ),
+            model=Span,
+        )
+
+
+class AsyncSpansResource(AsyncAPIResource):
+    @cached_property
+    def with_raw_response(self) -> AsyncSpansResourceWithRawResponse:
+        """
+        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 AsyncSpansResourceWithRawResponse(self)
+
+    @cached_property
+    def with_streaming_response(self) -> AsyncSpansResourceWithStreamingResponse:
+        """
+        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 AsyncSpansResourceWithStreamingResponse(self)
+
+    @overload
+    async def create(
+        self,
+        *,
+        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,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parent_id: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Create Span
+
+        Args:
+          trace_id: id for grouping traces together, uuid is recommended
+
+          parent_id: Reference to a parent span_id
+
+          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
+        """
+        ...
+
+    @overload
+    async def create(
+        self,
+        *,
+        application_interaction_id: str,
+        application_variant_id: str,
+        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,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parent_id: str | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Create Span
+
+        Args:
+          trace_id: id for grouping traces together, uuid is recommended
+
+          parent_id: Reference to a parent span_id
+
+          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
+        """
+        ...
+
+    @required_args(
+        ["name", "start_timestamp", "trace_id"],
+        ["application_interaction_id", "application_variant_id", "name", "start_timestamp", "trace_id"],
+    )
+    async def create(
+        self,
+        *,
+        name: str,
+        start_timestamp: Union[str, datetime],
+        trace_id: str,
+        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        input: Dict[str, object] | NotGiven = NOT_GIVEN,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        parent_id: 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.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        return await self._post(
+            "/v5/spans",
+            body=await async_maybe_transform(
+                {
+                    "name": name,
+                    "start_timestamp": start_timestamp,
+                    "trace_id": trace_id,
+                    "data": data,
+                    "end_timestamp": end_timestamp,
+                    "input": input,
+                    "output": output,
+                    "parent_id": parent_id,
+                    "application_interaction_id": application_interaction_id,
+                    "application_variant_id": application_variant_id,
+                },
+                span_create_params.SpanCreateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Span,
+        )
+
+    async def retrieve(
+        self,
+        span_id: str,
+        *,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Get Span
+
+        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
+        """
+        if not span_id:
+            raise ValueError(f"Expected a non-empty value for `span_id` but received {span_id!r}")
+        return await self._get(
+            f"/v5/spans/{span_id}",
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Span,
+        )
+
+    async def update(
+        self,
+        span_id: str,
+        *,
+        data: Dict[str, object] | NotGiven = NOT_GIVEN,
+        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        name: str | NotGiven = NOT_GIVEN,
+        output: Dict[str, object] | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> Span:
+        """
+        Update Span
+
+        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
+        """
+        if not span_id:
+            raise ValueError(f"Expected a non-empty value for `span_id` but received {span_id!r}")
+        return await self._patch(
+            f"/v5/spans/{span_id}",
+            body=await async_maybe_transform(
+                {
+                    "data": data,
+                    "end_timestamp": end_timestamp,
+                    "name": name,
+                    "output": output,
+                },
+                span_update_params.SpanUpdateParams,
+            ),
+            options=make_request_options(
+                extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
+            ),
+            cast_to=Span,
+        )
+
+    def list(
+        self,
+        *,
+        ending_before: Optional[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,
+        to_ts: int | NotGiven = NOT_GIVEN,
+        trace_id: Optional[str] | NotGiven = NOT_GIVEN,
+        # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
+        # The extra values given here take precedence over values defined on the client or passed to this method.
+        extra_headers: Headers | None = None,
+        extra_query: Query | None = None,
+        extra_body: Body | None = None,
+        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+    ) -> AsyncPaginator[Span, AsyncCursorPage[Span]]:
+        """
+        List Spans
+
+        Args:
+          from_ts: The starting (oldest) timestamp window in seconds.
+
+          to_ts: The ending (most recent) timestamp in seconds.
+
+          extra_headers: Send extra headers
+
+          extra_query: Add additional query parameters to the request
+
+          extra_body: Add additional JSON properties to the request
+
+          timeout: Override the client-level default timeout for this request, in seconds
+        """
+        return self._get_api_list(
+            "/v5/spans",
+            page=AsyncCursorPage[Span],
+            options=make_request_options(
+                extra_headers=extra_headers,
+                extra_query=extra_query,
+                extra_body=extra_body,
+                timeout=timeout,
+                query=maybe_transform(
+                    {
+                        "ending_before": ending_before,
+                        "from_ts": from_ts,
+                        "limit": limit,
+                        "parents_only": parents_only,
+                        "starting_after": starting_after,
+                        "to_ts": to_ts,
+                        "trace_id": trace_id,
+                    },
+                    span_list_params.SpanListParams,
+                ),
+            ),
+            model=Span,
+        )
+
+
+class SpansResourceWithRawResponse:
+    def __init__(self, spans: SpansResource) -> None:
+        self._spans = spans
+
+        self.create = to_raw_response_wrapper(
+            spans.create,
+        )
+        self.retrieve = to_raw_response_wrapper(
+            spans.retrieve,
+        )
+        self.update = to_raw_response_wrapper(
+            spans.update,
+        )
+        self.list = to_raw_response_wrapper(
+            spans.list,
+        )
+
+
+class AsyncSpansResourceWithRawResponse:
+    def __init__(self, spans: AsyncSpansResource) -> None:
+        self._spans = spans
+
+        self.create = async_to_raw_response_wrapper(
+            spans.create,
+        )
+        self.retrieve = async_to_raw_response_wrapper(
+            spans.retrieve,
+        )
+        self.update = async_to_raw_response_wrapper(
+            spans.update,
+        )
+        self.list = async_to_raw_response_wrapper(
+            spans.list,
+        )
+
+
+class SpansResourceWithStreamingResponse:
+    def __init__(self, spans: SpansResource) -> None:
+        self._spans = spans
+
+        self.create = to_streamed_response_wrapper(
+            spans.create,
+        )
+        self.retrieve = to_streamed_response_wrapper(
+            spans.retrieve,
+        )
+        self.update = to_streamed_response_wrapper(
+            spans.update,
+        )
+        self.list = to_streamed_response_wrapper(
+            spans.list,
+        )
+
+
+class AsyncSpansResourceWithStreamingResponse:
+    def __init__(self, spans: AsyncSpansResource) -> None:
+        self._spans = spans
+
+        self.create = async_to_streamed_response_wrapper(
+            spans.create,
+        )
+        self.retrieve = async_to_streamed_response_wrapper(
+            spans.retrieve,
+        )
+        self.update = async_to_streamed_response_wrapper(
+            spans.update,
+        )
+        self.list = async_to_streamed_response_wrapper(
+            spans.list,
+        )

scale_gp_beta/types/__init__.py

@@ -3,6 +3,7 @@
 from __future__ import annotations
 
 from .file import File as File
+from .span import Span as Span
 from .dataset import Dataset as Dataset
 from .file_list import FileList as FileList
 from .completion import Completion as Completion
@@ -12,10 +13,13 @@ from .evaluation_item import EvaluationItem as EvaluationItem
 from .evaluation_task import EvaluationTask as EvaluationTask
 from .inference_model import InferenceModel as InferenceModel
 from .file_list_params import FileListParams as FileListParams
+from .span_list_params import SpanListParams as SpanListParams
 from .model_list_params import ModelListParams as ModelListParams
 from .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_update_params import SpanUpdateParams as SpanUpdateParams
 from .dataset_list_params import DatasetListParams as DatasetListParams
 from .model_create_params import ModelCreateParams as ModelCreateParams
 from .model_update_params import ModelUpdateParams as ModelUpdateParams