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

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_create_params.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union, Iterable
+from typing import Dict, Union, Iterable
 from typing_extensions import Literal, Required, TypedDict
 
+from ..._types import SequenceNotStr
+
 __all__ = ["CompletionCreateParamsBase", "CompletionCreateParamsNonStreaming", "CompletionCreateParamsStreaming"]
 
 
@@ -67,7 +69,7 @@ class CompletionCreateParamsBase(TypedDict, total=False):
     dashboard.
     """
 
-    modalities: List[str]
+    modalities: SequenceNotStr[str]
     """Output types that you would like the model to generate for this request."""
 
     n: int
@@ -100,7 +102,7 @@ class CompletionCreateParamsBase(TypedDict, total=False):
     requests with same seed.
     """
 
-    stop: Union[str, List[str]]
+    stop: Union[str, SequenceNotStr[str]]
     """Up to 4 sequences where the API will stop generating further tokens."""
 
     store: bool

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/completion_create_params.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Union
+from typing import Dict, Union
 from typing_extensions import Literal, Required, TypedDict
 
+from .._types import SequenceNotStr
+
 __all__ = ["CompletionCreateParamsBase", "CompletionCreateParamsNonStreaming", "CompletionCreateParamsStreaming"]
 
 
@@ -12,7 +14,7 @@ class CompletionCreateParamsBase(TypedDict, total=False):
     model: Required[str]
     """model specified as `model_vendor/model`, for example `openai/gpt-4o`"""
 
-    prompt: Required[Union[str, List[str]]]
+    prompt: Required[Union[str, SequenceNotStr[str]]]
     """The prompt to generate completions for, encoded as a string"""
 
     best_of: int
@@ -58,7 +60,7 @@ class CompletionCreateParamsBase(TypedDict, total=False):
     Determinism is not guaranteed.
     """
 
-    stop: Union[str, List[str]]
+    stop: Union[str, SequenceNotStr[str]]
     """Up to 4 sequences where the API will stop generating further tokens."""
 
     stream_options: Dict[str, object]

scale_gp_beta/types/container.py

@@ -5,13 +5,13 @@ from __future__ import annotations
 from typing import TYPE_CHECKING, List, Union, Optional
 from typing_extensions import Literal, TypeAlias, TypeAliasType
 
-from .._compat import PYDANTIC_V2
+from .._compat import PYDANTIC_V1
 from .._models import BaseModel
 from .component import Component
 
 __all__ = ["Container", "Child"]
 
-if TYPE_CHECKING or PYDANTIC_V2:
+if TYPE_CHECKING or not PYDANTIC_V1:
     Child = TypeAliasType("Child", Union["Container", Component])
 else:
     Child: TypeAlias = Union["Container", Component]
@@ -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/container_param.py

@@ -5,12 +5,12 @@ from __future__ import annotations
 from typing import TYPE_CHECKING, Union, Iterable
 from typing_extensions import Literal, Required, TypeAlias, TypedDict, TypeAliasType
 
-from .._compat import PYDANTIC_V2
+from .._compat import PYDANTIC_V1
 from .component_param import ComponentParam
 
 __all__ = ["ContainerParam", "Child"]
 
-if TYPE_CHECKING or PYDANTIC_V2:
+if TYPE_CHECKING or not PYDANTIC_V1:
     Child = TypeAliasType("Child", Union["ContainerParam", ComponentParam])
 else:
     Child: TypeAlias = Union["ContainerParam", ComponentParam]

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_create_params.py

@@ -2,9 +2,11 @@
 
 from __future__ import annotations
 
-from typing import Dict, List, Iterable
+from typing import Dict, Iterable
 from typing_extensions import Required, TypedDict
 
+from .._types import SequenceNotStr
+
 __all__ = ["DatasetCreateParams"]
 
 
@@ -16,5 +18,5 @@ class DatasetCreateParams(TypedDict, total=False):
 
     description: str
 
-    tags: List[str]
+    tags: SequenceNotStr[str]
     """The tags associated with the entity"""

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/dataset_list_params.py

@@ -2,9 +2,10 @@
 
 from __future__ import annotations
 
-from typing import List
 from typing_extensions import Literal, TypedDict
 
+from .._types import SequenceNotStr
+
 __all__ = ["DatasetListParams"]
 
 
@@ -21,4 +22,4 @@ class DatasetListParams(TypedDict, total=False):
 
     starting_after: str
 
-    tags: List[str]
+    tags: SequenceNotStr[str]

scale_gp_beta/types/dataset_update_params.py

@@ -2,9 +2,10 @@
 
 from __future__ import annotations
 
-from typing import List
 from typing_extensions import TypedDict
 
+from .._types import SequenceNotStr
+
 __all__ = ["DatasetUpdateParams"]
 
 
@@ -13,5 +14,5 @@ class DatasetUpdateParams(TypedDict, total=False):
 
     name: str
 
-    tags: List[str]
+    tags: SequenceNotStr[str]
     """The tags associated with the entity"""

scale_gp_beta/types/evaluation.py

@@ -2,13 +2,13 @@
 
 from __future__ import annotations
 
-from typing import List, Optional
+from typing import Dict, List, Optional
 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]
 
@@ -33,6 +34,9 @@ class Evaluation(BaseModel):
 
     description: Optional[str] = None
 
+    metadata: Optional[Dict[str, object]] = None
+    """Metadata key-value pairs for the evaluation"""
+
     object: Optional[Literal["evaluation"]] = None
 
     tasks: Optional[List["EvaluationTask"]] = None
@@ -40,8 +44,3 @@ class Evaluation(BaseModel):
 
 
 from .evaluation_task import EvaluationTask
-
-if PYDANTIC_V2:
-    Evaluation.model_rebuild()
-else:
-    Evaluation.update_forward_refs()  # type: ignore