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

opik/rest_api/experiments/client.py

@@ -1,21 +1,28 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
-from ..core.client_wrapper import SyncClientWrapper
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..types.experiment_page_public import ExperimentPagePublic
-from ..core.pydantic_utilities import parse_obj_as
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-from ..types.json_node_write import JsonNodeWrite
-from ..types.prompt_version_link_write import PromptVersionLinkWrite
-from ..core.serialization import convert_and_respect_annotation_metadata
+from ..types.experiment_group_aggregations_response import ExperimentGroupAggregationsResponse
+from ..types.experiment_group_response import ExperimentGroupResponse
 from ..types.experiment_item import ExperimentItem
-from ..types.experiment_public import ExperimentPublic
-from ..core.jsonable_encoder import jsonable_encoder
-from ..errors.not_found_error import NotFoundError
+from ..types.experiment_item_bulk_record_experiment_item_bulk_write_view import (
+    ExperimentItemBulkRecordExperimentItemBulkWriteView,
+)
 from ..types.experiment_item_public import ExperimentItemPublic
-from ..core.client_wrapper import AsyncClientWrapper
+from ..types.experiment_page_public import ExperimentPagePublic
+from ..types.experiment_public import ExperimentPublic
+from ..types.experiment_score import ExperimentScore
+from ..types.experiment_score_write import ExperimentScoreWrite
+from ..types.json_list_string_write import JsonListStringWrite
+from ..types.json_node import JsonNode
+from ..types.prompt_version_link_write import PromptVersionLinkWrite
+from .raw_client import AsyncRawExperimentsClient, RawExperimentsClient
+from .types.experiment_update_status import ExperimentUpdateStatus
+from .types.experiment_update_type import ExperimentUpdateType
+from .types.experiment_write_status import ExperimentWriteStatus
+from .types.experiment_write_type import ExperimentWriteType
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -23,7 +30,18 @@ OMIT = typing.cast(typing.Any, ...)
 
 class ExperimentsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawExperimentsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawExperimentsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawExperimentsClient
+        """
+        return self._raw_client
 
     def find_experiments(
         self,
@@ -31,9 +49,13 @@ class ExperimentsClient:
         page: typing.Optional[int] = None,
         size: typing.Optional[int] = None,
         dataset_id: typing.Optional[str] = None,
+        optimization_id: typing.Optional[str] = None,
+        types: typing.Optional[str] = None,
         name: typing.Optional[str] = None,
         dataset_deleted: typing.Optional[bool] = None,
         prompt_id: typing.Optional[str] = None,
+        sorting: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ExperimentPagePublic:
         """
@@ -47,12 +69,20 @@ class ExperimentsClient:
 
         dataset_id : typing.Optional[str]
 
+        optimization_id : typing.Optional[str]
+
+        types : typing.Optional[str]
+
         name : typing.Optional[str]
 
         dataset_deleted : typing.Optional[bool]
 
         prompt_id : typing.Optional[str]
 
+        sorting : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -64,39 +94,23 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         client.experiments.find_experiments()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "datasetId": dataset_id,
-                "name": name,
-                "dataset_deleted": dataset_deleted,
-                "prompt_id": prompt_id,
-            },
+        _response = self._raw_client.find_experiments(
+            page=page,
+            size=size,
+            dataset_id=dataset_id,
+            optimization_id=optimization_id,
+            types=types,
+            name=name,
+            dataset_deleted=dataset_deleted,
+            prompt_id=prompt_id,
+            sorting=sorting,
+            filters=filters,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentPagePublic,
-                    parse_obj_as(
-                        type_=ExperimentPagePublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def create_experiment(
         self,
@@ -104,11 +118,13 @@ class ExperimentsClient:
         dataset_name: str,
         id: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[JsonNodeWrite] = OMIT,
+        metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        type: typing.Optional[ExperimentWriteType] = OMIT,
+        optimization_id: typing.Optional[str] = OMIT,
+        status: typing.Optional[ExperimentWriteStatus] = OMIT,
+        experiment_scores: typing.Optional[typing.Sequence[ExperimentScoreWrite]] = OMIT,
         prompt_version: typing.Optional[PromptVersionLinkWrite] = OMIT,
-        prompt_versions: typing.Optional[
-            typing.Sequence[PromptVersionLinkWrite]
-        ] = OMIT,
+        prompt_versions: typing.Optional[typing.Sequence[PromptVersionLinkWrite]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -122,7 +138,15 @@ class ExperimentsClient:
 
         name : typing.Optional[str]
 
-        metadata : typing.Optional[JsonNodeWrite]
+        metadata : typing.Optional[JsonListStringWrite]
+
+        type : typing.Optional[ExperimentWriteType]
+
+        optimization_id : typing.Optional[str]
+
+        status : typing.Optional[ExperimentWriteStatus]
+
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScoreWrite]]
 
         prompt_version : typing.Optional[PromptVersionLinkWrite]
 
@@ -138,47 +162,23 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.create_experiment(
-            dataset_name="dataset_name",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.create_experiment(dataset_name='dataset_name', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments",
-            method="POST",
-            json={
-                "id": id,
-                "dataset_name": dataset_name,
-                "name": name,
-                "metadata": metadata,
-                "prompt_version": convert_and_respect_annotation_metadata(
-                    object_=prompt_version,
-                    annotation=PromptVersionLinkWrite,
-                    direction="write",
-                ),
-                "prompt_versions": convert_and_respect_annotation_metadata(
-                    object_=prompt_versions,
-                    annotation=typing.Sequence[PromptVersionLinkWrite],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.create_experiment(
+            dataset_name=dataset_name,
+            id=id,
+            name=name,
+            metadata=metadata,
+            type=type,
+            optimization_id=optimization_id,
+            status=status,
+            experiment_scores=experiment_scores,
+            prompt_version=prompt_version,
+            prompt_versions=prompt_versions,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def create_experiment_items(
         self,
@@ -202,51 +202,18 @@ class ExperimentsClient:
 
         Examples
         --------
-        from Opik import ExperimentItem, OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.create_experiment_items(
-            experiment_items=[
-                ExperimentItem(
-                    experiment_id="experiment_id",
-                    dataset_item_id="dataset_item_id",
-                    trace_id="trace_id",
-                )
-            ],
-        )
+        from Opik import OpikApi
+        from Opik import ExperimentItem
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.create_experiment_items(experiment_items=[ExperimentItem(experiment_id='experiment_id', dataset_item_id='dataset_item_id', trace_id='trace_id', )], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/items",
-            method="POST",
-            json={
-                "experiment_items": convert_and_respect_annotation_metadata(
-                    object_=experiment_items,
-                    annotation=typing.Sequence[ExperimentItem],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.create_experiment_items(
+            experiment_items=experiment_items, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def delete_experiment_items(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete experiment items
@@ -265,40 +232,14 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.delete_experiment_items(
-            ids=["ids"],
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.delete_experiment_items(ids=['ids'], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/items/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.delete_experiment_items(ids=ids, request_options=request_options)
+        return _response.data
 
     def delete_experiments_by_id(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete experiments by id
@@ -317,40 +258,60 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.delete_experiments_by_id(ids=['ids'], )
+        """
+        _response = self._raw_client.delete_experiments_by_id(ids=ids, request_options=request_options)
+        return _response.data
 
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.delete_experiments_by_id(
-            ids=["ids"],
-        )
+    def experiment_items_bulk(
+        self,
+        *,
+        experiment_name: str,
+        dataset_name: str,
+        items: typing.Sequence[ExperimentItemBulkRecordExperimentItemBulkWriteView],
+        experiment_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Record experiment items in bulk with traces, spans, and feedback scores. Maximum request size is 4MB.
+
+        Parameters
+        ----------
+        experiment_name : str
+
+        dataset_name : str
+
+        items : typing.Sequence[ExperimentItemBulkRecordExperimentItemBulkWriteView]
+
+        experiment_id : typing.Optional[str]
+            Optional experiment ID. If provided, items will be added to the existing experiment and experimentName will be ignored. If not provided or experiment with that ID doesn't exist, a new experiment will be created with the given experimentName
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        from Opik import ExperimentItemBulkRecordExperimentItemBulkWriteView
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.experiment_items_bulk(experiment_name='experiment_name', dataset_name='dataset_name', items=[ExperimentItemBulkRecordExperimentItemBulkWriteView(dataset_item_id='dataset_item_id', )], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.experiment_items_bulk(
+            experiment_name=experiment_name,
+            dataset_name=dataset_name,
+            items=items,
+            experiment_id=experiment_id,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def find_feedback_score_names(
-        self,
-        *,
-        experiment_ids: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, experiment_ids: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[str]:
         """
         Find Feedback Score names
@@ -370,34 +331,121 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.find_feedback_score_names()
+        """
+        _response = self._raw_client.find_feedback_score_names(
+            experiment_ids=experiment_ids, request_options=request_options
+        )
+        return _response.data
+
+    def find_experiment_groups(
+        self,
+        *,
+        groups: typing.Optional[str] = None,
+        types: typing.Optional[str] = None,
+        name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ExperimentGroupResponse:
+        """
+        Find experiments grouped by specified fields
 
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
+        Parameters
+        ----------
+        groups : typing.Optional[str]
+
+        types : typing.Optional[str]
+
+        name : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ExperimentGroupResponse
+            Experiment groups
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.find_experiment_groups()
+        """
+        _response = self._raw_client.find_experiment_groups(
+            groups=groups, types=types, name=name, filters=filters, request_options=request_options
         )
-        client.experiments.find_feedback_score_names()
+        return _response.data
+
+    def find_experiment_groups_aggregations(
+        self,
+        *,
+        groups: typing.Optional[str] = None,
+        types: typing.Optional[str] = None,
+        name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ExperimentGroupAggregationsResponse:
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/feedback-scores/names",
-            method="GET",
-            params={
-                "experiment_ids": experiment_ids,
-            },
-            request_options=request_options,
+        Find experiments grouped by specified fields with aggregation metrics
+
+        Parameters
+        ----------
+        groups : typing.Optional[str]
+
+        types : typing.Optional[str]
+
+        name : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ExperimentGroupAggregationsResponse
+            Experiment groups with aggregations
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.find_experiment_groups_aggregations()
+        """
+        _response = self._raw_client.find_experiment_groups_aggregations(
+            groups=groups, types=types, name=name, filters=filters, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    parse_obj_as(
-                        type_=typing.List[str],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
+
+    def finish_experiments(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Finish experiments and trigger alert events
+
+        Parameters
+        ----------
+        ids : 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.experiments.finish_experiments(ids=['ids'], )
+        """
+        _response = self._raw_client.finish_experiments(ids=ids, request_options=request_options)
+        return _response.data
 
     def get_experiment_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -420,109 +468,64 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.get_experiment_by_id(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.get_experiment_by_id(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/experiments/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentPublic,
-                    parse_obj_as(
-                        type_=ExperimentPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def get_experiment_by_name(
-        self, *, name: str, request_options: typing.Optional[RequestOptions] = None
-    ) -> ExperimentPublic:
+        _response = self._raw_client.get_experiment_by_id(id, request_options=request_options)
+        return _response.data
+
+    def update_experiment(
+        self,
+        id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        metadata: typing.Optional[JsonNode] = OMIT,
+        type: typing.Optional[ExperimentUpdateType] = OMIT,
+        status: typing.Optional[ExperimentUpdateStatus] = OMIT,
+        experiment_scores: typing.Optional[typing.Sequence[ExperimentScore]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
         """
-        Get experiment by name
+        Update experiment by id
 
         Parameters
         ----------
-        name : str
+        id : str
+
+        name : typing.Optional[str]
+
+        metadata : typing.Optional[JsonNode]
+
+        type : typing.Optional[ExperimentUpdateType]
+
+        status : typing.Optional[ExperimentUpdateStatus]
+            The status of the experiment
+
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScore]]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        ExperimentPublic
-            Experiments resource
+        None
 
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.get_experiment_by_name(
-            name="name",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.update_experiment(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/retrieve",
-            method="POST",
-            json={
-                "name": name,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = self._raw_client.update_experiment(
+            id,
+            name=name,
+            metadata=metadata,
+            type=type,
+            status=status,
+            experiment_scores=experiment_scores,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentPublic,
-                    parse_obj_as(
-                        type_=ExperimentPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def get_experiment_item_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -545,43 +548,11 @@ class ExperimentsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.experiments.get_experiment_item_by_id(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.experiments.get_experiment_item_by_id(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/experiments/items/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentItemPublic,
-                    parse_obj_as(
-                        type_=ExperimentItemPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.get_experiment_item_by_id(id, request_options=request_options)
+        return _response.data
 
     def stream_experiment_items(
         self,
@@ -609,46 +580,67 @@ class ExperimentsClient:
         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.
 
-        Yields
-        ------
+        Returns
+        -------
         typing.Iterator[bytes]
             Experiment items stream or error during process
         """
-        with self._client_wrapper.httpx_client.stream(
-            "v1/private/experiments/items/stream",
-            method="POST",
-            json={
-                "experiment_name": experiment_name,
-                "limit": limit,
-                "last_retrieved_id": last_retrieved_id,
-                "truncate": truncate,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        with self._raw_client.stream_experiment_items(
+            experiment_name=experiment_name,
+            limit=limit,
+            last_retrieved_id=last_retrieved_id,
+            truncate=truncate,
             request_options=request_options,
-            omit=OMIT,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = (
-                        request_options.get("chunk_size", None)
-                        if request_options is not None
-                        else None
-                    )
-                    for _chunk in _response.iter_bytes(chunk_size=_chunk_size):
-                        yield _chunk
-                    return
-                _response.read()
-                _response_json = _response.json()
-            except JSONDecodeError:
-                raise ApiError(status_code=_response.status_code, body=_response.text)
-            raise ApiError(status_code=_response.status_code, body=_response_json)
+        ) as r:
+            yield from r.data
+
+    def stream_experiments(
+        self,
+        *,
+        name: str,
+        limit: typing.Optional[int] = OMIT,
+        last_retrieved_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Iterator[bytes]:
+        """
+        Stream experiments
+
+        Parameters
+        ----------
+        name : str
+
+        limit : typing.Optional[int]
+
+        last_retrieved_id : 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.
+
+        Returns
+        -------
+        typing.Iterator[bytes]
+            Experiments stream or error during process
+        """
+        with self._raw_client.stream_experiments(
+            name=name, limit=limit, last_retrieved_id=last_retrieved_id, request_options=request_options
+        ) as r:
+            yield from r.data
 
 
 class AsyncExperimentsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawExperimentsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawExperimentsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawExperimentsClient
+        """
+        return self._raw_client
 
     async def find_experiments(
         self,
@@ -656,9 +648,13 @@ class AsyncExperimentsClient:
         page: typing.Optional[int] = None,
         size: typing.Optional[int] = None,
         dataset_id: typing.Optional[str] = None,
+        optimization_id: typing.Optional[str] = None,
+        types: typing.Optional[str] = None,
         name: typing.Optional[str] = None,
         dataset_deleted: typing.Optional[bool] = None,
         prompt_id: typing.Optional[str] = None,
+        sorting: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ExperimentPagePublic:
         """
@@ -672,12 +668,20 @@ class AsyncExperimentsClient:
 
         dataset_id : typing.Optional[str]
 
+        optimization_id : typing.Optional[str]
+
+        types : typing.Optional[str]
+
         name : typing.Optional[str]
 
         dataset_deleted : typing.Optional[bool]
 
         prompt_id : typing.Optional[str]
 
+        sorting : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -688,48 +692,27 @@ class AsyncExperimentsClient:
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
             await client.experiments.find_experiments()
-
-
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "datasetId": dataset_id,
-                "name": name,
-                "dataset_deleted": dataset_deleted,
-                "prompt_id": prompt_id,
-            },
+        _response = await self._raw_client.find_experiments(
+            page=page,
+            size=size,
+            dataset_id=dataset_id,
+            optimization_id=optimization_id,
+            types=types,
+            name=name,
+            dataset_deleted=dataset_deleted,
+            prompt_id=prompt_id,
+            sorting=sorting,
+            filters=filters,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentPagePublic,
-                    parse_obj_as(
-                        type_=ExperimentPagePublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def create_experiment(
         self,
@@ -737,11 +720,13 @@ class AsyncExperimentsClient:
         dataset_name: str,
         id: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
-        metadata: typing.Optional[JsonNodeWrite] = OMIT,
+        metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        type: typing.Optional[ExperimentWriteType] = OMIT,
+        optimization_id: typing.Optional[str] = OMIT,
+        status: typing.Optional[ExperimentWriteStatus] = OMIT,
+        experiment_scores: typing.Optional[typing.Sequence[ExperimentScoreWrite]] = OMIT,
         prompt_version: typing.Optional[PromptVersionLinkWrite] = OMIT,
-        prompt_versions: typing.Optional[
-            typing.Sequence[PromptVersionLinkWrite]
-        ] = OMIT,
+        prompt_versions: typing.Optional[typing.Sequence[PromptVersionLinkWrite]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -755,7 +740,15 @@ class AsyncExperimentsClient:
 
         name : typing.Optional[str]
 
-        metadata : typing.Optional[JsonNodeWrite]
+        metadata : typing.Optional[JsonListStringWrite]
+
+        type : typing.Optional[ExperimentWriteType]
+
+        optimization_id : typing.Optional[str]
+
+        status : typing.Optional[ExperimentWriteStatus]
+
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScoreWrite]]
 
         prompt_version : typing.Optional[PromptVersionLinkWrite]
 
@@ -770,56 +763,27 @@ class AsyncExperimentsClient:
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.experiments.create_experiment(
-                dataset_name="dataset_name",
-            )
-
-
+            await client.experiments.create_experiment(dataset_name='dataset_name', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments",
-            method="POST",
-            json={
-                "id": id,
-                "dataset_name": dataset_name,
-                "name": name,
-                "metadata": metadata,
-                "prompt_version": convert_and_respect_annotation_metadata(
-                    object_=prompt_version,
-                    annotation=PromptVersionLinkWrite,
-                    direction="write",
-                ),
-                "prompt_versions": convert_and_respect_annotation_metadata(
-                    object_=prompt_versions,
-                    annotation=typing.Sequence[PromptVersionLinkWrite],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.create_experiment(
+            dataset_name=dataset_name,
+            id=id,
+            name=name,
+            metadata=metadata,
+            type=type,
+            optimization_id=optimization_id,
+            status=status,
+            experiment_scores=experiment_scores,
+            prompt_version=prompt_version,
+            prompt_versions=prompt_versions,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def create_experiment_items(
         self,
@@ -843,59 +807,21 @@ class AsyncExperimentsClient:
 
         Examples
         --------
+        from Opik import AsyncOpikApi
+        from Opik import ExperimentItem
         import asyncio
-
-        from Opik import AsyncOpikApi, ExperimentItem
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.experiments.create_experiment_items(
-                experiment_items=[
-                    ExperimentItem(
-                        experiment_id="experiment_id",
-                        dataset_item_id="dataset_item_id",
-                        trace_id="trace_id",
-                    )
-                ],
-            )
-
-
+            await client.experiments.create_experiment_items(experiment_items=[ExperimentItem(experiment_id='experiment_id', dataset_item_id='dataset_item_id', trace_id='trace_id', )], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/items",
-            method="POST",
-            json={
-                "experiment_items": convert_and_respect_annotation_metadata(
-                    object_=experiment_items,
-                    annotation=typing.Sequence[ExperimentItem],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.create_experiment_items(
+            experiment_items=experiment_items, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def delete_experiment_items(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete experiment items
@@ -913,49 +839,18 @@ class AsyncExperimentsClient:
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.experiments.delete_experiment_items(
-                ids=["ids"],
-            )
-
-
+            await client.experiments.delete_experiment_items(ids=['ids'], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/items/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.delete_experiment_items(ids=ids, request_options=request_options)
+        return _response.data
 
     async def delete_experiments_by_id(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete experiments by id
@@ -973,49 +868,67 @@ class AsyncExperimentsClient:
 
         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.experiments.delete_experiments_by_id(ids=['ids'], )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_experiments_by_id(ids=ids, request_options=request_options)
+        return _response.data
 
-        from Opik import AsyncOpikApi
+    async def experiment_items_bulk(
+        self,
+        *,
+        experiment_name: str,
+        dataset_name: str,
+        items: typing.Sequence[ExperimentItemBulkRecordExperimentItemBulkWriteView],
+        experiment_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Record experiment items in bulk with traces, spans, and feedback scores. Maximum request size is 4MB.
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        Parameters
+        ----------
+        experiment_name : str
 
+        dataset_name : str
 
-        async def main() -> None:
-            await client.experiments.delete_experiments_by_id(
-                ids=["ids"],
-            )
+        items : typing.Sequence[ExperimentItemBulkRecordExperimentItemBulkWriteView]
 
+        experiment_id : typing.Optional[str]
+            Optional experiment ID. If provided, items will be added to the existing experiment and experimentName will be ignored. If not provided or experiment with that ID doesn't exist, a new experiment will be created with the given experimentName
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
 
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        from Opik import ExperimentItemBulkRecordExperimentItemBulkWriteView
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.experiments.experiment_items_bulk(experiment_name='experiment_name', dataset_name='dataset_name', items=[ExperimentItemBulkRecordExperimentItemBulkWriteView(dataset_item_id='dataset_item_id', )], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.experiment_items_bulk(
+            experiment_name=experiment_name,
+            dataset_name=dataset_name,
+            items=items,
+            experiment_id=experiment_id,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def find_feedback_score_names(
-        self,
-        *,
-        experiment_ids: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, experiment_ids: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[str]:
         """
         Find Feedback Score names
@@ -1034,120 +947,144 @@ class AsyncExperimentsClient:
 
         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.experiments.find_feedback_score_names()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.find_feedback_score_names(
+            experiment_ids=experiment_ids, request_options=request_options
+        )
+        return _response.data
 
-        from Opik import AsyncOpikApi
+    async def find_experiment_groups(
+        self,
+        *,
+        groups: typing.Optional[str] = None,
+        types: typing.Optional[str] = None,
+        name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ExperimentGroupResponse:
+        """
+        Find experiments grouped by specified fields
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        Parameters
+        ----------
+        groups : typing.Optional[str]
 
+        types : typing.Optional[str]
 
-        async def main() -> None:
-            await client.experiments.find_feedback_score_names()
+        name : typing.Optional[str]
 
+        filters : typing.Optional[str]
 
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ExperimentGroupResponse
+            Experiment groups
+
+        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.experiments.find_experiment_groups()
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/feedback-scores/names",
-            method="GET",
-            params={
-                "experiment_ids": experiment_ids,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.find_experiment_groups(
+            groups=groups, types=types, name=name, filters=filters, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    parse_obj_as(
-                        type_=typing.List[str],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def get_experiment_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> ExperimentPublic:
+    async def find_experiment_groups_aggregations(
+        self,
+        *,
+        groups: typing.Optional[str] = None,
+        types: typing.Optional[str] = None,
+        name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> ExperimentGroupAggregationsResponse:
         """
-        Get experiment by id
+        Find experiments grouped by specified fields with aggregation metrics
 
         Parameters
         ----------
-        id : str
+        groups : typing.Optional[str]
+
+        types : typing.Optional[str]
+
+        name : typing.Optional[str]
+
+        filters : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        ExperimentPublic
-            Experiment resource
+        ExperimentGroupAggregationsResponse
+            Experiment groups with aggregations
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.experiments.find_experiment_groups_aggregations()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.find_experiment_groups_aggregations(
+            groups=groups, types=types, name=name, filters=filters, request_options=request_options
         )
+        return _response.data
 
+    async def finish_experiments(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Finish experiments and trigger alert events
 
-        async def main() -> None:
-            await client.experiments.get_experiment_by_id(
-                id="id",
-            )
+        Parameters
+        ----------
+        ids : 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.experiments.finish_experiments(ids=['ids'], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/experiments/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentPublic,
-                    parse_obj_as(
-                        type_=ExperimentPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    async def get_experiment_by_name(
-        self, *, name: str, request_options: typing.Optional[RequestOptions] = None
+        _response = await self._raw_client.finish_experiments(ids=ids, request_options=request_options)
+        return _response.data
+
+    async def get_experiment_by_id(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> ExperimentPublic:
         """
-        Get experiment by name
+        Get experiment by id
 
         Parameters
         ----------
-        name : str
+        id : str
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1155,63 +1092,75 @@ class AsyncExperimentsClient:
         Returns
         -------
         ExperimentPublic
-            Experiments resource
+            Experiment 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.experiments.get_experiment_by_id(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_experiment_by_id(id, request_options=request_options)
+        return _response.data
 
-        from Opik import AsyncOpikApi
+    async def update_experiment(
+        self,
+        id: str,
+        *,
+        name: typing.Optional[str] = OMIT,
+        metadata: typing.Optional[JsonNode] = OMIT,
+        type: typing.Optional[ExperimentUpdateType] = OMIT,
+        status: typing.Optional[ExperimentUpdateStatus] = OMIT,
+        experiment_scores: typing.Optional[typing.Sequence[ExperimentScore]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update experiment by id
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        Parameters
+        ----------
+        id : str
+
+        name : typing.Optional[str]
 
+        metadata : typing.Optional[JsonNode]
 
-        async def main() -> None:
-            await client.experiments.get_experiment_by_name(
-                name="name",
-            )
+        type : typing.Optional[ExperimentUpdateType]
+
+        status : typing.Optional[ExperimentUpdateStatus]
+            The status of the experiment
+
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScore]]
+
+        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.experiments.update_experiment(id='id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/experiments/retrieve",
-            method="POST",
-            json={
-                "name": name,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.update_experiment(
+            id,
+            name=name,
+            metadata=metadata,
+            type=type,
+            status=status,
+            experiment_scores=experiment_scores,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentPublic,
-                    parse_obj_as(
-                        type_=ExperimentPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def get_experiment_item_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -1233,52 +1182,15 @@ class AsyncExperimentsClient:
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.experiments.get_experiment_item_by_id(
-                id="id",
-            )
-
-
+            await client.experiments.get_experiment_item_by_id(id='id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/experiments/items/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ExperimentItemPublic,
-                    parse_obj_as(
-                        type_=ExperimentItemPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.get_experiment_item_by_id(id, request_options=request_options)
+        return _response.data
 
     async def stream_experiment_items(
         self,
@@ -1306,38 +1218,50 @@ class AsyncExperimentsClient:
         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.
 
-        Yields
-        ------
+        Returns
+        -------
         typing.AsyncIterator[bytes]
             Experiment items stream or error during process
         """
-        async with self._client_wrapper.httpx_client.stream(
-            "v1/private/experiments/items/stream",
-            method="POST",
-            json={
-                "experiment_name": experiment_name,
-                "limit": limit,
-                "last_retrieved_id": last_retrieved_id,
-                "truncate": truncate,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        async with self._raw_client.stream_experiment_items(
+            experiment_name=experiment_name,
+            limit=limit,
+            last_retrieved_id=last_retrieved_id,
+            truncate=truncate,
             request_options=request_options,
-            omit=OMIT,
-        ) as _response:
-            try:
-                if 200 <= _response.status_code < 300:
-                    _chunk_size = (
-                        request_options.get("chunk_size", None)
-                        if request_options is not None
-                        else None
-                    )
-                    async for _chunk in _response.aiter_bytes(chunk_size=_chunk_size):
-                        yield _chunk
-                    return
-                await _response.aread()
-                _response_json = _response.json()
-            except JSONDecodeError:
-                raise ApiError(status_code=_response.status_code, body=_response.text)
-            raise ApiError(status_code=_response.status_code, body=_response_json)
+        ) as r:
+            async for data in r.data:
+                yield data
+
+    async def stream_experiments(
+        self,
+        *,
+        name: str,
+        limit: typing.Optional[int] = OMIT,
+        last_retrieved_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.AsyncIterator[bytes]:
+        """
+        Stream experiments
+
+        Parameters
+        ----------
+        name : str
+
+        limit : typing.Optional[int]
+
+        last_retrieved_id : 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.
+
+        Returns
+        -------
+        typing.AsyncIterator[bytes]
+            Experiments stream or error during process
+        """
+        async with self._raw_client.stream_experiments(
+            name=name, limit=limit, last_retrieved_id=last_retrieved_id, request_options=request_options
+        ) as r:
+            async for data in r.data:
+                yield data