scale-gp-beta 0.1.0a28__py3-none-any.whl → 0.1.0a30__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_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

scale_gp_beta/types/evaluation_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__ = ["EvaluationItem"]
 
@@ -14,7 +15,8 @@ class EvaluationItem(BaseModel):
 
     created_at: datetime
 
-    created_by_user_id: str
+    created_by: Identity
+    """The identity that created the entity."""
 
     data: Dict[str, object]
 

scale_gp_beta/types/evaluation_task.py

@@ -5,8 +5,9 @@ from __future__ import annotations
 from typing import TYPE_CHECKING, Dict, List, Union, Optional
 from typing_extensions import Literal, Annotated, TypeAlias
 
+from pydantic import Field as FieldInfo
+
 from .._utils import PropertyInfo
-from .._compat import PYDANTIC_V2
 from .._models import BaseModel
 from .item_locator import ItemLocator
 
@@ -38,6 +39,8 @@ __all__ = [
     "AutoEvaluationQuestionTaskConfiguration",
     "AutoEvaluationGuidedDecodingEvaluationTask",
     "AutoEvaluationGuidedDecodingEvaluationTaskConfiguration",
+    "AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationStructuredOutputTaskRequestWithItemLocator",
+    "AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationGuidedDecodingTaskRequestWithItemLocator",
     "ContributorEvaluationQuestionTask",
     "ContributorEvaluationQuestionTaskConfiguration",
 ]
@@ -98,6 +101,7 @@ class ChatCompletionEvaluationTaskConfiguration(BaseModel):
 
     top_p: Union[float, ItemLocator, None] = 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.
@@ -325,7 +329,26 @@ class AutoEvaluationQuestionTask(BaseModel):
     task_type: Optional[Literal["auto_evaluation.question"]] = None
 
 
-class AutoEvaluationGuidedDecodingEvaluationTaskConfiguration(BaseModel):
+class AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationStructuredOutputTaskRequestWithItemLocator(
+    BaseModel
+):
+    model: str
+    """model specified as `model_vendor/model_name`"""
+
+    prompt: str
+
+    response_format: Dict[str, object]
+    """JSON schema used for structuring the model response"""
+
+    inference_args: Optional[Dict[str, object]] = None
+    """Additional arguments to pass to the inference request"""
+
+    system_prompt: Optional[str] = None
+
+
+class AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationGuidedDecodingTaskRequestWithItemLocator(
+    BaseModel
+):
     choices: List[str]
     """Choices array cannot be empty"""
 
@@ -340,6 +363,12 @@ class AutoEvaluationGuidedDecodingEvaluationTaskConfiguration(BaseModel):
     system_prompt: Optional[str] = None
 
 
+AutoEvaluationGuidedDecodingEvaluationTaskConfiguration: TypeAlias = Union[
+    AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationStructuredOutputTaskRequestWithItemLocator,
+    AutoEvaluationGuidedDecodingEvaluationTaskConfigurationAutoEvaluationGuidedDecodingTaskRequestWithItemLocator,
+]
+
+
 class AutoEvaluationGuidedDecodingEvaluationTask(BaseModel):
     configuration: AutoEvaluationGuidedDecodingEvaluationTaskConfiguration
 
@@ -387,56 +416,3 @@ EvaluationTask: TypeAlias = Annotated[
 ]
 
 from .container import Container
-
-if PYDANTIC_V2:
-    ChatCompletionEvaluationTask.model_rebuild()
-    ChatCompletionEvaluationTaskConfiguration.model_rebuild()
-    GenericInferenceEvaluationTask.model_rebuild()
-    GenericInferenceEvaluationTaskConfiguration.model_rebuild()
-    GenericInferenceEvaluationTaskConfigurationInferenceConfigurationLaunchInferenceConfiguration.model_rebuild()
-    ApplicationVariantV1EvaluationTask.model_rebuild()
-    ApplicationVariantV1EvaluationTaskConfiguration.model_rebuild()
-    ApplicationVariantV1EvaluationTaskConfigurationHistoryApplicationRequestResponsePairArray.model_rebuild()
-    ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverrides.model_rebuild()
-    ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverridesInitialState.model_rebuild()
-    ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverridesPartialTrace.model_rebuild()
-    MetricEvaluationTask.model_rebuild()
-    MetricEvaluationTaskConfigurationBleuScorerConfigWithItemLocator.model_rebuild()
-    MetricEvaluationTaskConfigurationMeteorScorerConfigWithItemLocator.model_rebuild()
-    MetricEvaluationTaskConfigurationCosineSimilarityScorerConfigWithItemLocator.model_rebuild()
-    MetricEvaluationTaskConfigurationF1ScorerConfigWithItemLocator.model_rebuild()
-    MetricEvaluationTaskConfigurationRougeScorer1ConfigWithItemLocator.model_rebuild()
-    MetricEvaluationTaskConfigurationRougeScorer2ConfigWithItemLocator.model_rebuild()
-    MetricEvaluationTaskConfigurationRougeScorerLConfigWithItemLocator.model_rebuild()
-    AutoEvaluationQuestionTask.model_rebuild()
-    AutoEvaluationQuestionTaskConfiguration.model_rebuild()
-    AutoEvaluationGuidedDecodingEvaluationTask.model_rebuild()
-    AutoEvaluationGuidedDecodingEvaluationTaskConfiguration.model_rebuild()
-    ContributorEvaluationQuestionTask.model_rebuild()
-    ContributorEvaluationQuestionTaskConfiguration.model_rebuild()
-else:
-    ChatCompletionEvaluationTask.update_forward_refs()  # type: ignore
-    ChatCompletionEvaluationTaskConfiguration.update_forward_refs()  # type: ignore
-    GenericInferenceEvaluationTask.update_forward_refs()  # type: ignore
-    GenericInferenceEvaluationTaskConfiguration.update_forward_refs()  # type: ignore
-    GenericInferenceEvaluationTaskConfigurationInferenceConfigurationLaunchInferenceConfiguration.update_forward_refs()  # type: ignore
-    ApplicationVariantV1EvaluationTask.update_forward_refs()  # type: ignore
-    ApplicationVariantV1EvaluationTaskConfiguration.update_forward_refs()  # type: ignore
-    ApplicationVariantV1EvaluationTaskConfigurationHistoryApplicationRequestResponsePairArray.update_forward_refs()  # type: ignore
-    ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverrides.update_forward_refs()  # type: ignore
-    ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverridesInitialState.update_forward_refs()  # type: ignore
-    ApplicationVariantV1EvaluationTaskConfigurationOverridesAgenticApplicationOverridesPartialTrace.update_forward_refs()  # type: ignore
-    MetricEvaluationTask.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationBleuScorerConfigWithItemLocator.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationMeteorScorerConfigWithItemLocator.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationCosineSimilarityScorerConfigWithItemLocator.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationF1ScorerConfigWithItemLocator.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationRougeScorer1ConfigWithItemLocator.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationRougeScorer2ConfigWithItemLocator.update_forward_refs()  # type: ignore
-    MetricEvaluationTaskConfigurationRougeScorerLConfigWithItemLocator.update_forward_refs()  # type: ignore
-    AutoEvaluationQuestionTask.update_forward_refs()  # type: ignore
-    AutoEvaluationQuestionTaskConfiguration.update_forward_refs()  # type: ignore
-    AutoEvaluationGuidedDecodingEvaluationTask.update_forward_refs()  # type: ignore
-    AutoEvaluationGuidedDecodingEvaluationTaskConfiguration.update_forward_refs()  # type: ignore
-    ContributorEvaluationQuestionTask.update_forward_refs()  # type: ignore
-    ContributorEvaluationQuestionTaskConfiguration.update_forward_refs()  # type: ignore