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

opik/rest_api/datasets/client.py

@@ -4,13 +4,25 @@ import typing
 
 from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
+from ..types.dataset_expansion_response import DatasetExpansionResponse
+from ..types.dataset_item_changes_public import DatasetItemChangesPublic
+from ..types.dataset_item_filter import DatasetItemFilter
 from ..types.dataset_item_page_compare import DatasetItemPageCompare
 from ..types.dataset_item_page_public import DatasetItemPagePublic
 from ..types.dataset_item_public import DatasetItemPublic
+from ..types.dataset_item_update import DatasetItemUpdate
 from ..types.dataset_item_write import DatasetItemWrite
+from ..types.dataset_item_write_source import DatasetItemWriteSource
 from ..types.dataset_page_public import DatasetPagePublic
 from ..types.dataset_public import DatasetPublic
+from ..types.dataset_version_diff import DatasetVersionDiff
+from ..types.dataset_version_page_public import DatasetVersionPagePublic
+from ..types.dataset_version_public import DatasetVersionPublic
+from ..types.json_node import JsonNode
 from ..types.page_columns import PageColumns
+from ..types.project_stats_public import ProjectStatsPublic
+from ..types.span_enrichment_options import SpanEnrichmentOptions
+from ..types.trace_enrichment_options import TraceEnrichmentOptions
 from .raw_client import AsyncRawDatasetsClient, RawDatasetsClient
 from .types.dataset_update_visibility import DatasetUpdateVisibility
 from .types.dataset_write_visibility import DatasetWriteVisibility
@@ -34,6 +46,104 @@ class DatasetsClient:
         """
         return self._raw_client
 
+    def apply_dataset_item_changes(
+        self,
+        id: str,
+        *,
+        request: DatasetItemChangesPublic,
+        override: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetVersionPublic:
+        """
+        Apply delta changes (add, edit, delete) to a dataset version with conflict detection.
+
+        This endpoint:
+        - Creates a new version with the applied changes
+        - Validates that baseVersion matches the latest version (unless override=true)
+        - Returns 409 Conflict if baseVersion is stale and override is not set
+
+        Use `override=true` query parameter to force version creation even with stale baseVersion.
+
+        Parameters
+        ----------
+        id : str
+
+        request : DatasetItemChangesPublic
+
+        override : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPublic
+            Version created successfully
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.apply_dataset_item_changes(id='id', request={'key': 'value'
+        }, )
+        """
+        _response = self._raw_client.apply_dataset_item_changes(
+            id, request=request, override=override, request_options=request_options
+        )
+        return _response.data
+
+    def batch_update_dataset_items(
+        self,
+        *,
+        update: DatasetItemUpdate,
+        ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        filters: typing.Optional[typing.Sequence[DatasetItemFilter]] = OMIT,
+        dataset_id: typing.Optional[str] = OMIT,
+        merge_tags: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update multiple dataset items
+
+        Parameters
+        ----------
+        update : DatasetItemUpdate
+
+        ids : typing.Optional[typing.Sequence[str]]
+            List of dataset item IDs to update (max 1000). Mutually exclusive with 'filters'.
+
+        filters : typing.Optional[typing.Sequence[DatasetItemFilter]]
+
+        dataset_id : typing.Optional[str]
+            Dataset ID. Required when using 'filters', optional when using 'ids'.
+
+        merge_tags : typing.Optional[bool]
+            If true, merge tags with existing tags instead of replacing them. Default: false. When using 'filters', this is automatically set to true.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        from Opik import DatasetItemUpdate
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.batch_update_dataset_items(update=DatasetItemUpdate(), )
+        """
+        _response = self._raw_client.batch_update_dataset_items(
+            update=update,
+            ids=ids,
+            filters=filters,
+            dataset_id=dataset_id,
+            merge_tags=merge_tags,
+            request_options=request_options,
+        )
+        return _response.data
+
     def find_datasets(
         self,
         *,
@@ -179,6 +289,119 @@ class DatasetsClient:
         )
         return _response.data
 
+    def create_dataset_items_from_csv(
+        self,
+        *,
+        file: typing.Dict[str, typing.Optional[typing.Any]],
+        dataset_id: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Create dataset items from uploaded CSV file. CSV should have headers in the first row. Processing happens asynchronously in batches.
+
+        Parameters
+        ----------
+        file : typing.Dict[str, typing.Optional[typing.Any]]
+
+        dataset_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.create_dataset_items_from_csv(file={'key': 'value'
+        }, dataset_id='dataset_id', )
+        """
+        _response = self._raw_client.create_dataset_items_from_csv(
+            file=file, dataset_id=dataset_id, request_options=request_options
+        )
+        return _response.data
+
+    def create_dataset_items_from_spans(
+        self,
+        dataset_id: str,
+        *,
+        span_ids: typing.Sequence[str],
+        enrichment_options: SpanEnrichmentOptions,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Create dataset items from spans with enriched metadata
+
+        Parameters
+        ----------
+        dataset_id : str
+
+        span_ids : typing.Sequence[str]
+            Set of span IDs to add to the dataset
+
+        enrichment_options : SpanEnrichmentOptions
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        from Opik import SpanEnrichmentOptions
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.create_dataset_items_from_spans(dataset_id='dataset_id', span_ids=['span_ids'], enrichment_options=SpanEnrichmentOptions(), )
+        """
+        _response = self._raw_client.create_dataset_items_from_spans(
+            dataset_id, span_ids=span_ids, enrichment_options=enrichment_options, request_options=request_options
+        )
+        return _response.data
+
+    def create_dataset_items_from_traces(
+        self,
+        dataset_id: str,
+        *,
+        trace_ids: typing.Sequence[str],
+        enrichment_options: TraceEnrichmentOptions,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Create dataset items from traces with enriched metadata
+
+        Parameters
+        ----------
+        dataset_id : str
+
+        trace_ids : typing.Sequence[str]
+            Set of trace IDs to add to the dataset
+
+        enrichment_options : TraceEnrichmentOptions
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        from Opik import TraceEnrichmentOptions
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.create_dataset_items_from_traces(dataset_id='dataset_id', trace_ids=['trace_ids'], enrichment_options=TraceEnrichmentOptions(), )
+        """
+        _response = self._raw_client.create_dataset_items_from_traces(
+            dataset_id, trace_ids=trace_ids, enrichment_options=enrichment_options, request_options=request_options
+        )
+        return _response.data
+
     def get_dataset_by_id(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> DatasetPublic:
         """
         Get dataset by id
@@ -298,14 +521,30 @@ class DatasetsClient:
         return _response.data
 
     def delete_dataset_items(
-        self, *, item_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        item_ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        dataset_id: typing.Optional[str] = OMIT,
+        filters: typing.Optional[typing.Sequence[DatasetItemFilter]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Delete dataset items
+        Delete dataset items using one of two modes:
+        1. **Delete by IDs**: Provide 'item_ids' to delete specific items by their IDs
+        2. **Delete by filters**: Provide 'dataset_id' with optional 'filters' to delete items matching criteria
+
+        When using filters, an empty 'filters' array will delete all items in the specified dataset.
 
         Parameters
         ----------
-        item_ids : typing.Sequence[str]
+        item_ids : typing.Optional[typing.Sequence[str]]
+            List of dataset item IDs to delete (max 1000). Use this to delete specific items by their IDs. Mutually exclusive with 'dataset_id' and 'filters'.
+
+        dataset_id : typing.Optional[str]
+            Dataset ID to scope the deletion. Required when using 'filters'. Mutually exclusive with 'item_ids'.
+
+        filters : typing.Optional[typing.Sequence[DatasetItemFilter]]
+            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'.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -318,9 +557,11 @@ class DatasetsClient:
         --------
         from Opik import OpikApi
         client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
-        client.datasets.delete_dataset_items(item_ids=['item_ids'], )
+        client.datasets.delete_dataset_items()
         """
-        _response = self._raw_client.delete_dataset_items(item_ids=item_ids, request_options=request_options)
+        _response = self._raw_client.delete_dataset_items(
+            item_ids=item_ids, dataset_id=dataset_id, filters=filters, request_options=request_options
+        )
         return _response.data
 
     def delete_datasets_batch(
@@ -349,6 +590,64 @@ class DatasetsClient:
         _response = self._raw_client.delete_datasets_batch(ids=ids, request_options=request_options)
         return _response.data
 
+    def expand_dataset(
+        self,
+        id: str,
+        *,
+        model: str,
+        sample_count: typing.Optional[int] = OMIT,
+        preserve_fields: typing.Optional[typing.Sequence[str]] = OMIT,
+        variation_instructions: typing.Optional[str] = OMIT,
+        custom_prompt: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetExpansionResponse:
+        """
+        Generate synthetic dataset samples using LLM based on existing data patterns
+
+        Parameters
+        ----------
+        id : str
+
+        model : str
+            The model to use for synthetic data generation
+
+        sample_count : typing.Optional[int]
+            Number of synthetic samples to generate
+
+        preserve_fields : typing.Optional[typing.Sequence[str]]
+            Fields to preserve patterns from original data
+
+        variation_instructions : typing.Optional[str]
+            Additional instructions for data variation
+
+        custom_prompt : typing.Optional[str]
+            Custom prompt to use for generation instead of auto-generated one
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetExpansionResponse
+            Generated synthetic samples
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.expand_dataset(id='id', model='gpt-4', )
+        """
+        _response = self._raw_client.expand_dataset(
+            id,
+            model=model,
+            sample_count=sample_count,
+            preserve_fields=preserve_fields,
+            variation_instructions=variation_instructions,
+            custom_prompt=custom_prompt,
+            request_options=request_options,
+        )
+        return _response.data
+
     def find_dataset_items_with_experiment_items(
         self,
         id: str,
@@ -357,6 +656,8 @@ class DatasetsClient:
         page: typing.Optional[int] = None,
         size: typing.Optional[int] = None,
         filters: typing.Optional[str] = None,
+        sorting: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
         truncate: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DatasetItemPageCompare:
@@ -375,6 +676,10 @@ class DatasetsClient:
 
         filters : typing.Optional[str]
 
+        sorting : typing.Optional[str]
+
+        search : typing.Optional[str]
+
         truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
@@ -397,6 +702,8 @@ class DatasetsClient:
             page=page,
             size=size,
             filters=filters,
+            sorting=sorting,
+            search=search,
             truncate=truncate,
             request_options=request_options,
         )
@@ -431,6 +738,44 @@ class DatasetsClient:
         )
         return _response.data
 
+    def get_dataset_experiment_items_stats(
+        self,
+        id: str,
+        *,
+        experiment_ids: str,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectStatsPublic:
+        """
+        Get experiment items stats for dataset
+
+        Parameters
+        ----------
+        id : str
+
+        experiment_ids : str
+
+        filters : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectStatsPublic
+            Experiment items stats resource
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.get_dataset_experiment_items_stats(id='id', experiment_ids='experiment_ids', )
+        """
+        _response = self._raw_client.get_dataset_experiment_items_stats(
+            id, experiment_ids=experiment_ids, filters=filters, request_options=request_options
+        )
+        return _response.data
+
     def get_dataset_item_by_id(
         self, item_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> DatasetItemPublic:
@@ -458,12 +803,71 @@ class DatasetsClient:
         _response = self._raw_client.get_dataset_item_by_id(item_id, request_options=request_options)
         return _response.data
 
+    def patch_dataset_item(
+        self,
+        item_id: str,
+        *,
+        source: DatasetItemWriteSource,
+        data: JsonNode,
+        id: typing.Optional[str] = OMIT,
+        trace_id: typing.Optional[str] = OMIT,
+        span_id: typing.Optional[str] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Partially update dataset item by id. Only provided fields will be updated.
+
+        Parameters
+        ----------
+        item_id : str
+
+        source : DatasetItemWriteSource
+
+        data : JsonNode
+
+        id : typing.Optional[str]
+
+        trace_id : typing.Optional[str]
+
+        span_id : typing.Optional[str]
+
+        tags : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.patch_dataset_item(item_id='itemId', source="manual", data={'key': 'value'
+        }, )
+        """
+        _response = self._raw_client.patch_dataset_item(
+            item_id,
+            source=source,
+            data=data,
+            id=id,
+            trace_id=trace_id,
+            span_id=span_id,
+            tags=tags,
+            request_options=request_options,
+        )
+        return _response.data
+
     def get_dataset_items(
         self,
         id: str,
         *,
         page: typing.Optional[int] = None,
         size: typing.Optional[int] = None,
+        version: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
         truncate: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DatasetItemPagePublic:
@@ -478,6 +882,10 @@ class DatasetsClient:
 
         size : typing.Optional[int]
 
+        version : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
         truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
@@ -495,12 +903,18 @@ class DatasetsClient:
         client.datasets.get_dataset_items(id='id', )
         """
         _response = self._raw_client.get_dataset_items(
-            id, page=page, size=size, truncate=truncate, request_options=request_options
-        )
-        return _response.data
-
-    def get_dataset_items_output_columns(
-        self,
+            id,
+            page=page,
+            size=size,
+            version=version,
+            filters=filters,
+            truncate=truncate,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def get_dataset_items_output_columns(
+        self,
         id: str,
         *,
         experiment_ids: typing.Optional[str] = None,
@@ -569,6 +983,208 @@ class DatasetsClient:
         ) as r:
             yield from r.data
 
+    def compare_dataset_versions(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetVersionDiff:
+        """
+        Compare the latest committed dataset version with the current draft state. This endpoint provides insights into changes made since the last version was committed. The comparison calculates additions, modifications, deletions, and unchanged items between the latest version snapshot and current draft.
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionDiff
+            Diff computed successfully
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.compare_dataset_versions(id='id', )
+        """
+        _response = self._raw_client.compare_dataset_versions(id, request_options=request_options)
+        return _response.data
+
+    def create_version_tag(
+        self, version_hash: str, id: str, *, tag: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Add a tag to a specific dataset version for easy reference (e.g., 'baseline', 'v1.0', 'production')
+
+        Parameters
+        ----------
+        version_hash : str
+
+        id : str
+
+        tag : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.create_version_tag(version_hash='versionHash', id='id', tag='tag', )
+        """
+        _response = self._raw_client.create_version_tag(version_hash, id, tag=tag, request_options=request_options)
+        return _response.data
+
+    def delete_version_tag(
+        self, version_hash: str, tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Remove a tag from a dataset version. The version itself is not deleted, only the tag reference.
+
+        Parameters
+        ----------
+        version_hash : str
+
+        tag : str
+
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.delete_version_tag(version_hash='versionHash', tag='tag', id='id', )
+        """
+        _response = self._raw_client.delete_version_tag(version_hash, tag, id, request_options=request_options)
+        return _response.data
+
+    def list_dataset_versions(
+        self,
+        id: str,
+        *,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetVersionPagePublic:
+        """
+        Get paginated list of versions for a dataset, ordered by creation time (newest first)
+
+        Parameters
+        ----------
+        id : str
+
+        page : typing.Optional[int]
+
+        size : typing.Optional[int]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPagePublic
+            Dataset versions
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.list_dataset_versions(id='id', )
+        """
+        _response = self._raw_client.list_dataset_versions(id, page=page, size=size, request_options=request_options)
+        return _response.data
+
+    def restore_dataset_version(
+        self, id: str, *, version_ref: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetVersionPublic:
+        """
+        Restores the dataset to a previous version state by creating a new version with items copied from the specified version. If the version is already the latest, returns it as-is (no-op).
+
+        Parameters
+        ----------
+        id : str
+
+        version_ref : str
+            Version hash or tag to restore from
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPublic
+            Version restored successfully
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.restore_dataset_version(id='id', version_ref='version_ref', )
+        """
+        _response = self._raw_client.restore_dataset_version(
+            id, version_ref=version_ref, request_options=request_options
+        )
+        return _response.data
+
+    def update_dataset_version(
+        self,
+        version_hash: str,
+        id: str,
+        *,
+        change_description: typing.Optional[str] = OMIT,
+        tags_to_add: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetVersionPublic:
+        """
+        Update a dataset version's change_description and/or add new tags
+
+        Parameters
+        ----------
+        version_hash : str
+
+        id : str
+
+        change_description : typing.Optional[str]
+            Optional description of changes in this version
+
+        tags_to_add : typing.Optional[typing.Sequence[str]]
+            Optional list of tags to add to this version
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPublic
+            Version updated successfully
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.datasets.update_dataset_version(version_hash='versionHash', id='id', )
+        """
+        _response = self._raw_client.update_dataset_version(
+            version_hash,
+            id,
+            change_description=change_description,
+            tags_to_add=tags_to_add,
+            request_options=request_options,
+        )
+        return _response.data
+
 
 class AsyncDatasetsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -585,6 +1201,110 @@ class AsyncDatasetsClient:
         """
         return self._raw_client
 
+    async def apply_dataset_item_changes(
+        self,
+        id: str,
+        *,
+        request: DatasetItemChangesPublic,
+        override: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetVersionPublic:
+        """
+        Apply delta changes (add, edit, delete) to a dataset version with conflict detection.
+
+        This endpoint:
+        - Creates a new version with the applied changes
+        - Validates that baseVersion matches the latest version (unless override=true)
+        - Returns 409 Conflict if baseVersion is stale and override is not set
+
+        Use `override=true` query parameter to force version creation even with stale baseVersion.
+
+        Parameters
+        ----------
+        id : str
+
+        request : DatasetItemChangesPublic
+
+        override : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPublic
+            Version created successfully
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.apply_dataset_item_changes(id='id', request={'key': 'value'
+            }, )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.apply_dataset_item_changes(
+            id, request=request, override=override, request_options=request_options
+        )
+        return _response.data
+
+    async def batch_update_dataset_items(
+        self,
+        *,
+        update: DatasetItemUpdate,
+        ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        filters: typing.Optional[typing.Sequence[DatasetItemFilter]] = OMIT,
+        dataset_id: typing.Optional[str] = OMIT,
+        merge_tags: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update multiple dataset items
+
+        Parameters
+        ----------
+        update : DatasetItemUpdate
+
+        ids : typing.Optional[typing.Sequence[str]]
+            List of dataset item IDs to update (max 1000). Mutually exclusive with 'filters'.
+
+        filters : typing.Optional[typing.Sequence[DatasetItemFilter]]
+
+        dataset_id : typing.Optional[str]
+            Dataset ID. Required when using 'filters', optional when using 'ids'.
+
+        merge_tags : typing.Optional[bool]
+            If true, merge tags with existing tags instead of replacing them. Default: false. When using 'filters', this is automatically set to true.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        from Opik import DatasetItemUpdate
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.batch_update_dataset_items(update=DatasetItemUpdate(), )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.batch_update_dataset_items(
+            update=update,
+            ids=ids,
+            filters=filters,
+            dataset_id=dataset_id,
+            merge_tags=merge_tags,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def find_datasets(
         self,
         *,
@@ -739,23 +1459,28 @@ class AsyncDatasetsClient:
         )
         return _response.data
 
-    async def get_dataset_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> DatasetPublic:
+    async def create_dataset_items_from_csv(
+        self,
+        *,
+        file: typing.Dict[str, typing.Optional[typing.Any]],
+        dataset_id: str,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
         """
-        Get dataset by id
+        Create dataset items from uploaded CSV file. CSV should have headers in the first row. Processing happens asynchronously in batches.
 
         Parameters
         ----------
-        id : str
+        file : typing.Dict[str, typing.Optional[typing.Any]]
+
+        dataset_id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        DatasetPublic
-            Dataset resource
+        None
 
         Examples
         --------
@@ -763,36 +1488,34 @@ class AsyncDatasetsClient:
         import asyncio
         client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.datasets.get_dataset_by_id(id='id', )
+            await client.datasets.create_dataset_items_from_csv(file={'key': 'value'
+            }, dataset_id='dataset_id', )
         asyncio.run(main())
         """
-        _response = await self._raw_client.get_dataset_by_id(id, request_options=request_options)
+        _response = await self._raw_client.create_dataset_items_from_csv(
+            file=file, dataset_id=dataset_id, request_options=request_options
+        )
         return _response.data
 
-    async def update_dataset(
+    async def create_dataset_items_from_spans(
         self,
-        id: str,
+        dataset_id: str,
         *,
-        name: str,
-        description: typing.Optional[str] = OMIT,
-        visibility: typing.Optional[DatasetUpdateVisibility] = OMIT,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        span_ids: typing.Sequence[str],
+        enrichment_options: SpanEnrichmentOptions,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Update dataset by id
+        Create dataset items from spans with enriched metadata
 
         Parameters
         ----------
-        id : str
-
-        name : str
+        dataset_id : str
 
-        description : typing.Optional[str]
-
-        visibility : typing.Optional[DatasetUpdateVisibility]
+        span_ids : typing.Sequence[str]
+            Set of span IDs to add to the dataset
 
-        tags : typing.Optional[typing.Sequence[str]]
+        enrichment_options : SpanEnrichmentOptions
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -804,18 +1527,137 @@ class AsyncDatasetsClient:
         Examples
         --------
         from Opik import AsyncOpikApi
+        from Opik import SpanEnrichmentOptions
         import asyncio
         client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.datasets.update_dataset(id='id', name='name', )
+            await client.datasets.create_dataset_items_from_spans(dataset_id='dataset_id', span_ids=['span_ids'], enrichment_options=SpanEnrichmentOptions(), )
         asyncio.run(main())
         """
-        _response = await self._raw_client.update_dataset(
-            id, name=name, description=description, visibility=visibility, tags=tags, request_options=request_options
+        _response = await self._raw_client.create_dataset_items_from_spans(
+            dataset_id, span_ids=span_ids, enrichment_options=enrichment_options, request_options=request_options
         )
         return _response.data
 
-    async def delete_dataset(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+    async def create_dataset_items_from_traces(
+        self,
+        dataset_id: str,
+        *,
+        trace_ids: typing.Sequence[str],
+        enrichment_options: TraceEnrichmentOptions,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Create dataset items from traces with enriched metadata
+
+        Parameters
+        ----------
+        dataset_id : str
+
+        trace_ids : typing.Sequence[str]
+            Set of trace IDs to add to the dataset
+
+        enrichment_options : TraceEnrichmentOptions
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        from Opik import TraceEnrichmentOptions
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.create_dataset_items_from_traces(dataset_id='dataset_id', trace_ids=['trace_ids'], enrichment_options=TraceEnrichmentOptions(), )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_dataset_items_from_traces(
+            dataset_id, trace_ids=trace_ids, enrichment_options=enrichment_options, request_options=request_options
+        )
+        return _response.data
+
+    async def get_dataset_by_id(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetPublic:
+        """
+        Get dataset by id
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetPublic
+            Dataset resource
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.get_dataset_by_id(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_dataset_by_id(id, request_options=request_options)
+        return _response.data
+
+    async def update_dataset(
+        self,
+        id: str,
+        *,
+        name: str,
+        description: typing.Optional[str] = OMIT,
+        visibility: typing.Optional[DatasetUpdateVisibility] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update dataset by id
+
+        Parameters
+        ----------
+        id : str
+
+        name : str
+
+        description : typing.Optional[str]
+
+        visibility : typing.Optional[DatasetUpdateVisibility]
+
+        tags : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.update_dataset(id='id', name='name', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.update_dataset(
+            id, name=name, description=description, visibility=visibility, tags=tags, request_options=request_options
+        )
+        return _response.data
+
+    async def delete_dataset(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
         Delete dataset by id
 
@@ -874,14 +1716,30 @@ class AsyncDatasetsClient:
         return _response.data
 
     async def delete_dataset_items(
-        self, *, item_ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+        self,
+        *,
+        item_ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        dataset_id: typing.Optional[str] = OMIT,
+        filters: typing.Optional[typing.Sequence[DatasetItemFilter]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Delete dataset items
+        Delete dataset items using one of two modes:
+        1. **Delete by IDs**: Provide 'item_ids' to delete specific items by their IDs
+        2. **Delete by filters**: Provide 'dataset_id' with optional 'filters' to delete items matching criteria
+
+        When using filters, an empty 'filters' array will delete all items in the specified dataset.
 
         Parameters
         ----------
-        item_ids : typing.Sequence[str]
+        item_ids : typing.Optional[typing.Sequence[str]]
+            List of dataset item IDs to delete (max 1000). Use this to delete specific items by their IDs. Mutually exclusive with 'dataset_id' and 'filters'.
+
+        dataset_id : typing.Optional[str]
+            Dataset ID to scope the deletion. Required when using 'filters'. Mutually exclusive with 'item_ids'.
+
+        filters : typing.Optional[typing.Sequence[DatasetItemFilter]]
+            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'.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -896,10 +1754,12 @@ class AsyncDatasetsClient:
         import asyncio
         client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.datasets.delete_dataset_items(item_ids=['item_ids'], )
+            await client.datasets.delete_dataset_items()
         asyncio.run(main())
         """
-        _response = await self._raw_client.delete_dataset_items(item_ids=item_ids, request_options=request_options)
+        _response = await self._raw_client.delete_dataset_items(
+            item_ids=item_ids, dataset_id=dataset_id, filters=filters, request_options=request_options
+        )
         return _response.data
 
     async def delete_datasets_batch(
@@ -931,6 +1791,67 @@ class AsyncDatasetsClient:
         _response = await self._raw_client.delete_datasets_batch(ids=ids, request_options=request_options)
         return _response.data
 
+    async def expand_dataset(
+        self,
+        id: str,
+        *,
+        model: str,
+        sample_count: typing.Optional[int] = OMIT,
+        preserve_fields: typing.Optional[typing.Sequence[str]] = OMIT,
+        variation_instructions: typing.Optional[str] = OMIT,
+        custom_prompt: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetExpansionResponse:
+        """
+        Generate synthetic dataset samples using LLM based on existing data patterns
+
+        Parameters
+        ----------
+        id : str
+
+        model : str
+            The model to use for synthetic data generation
+
+        sample_count : typing.Optional[int]
+            Number of synthetic samples to generate
+
+        preserve_fields : typing.Optional[typing.Sequence[str]]
+            Fields to preserve patterns from original data
+
+        variation_instructions : typing.Optional[str]
+            Additional instructions for data variation
+
+        custom_prompt : typing.Optional[str]
+            Custom prompt to use for generation instead of auto-generated one
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetExpansionResponse
+            Generated synthetic samples
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.expand_dataset(id='id', model='gpt-4', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.expand_dataset(
+            id,
+            model=model,
+            sample_count=sample_count,
+            preserve_fields=preserve_fields,
+            variation_instructions=variation_instructions,
+            custom_prompt=custom_prompt,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def find_dataset_items_with_experiment_items(
         self,
         id: str,
@@ -939,6 +1860,8 @@ class AsyncDatasetsClient:
         page: typing.Optional[int] = None,
         size: typing.Optional[int] = None,
         filters: typing.Optional[str] = None,
+        sorting: typing.Optional[str] = None,
+        search: typing.Optional[str] = None,
         truncate: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DatasetItemPageCompare:
@@ -957,6 +1880,10 @@ class AsyncDatasetsClient:
 
         filters : typing.Optional[str]
 
+        sorting : typing.Optional[str]
+
+        search : typing.Optional[str]
+
         truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
@@ -982,6 +1909,8 @@ class AsyncDatasetsClient:
             page=page,
             size=size,
             filters=filters,
+            sorting=sorting,
+            search=search,
             truncate=truncate,
             request_options=request_options,
         )
@@ -1019,6 +1948,47 @@ class AsyncDatasetsClient:
         )
         return _response.data
 
+    async def get_dataset_experiment_items_stats(
+        self,
+        id: str,
+        *,
+        experiment_ids: str,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectStatsPublic:
+        """
+        Get experiment items stats for dataset
+
+        Parameters
+        ----------
+        id : str
+
+        experiment_ids : str
+
+        filters : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectStatsPublic
+            Experiment items stats resource
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.get_dataset_experiment_items_stats(id='id', experiment_ids='experiment_ids', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_dataset_experiment_items_stats(
+            id, experiment_ids=experiment_ids, filters=filters, request_options=request_options
+        )
+        return _response.data
+
     async def get_dataset_item_by_id(
         self, item_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> DatasetItemPublic:
@@ -1049,12 +2019,74 @@ class AsyncDatasetsClient:
         _response = await self._raw_client.get_dataset_item_by_id(item_id, request_options=request_options)
         return _response.data
 
+    async def patch_dataset_item(
+        self,
+        item_id: str,
+        *,
+        source: DatasetItemWriteSource,
+        data: JsonNode,
+        id: typing.Optional[str] = OMIT,
+        trace_id: typing.Optional[str] = OMIT,
+        span_id: typing.Optional[str] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Partially update dataset item by id. Only provided fields will be updated.
+
+        Parameters
+        ----------
+        item_id : str
+
+        source : DatasetItemWriteSource
+
+        data : JsonNode
+
+        id : typing.Optional[str]
+
+        trace_id : typing.Optional[str]
+
+        span_id : typing.Optional[str]
+
+        tags : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.patch_dataset_item(item_id='itemId', source="manual", data={'key': 'value'
+            }, )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.patch_dataset_item(
+            item_id,
+            source=source,
+            data=data,
+            id=id,
+            trace_id=trace_id,
+            span_id=span_id,
+            tags=tags,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def get_dataset_items(
         self,
         id: str,
         *,
         page: typing.Optional[int] = None,
         size: typing.Optional[int] = None,
+        version: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
         truncate: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> DatasetItemPagePublic:
@@ -1069,6 +2101,10 @@ class AsyncDatasetsClient:
 
         size : typing.Optional[int]
 
+        version : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
         truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
@@ -1089,7 +2125,13 @@ class AsyncDatasetsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.get_dataset_items(
-            id, page=page, size=size, truncate=truncate, request_options=request_options
+            id,
+            page=page,
+            size=size,
+            version=version,
+            filters=filters,
+            truncate=truncate,
+            request_options=request_options,
         )
         return _response.data
 
@@ -1166,3 +2208,227 @@ class AsyncDatasetsClient:
         ) as r:
             async for data in r.data:
                 yield data
+
+    async def compare_dataset_versions(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetVersionDiff:
+        """
+        Compare the latest committed dataset version with the current draft state. This endpoint provides insights into changes made since the last version was committed. The comparison calculates additions, modifications, deletions, and unchanged items between the latest version snapshot and current draft.
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionDiff
+            Diff computed successfully
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.compare_dataset_versions(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.compare_dataset_versions(id, request_options=request_options)
+        return _response.data
+
+    async def create_version_tag(
+        self, version_hash: str, id: str, *, tag: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Add a tag to a specific dataset version for easy reference (e.g., 'baseline', 'v1.0', 'production')
+
+        Parameters
+        ----------
+        version_hash : str
+
+        id : str
+
+        tag : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.create_version_tag(version_hash='versionHash', id='id', tag='tag', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_version_tag(
+            version_hash, id, tag=tag, request_options=request_options
+        )
+        return _response.data
+
+    async def delete_version_tag(
+        self, version_hash: str, tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Remove a tag from a dataset version. The version itself is not deleted, only the tag reference.
+
+        Parameters
+        ----------
+        version_hash : str
+
+        tag : str
+
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.delete_version_tag(version_hash='versionHash', tag='tag', id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_version_tag(version_hash, tag, id, request_options=request_options)
+        return _response.data
+
+    async def list_dataset_versions(
+        self,
+        id: str,
+        *,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetVersionPagePublic:
+        """
+        Get paginated list of versions for a dataset, ordered by creation time (newest first)
+
+        Parameters
+        ----------
+        id : str
+
+        page : typing.Optional[int]
+
+        size : typing.Optional[int]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPagePublic
+            Dataset versions
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.list_dataset_versions(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.list_dataset_versions(
+            id, page=page, size=size, request_options=request_options
+        )
+        return _response.data
+
+    async def restore_dataset_version(
+        self, id: str, *, version_ref: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetVersionPublic:
+        """
+        Restores the dataset to a previous version state by creating a new version with items copied from the specified version. If the version is already the latest, returns it as-is (no-op).
+
+        Parameters
+        ----------
+        id : str
+
+        version_ref : str
+            Version hash or tag to restore from
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPublic
+            Version restored successfully
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.restore_dataset_version(id='id', version_ref='version_ref', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.restore_dataset_version(
+            id, version_ref=version_ref, request_options=request_options
+        )
+        return _response.data
+
+    async def update_dataset_version(
+        self,
+        version_hash: str,
+        id: str,
+        *,
+        change_description: typing.Optional[str] = OMIT,
+        tags_to_add: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> DatasetVersionPublic:
+        """
+        Update a dataset version's change_description and/or add new tags
+
+        Parameters
+        ----------
+        version_hash : str
+
+        id : str
+
+        change_description : typing.Optional[str]
+            Optional description of changes in this version
+
+        tags_to_add : typing.Optional[typing.Sequence[str]]
+            Optional list of tags to add to this version
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        DatasetVersionPublic
+            Version updated successfully
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.datasets.update_dataset_version(version_hash='versionHash', id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.update_dataset_version(
+            version_hash,
+            id,
+            change_description=change_description,
+            tags_to_add=tags_to_add,
+            request_options=request_options,
+        )
+        return _response.data