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

opik/rest_api/types/trace_public.py

@@ -1,25 +1,27 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
 import datetime as dt
-from .json_node_public import JsonNodePublic
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .comment_public import CommentPublic
 from .error_info_public import ErrorInfoPublic
 from .feedback_score_public import FeedbackScorePublic
-from .comment_public import CommentPublic
-import pydantic
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .guardrails_validation_public import GuardrailsValidationPublic
+from .json_list_string_public import JsonListStringPublic
+from .trace_public_visibility_mode import TracePublicVisibilityMode
 
 
 class TracePublic(UniversalBaseModel):
     id: typing.Optional[str] = None
     project_id: typing.Optional[str] = None
-    name: str
+    name: typing.Optional[str] = None
     start_time: dt.datetime
     end_time: typing.Optional[dt.datetime] = None
-    input: typing.Optional[JsonNodePublic] = None
-    output: typing.Optional[JsonNodePublic] = None
-    metadata: typing.Optional[JsonNodePublic] = None
+    input: typing.Optional[JsonListStringPublic] = None
+    output: typing.Optional[JsonListStringPublic] = None
+    metadata: typing.Optional[JsonListStringPublic] = None
     tags: typing.Optional[typing.List[str]] = None
     error_info: typing.Optional[ErrorInfoPublic] = None
     usage: typing.Optional[typing.Dict[str, int]] = None
@@ -28,19 +30,31 @@ class TracePublic(UniversalBaseModel):
     created_by: typing.Optional[str] = None
     last_updated_by: typing.Optional[str] = None
     feedback_scores: typing.Optional[typing.List[FeedbackScorePublic]] = None
+    span_feedback_scores: typing.Optional[typing.List[FeedbackScorePublic]] = pydantic.Field(default=None)
+    """
+    Aggregated feedback scores from all spans in this trace, averaged by score name
+    """
+
     comments: typing.Optional[typing.List[CommentPublic]] = None
+    guardrails_validations: typing.Optional[typing.List[GuardrailsValidationPublic]] = None
     total_estimated_cost: typing.Optional[float] = None
+    span_count: typing.Optional[int] = None
     duration: typing.Optional[float] = pydantic.Field(default=None)
     """
     Duration in milliseconds as a decimal number to support sub-millisecond precision
     """
 
     thread_id: typing.Optional[str] = None
+    visibility_mode: typing.Optional[TracePublicVisibilityMode] = None
+    llm_span_count: typing.Optional[int] = None
+    has_tool_spans: typing.Optional[bool] = None
+    providers: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    List of unique provider names from all spans in this trace, sorted alphabetically
+    """
 
     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/trace_public_visibility_mode.py

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

opik/rest_api/types/trace_thread.py

@@ -1,30 +1,39 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
 import datetime as dt
-from .json_node import JsonNode
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .comment import Comment
+from .feedback_score import FeedbackScore
+from .json_list_string import JsonListString
+from .trace_thread_status import TraceThreadStatus
 
 
 class TraceThread(UniversalBaseModel):
     id: typing.Optional[str] = None
     project_id: typing.Optional[str] = None
+    thread_model_id: typing.Optional[str] = None
     start_time: typing.Optional[dt.datetime] = None
     end_time: typing.Optional[dt.datetime] = None
     duration: typing.Optional[float] = None
-    first_message: typing.Optional[JsonNode] = None
-    last_message: typing.Optional[JsonNode] = None
+    first_message: typing.Optional[JsonListString] = None
+    last_message: typing.Optional[JsonListString] = None
+    feedback_scores: typing.Optional[typing.List[FeedbackScore]] = None
+    status: typing.Optional[TraceThreadStatus] = None
     number_of_messages: typing.Optional[int] = None
+    total_estimated_cost: typing.Optional[float] = None
+    usage: typing.Optional[typing.Dict[str, int]] = None
+    comments: typing.Optional[typing.List[Comment]] = None
+    tags: typing.Optional[typing.List[str]] = None
     last_updated_at: typing.Optional[dt.datetime] = None
+    last_updated_by: typing.Optional[str] = None
     created_by: typing.Optional[str] = None
     created_at: typing.Optional[dt.datetime] = 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/trace_thread_filter.py

@@ -0,0 +1,23 @@
+# 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 .trace_thread_filter_operator import TraceThreadFilterOperator
+
+
+class TraceThreadFilter(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[TraceThreadFilterOperator] = None
+    key: typing.Optional[str] = None
+    value: 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/trace_thread_filter_operator.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TraceThreadFilterOperator = typing.Union[
+    typing.Literal[
+        "contains",
+        "not_contains",
+        "starts_with",
+        "ends_with",
+        "=",
+        "!=",
+        ">",
+        ">=",
+        "<",
+        "<=",
+        "is_empty",
+        "is_not_empty",
+    ],
+    typing.Any,
+]

opik/rest_api/types/trace_thread_filter_public.py

@@ -0,0 +1,23 @@
+# 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 .trace_thread_filter_public_operator import TraceThreadFilterPublicOperator
+
+
+class TraceThreadFilterPublic(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[TraceThreadFilterPublicOperator] = None
+    key: typing.Optional[str] = None
+    value: 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/trace_thread_filter_public_operator.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TraceThreadFilterPublicOperator = typing.Union[
+    typing.Literal[
+        "contains",
+        "not_contains",
+        "starts_with",
+        "ends_with",
+        "=",
+        "!=",
+        ">",
+        ">=",
+        "<",
+        "<=",
+        "is_empty",
+        "is_not_empty",
+    ],
+    typing.Any,
+]

opik/rest_api/types/trace_thread_filter_write.py

@@ -0,0 +1,23 @@
+# 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 .trace_thread_filter_write_operator import TraceThreadFilterWriteOperator
+
+
+class TraceThreadFilterWrite(UniversalBaseModel):
+    field: typing.Optional[str] = None
+    operator: typing.Optional[TraceThreadFilterWriteOperator] = None
+    key: typing.Optional[str] = None
+    value: 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/trace_thread_filter_write_operator.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TraceThreadFilterWriteOperator = typing.Union[
+    typing.Literal[
+        "contains",
+        "not_contains",
+        "starts_with",
+        "ends_with",
+        "=",
+        "!=",
+        ">",
+        ">=",
+        "<",
+        "<=",
+        "is_empty",
+        "is_not_empty",
+    ],
+    typing.Any,
+]

opik/rest_api/types/trace_thread_identifier.py

@@ -0,0 +1,22 @@
+# 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 TraceThreadIdentifier(UniversalBaseModel):
+    project_name: typing.Optional[str] = None
+    project_id: typing.Optional[str] = None
+    thread_id: str
+    truncate: typing.Optional[bool] = 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/trace_thread_llm_as_judge_code.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+from .llm_as_judge_message import LlmAsJudgeMessage
+from .llm_as_judge_model_parameters import LlmAsJudgeModelParameters
+from .llm_as_judge_output_schema import LlmAsJudgeOutputSchema
+
+
+class TraceThreadLlmAsJudgeCode(UniversalBaseModel):
+    model: LlmAsJudgeModelParameters
+    messages: typing.List[LlmAsJudgeMessage]
+    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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_thread_llm_as_judge_code_public.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+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 TraceThreadLlmAsJudgeCodePublic(UniversalBaseModel):
+    model: LlmAsJudgeModelParametersPublic
+    messages: typing.List[LlmAsJudgeMessagePublic]
+    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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_thread_llm_as_judge_code_write.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+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 TraceThreadLlmAsJudgeCodeWrite(UniversalBaseModel):
+    model: LlmAsJudgeModelParametersWrite
+    messages: typing.List[LlmAsJudgeMessageWrite]
+    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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/trace_thread_page.py

@@ -1,10 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .trace_thread import TraceThread
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+import typing_extensions
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
+from .trace_thread import TraceThread
 
 
 class TraceThreadPage(UniversalBaseModel):
@@ -12,11 +14,12 @@ class TraceThreadPage(UniversalBaseModel):
     size: typing.Optional[int] = None
     total: typing.Optional[int] = None
     content: typing.Optional[typing.List[TraceThread]] = None
+    sortable_by: typing_extensions.Annotated[typing.Optional[typing.List[str]], FieldMetadata(alias="sortableBy")] = (
+        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/trace_thread_status.py

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

opik/rest_api/types/trace_thread_update.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 TraceThreadUpdate(UniversalBaseModel):
+    tags: 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/trace_thread_user_defined_metric_python_code.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 TraceThreadUserDefinedMetricPythonCode(UniversalBaseModel):
+    metric: 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/trace_thread_user_defined_metric_python_code_public.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 TraceThreadUserDefinedMetricPythonCodePublic(UniversalBaseModel):
+    metric: 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/trace_thread_user_defined_metric_python_code_write.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 TraceThreadUserDefinedMetricPythonCodeWrite(UniversalBaseModel):
+    metric: 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/trace_update.py

@@ -0,0 +1,39 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .error_info import ErrorInfo
+from .json_list_string import JsonListString
+
+
+class TraceUpdate(UniversalBaseModel):
+    project_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null and project_id not specified, Default Project is assumed
+    """
+
+    project_id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    If null and project_name not specified, Default Project is assumed
+    """
+
+    name: typing.Optional[str] = None
+    end_time: typing.Optional[dt.datetime] = None
+    input: typing.Optional[JsonListString] = None
+    output: typing.Optional[JsonListString] = None
+    metadata: typing.Optional[JsonListString] = None
+    tags: typing.Optional[typing.List[str]] = None
+    error_info: typing.Optional[ErrorInfo] = None
+    thread_id: 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/trace_visibility_mode.py

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

opik/rest_api/types/trace_write.py

@@ -1,12 +1,12 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
 import typing
+
 import pydantic
-import datetime as dt
-from .json_node_write import JsonNodeWrite
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .error_info_write import ErrorInfoWrite
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+from .json_list_string_write import JsonListStringWrite
 
 
 class TraceWrite(UniversalBaseModel):
@@ -16,20 +16,19 @@ class TraceWrite(UniversalBaseModel):
     If null, the default project is used
     """
 
-    name: str
+    name: typing.Optional[str] = None
     start_time: dt.datetime
     end_time: typing.Optional[dt.datetime] = None
-    input: typing.Optional[JsonNodeWrite] = None
-    output: typing.Optional[JsonNodeWrite] = None
-    metadata: typing.Optional[JsonNodeWrite] = None
+    input: typing.Optional[JsonListStringWrite] = None
+    output: typing.Optional[JsonListStringWrite] = None
+    metadata: typing.Optional[JsonListStringWrite] = None
     tags: typing.Optional[typing.List[str]] = None
     error_info: typing.Optional[ErrorInfoWrite] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
     thread_id: 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/usage.py

@@ -1,22 +1,22 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from ..core.pydantic_utilities import UniversalBaseModel
 import typing
-from .completion_tokens_details import CompletionTokensDetails
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
+
 import pydantic
+from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .completion_tokens_details import CompletionTokensDetails
+from .prompt_tokens_details import PromptTokensDetails
 
 
 class Usage(UniversalBaseModel):
     total_tokens: typing.Optional[int] = None
     prompt_tokens: typing.Optional[int] = None
+    prompt_tokens_details: typing.Optional[PromptTokensDetails] = None
     completion_tokens: typing.Optional[int] = None
     completion_tokens_details: typing.Optional[CompletionTokensDetails] = 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/user_defined_metric_python_code.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 UserDefinedMetricPythonCode(UniversalBaseModel):
@@ -11,9 +11,7 @@ class UserDefinedMetricPythonCode(UniversalBaseModel):
     arguments: typing.Dict[str, str]
 
     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/user_defined_metric_python_code_public.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 UserDefinedMetricPythonCodePublic(UniversalBaseModel):
@@ -11,9 +11,7 @@ class UserDefinedMetricPythonCodePublic(UniversalBaseModel):
     arguments: typing.Dict[str, str]
 
     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: