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

scale_gp_beta/resources/spans.py

@@ -17,7 +17,7 @@ from ..types import (
     span_update_params,
     span_upsert_batch_params,
 )
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -64,23 +64,23 @@ class SpansResource(SyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        id: str | NotGiven = NOT_GIVEN,
-        application_interaction_id: str | NotGiven = NOT_GIVEN,
-        application_variant_id: str | NotGiven = NOT_GIVEN,
-        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        group_id: str | NotGiven = NOT_GIVEN,
-        input: Dict[str, object] | NotGiven = NOT_GIVEN,
-        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        output: Dict[str, object] | NotGiven = NOT_GIVEN,
-        parent_id: str | NotGiven = NOT_GIVEN,
-        status: SpanStatus | NotGiven = NOT_GIVEN,
-        type: SpanType | NotGiven = NOT_GIVEN,
+        id: str | Omit = omit,
+        application_interaction_id: str | Omit = omit,
+        application_variant_id: str | Omit = omit,
+        end_timestamp: Union[str, datetime] | Omit = omit,
+        group_id: str | Omit = omit,
+        input: Dict[str, object] | Omit = omit,
+        metadata: Dict[str, object] | Omit = omit,
+        output: Dict[str, object] | Omit = omit,
+        parent_id: str | Omit = omit,
+        status: SpanStatus | Omit = omit,
+        type: SpanType | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Span:
         """
         Create Span
@@ -142,7 +142,7 @@ class SpansResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Span:
         """
         Get Span
@@ -170,17 +170,17 @@ class SpansResource(SyncAPIResource):
         self,
         span_id: str,
         *,
-        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        name: str | NotGiven = NOT_GIVEN,
-        output: Dict[str, object] | NotGiven = NOT_GIVEN,
-        status: SpanStatus | NotGiven = NOT_GIVEN,
+        end_timestamp: Union[str, datetime] | Omit = omit,
+        metadata: Dict[str, object] | Omit = omit,
+        name: str | Omit = omit,
+        output: Dict[str, object] | Omit = omit,
+        status: SpanStatus | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Span:
         """
         Update Span
@@ -223,7 +223,7 @@ class SpansResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SpanBatchResponse:
         """
         Create Spans in Batch
@@ -249,30 +249,30 @@ class SpansResource(SyncAPIResource):
     def search(
         self,
         *,
-        ending_before: str | NotGiven = NOT_GIVEN,
-        from_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
-        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_texts: List[str] | NotGiven = NOT_GIVEN,
-        span_ids: List[str] | NotGiven = NOT_GIVEN,
-        statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
-        trace_ids: List[str] | NotGiven = NOT_GIVEN,
-        types: List[SpanType] | NotGiven = NOT_GIVEN,
+        ending_before: str | Omit = omit,
+        from_ts: Union[str, datetime] | Omit = omit,
+        limit: int | Omit = omit,
+        sort_order: Literal["asc", "desc"] | Omit = omit,
+        starting_after: str | Omit = omit,
+        to_ts: Union[str, datetime] | Omit = omit,
+        application_variant_ids: SequenceNotStr[str] | Omit = omit,
+        excluded_span_ids: SequenceNotStr[str] | Omit = omit,
+        excluded_trace_ids: SequenceNotStr[str] | Omit = omit,
+        extra_metadata: Dict[str, object] | Omit = omit,
+        group_id: str | Omit = omit,
+        names: SequenceNotStr[str] | Omit = omit,
+        parents_only: bool | Omit = omit,
+        search_texts: SequenceNotStr[str] | Omit = omit,
+        span_ids: SequenceNotStr[str] | Omit = omit,
+        statuses: List[SpanStatus] | Omit = omit,
+        trace_ids: SequenceNotStr[str] | Omit = omit,
+        types: List[SpanType] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SyncCursorPage[Span]:
         """
         Search and list spans
@@ -362,7 +362,7 @@ class SpansResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SpanUpsertBatchResponse:
         """
         Upsert Spans in Batch
@@ -412,23 +412,23 @@ class AsyncSpansResource(AsyncAPIResource):
         name: str,
         start_timestamp: Union[str, datetime],
         trace_id: str,
-        id: str | NotGiven = NOT_GIVEN,
-        application_interaction_id: str | NotGiven = NOT_GIVEN,
-        application_variant_id: str | NotGiven = NOT_GIVEN,
-        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        group_id: str | NotGiven = NOT_GIVEN,
-        input: Dict[str, object] | NotGiven = NOT_GIVEN,
-        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        output: Dict[str, object] | NotGiven = NOT_GIVEN,
-        parent_id: str | NotGiven = NOT_GIVEN,
-        status: SpanStatus | NotGiven = NOT_GIVEN,
-        type: SpanType | NotGiven = NOT_GIVEN,
+        id: str | Omit = omit,
+        application_interaction_id: str | Omit = omit,
+        application_variant_id: str | Omit = omit,
+        end_timestamp: Union[str, datetime] | Omit = omit,
+        group_id: str | Omit = omit,
+        input: Dict[str, object] | Omit = omit,
+        metadata: Dict[str, object] | Omit = omit,
+        output: Dict[str, object] | Omit = omit,
+        parent_id: str | Omit = omit,
+        status: SpanStatus | Omit = omit,
+        type: SpanType | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Span:
         """
         Create Span
@@ -490,7 +490,7 @@ class AsyncSpansResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Span:
         """
         Get Span
@@ -518,17 +518,17 @@ class AsyncSpansResource(AsyncAPIResource):
         self,
         span_id: str,
         *,
-        end_timestamp: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        metadata: Dict[str, object] | NotGiven = NOT_GIVEN,
-        name: str | NotGiven = NOT_GIVEN,
-        output: Dict[str, object] | NotGiven = NOT_GIVEN,
-        status: SpanStatus | NotGiven = NOT_GIVEN,
+        end_timestamp: Union[str, datetime] | Omit = omit,
+        metadata: Dict[str, object] | Omit = omit,
+        name: str | Omit = omit,
+        output: Dict[str, object] | Omit = omit,
+        status: SpanStatus | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Span:
         """
         Update Span
@@ -571,7 +571,7 @@ class AsyncSpansResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SpanBatchResponse:
         """
         Create Spans in Batch
@@ -597,30 +597,30 @@ class AsyncSpansResource(AsyncAPIResource):
     def search(
         self,
         *,
-        ending_before: str | NotGiven = NOT_GIVEN,
-        from_ts: Union[str, datetime] | NotGiven = NOT_GIVEN,
-        limit: int | NotGiven = NOT_GIVEN,
-        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_texts: List[str] | NotGiven = NOT_GIVEN,
-        span_ids: List[str] | NotGiven = NOT_GIVEN,
-        statuses: List[SpanStatus] | NotGiven = NOT_GIVEN,
-        trace_ids: List[str] | NotGiven = NOT_GIVEN,
-        types: List[SpanType] | NotGiven = NOT_GIVEN,
+        ending_before: str | Omit = omit,
+        from_ts: Union[str, datetime] | Omit = omit,
+        limit: int | Omit = omit,
+        sort_order: Literal["asc", "desc"] | Omit = omit,
+        starting_after: str | Omit = omit,
+        to_ts: Union[str, datetime] | Omit = omit,
+        application_variant_ids: SequenceNotStr[str] | Omit = omit,
+        excluded_span_ids: SequenceNotStr[str] | Omit = omit,
+        excluded_trace_ids: SequenceNotStr[str] | Omit = omit,
+        extra_metadata: Dict[str, object] | Omit = omit,
+        group_id: str | Omit = omit,
+        names: SequenceNotStr[str] | Omit = omit,
+        parents_only: bool | Omit = omit,
+        search_texts: SequenceNotStr[str] | Omit = omit,
+        span_ids: SequenceNotStr[str] | Omit = omit,
+        statuses: List[SpanStatus] | Omit = omit,
+        trace_ids: SequenceNotStr[str] | Omit = omit,
+        types: List[SpanType] | Omit = omit,
         # 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,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AsyncPaginator[Span, AsyncCursorPage[Span]]:
         """
         Search and list spans
@@ -710,7 +710,7 @@ class AsyncSpansResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SpanUpsertBatchResponse:
         """
         Upsert Spans in Batch

scale_gp_beta/types/__init__.py

@@ -75,9 +75,9 @@ from .dataset_item_batch_create_response import DatasetItemBatchCreateResponse a
 # 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:
+if _compat.PYDANTIC_V1:
     evaluation.Evaluation.update_forward_refs()  # type: ignore
     container.Container.update_forward_refs()  # type: ignore
+else:
+    evaluation.Evaluation.model_rebuild(_parent_namespace_depth=0)
+    container.Container.model_rebuild(_parent_namespace_depth=0)

scale_gp_beta/types/chat/chat_completion.py

@@ -41,12 +41,17 @@ class ChoiceMessageAnnotationURLCitation(BaseModel):
 
     url: str
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageAnnotation(BaseModel):
@@ -54,12 +59,17 @@ class ChoiceMessageAnnotation(BaseModel):
 
     url_citation: ChoiceMessageAnnotationURLCitation
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageAudio(BaseModel):
@@ -71,12 +81,17 @@ class ChoiceMessageAudio(BaseModel):
 
     transcript: str
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageFunctionCall(BaseModel):
@@ -84,12 +99,17 @@ class ChoiceMessageFunctionCall(BaseModel):
 
     name: str
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageToolCallChatCompletionMessageFunctionToolCallFunction(BaseModel):
@@ -97,12 +117,17 @@ class ChoiceMessageToolCallChatCompletionMessageFunctionToolCallFunction(BaseMod
 
     name: str
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageToolCallChatCompletionMessageFunctionToolCall(BaseModel):
@@ -112,12 +137,17 @@ class ChoiceMessageToolCallChatCompletionMessageFunctionToolCall(BaseModel):
 
     type: Literal["function"]
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageToolCallChatCompletionMessageCustomToolCallCustom(BaseModel):
@@ -125,12 +155,17 @@ class ChoiceMessageToolCallChatCompletionMessageCustomToolCallCustom(BaseModel):
 
     name: str
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceMessageToolCallChatCompletionMessageCustomToolCall(BaseModel):
@@ -140,12 +175,17 @@ class ChoiceMessageToolCallChatCompletionMessageCustomToolCall(BaseModel):
 
     type: Literal["custom"]
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 ChoiceMessageToolCall: TypeAlias = Union[
@@ -168,12 +208,17 @@ class ChoiceMessage(BaseModel):
 
     tool_calls: Optional[List[ChoiceMessageToolCall]] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceLogprobsContentTopLogprob(BaseModel):
@@ -183,12 +228,17 @@ class ChoiceLogprobsContentTopLogprob(BaseModel):
 
     bytes: Optional[List[int]] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceLogprobsContent(BaseModel):
@@ -200,12 +250,17 @@ class ChoiceLogprobsContent(BaseModel):
 
     bytes: Optional[List[int]] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceLogprobsRefusalTopLogprob(BaseModel):
@@ -215,12 +270,17 @@ class ChoiceLogprobsRefusalTopLogprob(BaseModel):
 
     bytes: Optional[List[int]] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceLogprobsRefusal(BaseModel):
@@ -232,12 +292,17 @@ class ChoiceLogprobsRefusal(BaseModel):
 
     bytes: Optional[List[int]] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChoiceLogprobs(BaseModel):
@@ -245,12 +310,17 @@ class ChoiceLogprobs(BaseModel):
 
     refusal: Optional[List[ChoiceLogprobsRefusal]] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class Choice(BaseModel):
@@ -262,12 +332,17 @@ class Choice(BaseModel):
 
     logprobs: Optional[ChoiceLogprobs] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class UsageCompletionTokensDetails(BaseModel):
@@ -279,12 +354,17 @@ class UsageCompletionTokensDetails(BaseModel):
 
     rejected_prediction_tokens: Optional[int] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class UsagePromptTokensDetails(BaseModel):
@@ -292,12 +372,17 @@ class UsagePromptTokensDetails(BaseModel):
 
     cached_tokens: Optional[int] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class Usage(BaseModel):
@@ -311,12 +396,17 @@ class Usage(BaseModel):
 
     prompt_tokens_details: Optional[UsagePromptTokensDetails] = None
 
-    __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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: ...
+    else:
+        __pydantic_extra__: Dict[str, object]
 
 
 class ChatCompletion(BaseModel):
@@ -336,9 +426,14 @@ class ChatCompletion(BaseModel):
 
     usage: Optional[Usage] = None
 
-    __pydantic_extra__: Dict[str, builtins.object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
     if TYPE_CHECKING:
+        # Some versions of Pydantic <2.8.0 have a bug and don’t allow assigning a
+        # value to this field, so for compatibility we avoid doing it at runtime.
+        __pydantic_extra__: Dict[str, builtins.object] = FieldInfo(init=False)  # pyright: ignore[reportIncompatibleVariableOverride]
+
         # 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) -> builtins.object: ...
+    else:
+        __pydantic_extra__: Dict[str, builtins.object]