opik 1.6.4__py3-none-any.whl → 1.9.71__py3-none-any.whl

opik/rest_api/types/feedback_score_public_source.py

@@ -2,6 +2,4 @@
 
 import typing
 
-FeedbackScorePublicSource = typing.Union[
-    typing.Literal["ui", "sdk", "online_scoring"], typing.Any
-]
+FeedbackScorePublicSource = typing.Union[typing.Literal["ui", "sdk", "online_scoring"], typing.Any]

opik/rest_api/types/feedback_score_source.py

@@ -2,6 +2,4 @@
 
 import typing
 
-FeedbackScoreSource = typing.Union[
-    typing.Literal["ui", "sdk", "online_scoring"], typing.Any
-]
+FeedbackScoreSource = typing.Union[typing.Literal["ui", "sdk", "online_scoring"], typing.Any]

opik/rest_api/types/feedback_update.py

@@ -1,22 +1,26 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
+
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
-from .numerical_feedback_detail_update import NumericalFeedbackDetailUpdate
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .boolean_feedback_detail_update import BooleanFeedbackDetailUpdate
 from .categorical_feedback_detail_update import CategoricalFeedbackDetailUpdate
+from .numerical_feedback_detail_update import NumericalFeedbackDetailUpdate
 
 
 class Base(UniversalBaseModel):
     id: typing.Optional[str] = None
     name: str
+    description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Optional description for the feedback definition
+    """
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
@@ -30,9 +34,7 @@ class FeedbackUpdate_Numerical(Base):
     details: typing.Optional[NumericalFeedbackDetailUpdate] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
@@ -46,9 +48,21 @@ class FeedbackUpdate_Categorical(Base):
     details: typing.Optional[CategoricalFeedbackDetailUpdate] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+class FeedbackUpdate_Boolean(Base):
+    type: typing.Literal["boolean"] = "boolean"
+    details: typing.Optional[BooleanFeedbackDetailUpdate] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:
@@ -57,4 +71,4 @@ class FeedbackUpdate_Categorical(Base):
             extra = pydantic.Extra.allow
 
 
-FeedbackUpdate = typing.Union[FeedbackUpdate_Numerical, FeedbackUpdate_Categorical]
+FeedbackUpdate = typing.Union[FeedbackUpdate_Numerical, FeedbackUpdate_Categorical, FeedbackUpdate_Boolean]

opik/rest_api/types/function.py

@@ -1,22 +1,19 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .json_object_schema import JsonObjectSchema
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class Function(UniversalBaseModel):
     name: typing.Optional[str] = None
     description: typing.Optional[str] = None
     strict: typing.Optional[bool] = None
-    parameters: typing.Optional[JsonObjectSchema] = None
+    parameters: typing.Optional[typing.Dict[str, typing.Dict[str, typing.Optional[typing.Any]]]] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/function_call.py

@@ -1,9 +1,9 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 
 
 class FunctionCall(UniversalBaseModel):
@@ -11,9 +11,7 @@ class FunctionCall(UniversalBaseModel):
     arguments: typing.Optional[str] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/group_content.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class GroupContent(UniversalBaseModel):
+    label: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/group_content_with_aggregations.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .aggregation_data import AggregationData
+
+
+class GroupContentWithAggregations(UniversalBaseModel):
+    label: typing.Optional[str] = None
+    aggregations: typing.Optional[AggregationData] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/group_detail.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class GroupDetail(UniversalBaseModel):
+    group_sorting: typing.Optional[typing.List[str]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/group_details.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .group_detail import GroupDetail
+
+
+class GroupDetails(UniversalBaseModel):
+    groups_details: typing.Optional[typing.List[GroupDetail]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/guardrail.py

@@ -0,0 +1,34 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .guardrail_name import GuardrailName
+from .guardrail_result import GuardrailResult
+from .json_node import JsonNode
+
+
+class Guardrail(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    entity_id: str
+    secondary_id: str
+    project_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null, the default project is used
+    """
+
+    project_id: typing.Optional[str] = None
+    name: GuardrailName
+    result: GuardrailResult
+    config: JsonNode
+    details: JsonNode
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/guardrail_batch.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .guardrail import Guardrail
+
+
+class GuardrailBatch(UniversalBaseModel):
+    guardrails: typing.List[Guardrail]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/guardrail_name.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+GuardrailName = typing.Union[typing.Literal["TOPIC", "PII"], typing.Any]

opik/rest_api/types/guardrail_result.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+GuardrailResult = typing.Union[typing.Literal["passed", "failed"], typing.Any]

opik/rest_api/types/guardrail_write.py

@@ -0,0 +1,33 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .guardrail_write_name import GuardrailWriteName
+from .guardrail_write_result import GuardrailWriteResult
+from .json_node import JsonNode
+
+
+class GuardrailWrite(UniversalBaseModel):
+    entity_id: str
+    secondary_id: str
+    project_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null, the default project is used
+    """
+
+    project_id: typing.Optional[str] = None
+    name: GuardrailWriteName
+    result: GuardrailWriteResult
+    config: JsonNode
+    details: JsonNode
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/guardrail_write_name.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+GuardrailWriteName = typing.Union[typing.Literal["TOPIC", "PII"], typing.Any]

opik/rest_api/types/guardrail_write_result.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+GuardrailWriteResult = typing.Union[typing.Literal["passed", "failed"], typing.Any]

opik/rest_api/types/guardrails_validation.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .check import Check
+
+
+class GuardrailsValidation(UniversalBaseModel):
+    span_id: typing.Optional[str] = None
+    checks: typing.Optional[typing.List[Check]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/guardrails_validation_public.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .check_public import CheckPublic
+
+
+class GuardrailsValidationPublic(UniversalBaseModel):
+    span_id: typing.Optional[str] = None
+    checks: typing.Optional[typing.List[CheckPublic]] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/ids_holder.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class IdsHolder(UniversalBaseModel):
+    ids: typing.List[str]
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/image_url.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ImageUrl(UniversalBaseModel):
+    url: str
+    detail: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/image_url_public.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ImageUrlPublic(UniversalBaseModel):
+    url: str
+    detail: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/image_url_write.py

@@ -0,0 +1,20 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+
+
+class ImageUrlWrite(UniversalBaseModel):
+    url: str
+    detail: typing.Optional[str] = None
+
+    if IS_PYDANTIC_V2:
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/json_list_string.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+JsonListString = typing.Union[
+    typing.Dict[str, typing.Optional[typing.Any]], typing.List[typing.Dict[str, typing.Optional[typing.Any]]], str
+]

opik/rest_api/types/json_list_string_compare.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+JsonListStringCompare = typing.Union[
+    typing.Dict[str, typing.Optional[typing.Any]], typing.List[typing.Dict[str, typing.Optional[typing.Any]]], str
+]

opik/rest_api/types/json_list_string_experiment_item_bulk_write_view.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+JsonListStringExperimentItemBulkWriteView = typing.Union[
+    typing.Dict[str, typing.Optional[typing.Any]], typing.List[typing.Dict[str, typing.Optional[typing.Any]]], str
+]

opik/rest_api/types/json_list_string_public.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+JsonListStringPublic = typing.Union[
+    typing.Dict[str, typing.Optional[typing.Any]], typing.List[typing.Dict[str, typing.Optional[typing.Any]]], str
+]

opik/rest_api/types/json_list_string_write.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+JsonListStringWrite = typing.Union[
+    typing.Dict[str, typing.Optional[typing.Any]], typing.List[typing.Dict[str, typing.Optional[typing.Any]]], str
+]

opik/rest_api/types/json_schema.py

@@ -1,25 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
+
+import pydantic
 import typing_extensions
-from .json_object_schema import JsonObjectSchema
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
 
 
 class JsonSchema(UniversalBaseModel):
     name: typing.Optional[str] = None
     strict: typing.Optional[bool] = None
     schema_: typing_extensions.Annotated[
-        typing.Optional[JsonObjectSchema], FieldMetadata(alias="schema")
+        typing.Optional[typing.Dict[str, typing.Dict[str, typing.Optional[typing.Any]]]], FieldMetadata(alias="schema")
     ] = None
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/llm_as_judge_code.py

@@ -1,28 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .llm_as_judge_model_parameters import LlmAsJudgeModelParameters
 import typing
-from .llm_as_judge_message import LlmAsJudgeMessage
+
+import pydantic
 import typing_extensions
-from .llm_as_judge_output_schema import LlmAsJudgeOutputSchema
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .llm_as_judge_message import LlmAsJudgeMessage
+from .llm_as_judge_model_parameters import LlmAsJudgeModelParameters
+from .llm_as_judge_output_schema import LlmAsJudgeOutputSchema
 
 
 class LlmAsJudgeCode(UniversalBaseModel):
     model: LlmAsJudgeModelParameters
     messages: typing.List[LlmAsJudgeMessage]
     variables: typing.Dict[str, str]
-    schema_: typing_extensions.Annotated[
-        typing.List[LlmAsJudgeOutputSchema], FieldMetadata(alias="schema")
-    ]
+    schema_: typing_extensions.Annotated[typing.List[LlmAsJudgeOutputSchema], FieldMetadata(alias="schema")]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/llm_as_judge_code_public.py

@@ -1,28 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .llm_as_judge_model_parameters_public import LlmAsJudgeModelParametersPublic
 import typing
-from .llm_as_judge_message_public import LlmAsJudgeMessagePublic
+
+import pydantic
 import typing_extensions
-from .llm_as_judge_output_schema_public import LlmAsJudgeOutputSchemaPublic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .llm_as_judge_message_public import LlmAsJudgeMessagePublic
+from .llm_as_judge_model_parameters_public import LlmAsJudgeModelParametersPublic
+from .llm_as_judge_output_schema_public import LlmAsJudgeOutputSchemaPublic
 
 
 class LlmAsJudgeCodePublic(UniversalBaseModel):
     model: LlmAsJudgeModelParametersPublic
     messages: typing.List[LlmAsJudgeMessagePublic]
     variables: typing.Dict[str, str]
-    schema_: typing_extensions.Annotated[
-        typing.List[LlmAsJudgeOutputSchemaPublic], FieldMetadata(alias="schema")
-    ]
+    schema_: typing_extensions.Annotated[typing.List[LlmAsJudgeOutputSchemaPublic], FieldMetadata(alias="schema")]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config:

opik/rest_api/types/llm_as_judge_code_write.py

@@ -1,28 +1,24 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-from .llm_as_judge_model_parameters_write import LlmAsJudgeModelParametersWrite
 import typing
-from .llm_as_judge_message_write import LlmAsJudgeMessageWrite
+
+import pydantic
 import typing_extensions
-from .llm_as_judge_output_schema_write import LlmAsJudgeOutputSchemaWrite
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
+from .llm_as_judge_message_write import LlmAsJudgeMessageWrite
+from .llm_as_judge_model_parameters_write import LlmAsJudgeModelParametersWrite
+from .llm_as_judge_output_schema_write import LlmAsJudgeOutputSchemaWrite
 
 
 class LlmAsJudgeCodeWrite(UniversalBaseModel):
     model: LlmAsJudgeModelParametersWrite
     messages: typing.List[LlmAsJudgeMessageWrite]
     variables: typing.Dict[str, str]
-    schema_: typing_extensions.Annotated[
-        typing.List[LlmAsJudgeOutputSchemaWrite], FieldMetadata(alias="schema")
-    ]
+    schema_: typing_extensions.Annotated[typing.List[LlmAsJudgeOutputSchemaWrite], FieldMetadata(alias="schema")]
 
     if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
+        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2
     else:
 
         class Config: