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

opik/rest_api/types/dashboard_public.py

@@ -0,0 +1,30 @@
+# 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 .json_node_public import JsonNodePublic
+
+
+class DashboardPublic(UniversalBaseModel):
+    id: typing.Optional[str] = None
+    workspace_id: typing.Optional[str] = None
+    name: str
+    slug: typing.Optional[str] = None
+    description: typing.Optional[str] = None
+    config: JsonNodePublic
+    created_by: typing.Optional[str] = None
+    last_updated_by: typing.Optional[str] = None
+    created_at: typing.Optional[dt.datetime] = None
+    last_updated_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
+    else:
+
+        class Config:
+            frozen = True
+            smart_union = True
+            extra = pydantic.Extra.allow

opik/rest_api/types/dataset.py

@@ -5,6 +5,8 @@ import typing
 
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from .dataset_status import DatasetStatus
+from .dataset_version_summary import DatasetVersionSummary
 from .dataset_visibility import DatasetVisibility
 
 
@@ -25,6 +27,8 @@ 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
+    latest_version: typing.Optional[DatasetVersionSummary] = 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_expansion.py

@@ -0,0 +1,42 @@
+# 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 DatasetExpansion(UniversalBaseModel):
+    model: str = pydantic.Field()
+    """
+    The model to use for synthetic data generation
+    """
+
+    sample_count: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Number of synthetic samples to generate
+    """
+
+    preserve_fields: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Fields to preserve patterns from original data
+    """
+
+    variation_instructions: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Additional instructions for data variation
+    """
+
+    custom_prompt: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Custom prompt to use for generation instead of auto-generated one
+    """
+
+    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_expansion_response.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 .dataset_item import DatasetItem
+
+
+class DatasetExpansionResponse(UniversalBaseModel):
+    generated_samples: typing.Optional[typing.List[DatasetItem]] = pydantic.Field(default=None)
+    """
+    List of generated synthetic dataset items
+    """
+
+    model: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Model used for generation
+    """
+
+    total_generated: typing.Optional[int] = pydantic.Field(default=None)
+    """
+    Total number of samples generated
+    """
+
+    generation_time: typing.Optional[dt.datetime] = pydantic.Field(default=None)
+    """
+    Generation timestamp
+    """
+
+    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_item.py

@@ -12,10 +12,12 @@ 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
     data: JsonNode
+    tags: typing.Optional[typing.List[str]] = None
     experiment_items: typing.Optional[typing.List[ExperimentItem]] = None
     dataset_id: typing.Optional[str] = None
     created_at: typing.Optional[dt.datetime] = None

opik/rest_api/types/dataset_item_changes_public.py

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

opik/rest_api/types/dataset_item_compare.py

@@ -12,10 +12,12 @@ 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
     data: JsonNode
+    tags: typing.Optional[typing.List[str]] = None
     experiment_items: typing.Optional[typing.List[ExperimentItemCompare]] = None
     dataset_id: typing.Optional[str] = None
     created_at: typing.Optional[dt.datetime] = None

opik/rest_api/types/dataset_item_filter.py

@@ -0,0 +1,27 @@
+# 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_item_filter_operator import DatasetItemFilterOperator
+
+
+class DatasetItemFilter(UniversalBaseModel):
+    """
+    Filters to select dataset items to delete within the specified dataset. Must be used with 'dataset_id'. Mutually exclusive with 'item_ids'. Empty array means 'delete all items in the dataset'.
+    """
+
+    field: typing.Optional[str] = None
+    operator: typing.Optional[DatasetItemFilterOperator] = 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/dataset_item_filter_operator.py

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

opik/rest_api/types/dataset_item_page_compare.py

@@ -3,7 +3,9 @@
 import typing
 
 import pydantic
+import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
 from .column_compare import ColumnCompare
 from .dataset_item_compare import DatasetItemCompare
 
@@ -14,6 +16,9 @@ class DatasetItemPageCompare(UniversalBaseModel):
     size: typing.Optional[int] = None
     total: typing.Optional[int] = None
     columns: typing.Optional[typing.List[ColumnCompare]] = 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

opik/rest_api/types/dataset_item_page_public.py

@@ -3,7 +3,9 @@
 import typing
 
 import pydantic
+import typing_extensions
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
+from ..core.serialization import FieldMetadata
 from .column_public import ColumnPublic
 from .dataset_item_public import DatasetItemPublic
 
@@ -14,6 +16,9 @@ class DatasetItemPagePublic(UniversalBaseModel):
     size: typing.Optional[int] = None
     total: typing.Optional[int] = None
     columns: typing.Optional[typing.List[ColumnPublic]] = 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

opik/rest_api/types/dataset_item_public.py

@@ -12,10 +12,12 @@ 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
     data: JsonNode
+    tags: typing.Optional[typing.List[str]] = None
     experiment_items: typing.Optional[typing.List[ExperimentItemPublic]] = None
     dataset_id: typing.Optional[str] = None
     created_at: typing.Optional[dt.datetime] = None

opik/rest_api/types/dataset_item_update.py

@@ -0,0 +1,39 @@
+# 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 .json_node import JsonNode
+
+
+class DatasetItemUpdate(UniversalBaseModel):
+    """
+    Dataset item update request
+    """
+
+    input: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Dataset item input
+    """
+
+    expected_output: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Dataset item expected output
+    """
+
+    metadata: typing.Optional[JsonNode] = None
+    data: typing.Optional[JsonNode] = None
+    tags: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Tags
+    """
+
+    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_item_write.py

@@ -14,6 +14,7 @@ class DatasetItemWrite(UniversalBaseModel):
     span_id: typing.Optional[str] = None
     source: DatasetItemWriteSource
     data: JsonNode
+    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

opik/rest_api/types/dataset_public.py

@@ -5,7 +5,9 @@ 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
+from .dataset_version_summary_public import DatasetVersionSummaryPublic
 
 
 class DatasetPublic(UniversalBaseModel):
@@ -25,6 +27,8 @@ 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
+    latest_version: typing.Optional[DatasetVersionSummaryPublic] = 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,59 @@
+# 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
+    is_latest: typing.Optional[bool] = pydantic.Field(default=None)
+    """
+    Indicates whether this is the latest version of the dataset
+    """
+
+    version_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Sequential version name formatted as 'v1', 'v2', etc.
+    """
+
+    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/dataset_version_summary.py

@@ -0,0 +1,46 @@
+# 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 DatasetVersionSummary(UniversalBaseModel):
+    """
+    Summary of the latest dataset version
+    """
+
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Unique identifier of the version
+    """
+
+    version_hash: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Hash of the version content
+    """
+
+    version_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Sequential version name formatted as 'v1', 'v2', etc.
+    """
+
+    change_description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Description of changes in this version
+    """
+
+    tags: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Tags associated with this version
+    """
+
+    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_summary_public.py

@@ -0,0 +1,46 @@
+# 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 DatasetVersionSummaryPublic(UniversalBaseModel):
+    """
+    Summary of the latest dataset version
+    """
+
+    id: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Unique identifier of the version
+    """
+
+    version_hash: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Hash of the version content
+    """
+
+    version_name: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Sequential version name formatted as 'v1', 'v2', etc.
+    """
+
+    change_description: typing.Optional[str] = pydantic.Field(default=None)
+    """
+    Description of changes in this version
+    """
+
+    tags: typing.Optional[typing.List[str]] = pydantic.Field(default=None)
+    """
+    Tags associated with this version
+    """
+
+    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,9 +6,11 @@ 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
-from .json_node import JsonNode
+from .json_list_string import JsonListString
 from .percentage_values import PercentageValues
 from .prompt_version_link import PromptVersionLink
 
@@ -17,8 +19,9 @@ class Experiment(UniversalBaseModel):
     id: typing.Optional[str] = None
     dataset_name: str
     dataset_id: typing.Optional[str] = None
+    project_id: typing.Optional[str] = None
     name: typing.Optional[str] = None
-    metadata: typing.Optional[JsonNode] = None
+    metadata: typing.Optional[JsonListString] = None
     type: typing.Optional[ExperimentType] = None
     optimization_id: typing.Optional[str] = None
     feedback_scores: typing.Optional[typing.List[FeedbackScoreAverage]] = None
@@ -32,6 +35,8 @@ class Experiment(UniversalBaseModel):
     last_updated_at: typing.Optional[dt.datetime] = None
     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
 

opik/rest_api/types/experiment_group_response.py

@@ -5,10 +5,12 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .group_content import GroupContent
+from .group_details import GroupDetails
 
 
 class ExperimentGroupResponse(UniversalBaseModel):
     content: typing.Optional[typing.Dict[str, GroupContent]] = None
+    details: typing.Optional[GroupDetails] = 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/experiment_public.py

@@ -6,9 +6,11 @@ import typing
 import pydantic
 from ..core.pydantic_utilities import IS_PYDANTIC_V2, UniversalBaseModel
 from .comment_public import CommentPublic
+from .experiment_public_status import ExperimentPublicStatus
 from .experiment_public_type import ExperimentPublicType
+from .experiment_score_public import ExperimentScorePublic
 from .feedback_score_average_public import FeedbackScoreAveragePublic
-from .json_node_public import JsonNodePublic
+from .json_list_string_public import JsonListStringPublic
 from .percentage_values_public import PercentageValuesPublic
 from .prompt_version_link_public import PromptVersionLinkPublic
 
@@ -17,8 +19,9 @@ class ExperimentPublic(UniversalBaseModel):
     id: typing.Optional[str] = None
     dataset_name: str
     dataset_id: typing.Optional[str] = None
+    project_id: typing.Optional[str] = None
     name: typing.Optional[str] = None
-    metadata: typing.Optional[JsonNodePublic] = None
+    metadata: typing.Optional[JsonListStringPublic] = None
     type: typing.Optional[ExperimentPublicType] = None
     optimization_id: typing.Optional[str] = None
     feedback_scores: typing.Optional[typing.List[FeedbackScoreAveragePublic]] = None
@@ -32,6 +35,8 @@ class ExperimentPublic(UniversalBaseModel):
     last_updated_at: typing.Optional[dt.datetime] = None
     created_by: typing.Optional[str] = None
     last_updated_by: typing.Optional[str] = None
+    status: typing.Optional[ExperimentPublicStatus] = None
+    experiment_scores: typing.Optional[typing.List[ExperimentScorePublic]] = None
     prompt_version: typing.Optional[PromptVersionLinkPublic] = None
     prompt_versions: typing.Optional[typing.List[PromptVersionLinkPublic]] = None
 

opik/rest_api/types/experiment_public_status.py

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