opik 1.9.26__py3-none-any.whl → 1.9.39__py3-none-any.whl

opik/rest_api/types/automation_rule_evaluator_span_llm_as_judge_public.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
+from .span_filter_public import SpanFilterPublic
+from .span_llm_as_judge_code_public import SpanLlmAsJudgeCodePublic
+
+
+class AutomationRuleEvaluatorSpanLlmAsJudgePublic(UniversalBaseModel):
+    filters: typing.Optional[typing.List[SpanFilterPublic]] = None
+    code: typing.Optional[SpanLlmAsJudgeCodePublic] = 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/automation_rule_evaluator_span_llm_as_judge_write.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
+from .span_filter_write import SpanFilterWrite
+from .span_llm_as_judge_code_write import SpanLlmAsJudgeCodeWrite
+
+
+class AutomationRuleEvaluatorSpanLlmAsJudgeWrite(UniversalBaseModel):
+    filters: typing.Optional[typing.List[SpanFilterWrite]] = None
+    code: typing.Optional[SpanLlmAsJudgeCodeWrite] = 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/automation_rule_evaluator_trace_thread_llm_as_judge.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter import TraceThreadFilter
 from .trace_thread_llm_as_judge_code import TraceThreadLlmAsJudgeCode
 
 
 class AutomationRuleEvaluatorTraceThreadLlmAsJudge(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilter]] = None
     code: typing.Optional[TraceThreadLlmAsJudgeCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_trace_thread_llm_as_judge_public.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter_public import TraceThreadFilterPublic
 from .trace_thread_llm_as_judge_code_public import TraceThreadLlmAsJudgeCodePublic
 
 
 class AutomationRuleEvaluatorTraceThreadLlmAsJudgePublic(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilterPublic]] = None
     code: typing.Optional[TraceThreadLlmAsJudgeCodePublic] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_trace_thread_llm_as_judge_write.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter_write import TraceThreadFilterWrite
 from .trace_thread_llm_as_judge_code_write import TraceThreadLlmAsJudgeCodeWrite
 
 
 class AutomationRuleEvaluatorTraceThreadLlmAsJudgeWrite(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilterWrite]] = None
     code: typing.Optional[TraceThreadLlmAsJudgeCodeWrite] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_trace_thread_user_defined_metric_python.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter import TraceThreadFilter
 from .trace_thread_user_defined_metric_python_code import TraceThreadUserDefinedMetricPythonCode
 
 
 class AutomationRuleEvaluatorTraceThreadUserDefinedMetricPython(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilter]] = None
     code: typing.Optional[TraceThreadUserDefinedMetricPythonCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_trace_thread_user_defined_metric_python_public.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter_public import TraceThreadFilterPublic
 from .trace_thread_user_defined_metric_python_code_public import TraceThreadUserDefinedMetricPythonCodePublic
 
 
 class AutomationRuleEvaluatorTraceThreadUserDefinedMetricPythonPublic(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilterPublic]] = None
     code: typing.Optional[TraceThreadUserDefinedMetricPythonCodePublic] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_trace_thread_user_defined_metric_python_write.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter_write import TraceThreadFilterWrite
 from .trace_thread_user_defined_metric_python_code_write import TraceThreadUserDefinedMetricPythonCodeWrite
 
 
 class AutomationRuleEvaluatorTraceThreadUserDefinedMetricPythonWrite(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilterWrite]] = None
     code: typing.Optional[TraceThreadUserDefinedMetricPythonCodeWrite] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_update.py

@@ -7,7 +7,10 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .llm_as_judge_code import LlmAsJudgeCode
+from .span_filter import SpanFilter
+from .span_llm_as_judge_code import SpanLlmAsJudgeCode
 from .trace_filter import TraceFilter
+from .trace_thread_filter import TraceThreadFilter
 from .trace_thread_llm_as_judge_code import TraceThreadLlmAsJudgeCode
 from .trace_thread_user_defined_metric_python_code import TraceThreadUserDefinedMetricPythonCode
 from .user_defined_metric_python_code import UserDefinedMetricPythonCode
@@ -17,7 +20,6 @@ class Base(UniversalBaseModel):
     name: str
     sampling_rate: typing.Optional[float] = None
     enabled: typing.Optional[bool] = None
-    filters: typing.Optional[typing.List[TraceFilter]] = None
     project_id: str
     action: typing.Literal["evaluator"] = "evaluator"
 
@@ -33,6 +35,7 @@ class Base(UniversalBaseModel):
 
 class AutomationRuleEvaluatorUpdate_LlmAsJudge(Base):
     type: typing.Literal["llm_as_judge"] = "llm_as_judge"
+    filters: typing.Optional[typing.List[TraceFilter]] = None
     code: typing.Optional[LlmAsJudgeCode] = None
 
     if IS_PYDANTIC_V2:
@@ -47,6 +50,7 @@ class AutomationRuleEvaluatorUpdate_LlmAsJudge(Base):
 
 class AutomationRuleEvaluatorUpdate_UserDefinedMetricPython(Base):
     type: typing.Literal["user_defined_metric_python"] = "user_defined_metric_python"
+    filters: typing.Optional[typing.List[TraceFilter]] = None
     code: typing.Optional[UserDefinedMetricPythonCode] = None
 
     if IS_PYDANTIC_V2:
@@ -61,6 +65,7 @@ class AutomationRuleEvaluatorUpdate_UserDefinedMetricPython(Base):
 
 class AutomationRuleEvaluatorUpdate_TraceThreadLlmAsJudge(Base):
     type: typing.Literal["trace_thread_llm_as_judge"] = "trace_thread_llm_as_judge"
+    filters: typing.Optional[typing.List[TraceThreadFilter]] = None
     code: typing.Optional[TraceThreadLlmAsJudgeCode] = None
 
     if IS_PYDANTIC_V2:
@@ -75,6 +80,7 @@ class AutomationRuleEvaluatorUpdate_TraceThreadLlmAsJudge(Base):
 
 class AutomationRuleEvaluatorUpdate_TraceThreadUserDefinedMetricPython(Base):
     type: typing.Literal["trace_thread_user_defined_metric_python"] = "trace_thread_user_defined_metric_python"
+    filters: typing.Optional[typing.List[TraceThreadFilter]] = None
     code: typing.Optional[TraceThreadUserDefinedMetricPythonCode] = None
 
     if IS_PYDANTIC_V2:
@@ -87,9 +93,25 @@ class AutomationRuleEvaluatorUpdate_TraceThreadUserDefinedMetricPython(Base):
             extra = pydantic.Extra.allow
 
 
+class AutomationRuleEvaluatorUpdate_SpanLlmAsJudge(Base):
+    type: typing.Literal["span_llm_as_judge"] = "span_llm_as_judge"
+    filters: typing.Optional[typing.List[SpanFilter]] = None
+    code: typing.Optional[SpanLlmAsJudgeCode] = 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
+
+
 AutomationRuleEvaluatorUpdate = typing.Union[
     AutomationRuleEvaluatorUpdate_LlmAsJudge,
     AutomationRuleEvaluatorUpdate_UserDefinedMetricPython,
     AutomationRuleEvaluatorUpdate_TraceThreadLlmAsJudge,
     AutomationRuleEvaluatorUpdate_TraceThreadUserDefinedMetricPython,
+    AutomationRuleEvaluatorUpdate_SpanLlmAsJudge,
 ]

opik/rest_api/types/automation_rule_evaluator_update_llm_as_judge.py

@@ -5,9 +5,11 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .llm_as_judge_code import LlmAsJudgeCode
+from .trace_filter import TraceFilter
 
 
 class AutomationRuleEvaluatorUpdateLlmAsJudge(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceFilter]] = None
     code: typing.Optional[LlmAsJudgeCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_update_span_llm_as_judge.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
+from .span_filter import SpanFilter
+from .span_llm_as_judge_code import SpanLlmAsJudgeCode
+
+
+class AutomationRuleEvaluatorUpdateSpanLlmAsJudge(UniversalBaseModel):
+    filters: typing.Optional[typing.List[SpanFilter]] = None
+    code: typing.Optional[SpanLlmAsJudgeCode] = 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/automation_rule_evaluator_update_trace_thread_llm_as_judge.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter import TraceThreadFilter
 from .trace_thread_llm_as_judge_code import TraceThreadLlmAsJudgeCode
 
 
 class AutomationRuleEvaluatorUpdateTraceThreadLlmAsJudge(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilter]] = None
     code: typing.Optional[TraceThreadLlmAsJudgeCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_update_trace_thread_user_defined_metric_python.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_thread_filter import TraceThreadFilter
 from .trace_thread_user_defined_metric_python_code import TraceThreadUserDefinedMetricPythonCode
 
 
 class AutomationRuleEvaluatorUpdateTraceThreadUserDefinedMetricPython(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceThreadFilter]] = None
     code: typing.Optional[TraceThreadUserDefinedMetricPythonCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_update_user_defined_metric_python.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter import TraceFilter
 from .user_defined_metric_python_code import UserDefinedMetricPythonCode
 
 
 class AutomationRuleEvaluatorUpdateUserDefinedMetricPython(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceFilter]] = None
     code: typing.Optional[UserDefinedMetricPythonCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_user_defined_metric_python.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter import TraceFilter
 from .user_defined_metric_python_code import UserDefinedMetricPythonCode
 
 
 class AutomationRuleEvaluatorUserDefinedMetricPython(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceFilter]] = None
     code: typing.Optional[UserDefinedMetricPythonCode] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_user_defined_metric_python_public.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter_public import TraceFilterPublic
 from .user_defined_metric_python_code_public import UserDefinedMetricPythonCodePublic
 
 
 class AutomationRuleEvaluatorUserDefinedMetricPythonPublic(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceFilterPublic]] = None
     code: typing.Optional[UserDefinedMetricPythonCodePublic] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_user_defined_metric_python_write.py

@@ -4,10 +4,12 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .trace_filter_write import TraceFilterWrite
 from .user_defined_metric_python_code_write import UserDefinedMetricPythonCodeWrite
 
 
 class AutomationRuleEvaluatorUserDefinedMetricPythonWrite(UniversalBaseModel):
+    filters: typing.Optional[typing.List[TraceFilterWrite]] = None
     code: typing.Optional[UserDefinedMetricPythonCodeWrite] = None
 
     if IS_PYDANTIC_V2:

opik/rest_api/types/automation_rule_evaluator_write.py

@@ -7,7 +7,10 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .llm_as_judge_code_write import LlmAsJudgeCodeWrite
+from .span_filter_write import SpanFilterWrite
+from .span_llm_as_judge_code_write import SpanLlmAsJudgeCodeWrite
 from .trace_filter_write import TraceFilterWrite
+from .trace_thread_filter_write import TraceThreadFilterWrite
 from .trace_thread_llm_as_judge_code_write import TraceThreadLlmAsJudgeCodeWrite
 from .trace_thread_user_defined_metric_python_code_write import TraceThreadUserDefinedMetricPythonCodeWrite
 from .user_defined_metric_python_code_write import UserDefinedMetricPythonCodeWrite
@@ -18,7 +21,6 @@ class Base(UniversalBaseModel):
     name: str
     sampling_rate: typing.Optional[float] = None
     enabled: typing.Optional[bool] = None
-    filters: typing.Optional[typing.List[TraceFilterWrite]] = None
     action: typing.Literal["evaluator"] = "evaluator"
 
     if IS_PYDANTIC_V2:
@@ -33,6 +35,7 @@ class Base(UniversalBaseModel):
 
 class AutomationRuleEvaluatorWrite_LlmAsJudge(Base):
     type: typing.Literal["llm_as_judge"] = "llm_as_judge"
+    filters: typing.Optional[typing.List[TraceFilterWrite]] = None
     code: typing.Optional[LlmAsJudgeCodeWrite] = None
 
     if IS_PYDANTIC_V2:
@@ -47,6 +50,7 @@ class AutomationRuleEvaluatorWrite_LlmAsJudge(Base):
 
 class AutomationRuleEvaluatorWrite_UserDefinedMetricPython(Base):
     type: typing.Literal["user_defined_metric_python"] = "user_defined_metric_python"
+    filters: typing.Optional[typing.List[TraceFilterWrite]] = None
     code: typing.Optional[UserDefinedMetricPythonCodeWrite] = None
 
     if IS_PYDANTIC_V2:
@@ -61,6 +65,7 @@ class AutomationRuleEvaluatorWrite_UserDefinedMetricPython(Base):
 
 class AutomationRuleEvaluatorWrite_TraceThreadLlmAsJudge(Base):
     type: typing.Literal["trace_thread_llm_as_judge"] = "trace_thread_llm_as_judge"
+    filters: typing.Optional[typing.List[TraceThreadFilterWrite]] = None
     code: typing.Optional[TraceThreadLlmAsJudgeCodeWrite] = None
 
     if IS_PYDANTIC_V2:
@@ -75,6 +80,7 @@ class AutomationRuleEvaluatorWrite_TraceThreadLlmAsJudge(Base):
 
 class AutomationRuleEvaluatorWrite_TraceThreadUserDefinedMetricPython(Base):
     type: typing.Literal["trace_thread_user_defined_metric_python"] = "trace_thread_user_defined_metric_python"
+    filters: typing.Optional[typing.List[TraceThreadFilterWrite]] = None
     code: typing.Optional[TraceThreadUserDefinedMetricPythonCodeWrite] = None
 
     if IS_PYDANTIC_V2:
@@ -87,9 +93,25 @@ class AutomationRuleEvaluatorWrite_TraceThreadUserDefinedMetricPython(Base):
             extra = pydantic.Extra.allow
 
 
+class AutomationRuleEvaluatorWrite_SpanLlmAsJudge(Base):
+    type: typing.Literal["span_llm_as_judge"] = "span_llm_as_judge"
+    filters: typing.Optional[typing.List[SpanFilterWrite]] = None
+    code: typing.Optional[SpanLlmAsJudgeCodeWrite] = 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
+
+
 AutomationRuleEvaluatorWrite = typing.Union[
     AutomationRuleEvaluatorWrite_LlmAsJudge,
     AutomationRuleEvaluatorWrite_UserDefinedMetricPython,
     AutomationRuleEvaluatorWrite_TraceThreadLlmAsJudge,
     AutomationRuleEvaluatorWrite_TraceThreadUserDefinedMetricPython,
+    AutomationRuleEvaluatorWrite_SpanLlmAsJudge,
 ]

opik/rest_api/types/dashboard_page_public.py

@@ -12,6 +12,7 @@ class DashboardPagePublic(UniversalBaseModel):
     page: typing.Optional[int] = None
     size: typing.Optional[int] = None
     total: typing.Optional[int] = None
+    sortable_by: 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

opik/rest_api/types/dataset.py

@@ -5,6 +5,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .dataset_status import DatasetStatus
 from .dataset_visibility import DatasetVisibility
 
 
@@ -25,6 +26,7 @@ class Dataset(UniversalBaseModel):
     last_created_experiment_at: typing.Optional[dt.datetime] = None
     most_recent_optimization_at: typing.Optional[dt.datetime] = None
     last_created_optimization_at: typing.Optional[dt.datetime] = None
+    status: typing.Optional[DatasetStatus] = 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/dataset_item.py

@@ -12,6 +12,7 @@ from .json_node import JsonNode
 
 class DatasetItem(UniversalBaseModel):
     id: typing.Optional[str] = None
+    draft_item_id: typing.Optional[str] = None
     trace_id: typing.Optional[str] = None
     span_id: typing.Optional[str] = None
     source: DatasetItemSource

opik/rest_api/types/dataset_item_compare.py

@@ -12,6 +12,7 @@ from .json_node import JsonNode
 
 class DatasetItemCompare(UniversalBaseModel):
     id: typing.Optional[str] = None
+    draft_item_id: typing.Optional[str] = None
     trace_id: typing.Optional[str] = None
     span_id: typing.Optional[str] = None
     source: DatasetItemCompareSource

opik/rest_api/types/dataset_item_page_compare.py

@@ -19,6 +19,7 @@ class DatasetItemPageCompare(UniversalBaseModel):
     sortable_by: typing_extensions.Annotated[typing.Optional[typing.List[str]], FieldMetadata(alias="sortableBy")] = (
         None
     )
+    has_draft: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="hasDraft")] = 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/dataset_item_page_public.py

@@ -19,6 +19,7 @@ class DatasetItemPagePublic(UniversalBaseModel):
     sortable_by: typing_extensions.Annotated[typing.Optional[typing.List[str]], FieldMetadata(alias="sortableBy")] = (
         None
     )
+    has_draft: typing_extensions.Annotated[typing.Optional[bool], FieldMetadata(alias="hasDraft")] = 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/dataset_item_public.py

@@ -12,6 +12,7 @@ from .json_node import JsonNode
 
 class DatasetItemPublic(UniversalBaseModel):
     id: typing.Optional[str] = None
+    draft_item_id: typing.Optional[str] = None
     trace_id: typing.Optional[str] = None
     span_id: typing.Optional[str] = None
     source: DatasetItemPublicSource

opik/rest_api/types/dataset_public.py

@@ -5,6 +5,7 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .dataset_public_status import DatasetPublicStatus
 from .dataset_public_visibility import DatasetPublicVisibility
 
 
@@ -25,6 +26,7 @@ class DatasetPublic(UniversalBaseModel):
     last_created_experiment_at: typing.Optional[dt.datetime] = None
     most_recent_optimization_at: typing.Optional[dt.datetime] = None
     last_created_optimization_at: typing.Optional[dt.datetime] = None
+    status: typing.Optional[DatasetPublicStatus] = 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/dataset_public_status.py

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

opik/rest_api/types/dataset_status.py

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

opik/rest_api/types/dataset_version_diff.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
+from .dataset_version_diff_stats import DatasetVersionDiffStats
+
+
+class DatasetVersionDiff(UniversalBaseModel):
+    from_version: typing.Optional[str] = None
+    to_version: typing.Optional[str] = None
+    statistics: typing.Optional[DatasetVersionDiffStats] = 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/dataset_version_diff_stats.py

@@ -0,0 +1,24 @@
+# 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
+
+
+class DatasetVersionDiffStats(UniversalBaseModel):
+    items_added: typing_extensions.Annotated[typing.Optional[int], FieldMetadata(alias="itemsAdded")] = None
+    items_modified: typing_extensions.Annotated[typing.Optional[int], FieldMetadata(alias="itemsModified")] = None
+    items_deleted: typing_extensions.Annotated[typing.Optional[int], FieldMetadata(alias="itemsDeleted")] = None
+    items_unchanged: typing_extensions.Annotated[typing.Optional[int], FieldMetadata(alias="itemsUnchanged")] = 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/dataset_version_page_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 .dataset_version_public import DatasetVersionPublic
+
+
+class DatasetVersionPagePublic(UniversalBaseModel):
+    content: typing.Optional[typing.List[DatasetVersionPublic]] = None
+    page: typing.Optional[int] = None
+    size: typing.Optional[int] = None
+    total: typing.Optional[int] = 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/dataset_version_public.py

@@ -0,0 +1,49 @@
+# 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
+
+
+class DatasetVersionPublic(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    dataset_id: typing.Optional[str] = None
+    version_hash: typing.Optional[str] = None
+    tags: typing.Optional[typing.List[str]] = None
+    items_total: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Total number of items in this version
+    """
+
+    items_added: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of items added since last version
+    """
+
+    items_modified: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of items modified since last version
+    """
+
+    items_deleted: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of items deleted since last version
+    """
+
+    change_description: typing.Optional[str] = None
+    metadata: typing.Optional[typing.Dict[str, str]] = 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

opik/rest_api/types/experiment.py

@@ -6,6 +6,7 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .comment import Comment
+from .experiment_score import ExperimentScore
 from .experiment_status import ExperimentStatus
 from .experiment_type import ExperimentType
 from .feedback_score_average import FeedbackScoreAverage
@@ -34,6 +35,7 @@ class Experiment(UniversalBaseModel):
     created_by: typing.Optional[str] = None
     last_updated_by: typing.Optional[str] = None
     status: typing.Optional[ExperimentStatus] = None
+    experiment_scores: typing.Optional[typing.List[ExperimentScore]] = None
     prompt_version: typing.Optional[PromptVersionLink] = None
     prompt_versions: typing.Optional[typing.List[PromptVersionLink]] = None