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

opik/rest_api/datasets/client.py

@@ -5,6 +5,7 @@ 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
@@ -45,12 +46,59 @@ 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:
@@ -66,6 +114,9 @@ class DatasetsClient:
 
         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.
 
@@ -84,7 +135,12 @@ class DatasetsClient:
         client.datasets.batch_update_dataset_items(update=DatasetItemUpdate(), )
         """
         _response = self._raw_client.batch_update_dataset_items(
-            update=update, ids=ids, filters=filters, merge_tags=merge_tags, request_options=request_options
+            update=update,
+            ids=ids,
+            filters=filters,
+            dataset_id=dataset_id,
+            merge_tags=merge_tags,
+            request_options=request_options,
         )
         return _response.data
 
@@ -198,6 +254,7 @@ class DatasetsClient:
         items: typing.Sequence[DatasetItemWrite],
         dataset_name: typing.Optional[str] = OMIT,
         dataset_id: typing.Optional[str] = OMIT,
+        batch_group_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -213,6 +270,9 @@ class DatasetsClient:
         dataset_id : typing.Optional[str]
             If null, dataset_name must be provided
 
+        batch_group_id : typing.Optional[str]
+            Optional batch group ID to group multiple batches into a single dataset version. If null, mutates the latest version instead of creating a new one.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -229,7 +289,11 @@ class DatasetsClient:
         }, )], )
         """
         _response = self._raw_client.create_or_update_dataset_items(
-            items=items, dataset_name=dataset_name, dataset_id=dataset_id, request_options=request_options
+            items=items,
+            dataset_name=dataset_name,
+            dataset_id=dataset_id,
+            batch_group_id=batch_group_id,
+            request_options=request_options,
         )
         return _response.data
 
@@ -465,14 +529,34 @@ 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,
+        batch_group_id: typing.Optional[str] = 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'.
+
+        batch_group_id : typing.Optional[str]
+            Optional batch group ID to group multiple delete operations into a single dataset version. If null, mutates the latest version instead of creating a new one.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -485,9 +569,15 @@ 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,
+            batch_group_id=batch_group_id,
+            request_options=request_options,
+        )
         return _response.data
 
     def delete_datasets_batch(
@@ -880,6 +970,7 @@ class DatasetsClient:
         dataset_name: str,
         last_retrieved_id: typing.Optional[str] = OMIT,
         steam_limit: typing.Optional[int] = OMIT,
+        dataset_version: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.Iterator[bytes]:
         """
@@ -893,6 +984,8 @@ class DatasetsClient:
 
         steam_limit : typing.Optional[int]
 
+        dataset_version : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
@@ -905,6 +998,7 @@ class DatasetsClient:
             dataset_name=dataset_name,
             last_retrieved_id=last_retrieved_id,
             steam_limit=steam_limit,
+            dataset_version=dataset_version,
             request_options=request_options,
         ) as r:
             yield from r.data
@@ -966,127 +1060,127 @@ class DatasetsClient:
         _response = self._raw_client.create_version_tag(version_hash, id, tag=tag, 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:
+    def delete_version_tag(
+        self, version_hash: str, tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
-        Get paginated list of versions for a dataset, ordered by creation time (newest first)
+        Remove a tag from a dataset version. The version itself is not deleted, only the tag reference.
 
         Parameters
         ----------
-        id : str
+        version_hash : str
 
-        page : typing.Optional[int]
+        tag : str
 
-        size : typing.Optional[int]
+        id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        DatasetVersionPagePublic
-            Dataset versions
+        None
 
         Examples
         --------
         from Opik import OpikApi
         client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
-        client.datasets.list_dataset_versions(id='id', )
+        client.datasets.delete_version_tag(version_hash='versionHash', tag='tag', id='id', )
         """
-        _response = self._raw_client.list_dataset_versions(id, page=page, size=size, request_options=request_options)
+        _response = self._raw_client.delete_version_tag(version_hash, tag, id, request_options=request_options)
         return _response.data
 
-    def create_dataset_version(
+    def list_dataset_versions(
         self,
         id: str,
         *,
-        tag: typing.Optional[str] = OMIT,
-        change_description: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[typing.Dict[str, str]] = OMIT,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> DatasetVersionPagePublic:
         """
-        Create a new immutable version of the dataset by snapshotting the current state
+        Get paginated list of versions for a dataset, ordered by creation time (newest first)
 
         Parameters
         ----------
         id : str
 
-        tag : typing.Optional[str]
-            Optional tag for this version
-
-        change_description : typing.Optional[str]
-            Optional description of changes in this version
+        page : typing.Optional[int]
 
-        metadata : typing.Optional[typing.Dict[str, str]]
-            Optional user-defined metadata
+        size : typing.Optional[int]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        DatasetVersionPagePublic
+            Dataset versions
 
         Examples
         --------
         from Opik import OpikApi
         client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
-        client.datasets.create_dataset_version(id='id', )
+        client.datasets.list_dataset_versions(id='id', )
         """
-        _response = self._raw_client.create_dataset_version(
-            id, tag=tag, change_description=change_description, metadata=metadata, request_options=request_options
-        )
+        _response = self._raw_client.list_dataset_versions(id, page=page, size=size, 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:
+    def restore_dataset_version(
+        self, id: str, *, version_ref: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetVersionPublic:
         """
-        Remove a tag from a dataset version. The version itself is not deleted, only the tag reference.
+        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
         ----------
-        version_hash : str
-
-        tag : str
-
         id : str
 
+        version_ref : str
+            Version hash or tag to restore from
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        DatasetVersionPublic
+            Version restored successfully
 
         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', )
+        client.datasets.restore_dataset_version(id='id', version_ref='version_ref', )
         """
-        _response = self._raw_client.delete_version_tag(version_hash, tag, id, request_options=request_options)
+        _response = self._raw_client.restore_dataset_version(
+            id, version_ref=version_ref, request_options=request_options
+        )
         return _response.data
 
-    def restore_dataset_version(
-        self, id: str, *, version_ref: str, request_options: typing.Optional[RequestOptions] = None
+    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:
         """
-        Restores the dataset to a previous version state. All draft items are replaced with items from the specified version. If the version is not the latest, a new version snapshot is created. If the version is the latest, only draft items are replaced (revert functionality).
+        Update a dataset version's change_description and/or add new tags
 
         Parameters
         ----------
+        version_hash : str
+
         id : str
 
-        version_ref : str
-            Version hash or tag to restore from
+        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.
@@ -1094,16 +1188,20 @@ class DatasetsClient:
         Returns
         -------
         DatasetVersionPublic
-            Version restored successfully
+            Version updated 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', )
+        client.datasets.update_dataset_version(version_hash='versionHash', id='id', )
         """
-        _response = self._raw_client.restore_dataset_version(
-            id, version_ref=version_ref, request_options=request_options
+        _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
 
@@ -1123,12 +1221,62 @@ 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:
@@ -1144,6 +1292,9 @@ class AsyncDatasetsClient:
 
         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.
 
@@ -1165,7 +1316,12 @@ class AsyncDatasetsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.batch_update_dataset_items(
-            update=update, ids=ids, filters=filters, merge_tags=merge_tags, request_options=request_options
+            update=update,
+            ids=ids,
+            filters=filters,
+            dataset_id=dataset_id,
+            merge_tags=merge_tags,
+            request_options=request_options,
         )
         return _response.data
 
@@ -1285,6 +1441,7 @@ class AsyncDatasetsClient:
         items: typing.Sequence[DatasetItemWrite],
         dataset_name: typing.Optional[str] = OMIT,
         dataset_id: typing.Optional[str] = OMIT,
+        batch_group_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -1300,6 +1457,9 @@ class AsyncDatasetsClient:
         dataset_id : typing.Optional[str]
             If null, dataset_name must be provided
 
+        batch_group_id : typing.Optional[str]
+            Optional batch group ID to group multiple batches into a single dataset version. If null, mutates the latest version instead of creating a new one.
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1319,7 +1479,11 @@ class AsyncDatasetsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.create_or_update_dataset_items(
-            items=items, dataset_name=dataset_name, dataset_id=dataset_id, request_options=request_options
+            items=items,
+            dataset_name=dataset_name,
+            dataset_id=dataset_id,
+            batch_group_id=batch_group_id,
+            request_options=request_options,
         )
         return _response.data
 
@@ -1580,14 +1744,34 @@ 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,
+        batch_group_id: typing.Optional[str] = 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'.
+
+        batch_group_id : typing.Optional[str]
+            Optional batch group ID to group multiple delete operations into a single dataset version. If null, mutates the latest version instead of creating a new one.
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1602,10 +1786,16 @@ 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,
+            batch_group_id=batch_group_id,
+            request_options=request_options,
+        )
         return _response.data
 
     async def delete_datasets_batch(
@@ -2025,6 +2215,7 @@ class AsyncDatasetsClient:
         dataset_name: str,
         last_retrieved_id: typing.Optional[str] = OMIT,
         steam_limit: typing.Optional[int] = OMIT,
+        dataset_version: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.AsyncIterator[bytes]:
         """
@@ -2038,6 +2229,8 @@ class AsyncDatasetsClient:
 
         steam_limit : typing.Optional[int]
 
+        dataset_version : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
@@ -2050,6 +2243,7 @@ class AsyncDatasetsClient:
             dataset_name=dataset_name,
             last_retrieved_id=last_retrieved_id,
             steam_limit=steam_limit,
+            dataset_version=dataset_version,
             request_options=request_options,
         ) as r:
             async for data in r.data:
@@ -2120,32 +2314,26 @@ class AsyncDatasetsClient:
         )
         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:
+    async def delete_version_tag(
+        self, version_hash: str, tag: str, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
-        Get paginated list of versions for a dataset, ordered by creation time (newest first)
+        Remove a tag from a dataset version. The version itself is not deleted, only the tag reference.
 
         Parameters
         ----------
-        id : str
+        version_hash : str
 
-        page : typing.Optional[int]
+        tag : str
 
-        size : typing.Optional[int]
+        id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        DatasetVersionPagePublic
-            Dataset versions
+        None
 
         Examples
         --------
@@ -2153,45 +2341,38 @@ class AsyncDatasetsClient:
         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', )
+            await client.datasets.delete_version_tag(version_hash='versionHash', tag='tag', id='id', )
         asyncio.run(main())
         """
-        _response = await self._raw_client.list_dataset_versions(
-            id, page=page, size=size, request_options=request_options
-        )
+        _response = await self._raw_client.delete_version_tag(version_hash, tag, id, request_options=request_options)
         return _response.data
 
-    async def create_dataset_version(
+    async def list_dataset_versions(
         self,
         id: str,
         *,
-        tag: typing.Optional[str] = OMIT,
-        change_description: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[typing.Dict[str, str]] = OMIT,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> DatasetVersionPagePublic:
         """
-        Create a new immutable version of the dataset by snapshotting the current state
+        Get paginated list of versions for a dataset, ordered by creation time (newest first)
 
         Parameters
         ----------
         id : str
 
-        tag : typing.Optional[str]
-            Optional tag for this version
-
-        change_description : typing.Optional[str]
-            Optional description of changes in this version
+        page : typing.Optional[int]
 
-        metadata : typing.Optional[typing.Dict[str, str]]
-            Optional user-defined metadata
+        size : typing.Optional[int]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        DatasetVersionPagePublic
+            Dataset versions
 
         Examples
         --------
@@ -2199,34 +2380,34 @@ class AsyncDatasetsClient:
         import asyncio
         client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.datasets.create_dataset_version(id='id', )
+            await client.datasets.list_dataset_versions(id='id', )
         asyncio.run(main())
         """
-        _response = await self._raw_client.create_dataset_version(
-            id, tag=tag, change_description=change_description, metadata=metadata, request_options=request_options
+        _response = await self._raw_client.list_dataset_versions(
+            id, page=page, size=size, 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:
+    async def restore_dataset_version(
+        self, id: str, *, version_ref: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> DatasetVersionPublic:
         """
-        Remove a tag from a dataset version. The version itself is not deleted, only the tag reference.
+        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
         ----------
-        version_hash : str
-
-        tag : str
-
         id : str
 
+        version_ref : str
+            Version hash or tag to restore from
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        DatasetVersionPublic
+            Version restored successfully
 
         Examples
         --------
@@ -2234,24 +2415,37 @@ class AsyncDatasetsClient:
         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', )
+            await client.datasets.restore_dataset_version(id='id', version_ref='version_ref', )
         asyncio.run(main())
         """
-        _response = await self._raw_client.delete_version_tag(version_hash, tag, id, request_options=request_options)
+        _response = await self._raw_client.restore_dataset_version(
+            id, version_ref=version_ref, request_options=request_options
+        )
         return _response.data
 
-    async def restore_dataset_version(
-        self, id: str, *, version_ref: str, request_options: typing.Optional[RequestOptions] = None
+    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:
         """
-        Restores the dataset to a previous version state. All draft items are replaced with items from the specified version. If the version is not the latest, a new version snapshot is created. If the version is the latest, only draft items are replaced (revert functionality).
+        Update a dataset version's change_description and/or add new tags
 
         Parameters
         ----------
+        version_hash : str
+
         id : str
 
-        version_ref : str
-            Version hash or tag to restore from
+        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.
@@ -2259,7 +2453,7 @@ class AsyncDatasetsClient:
         Returns
         -------
         DatasetVersionPublic
-            Version restored successfully
+            Version updated successfully
 
         Examples
         --------
@@ -2267,10 +2461,14 @@ class AsyncDatasetsClient:
         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', )
+            await client.datasets.update_dataset_version(version_hash='versionHash', id='id', )
         asyncio.run(main())
         """
-        _response = await self._raw_client.restore_dataset_version(
-            id, version_ref=version_ref, request_options=request_options
+        _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