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

scale_gp_beta/resources/spans.py

@@ -255,17 +255,16 @@ class SpansResource(SyncAPIResource):
         sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
         starting_after: str | NotGiven = NOT_GIVEN,
         to_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        application_variant_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_span_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_trace_ids: List[str] | NotGiven = NOT_GIVEN,
         extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         group_id: str | NotGiven = NOT_GIVEN,
         names: List[str] | NotGiven = NOT_GIVEN,
         parents_only: bool | NotGiven = NOT_GIVEN,
-        search_text: str | NotGiven = NOT_GIVEN,
-        span_id: str | NotGiven = NOT_GIVEN,
+        search_texts: List[str] | NotGiven = NOT_GIVEN,
         span_ids: List[str] | NotGiven = NOT_GIVEN,
         statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
-        trace_id: str | NotGiven = NOT_GIVEN,
         trace_ids: List[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.
@@ -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,17 +603,16 @@ class AsyncSpansResource(AsyncAPIResource):
         sort_order: Literal["asc", "desc"] | NotGiven = NOT_GIVEN,
         starting_after: str | NotGiven = NOT_GIVEN,
         to_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
+        application_variant_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_span_ids: List[str] | NotGiven = NOT_GIVEN,
         excluded_trace_ids: List[str] | NotGiven = NOT_GIVEN,
         extra_metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
         group_id: str | NotGiven = NOT_GIVEN,
         names: List[str] | NotGiven = NOT_GIVEN,
         parents_only: bool | NotGiven = NOT_GIVEN,
-        search_text: str | NotGiven = NOT_GIVEN,
-        span_id: str | NotGiven = NOT_GIVEN,
+        search_texts: List[str] | NotGiven = NOT_GIVEN,
         span_ids: List[str] | NotGiven = NOT_GIVEN,
         statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
-        trace_id: str | NotGiven = NOT_GIVEN,
         trace_ids: List[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.
@@ -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_V2:
+    evaluation.Evaluation.model_rebuild(_parent_namespace_depth=0)
+    container.Container.model_rebuild(_parent_namespace_depth=0)
+else:
+    evaluation.Evaluation.update_forward_refs()  # type: ignore
+    container.Container.update_forward_refs()  # type: ignore

scale_gp_beta/types/chat/chat_completion.py

@@ -1,8 +1,10 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import builtins
-from typing import TYPE_CHECKING, List, Optional
-from typing_extensions import Literal
+from typing import TYPE_CHECKING, Dict, List, Union, Optional
+from typing_extensions import Literal, TypeAlias
+
+from pydantic import Field as FieldInfo
 
 from ..._models import BaseModel
 
@@ -15,7 +17,10 @@ __all__ = [
     "ChoiceMessageAudio",
     "ChoiceMessageFunctionCall",
     "ChoiceMessageToolCall",
-    "ChoiceMessageToolCallFunction",
+    "ChoiceMessageToolCallChatCompletionMessageFunctionToolCall",
+    "ChoiceMessageToolCallChatCompletionMessageFunctionToolCallFunction",
+    "ChoiceMessageToolCallChatCompletionMessageCustomToolCall",
+    "ChoiceMessageToolCallChatCompletionMessageCustomToolCallCustom",
     "ChoiceLogprobs",
     "ChoiceLogprobsContent",
     "ChoiceLogprobsContentTopLogprob",
@@ -36,6 +41,7 @@ class ChoiceMessageAnnotationURLCitation(BaseModel):
 
     url: str
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -48,6 +54,7 @@ class ChoiceMessageAnnotation(BaseModel):
 
     url_citation: ChoiceMessageAnnotationURLCitation
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -64,6 +71,7 @@ class ChoiceMessageAudio(BaseModel):
 
     transcript: str
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -76,6 +84,7 @@ class ChoiceMessageFunctionCall(BaseModel):
 
     name: str
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -83,11 +92,12 @@ class ChoiceMessageFunctionCall(BaseModel):
         def __getattr__(self, attr: str) -> object: ...
 
 
-class ChoiceMessageToolCallFunction(BaseModel):
+class ChoiceMessageToolCallChatCompletionMessageFunctionToolCallFunction(BaseModel):
     arguments: str
 
     name: str
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -95,13 +105,14 @@ class ChoiceMessageToolCallFunction(BaseModel):
         def __getattr__(self, attr: str) -> object: ...
 
 
-class ChoiceMessageToolCall(BaseModel):
+class ChoiceMessageToolCallChatCompletionMessageFunctionToolCall(BaseModel):
     id: str
 
-    function: ChoiceMessageToolCallFunction
+    function: ChoiceMessageToolCallChatCompletionMessageFunctionToolCallFunction
 
     type: Literal["function"]
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -109,6 +120,39 @@ class ChoiceMessageToolCall(BaseModel):
         def __getattr__(self, attr: str) -> object: ...
 
 
+class ChoiceMessageToolCallChatCompletionMessageCustomToolCallCustom(BaseModel):
+    input: str
+
+    name: str
+
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+    if TYPE_CHECKING:
+        # Stub to indicate that arbitrary properties are accepted.
+        # To access properties that are not valid identifiers you can use `getattr`, e.g.
+        # `getattr(obj, '$type')`
+        def __getattr__(self, attr: str) -> object: ...
+
+
+class ChoiceMessageToolCallChatCompletionMessageCustomToolCall(BaseModel):
+    id: str
+
+    custom: ChoiceMessageToolCallChatCompletionMessageCustomToolCallCustom
+
+    type: Literal["custom"]
+
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+    if TYPE_CHECKING:
+        # Stub to indicate that arbitrary properties are accepted.
+        # To access properties that are not valid identifiers you can use `getattr`, e.g.
+        # `getattr(obj, '$type')`
+        def __getattr__(self, attr: str) -> object: ...
+
+
+ChoiceMessageToolCall: TypeAlias = Union[
+    ChoiceMessageToolCallChatCompletionMessageFunctionToolCall, ChoiceMessageToolCallChatCompletionMessageCustomToolCall
+]
+
+
 class ChoiceMessage(BaseModel):
     role: Literal["assistant"]
 
@@ -124,6 +168,7 @@ class ChoiceMessage(BaseModel):
 
     tool_calls: Optional[List[ChoiceMessageToolCall]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -138,6 +183,7 @@ class ChoiceLogprobsContentTopLogprob(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -154,6 +200,7 @@ class ChoiceLogprobsContent(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -168,6 +215,7 @@ class ChoiceLogprobsRefusalTopLogprob(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -184,6 +232,7 @@ class ChoiceLogprobsRefusal(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -196,6 +245,7 @@ class ChoiceLogprobs(BaseModel):
 
     refusal: Optional[List[ChoiceLogprobsRefusal]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -212,6 +262,7 @@ class Choice(BaseModel):
 
     logprobs: Optional[ChoiceLogprobs] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -228,6 +279,7 @@ class UsageCompletionTokensDetails(BaseModel):
 
     rejected_prediction_tokens: Optional[int] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -240,6 +292,7 @@ class UsagePromptTokensDetails(BaseModel):
 
     cached_tokens: Optional[int] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -258,6 +311,7 @@ class Usage(BaseModel):
 
     prompt_tokens_details: Optional[UsagePromptTokensDetails] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -282,6 +336,7 @@ class ChatCompletion(BaseModel):
 
     usage: Optional[Usage] = None
 
+    __pydantic_extra__: Dict[str, builtins.object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.

scale_gp_beta/types/chat/chat_completion_chunk.py

@@ -1,9 +1,11 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 import builtins
-from typing import TYPE_CHECKING, List, Optional
+from typing import TYPE_CHECKING, Dict, List, Optional
 from typing_extensions import Literal
 
+from pydantic import Field as FieldInfo
+
 from ..._models import BaseModel
 
 __all__ = [
@@ -29,6 +31,7 @@ class ChoiceDeltaFunctionCall(BaseModel):
 
     name: Optional[str] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -41,6 +44,7 @@ class ChoiceDeltaToolCallFunction(BaseModel):
 
     name: Optional[str] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -57,6 +61,7 @@ class ChoiceDeltaToolCall(BaseModel):
 
     type: Optional[Literal["function"]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -75,6 +80,7 @@ class ChoiceDelta(BaseModel):
 
     tool_calls: Optional[List[ChoiceDeltaToolCall]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -89,6 +95,7 @@ class ChoiceLogprobsContentTopLogprob(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -105,6 +112,7 @@ class ChoiceLogprobsContent(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -119,6 +127,7 @@ class ChoiceLogprobsRefusalTopLogprob(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -135,6 +144,7 @@ class ChoiceLogprobsRefusal(BaseModel):
 
     bytes: Optional[List[int]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -147,6 +157,7 @@ class ChoiceLogprobs(BaseModel):
 
     refusal: Optional[List[ChoiceLogprobsRefusal]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -163,6 +174,7 @@ class Choice(BaseModel):
 
     logprobs: Optional[ChoiceLogprobs] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -179,6 +191,7 @@ class UsageCompletionTokensDetails(BaseModel):
 
     rejected_prediction_tokens: Optional[int] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -191,6 +204,7 @@ class UsagePromptTokensDetails(BaseModel):
 
     cached_tokens: Optional[int] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -209,6 +223,7 @@ class Usage(BaseModel):
 
     prompt_tokens_details: Optional[UsagePromptTokensDetails] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -233,6 +248,7 @@ class ChatCompletionChunk(BaseModel):
 
     usage: Optional[Usage] = None
 
+    __pydantic_extra__: Dict[str, builtins.object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.

scale_gp_beta/types/chat/completion_models_params.py

@@ -8,6 +8,8 @@ __all__ = ["CompletionModelsParams"]
 
 
 class CompletionModelsParams(TypedDict, total=False):
+    check_availability: bool
+
     ending_before: str
 
     limit: int

scale_gp_beta/types/chat/model_definition.py

@@ -1,5 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
+from typing import Optional
 from typing_extensions import Literal
 
 from pydantic import Field as FieldInfo
@@ -31,3 +32,8 @@ class ModelDefinition(BaseModel):
         "fireworks_ai",
     ] = FieldInfo(alias="model_vendor")
     """model vendor, for example `openai`"""
+
+    availability: Optional[Literal["unknown", "available", "unavailable"]] = FieldInfo(
+        alias="model_availability", default=None
+    )
+    """model availability indicating availability status, for example `available`"""

scale_gp_beta/types/completion.py

@@ -4,6 +4,8 @@ import builtins
 from typing import TYPE_CHECKING, Dict, List, Optional
 from typing_extensions import Literal
 
+from pydantic import Field as FieldInfo
+
 from .._models import BaseModel
 
 __all__ = [
@@ -25,6 +27,7 @@ class ChoiceLogprobs(BaseModel):
 
     top_logprobs: Optional[List[Dict[str, float]]] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -41,6 +44,7 @@ class Choice(BaseModel):
 
     logprobs: Optional[ChoiceLogprobs] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -57,6 +61,7 @@ class UsageCompletionTokensDetails(BaseModel):
 
     rejected_prediction_tokens: Optional[int] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -69,6 +74,7 @@ class UsagePromptTokensDetails(BaseModel):
 
     cached_tokens: Optional[int] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -87,6 +93,7 @@ class Usage(BaseModel):
 
     prompt_tokens_details: Optional[UsagePromptTokensDetails] = None
 
+    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.
@@ -109,6 +116,7 @@ class Completion(BaseModel):
 
     usage: Optional[Usage] = None
 
+    __pydantic_extra__: Dict[str, builtins.object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
         # Stub to indicate that arbitrary properties are accepted.
         # To access properties that are not valid identifiers you can use `getattr`, e.g.

scale_gp_beta/types/container.py

@@ -27,9 +27,3 @@ class Container(BaseModel):
     Based on CSS `flex-direction` (see:
     https://developer.mozilla.org/en-US/docs/Web/CSS/flex-direction)
     """
-
-
-if PYDANTIC_V2:
-    Container.model_rebuild()
-else:
-    Container.update_forward_refs()  # type: ignore

scale_gp_beta/types/dataset.py

@@ -5,6 +5,7 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .shared.identity import Identity
 
 __all__ = ["Dataset"]
 
@@ -14,7 +15,8 @@ class Dataset(BaseModel):
 
     created_at: datetime
 
-    created_by_user_id: str
+    created_by: Identity
+    """The identity that created the entity."""
 
     current_version_num: int
 

scale_gp_beta/types/dataset_item.py

@@ -5,6 +5,7 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .._models import BaseModel
+from .shared.identity import Identity
 
 __all__ = ["DatasetItem"]
 
@@ -16,7 +17,8 @@ class DatasetItem(BaseModel):
 
     created_at: datetime
 
-    created_by_user_id: str
+    created_by: Identity
+    """The identity that created the entity."""
 
     data: object
 

scale_gp_beta/types/evaluation.py

@@ -7,8 +7,8 @@ from datetime import datetime
 from typing_extensions import Literal
 
 from .dataset import Dataset
-from .._compat import PYDANTIC_V2
 from .._models import BaseModel
+from .shared.identity import Identity
 
 __all__ = ["Evaluation"]
 
@@ -18,7 +18,8 @@ class Evaluation(BaseModel):
 
     created_at: datetime
 
-    created_by_user_id: str
+    created_by: Identity
+    """The identity that created the entity."""
 
     datasets: List[Dataset]
 
@@ -40,8 +41,3 @@ class Evaluation(BaseModel):
 
 
 from .evaluation_task import EvaluationTask
-
-if PYDANTIC_V2:
-    Evaluation.model_rebuild()
-else:
-    Evaluation.update_forward_refs()  # type: ignore