scale-gp-beta 0.1.0a28__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.0a28 → scale_gp_beta-0.1.0a29}/CHANGELOG.md

@@ -1,5 +1,20 @@
 # 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)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: scale-gp-beta
-Version: 0.1.0a28
+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

{scale_gp_beta-0.1.0a28 → 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.0a28 → scale_gp_beta-0.1.0a29}/pyproject.toml

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

{scale_gp_beta-0.1.0a28 → 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.28"  # x-release-please-version
+__version__ = "0.1.0-alpha.29"  # x-release-please-version

{scale_gp_beta-0.1.0a28 → 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.0a28 → 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.0a28 → 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.0a28 → 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.0a28 → 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,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,
@@ -287,39 +264,10 @@ class SpansResource(SyncAPIResource):
         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",
-                "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,
@@ -477,35 +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",
-            "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 +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,
@@ -695,39 +616,10 @@ class AsyncSpansResource(AsyncAPIResource):
         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",
-                "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,

{scale_gp_beta-0.1.0a28 → 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.0a28 → 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,34 +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",
-            "GUARDRAIL",
-            "TRACER",
-            "AGENT_TRACER",
-            "AGENT_WORKFLOW",
-            "STANDALONE",
-        ]
-    ] = None
+    status: Optional[SpanStatus] = None
+
+    type: Optional[SpanType] = None

{scale_gp_beta-0.1.0a28 → 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,32 +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",
-        "GUARDRAIL",
-        "TRACER",
-        "AGENT_TRACER",
-        "AGENT_WORKFLOW",
-        "STANDALONE",
-    ]
+    status: SpanStatus
+
+    type: SpanType

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/types/span_create_params.py

@@ -4,9 +4,11 @@ from __future__ import annotations
 
 from typing import Dict, Union
 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__ = ["SpanCreateParams"]
 
@@ -42,32 +44,6 @@ class SpanCreateParams(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",
-        "GUARDRAIL",
-        "TRACER",
-        "AGENT_TRACER",
-        "AGENT_WORKFLOW",
-        "STANDALONE",
-    ]
+    status: SpanStatus
+
+    type: SpanType

{scale_gp_beta-0.1.0a28 → scale_gp_beta-0.1.0a29}/src/scale_gp_beta/types/span_search_params.py

@@ -7,6 +7,8 @@ from datetime import datetime
 from typing_extensions import Literal, Annotated, TypedDict
 
 from .._utils import PropertyInfo
+from .span_type import SpanType
+from .span_status import SpanStatus
 
 __all__ = ["SpanSearchParams"]
 
@@ -53,7 +55,7 @@ class SpanSearchParams(TypedDict, total=False):
     span_ids: List[str]
     """Filter by span IDs"""
 
-    statuses: List[Literal["SUCCESS", "ERROR", "CANCELED"]]
+    statuses: List[SpanStatus]
     """Filter on span status"""
 
     trace_id: str
@@ -62,32 +64,4 @@ class SpanSearchParams(TypedDict, total=False):
     trace_ids: List[str]
     """Filter by trace IDs"""
 
-    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",
-        ]
-    ]
+    types: List[SpanType]

scale_gp_beta-0.1.0a29/src/scale_gp_beta/types/span_status.py

@@ -0,0 +1,7 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing_extensions import Literal, TypeAlias
+
+__all__ = ["SpanStatus"]
+
+SpanStatus: TypeAlias = Literal["SUCCESS", "ERROR", "CANCELED"]