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

opik/rest_api/types/experiment_score.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 ExperimentScore(UniversalBaseModel):
+    name: str
+    value: float
+
+    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/experiment_score_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 ExperimentScorePublic(UniversalBaseModel):
+    name: str
+    value: float
+
+    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/experiment_score_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 ExperimentScoreWrite(UniversalBaseModel):
+    name: str
+    value: float
+
+    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/experiment_status.py

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

opik/rest_api/types/feedback.py

@@ -9,6 +9,7 @@ import pydantic
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .boolean_feedback_detail import BooleanFeedbackDetail
 from .categorical_feedback_detail import CategoricalFeedbackDetail
 from .numerical_feedback_detail import NumericalFeedbackDetail
 
@@ -16,6 +17,11 @@ from .numerical_feedback_detail import NumericalFeedbackDetail
 class Base(UniversalBaseModel):
     id: typing.Optional[str] = None
     name: str
+    description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Optional description for the feedback definition
+    """
+
     created_at: typing_extensions.Annotated[typing.Optional[dt.datetime], FieldMetadata(alias="createdAt")] = None
     created_by: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="createdBy")] = None
     last_updated_at: typing_extensions.Annotated[typing.Optional[dt.datetime], FieldMetadata(alias="lastUpdatedAt")] = (
@@ -69,4 +75,22 @@ class Feedback_Categorical(Base):
             extra = pydantic.Extra.allow
 
 
-Feedback = typing.Union[Feedback_Numerical, Feedback_Categorical]
+class Feedback_Boolean(Base):
+    type: typing.Literal["boolean"] = "boolean"
+    details: typing.Optional[BooleanFeedbackDetail] = None
+    created_at: typing.Optional[dt.datetime] = None
+    created_by: typing.Optional[str] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
+    last_updated_by: 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
+
+
+Feedback = typing.Union[Feedback_Numerical, Feedback_Categorical, Feedback_Boolean]

opik/rest_api/types/feedback_create.py

@@ -6,6 +6,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .boolean_feedback_detail_create import BooleanFeedbackDetailCreate
 from .categorical_feedback_detail_create import CategoricalFeedbackDetailCreate
 from .numerical_feedback_detail_create import NumericalFeedbackDetailCreate
 
@@ -13,6 +14,10 @@ from .numerical_feedback_detail_create import NumericalFeedbackDetailCreate
 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
@@ -52,4 +57,18 @@ class FeedbackCreate_Categorical(Base):
             extra = pydantic.Extra.allow
 
 
-FeedbackCreate = typing.Union[FeedbackCreate_Numerical, FeedbackCreate_Categorical]
+class FeedbackCreate_Boolean(Base):
+    type: typing.Literal["boolean"] = "boolean"
+    details: typing.Optional[BooleanFeedbackDetailCreate] = 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
+
+
+FeedbackCreate = typing.Union[FeedbackCreate_Numerical, FeedbackCreate_Categorical, FeedbackCreate_Boolean]

opik/rest_api/types/feedback_object_public.py

@@ -9,6 +9,7 @@ import pydantic
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .boolean_feedback_detail_public import BooleanFeedbackDetailPublic
 from .categorical_feedback_detail_public import CategoricalFeedbackDetailPublic
 from .numerical_feedback_detail_public import NumericalFeedbackDetailPublic
 
@@ -16,6 +17,11 @@ from .numerical_feedback_detail_public import NumericalFeedbackDetailPublic
 class Base(UniversalBaseModel):
     id: typing.Optional[str] = None
     name: str
+    description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Optional description for the feedback definition
+    """
+
     created_at: typing_extensions.Annotated[typing.Optional[dt.datetime], FieldMetadata(alias="createdAt")] = None
     created_by: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="createdBy")] = None
     last_updated_at: typing_extensions.Annotated[typing.Optional[dt.datetime], FieldMetadata(alias="lastUpdatedAt")] = (
@@ -69,4 +75,24 @@ class FeedbackObjectPublic_Categorical(Base):
             extra = pydantic.Extra.allow
 
 
-FeedbackObjectPublic = typing.Union[FeedbackObjectPublic_Numerical, FeedbackObjectPublic_Categorical]
+class FeedbackObjectPublic_Boolean(Base):
+    type: typing.Literal["boolean"] = "boolean"
+    details: typing.Optional[BooleanFeedbackDetailPublic] = None
+    created_at: typing.Optional[dt.datetime] = None
+    created_by: typing.Optional[str] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
+    last_updated_by: 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
+
+
+FeedbackObjectPublic = typing.Union[
+    FeedbackObjectPublic_Numerical, FeedbackObjectPublic_Categorical, FeedbackObjectPublic_Boolean
+]

opik/rest_api/types/feedback_public.py

@@ -9,6 +9,7 @@ import pydantic
 import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from ..core.serialization import FieldMetadata
+from .boolean_feedback_detail_public import BooleanFeedbackDetailPublic
 from .categorical_feedback_detail_public import CategoricalFeedbackDetailPublic
 from .numerical_feedback_detail_public import NumericalFeedbackDetailPublic
 
@@ -16,6 +17,11 @@ from .numerical_feedback_detail_public import NumericalFeedbackDetailPublic
 class Base(UniversalBaseModel):
     id: typing.Optional[str] = None
     name: str
+    description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Optional description for the feedback definition
+    """
+
     created_at: typing_extensions.Annotated[typing.Optional[dt.datetime], FieldMetadata(alias="createdAt")] = None
     created_by: typing_extensions.Annotated[typing.Optional[str], FieldMetadata(alias="createdBy")] = None
     last_updated_at: typing_extensions.Annotated[typing.Optional[dt.datetime], FieldMetadata(alias="lastUpdatedAt")] = (
@@ -69,4 +75,22 @@ class FeedbackPublic_Categorical(Base):
             extra = pydantic.Extra.allow
 
 
-FeedbackPublic = typing.Union[FeedbackPublic_Numerical, FeedbackPublic_Categorical]
+class FeedbackPublic_Boolean(Base):
+    type: typing.Literal["boolean"] = "boolean"
+    details: typing.Optional[BooleanFeedbackDetailPublic] = None
+    created_at: typing.Optional[dt.datetime] = None
+    created_by: typing.Optional[str] = None
+    last_updated_at: typing.Optional[dt.datetime] = None
+    last_updated_by: 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
+
+
+FeedbackPublic = typing.Union[FeedbackPublic_Numerical, FeedbackPublic_Categorical, FeedbackPublic_Boolean]

opik/rest_api/types/feedback_score_batch_item.py

@@ -8,18 +8,19 @@ from .feedback_score_batch_item_source import FeedbackScoreBatchItemSource
 
 
 class FeedbackScoreBatchItem(UniversalBaseModel):
-    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: str
     category_name: typing.Optional[str] = None
     value: float
     reason: typing.Optional[str] = None
     source: FeedbackScoreBatchItemSource
     author: typing.Optional[str] = None
+    id: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/feedback_score_batch_item_thread.py

@@ -8,18 +8,19 @@ from .feedback_score_batch_item_thread_source import FeedbackScoreBatchItemThrea
 
 
 class FeedbackScoreBatchItemThread(UniversalBaseModel):
-    thread_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: str
     category_name: typing.Optional[str] = None
     value: float
     reason: typing.Optional[str] = None
     source: FeedbackScoreBatchItemThreadSource
     author: typing.Optional[str] = None
+    thread_id: str
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/feedback_score_public.py

@@ -10,6 +10,10 @@ from .value_entry_public import ValueEntryPublic
 
 
 class FeedbackScorePublic(UniversalBaseModel):
+    """
+    Aggregated feedback scores from all spans in this trace, averaged by score name
+    """
+
     name: str
     category_name: typing.Optional[str] = None
     value: float

opik/rest_api/types/feedback_update.py

@@ -6,6 +6,7 @@ import typing
 
 import pydantic
 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
 
@@ -13,6 +14,10 @@ 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
@@ -52,4 +57,18 @@ class FeedbackUpdate_Categorical(Base):
             extra = pydantic.Extra.allow
 
 
-FeedbackUpdate = typing.Union[FeedbackUpdate_Numerical, FeedbackUpdate_Categorical]
+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:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow
+
+
+FeedbackUpdate = typing.Union[FeedbackUpdate_Numerical, FeedbackUpdate_Categorical, FeedbackUpdate_Boolean]

opik/rest_api/types/group_content_with_aggregations.py

@@ -8,6 +8,7 @@ from .aggregation_data import AggregationData
 
 
 class GroupContentWithAggregations(UniversalBaseModel):
+    label: typing.Optional[str] = None
     aggregations: typing.Optional[AggregationData] = None
 
     if IS_PYDANTIC_V2:

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

@@ -18,6 +18,7 @@ class Guardrail(UniversalBaseModel):
     If null, the default project is used
     """
 
+    project_id: typing.Optional[str] = None
     name: GuardrailName
     result: GuardrailResult
     config: JsonNode

opik/rest_api/types/guardrail_write.py

@@ -17,6 +17,7 @@ class GuardrailWrite(UniversalBaseModel):
     If null, the default project is used
     """
 
+    project_id: typing.Optional[str] = None
     name: GuardrailWriteName
     result: GuardrailWriteResult
     config: JsonNode

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

@@ -4,12 +4,16 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .llm_as_judge_message_content import LlmAsJudgeMessageContent
 from .llm_as_judge_message_role import LlmAsJudgeMessageRole
 
 
 class LlmAsJudgeMessage(UniversalBaseModel):
     role: LlmAsJudgeMessageRole
-    content: str
+    content: typing.Optional[str] = None
+    content_array: typing.Optional[typing.List[LlmAsJudgeMessageContent]] = None
+    string_content: typing.Optional[bool] = None
+    structured_content: typing.Optional[bool] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/llm_as_judge_message_content.py

@@ -0,0 +1,26 @@
+# 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 .audio_url import AudioUrl
+from .image_url import ImageUrl
+from .video_url import VideoUrl
+
+
+class LlmAsJudgeMessageContent(UniversalBaseModel):
+    type: str
+    text: typing.Optional[str] = None
+    image_url: typing.Optional[ImageUrl] = None
+    video_url: typing.Optional[VideoUrl] = None
+    audio_url: typing.Optional[AudioUrl] = 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/llm_as_judge_message_content_public.py

@@ -0,0 +1,26 @@
+# 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 .audio_url_public import AudioUrlPublic
+from .image_url_public import ImageUrlPublic
+from .video_url_public import VideoUrlPublic
+
+
+class LlmAsJudgeMessageContentPublic(UniversalBaseModel):
+    type: str
+    text: typing.Optional[str] = None
+    image_url: typing.Optional[ImageUrlPublic] = None
+    video_url: typing.Optional[VideoUrlPublic] = None
+    audio_url: typing.Optional[AudioUrlPublic] = 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/llm_as_judge_message_content_write.py

@@ -0,0 +1,26 @@
+# 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 .audio_url_write import AudioUrlWrite
+from .image_url_write import ImageUrlWrite
+from .video_url_write import VideoUrlWrite
+
+
+class LlmAsJudgeMessageContentWrite(UniversalBaseModel):
+    type: str
+    text: typing.Optional[str] = None
+    image_url: typing.Optional[ImageUrlWrite] = None
+    video_url: typing.Optional[VideoUrlWrite] = None
+    audio_url: typing.Optional[AudioUrlWrite] = 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/llm_as_judge_message_public.py

@@ -4,12 +4,16 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .llm_as_judge_message_content_public import LlmAsJudgeMessageContentPublic
 from .llm_as_judge_message_public_role import LlmAsJudgeMessagePublicRole
 
 
 class LlmAsJudgeMessagePublic(UniversalBaseModel):
     role: LlmAsJudgeMessagePublicRole
-    content: str
+    content: typing.Optional[str] = None
+    content_array: typing.Optional[typing.List[LlmAsJudgeMessageContentPublic]] = None
+    string_content: typing.Optional[bool] = None
+    structured_content: typing.Optional[bool] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/llm_as_judge_message_write.py

@@ -4,12 +4,16 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .llm_as_judge_message_content_write import LlmAsJudgeMessageContentWrite
 from .llm_as_judge_message_write_role import LlmAsJudgeMessageWriteRole
 
 
 class LlmAsJudgeMessageWrite(UniversalBaseModel):
     role: LlmAsJudgeMessageWriteRole
-    content: str
+    content: typing.Optional[str] = None
+    content_array: typing.Optional[typing.List[LlmAsJudgeMessageContentWrite]] = None
+    string_content: typing.Optional[bool] = None
+    structured_content: typing.Optional[bool] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/llm_as_judge_model_parameters.py

@@ -4,11 +4,14 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node import JsonNode
 
 
 class LlmAsJudgeModelParameters(UniversalBaseModel):
     name: str
     temperature: float
+    seed: typing.Optional[int] = None
+    custom_parameters: typing.Optional[JsonNode] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2

opik/rest_api/types/llm_as_judge_model_parameters_public.py

@@ -4,11 +4,14 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .json_node_public import JsonNodePublic
 
 
 class LlmAsJudgeModelParametersPublic(UniversalBaseModel):
     name: str
     temperature: float
+    seed: typing.Optional[int] = None
+    custom_parameters: typing.Optional[JsonNodePublic] = None
 
     if IS_PYDANTIC_V2:
         model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(extra="allow", frozen=True)  # type: ignore # Pydantic v2