scale-gp-beta 0.1.0a27__tar.gz → 0.1.0a29__tar.gz

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

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

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/CHANGELOG.md

@@ -1,5 +1,38 @@
 # Changelog
 
+## 0.1.0-alpha.29 (2025-07-14)
+
+Full Changelog: [v0.1.0-alpha.28...v0.1.0-alpha.29](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.28...v0.1.0-alpha.29)
+
+### Features
+
+* **api:** Add application_operation_type ([2455e20](https://github.com/scaleapi/sgp-python-beta/commit/2455e20838b22637ddce2a8e523f113d43acab0e))
+* **api:** Add span status to span models ([c3fb91c](https://github.com/scaleapi/sgp-python-beta/commit/c3fb91c773d6ef11e1cc02e4546ff1923fc25c82))
+* **api:** change name of app op type to span_type ([2285f10](https://github.com/scaleapi/sgp-python-beta/commit/2285f10ca3f1d51d3733996724d24daa09e83b88))
+
+
+### Bug Fixes
+
+* Use stainless types auto generated types for Span status and span type. ([#125](https://github.com/scaleapi/sgp-python-beta/issues/125)) ([6a3b070](https://github.com/scaleapi/sgp-python-beta/commit/6a3b07040137cb283711399bad76bedf78affcaf))
+
+## 0.1.0-alpha.28 (2025-07-11)
+
+Full Changelog: [v0.1.0-alpha.27...v0.1.0-alpha.28](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.27...v0.1.0-alpha.28)
+
+### Features
+
+* **api:** api update ([63befc8](https://github.com/scaleapi/sgp-python-beta/commit/63befc8168d56b166e43bdcfb804033cce9b1052))
+
+
+### Bug Fixes
+
+* **client:** don't send Content-Type header on GET requests ([9ca896f](https://github.com/scaleapi/sgp-python-beta/commit/9ca896fb0adf0a643cc071ad97a956752d63e42a))
+
+
+### Chores
+
+* **readme:** fix version rendering on pypi ([f4d2382](https://github.com/scaleapi/sgp-python-beta/commit/f4d23824e3d9238635cbefc406bd525a652ad8cc))
+
 ## 0.1.0-alpha.27 (2025-07-10)
 
 Full Changelog: [v0.1.0-alpha.26...v0.1.0-alpha.27](https://github.com/scaleapi/sgp-python-beta/compare/v0.1.0-alpha.26...v0.1.0-alpha.27)

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: scale-gp-beta
-Version: 0.1.0a27
+Version: 0.1.0a29
 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
@@ -30,12 +30,13 @@ Requires-Dist: sniffio
 Requires-Dist: typing-extensions<5,>=4.10
 Provides-Extra: aiohttp
 Requires-Dist: aiohttp; extra == 'aiohttp'
-Requires-Dist: httpx-aiohttp>=0.1.6; extra == 'aiohttp'
+Requires-Dist: httpx-aiohttp>=0.1.8; extra == 'aiohttp'
 Description-Content-Type: text/markdown
 
 # Scale GP Python API library
 
-[![PyPI version](https://github.com/scaleapi/sgp-python-beta/tree/main/<https://img.shields.io/pypi/v/scale-gp-beta.svg?label=pypi%20(stable)>)](https://pypi.org/project/scale-gp-beta/)
+<!-- prettier-ignore -->
+[![PyPI version](https://img.shields.io/pypi/v/scale-gp-beta.svg?label=pypi%20(stable))](https://pypi.org/project/scale-gp-beta/)
 
 The Scale GP Python library provides convenient access to the Scale GP REST API from any Python 3.8+
 application. The library includes type definitions for all request params and response fields,

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/README.md

@@ -1,6 +1,7 @@
 # Scale GP Python API library
 
-[![PyPI version](<https://img.shields.io/pypi/v/scale-gp-beta.svg?label=pypi%20(stable)>)](https://pypi.org/project/scale-gp-beta/)
+<!-- prettier-ignore -->
+[![PyPI version](https://img.shields.io/pypi/v/scale-gp-beta.svg?label=pypi%20(stable))](https://pypi.org/project/scale-gp-beta/)
 
 The Scale GP Python library provides convenient access to the Scale GP REST API from any Python 3.8+
 application. The library includes type definitions for all request params and response fields,

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/api.md

@@ -171,7 +171,13 @@ Methods:
 Types:
 
 ```python
-from scale_gp_beta.types import Span, SpanBatchResponse, SpanUpsertBatchResponse
+from scale_gp_beta.types import (
+    Span,
+    SpanStatus,
+    SpanType,
+    SpanBatchResponse,
+    SpanUpsertBatchResponse,
+)
 ```
 
 Methods:

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "scale-gp-beta"
-version = "0.1.0-alpha.27"
+version = "0.1.0-alpha.29"
 description = "The official Python library for the Scale GP API"
 dynamic = ["readme"]
 license = "Apache-2.0"
@@ -39,7 +39,7 @@ Homepage = "https://github.com/scaleapi/sgp-python-beta"
 Repository = "https://github.com/scaleapi/sgp-python-beta"
 
 [project.optional-dependencies]
-aiohttp = ["aiohttp", "httpx_aiohttp>=0.1.6"]
+aiohttp = ["aiohttp", "httpx_aiohttp>=0.1.8"]
 
 [tool.rye]
 managed = true

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/_base_client.py

@@ -529,6 +529,15 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
             # work around https://github.com/encode/httpx/discussions/2880
             kwargs["extensions"] = {"sni_hostname": prepared_url.host.replace("_", "-")}
 
+        is_body_allowed = options.method.lower() != "get"
+
+        if is_body_allowed:
+            kwargs["json"] = json_data if is_given(json_data) else None
+            kwargs["files"] = files
+        else:
+            headers.pop("Content-Type", None)
+            kwargs.pop("data", None)
+
         # TODO: report this error to httpx
         return self._client.build_request(  # pyright: ignore[reportUnknownMemberType]
             headers=headers,
@@ -540,8 +549,6 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
             # so that passing a `TypedDict` doesn't cause an error.
             # https://github.com/microsoft/pyright/issues/3526#event-6715453066
             params=self.qs.stringify(cast(Mapping[str, Any], params)) if params else None,
-            json=json_data if is_given(json_data) else None,
-            files=files,
             **kwargs,
         )
 

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/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.27"  # x-release-please-version
+__version__ = "0.1.0-alpha.29"  # x-release-please-version

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/lib/tracing/integrations/openai/openai_span_type_map.py

@@ -1,8 +1,8 @@
 from typing import Dict, Optional
 
-from scale_gp_beta.lib.tracing.types import SpanTypeLiterals
+from scale_gp_beta.types import SpanType
 
-OPENAI_SPAN_TYPE_MAP: Dict[str, SpanTypeLiterals] = {
+OPENAI_SPAN_TYPE_MAP: Dict[str, SpanType] = {
     "generation": "COMPLETION",
     "agent": "AGENT_WORKFLOW",
     "function": "CODE_EXECUTION",
@@ -17,7 +17,7 @@ OPENAI_SPAN_TYPE_MAP: Dict[str, SpanTypeLiterals] = {
 }
 
 
-def openai_span_type_map(span_type: Optional[str]) -> SpanTypeLiterals:
+def openai_span_type_map(span_type: Optional[str]) -> SpanType:
     """
     Maps an OpenAI span type string to its corresponding SGP SpanTypeLiteral.
 

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/lib/tracing/span.py

@@ -6,11 +6,12 @@ from typing import TYPE_CHECKING, Type, Optional
 from threading import RLock
 from typing_extensions import override
 
+from scale_gp_beta.types import SpanType, SpanStatus
 from scale_gp_beta.types.span_upsert_batch_params import Item as SpanCreateRequest
 
 from .util import iso_timestamp, generate_span_id
 from .scope import Scope
-from .types import SpanInputParam, SpanOutputParam, SpanTypeLiterals, SpanMetadataParam, SpanStatusLiterals
+from .types import SpanInputParam, SpanOutputParam, SpanMetadataParam
 from .exceptions import ParamsCreationError
 
 if TYPE_CHECKING:
@@ -59,7 +60,7 @@ class BaseSpan:
         input: Optional[SpanInputParam] = None,
         output: Optional[SpanOutputParam] = None,
         metadata: Optional[SpanMetadataParam] = None,
-        span_type: SpanTypeLiterals = "STANDALONE"
+        span_type: SpanType = "STANDALONE"
     ):
         self._name = name
         self._trace_id: str = trace_id or "no_trace_id"
@@ -71,8 +72,8 @@ class BaseSpan:
         self._input: SpanInputParam = input or {}
         self._output: SpanOutputParam = output or {}
         self._metadata: SpanMetadataParam = metadata or {}
-        self._span_type: SpanTypeLiterals = span_type
-        self._status: SpanStatusLiterals = "SUCCESS"
+        self._span_type: SpanType = span_type
+        self._status: SpanStatus = "SUCCESS"
         self._queue_manager = queue_manager
 
         self._contextvar_token: Optional[contextvars.Token[Optional[BaseSpan]]] = None
@@ -108,11 +109,11 @@ class BaseSpan:
         return self._parent_span_id
 
     @property
-    def status(self) -> SpanStatusLiterals:
+    def status(self) -> SpanStatus:
         return self._status
 
     @property
-    def span_type(self) -> SpanTypeLiterals:
+    def span_type(self) -> SpanType:
         return self._span_type
 
     # with setters
@@ -291,7 +292,7 @@ class Span(BaseSpan):
         input: Optional[SpanInputParam] = None,
         output: Optional[SpanOutputParam] = None,
         metadata: Optional[SpanMetadataParam] = None,
-        span_type: SpanTypeLiterals = "STANDALONE",
+        span_type: SpanType = "STANDALONE",
     ):
         super().__init__(name, trace_id, queue_manager, span_id, parent_span_id, group_id, input, output, metadata, span_type)
         self._queue_manager: TraceQueueManager = queue_manager

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/lib/tracing/trace.py

@@ -4,10 +4,12 @@ from types import TracebackType
 from typing import Type, Optional
 from typing_extensions import override
 
+from scale_gp_beta.types import SpanType, SpanStatus
+
 from .span import Span, BaseSpan, NoOpSpan
 from .util import generate_trace_id
 from .scope import Scope
-from .types import SpanInputParam, SpanOutputParam, SpanTypeLiterals, SpanMetadataParam, SpanStatusLiterals
+from .types import SpanInputParam, SpanOutputParam, SpanMetadataParam
 from .trace_queue_manager import TraceQueueManager
 
 log: logging.Logger = logging.getLogger(__name__)
@@ -79,11 +81,11 @@ class BaseTrace:
         return self.root_span.group_id
 
     @property
-    def span_type(self) -> SpanTypeLiterals:
+    def span_type(self) -> SpanType:
         return self.root_span.span_type
 
     @property
-    def status(self) -> SpanStatusLiterals:
+    def status(self) -> SpanStatus:
         return self.root_span.status
 
     def set_error(
@@ -128,7 +130,7 @@ class NoOpTrace(BaseTrace):
             trace_id: Optional[str] = None,
             span_id: Optional[str] = None,
             group_id: Optional[str] = None,
-            span_type: SpanTypeLiterals = "TRACER",
+            span_type: SpanType = "TRACER",
             input: Optional[SpanInputParam] = None,
             output: Optional[SpanOutputParam] = None,
             metadata: Optional[SpanMetadataParam] = None,
@@ -164,7 +166,7 @@ class Trace(BaseTrace):
             trace_id: Optional[str] = None,
             span_id: Optional[str] = None,
             group_id: Optional[str] = None,
-            span_type: SpanTypeLiterals = "TRACER",
+            span_type: SpanType = "TRACER",
             input: Optional[SpanInputParam] = None,
             output: Optional[SpanOutputParam] = None,
             metadata: Optional[SpanMetadataParam] = None,

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/lib/tracing/tracing.py

@@ -1,11 +1,13 @@
 import logging
 from typing import Optional
 
+from scale_gp_beta.types import SpanType
+
 from .span import Span, BaseSpan, NoOpSpan
 from .util import is_disabled
 from .scope import Scope
 from .trace import Trace, BaseTrace, NoOpTrace
-from .types import SpanInputParam, SpanOutputParam, SpanTypeLiterals, SpanMetadataParam
+from .types import SpanInputParam, SpanOutputParam, SpanMetadataParam
 from .trace_queue_manager import TraceQueueManager, tracing_queue_manager
 
 log: logging.Logger = logging.getLogger(__name__)
@@ -51,7 +53,7 @@ def flush_queue() -> None:
 
 def create_trace(
         name: str,
-        span_type: SpanTypeLiterals = "TRACER",
+        span_type: SpanType = "TRACER",
         input: Optional[SpanInputParam] = None,
         output: Optional[SpanOutputParam] = None,
         metadata: Optional[SpanMetadataParam] = None,
@@ -74,7 +76,7 @@ def create_trace(
 
     Args:
         name: The name of the trace.
-        span_type (Optional[SpanTypeLiterals]): Type of root span.
+        span_type (Optional[SpanType]): Type of root span.
         input (Optional[SpanInputParam]): Input of root span.
         output (Optional[SpanOutputParam]): Output of root span.
         metadata (Optional[SpanMetadataParam]): An optional, user-defined metadata.
@@ -131,7 +133,7 @@ def create_trace(
 
 def create_span(
     name: str,
-    span_type: SpanTypeLiterals = "STANDALONE",
+    span_type: SpanType = "STANDALONE",
     input: Optional[SpanInputParam] = None,
     output: Optional[SpanOutputParam] = None,
     metadata: Optional[SpanMetadataParam] = None,
@@ -160,7 +162,7 @@ def create_span(
     Args:
         name (str): A descriptive name for the span (e.g., "database_query",
                     "http_request").
-        span_type (SpanTypeLiterals): The type of the span.
+        span_type (SpanType): The type of the span.
         input (Optional[SpanInputParam], optional): A dictionary containing
             input data or parameters relevant to this span's operation. Defaults to None.
         output (Optional[SpanOutputParam], optional): A dictionary containing

scale_gp_beta-0.1.0a29/src/scale_gp_beta/lib/tracing/types.py

@@ -0,0 +1,5 @@
+from typing_extensions import Any, Dict
+
+SpanInputParam = Dict[str, Any]
+SpanOutputParam = Dict[str, Any]
+SpanMetadataParam = Dict[str, Any]

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/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,34 +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",
-            "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,
@@ -196,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,
@@ -283,40 +261,13 @@ class SpansResource(SyncAPIResource):
         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,
         span_ids: List[str] | NotGiven = NOT_GIVEN,
-        statuses: List[Literal["SUCCESS", "ERROR", "CANCELED"]] | NotGiven = NOT_GIVEN,
+        statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
         trace_id: str | 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",
-                "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,
@@ -344,6 +295,8 @@ 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)
 
           span_ids: Filter by span IDs
@@ -373,6 +326,7 @@ class SpansResource(SyncAPIResource):
                     "group_id": group_id,
                     "names": names,
                     "parents_only": parents_only,
+                    "search_text": search_text,
                     "span_id": span_id,
                     "span_ids": span_ids,
                     "statuses": statuses,
@@ -471,34 +425,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",
-            "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,
@@ -598,7 +526,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,
@@ -685,40 +613,13 @@ class AsyncSpansResource(AsyncAPIResource):
         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,
         span_ids: List[str] | NotGiven = NOT_GIVEN,
-        statuses: List[Literal["SUCCESS", "ERROR", "CANCELED"]] | NotGiven = NOT_GIVEN,
+        statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
         trace_id: str | 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",
-                "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,
@@ -746,6 +647,8 @@ 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)
 
           span_ids: Filter by span IDs
@@ -775,6 +678,7 @@ class AsyncSpansResource(AsyncAPIResource):
                     "group_id": group_id,
                     "names": names,
                     "parents_only": parents_only,
+                    "search_text": search_text,
                     "span_id": span_id,
                     "span_ids": span_ids,
                     "statuses": statuses,

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/types/__init__.py

@@ -9,8 +9,10 @@ from .question import Question as Question
 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

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/types/span.py

@@ -6,6 +6,8 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .span_type import SpanType
+from .span_status import SpanStatus
 
 __all__ = ["Span"]
 
@@ -46,33 +48,6 @@ class Span(BaseModel):
     parent_id: Optional[str] = None
     """Reference to a parent span_id"""
 
-    status: Optional[Literal["SUCCESS", "ERROR", "CANCELED"]] = None
-
-    type: Optional[
-        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",
-            "TRACER",
-            "AGENT_TRACER",
-            "AGENT_WORKFLOW",
-            "STANDALONE",
-        ]
-    ] = None
+    status: Optional[SpanStatus] = None
+
+    type: Optional[SpanType] = None

{scale_gp_beta-0.1.0a27 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/types/span_batch_params.py

@@ -4,9 +4,11 @@ from __future__ import annotations
 
 from typing import Dict, Union, Iterable
 from datetime import datetime
-from typing_extensions import Literal, Required, Annotated, TypedDict
+from typing_extensions import Required, Annotated, TypedDict
 
 from .._utils import PropertyInfo
+from .span_type import SpanType
+from .span_status import SpanStatus
 
 __all__ = ["SpanBatchParams", "Item"]
 
@@ -46,31 +48,6 @@ class Item(TypedDict, total=False):
     parent_id: str
     """Reference to a parent span_id"""
 
-    status: Literal["SUCCESS", "ERROR", "CANCELED"]
-
-    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",
-        "TRACER",
-        "AGENT_TRACER",
-        "AGENT_WORKFLOW",
-        "STANDALONE",
-    ]
+    status: SpanStatus
+
+    type: SpanType