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

opik/rest_api/experiments/client.py

@@ -13,6 +13,8 @@ from ..types.experiment_item_bulk_record_experiment_item_bulk_write_view import
 from ..types.experiment_item_public import ExperimentItemPublic
 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
@@ -120,6 +122,7 @@ class ExperimentsClient:
         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,
         request_options: typing.Optional[RequestOptions] = None,
@@ -143,6 +146,8 @@ class ExperimentsClient:
 
         status : typing.Optional[ExperimentWriteStatus]
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScoreWrite]]
+
         prompt_version : typing.Optional[PromptVersionLinkWrite]
 
         prompt_versions : typing.Optional[typing.Sequence[PromptVersionLinkWrite]]
@@ -168,6 +173,7 @@ class ExperimentsClient:
             type=type,
             optimization_id=optimization_id,
             status=status,
+            experiment_scores=experiment_scores,
             prompt_version=prompt_version,
             prompt_versions=prompt_versions,
             request_options=request_options,
@@ -476,6 +482,7 @@ class ExperimentsClient:
         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:
         """
@@ -494,6 +501,8 @@ class ExperimentsClient:
         status : typing.Optional[ExperimentUpdateStatus]
             The status of the experiment
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScore]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -508,7 +517,13 @@ class ExperimentsClient:
         client.experiments.update_experiment(id='id', )
         """
         _response = self._raw_client.update_experiment(
-            id, name=name, metadata=metadata, type=type, status=status, request_options=request_options
+            id,
+            name=name,
+            metadata=metadata,
+            type=type,
+            status=status,
+            experiment_scores=experiment_scores,
+            request_options=request_options,
         )
         return _response.data
 
@@ -709,6 +724,7 @@ class AsyncExperimentsClient:
         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,
         request_options: typing.Optional[RequestOptions] = None,
@@ -732,6 +748,8 @@ class AsyncExperimentsClient:
 
         status : typing.Optional[ExperimentWriteStatus]
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScoreWrite]]
+
         prompt_version : typing.Optional[PromptVersionLinkWrite]
 
         prompt_versions : typing.Optional[typing.Sequence[PromptVersionLinkWrite]]
@@ -760,6 +778,7 @@ class AsyncExperimentsClient:
             type=type,
             optimization_id=optimization_id,
             status=status,
+            experiment_scores=experiment_scores,
             prompt_version=prompt_version,
             prompt_versions=prompt_versions,
             request_options=request_options,
@@ -1095,6 +1114,7 @@ class AsyncExperimentsClient:
         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:
         """
@@ -1113,6 +1133,8 @@ class AsyncExperimentsClient:
         status : typing.Optional[ExperimentUpdateStatus]
             The status of the experiment
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScore]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1130,7 +1152,13 @@ class AsyncExperimentsClient:
         asyncio.run(main())
         """
         _response = await self._raw_client.update_experiment(
-            id, name=name, metadata=metadata, type=type, status=status, request_options=request_options
+            id,
+            name=name,
+            metadata=metadata,
+            type=type,
+            status=status,
+            experiment_scores=experiment_scores,
+            request_options=request_options,
         )
         return _response.data
 

opik/rest_api/experiments/raw_client.py

@@ -24,6 +24,8 @@ from ..types.experiment_item_bulk_record_experiment_item_bulk_write_view import
 from ..types.experiment_item_public import ExperimentItemPublic
 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
@@ -141,6 +143,7 @@ class RawExperimentsClient:
         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,
         request_options: typing.Optional[RequestOptions] = None,
@@ -164,6 +167,8 @@ class RawExperimentsClient:
 
         status : typing.Optional[ExperimentWriteStatus]
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScoreWrite]]
+
         prompt_version : typing.Optional[PromptVersionLinkWrite]
 
         prompt_versions : typing.Optional[typing.Sequence[PromptVersionLinkWrite]]
@@ -188,6 +193,9 @@ class RawExperimentsClient:
                 "type": type,
                 "optimization_id": optimization_id,
                 "status": status,
+                "experiment_scores": convert_and_respect_annotation_metadata(
+                    object_=experiment_scores, annotation=typing.Sequence[ExperimentScoreWrite], direction="write"
+                ),
                 "prompt_version": convert_and_respect_annotation_metadata(
                     object_=prompt_version, annotation=PromptVersionLinkWrite, direction="write"
                 ),
@@ -695,6 +703,7 @@ class RawExperimentsClient:
         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,
     ) -> HttpResponse[None]:
         """
@@ -713,6 +722,8 @@ class RawExperimentsClient:
         status : typing.Optional[ExperimentUpdateStatus]
             The status of the experiment
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScore]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -728,6 +739,9 @@ class RawExperimentsClient:
                 "metadata": metadata,
                 "type": type,
                 "status": status,
+                "experiment_scores": convert_and_respect_annotation_metadata(
+                    object_=experiment_scores, annotation=typing.Sequence[ExperimentScore], direction="write"
+                ),
             },
             headers={
                 "content-type": "application/json",
@@ -1045,6 +1059,7 @@ class AsyncRawExperimentsClient:
         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,
         request_options: typing.Optional[RequestOptions] = None,
@@ -1068,6 +1083,8 @@ class AsyncRawExperimentsClient:
 
         status : typing.Optional[ExperimentWriteStatus]
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScoreWrite]]
+
         prompt_version : typing.Optional[PromptVersionLinkWrite]
 
         prompt_versions : typing.Optional[typing.Sequence[PromptVersionLinkWrite]]
@@ -1092,6 +1109,9 @@ class AsyncRawExperimentsClient:
                 "type": type,
                 "optimization_id": optimization_id,
                 "status": status,
+                "experiment_scores": convert_and_respect_annotation_metadata(
+                    object_=experiment_scores, annotation=typing.Sequence[ExperimentScoreWrite], direction="write"
+                ),
                 "prompt_version": convert_and_respect_annotation_metadata(
                     object_=prompt_version, annotation=PromptVersionLinkWrite, direction="write"
                 ),
@@ -1599,6 +1619,7 @@ class AsyncRawExperimentsClient:
         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,
     ) -> AsyncHttpResponse[None]:
         """
@@ -1617,6 +1638,8 @@ class AsyncRawExperimentsClient:
         status : typing.Optional[ExperimentUpdateStatus]
             The status of the experiment
 
+        experiment_scores : typing.Optional[typing.Sequence[ExperimentScore]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1632,6 +1655,9 @@ class AsyncRawExperimentsClient:
                 "metadata": metadata,
                 "type": type,
                 "status": status,
+                "experiment_scores": convert_and_respect_annotation_metadata(
+                    object_=experiment_scores, annotation=typing.Sequence[ExperimentScore], direction="write"
+                ),
             },
             headers={
                 "content-type": "application/json",

opik/rest_api/feedback_definitions/types/find_feedback_definitions_request_type.py

@@ -2,4 +2,4 @@
 
 import typing
 
-FindFeedbackDefinitionsRequestType = typing.Union[typing.Literal["numerical", "categorical"], typing.Any]
+FindFeedbackDefinitionsRequestType = typing.Union[typing.Literal["numerical", "categorical", "boolean"], typing.Any]

opik/rest_api/optimizations/client.py

@@ -8,6 +8,8 @@ from ..core.request_options import RequestOptions
 from ..types.json_list_string_write import JsonListStringWrite
 from ..types.optimization_page_public import OptimizationPagePublic
 from ..types.optimization_public import OptimizationPublic
+from ..types.optimization_studio_config_write import OptimizationStudioConfigWrite
+from ..types.optimization_studio_log import OptimizationStudioLog
 from ..types.optimization_write_status import OptimizationWriteStatus
 from .raw_client import AsyncRawOptimizationsClient, RawOptimizationsClient
 from .types.optimization_update_status import OptimizationUpdateStatus
@@ -31,6 +33,30 @@ class OptimizationsClient:
         """
         return self._raw_client
 
+    def cancel_studio_optimizations(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Cancel Studio optimizations by id
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.optimizations.cancel_studio_optimizations(id='id', )
+        """
+        _response = self._raw_client.cancel_studio_optimizations(id, request_options=request_options)
+        return _response.data
+
     def find_optimizations(
         self,
         *,
@@ -39,6 +65,7 @@ class OptimizationsClient:
         dataset_id: typing.Optional[str] = None,
         name: typing.Optional[str] = None,
         dataset_deleted: typing.Optional[bool] = None,
+        filters: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> OptimizationPagePublic:
         """
@@ -56,6 +83,8 @@ class OptimizationsClient:
 
         dataset_deleted : typing.Optional[bool]
 
+        filters : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -76,6 +105,7 @@ class OptimizationsClient:
             dataset_id=dataset_id,
             name=name,
             dataset_deleted=dataset_deleted,
+            filters=filters,
             request_options=request_options,
         )
         return _response.data
@@ -89,6 +119,7 @@ class OptimizationsClient:
         id: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
         metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        studio_config: typing.Optional[OptimizationStudioConfigWrite] = OMIT,
         last_updated_at: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -109,6 +140,8 @@ class OptimizationsClient:
 
         metadata : typing.Optional[JsonListStringWrite]
 
+        studio_config : typing.Optional[OptimizationStudioConfigWrite]
+
         last_updated_at : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
@@ -131,6 +164,7 @@ class OptimizationsClient:
             id=id,
             name=name,
             metadata=metadata,
+            studio_config=studio_config,
             last_updated_at=last_updated_at,
             request_options=request_options,
         )
@@ -145,6 +179,7 @@ class OptimizationsClient:
         id: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
         metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        studio_config: typing.Optional[OptimizationStudioConfigWrite] = OMIT,
         last_updated_at: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -165,6 +200,8 @@ class OptimizationsClient:
 
         metadata : typing.Optional[JsonListStringWrite]
 
+        studio_config : typing.Optional[OptimizationStudioConfigWrite]
+
         last_updated_at : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
@@ -187,6 +224,7 @@ class OptimizationsClient:
             id=id,
             name=name,
             metadata=metadata,
+            studio_config=studio_config,
             last_updated_at=last_updated_at,
             request_options=request_options,
         )
@@ -218,6 +256,59 @@ class OptimizationsClient:
         _response = self._raw_client.delete_optimizations_by_id(ids=ids, request_options=request_options)
         return _response.data
 
+    def find_studio_optimizations(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
+        dataset_id: typing.Optional[str] = None,
+        name: typing.Optional[str] = None,
+        dataset_deleted: typing.Optional[bool] = None,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> OptimizationPagePublic:
+        """
+        Find Studio optimizations
+
+        Parameters
+        ----------
+        page : typing.Optional[int]
+
+        size : typing.Optional[int]
+
+        dataset_id : typing.Optional[str]
+
+        name : typing.Optional[str]
+
+        dataset_deleted : typing.Optional[bool]
+
+        filters : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OptimizationPagePublic
+            Studio optimizations resource
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.optimizations.find_studio_optimizations()
+        """
+        _response = self._raw_client.find_studio_optimizations(
+            page=page,
+            size=size,
+            dataset_id=dataset_id,
+            name=name,
+            dataset_deleted=dataset_deleted,
+            filters=filters,
+            request_options=request_options,
+        )
+        return _response.data
+
     def get_optimization_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> OptimizationPublic:
@@ -282,6 +373,60 @@ class OptimizationsClient:
         )
         return _response.data
 
+    def get_studio_optimization_by_id(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OptimizationPublic:
+        """
+        Get Studio optimization with config included
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OptimizationPublic
+            Studio optimization resource
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.optimizations.get_studio_optimization_by_id(id='id', )
+        """
+        _response = self._raw_client.get_studio_optimization_by_id(id, request_options=request_options)
+        return _response.data
+
+    def get_studio_optimization_logs(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OptimizationStudioLog:
+        """
+        Get presigned S3 URL for downloading optimization logs
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OptimizationStudioLog
+            Logs response
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.optimizations.get_studio_optimization_logs(id='id', )
+        """
+        _response = self._raw_client.get_studio_optimization_logs(id, request_options=request_options)
+        return _response.data
+
 
 class AsyncOptimizationsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -298,6 +443,35 @@ class AsyncOptimizationsClient:
         """
         return self._raw_client
 
+    async def cancel_studio_optimizations(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Cancel Studio optimizations by id
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.optimizations.cancel_studio_optimizations(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.cancel_studio_optimizations(id, request_options=request_options)
+        return _response.data
+
     async def find_optimizations(
         self,
         *,
@@ -306,6 +480,7 @@ class AsyncOptimizationsClient:
         dataset_id: typing.Optional[str] = None,
         name: typing.Optional[str] = None,
         dataset_deleted: typing.Optional[bool] = None,
+        filters: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> OptimizationPagePublic:
         """
@@ -323,6 +498,8 @@ class AsyncOptimizationsClient:
 
         dataset_deleted : typing.Optional[bool]
 
+        filters : typing.Optional[str]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -346,6 +523,7 @@ class AsyncOptimizationsClient:
             dataset_id=dataset_id,
             name=name,
             dataset_deleted=dataset_deleted,
+            filters=filters,
             request_options=request_options,
         )
         return _response.data
@@ -359,6 +537,7 @@ class AsyncOptimizationsClient:
         id: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
         metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        studio_config: typing.Optional[OptimizationStudioConfigWrite] = OMIT,
         last_updated_at: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -379,6 +558,8 @@ class AsyncOptimizationsClient:
 
         metadata : typing.Optional[JsonListStringWrite]
 
+        studio_config : typing.Optional[OptimizationStudioConfigWrite]
+
         last_updated_at : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
@@ -404,6 +585,7 @@ class AsyncOptimizationsClient:
             id=id,
             name=name,
             metadata=metadata,
+            studio_config=studio_config,
             last_updated_at=last_updated_at,
             request_options=request_options,
         )
@@ -418,6 +600,7 @@ class AsyncOptimizationsClient:
         id: typing.Optional[str] = OMIT,
         name: typing.Optional[str] = OMIT,
         metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        studio_config: typing.Optional[OptimizationStudioConfigWrite] = OMIT,
         last_updated_at: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -438,6 +621,8 @@ class AsyncOptimizationsClient:
 
         metadata : typing.Optional[JsonListStringWrite]
 
+        studio_config : typing.Optional[OptimizationStudioConfigWrite]
+
         last_updated_at : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
@@ -463,6 +648,7 @@ class AsyncOptimizationsClient:
             id=id,
             name=name,
             metadata=metadata,
+            studio_config=studio_config,
             last_updated_at=last_updated_at,
             request_options=request_options,
         )
@@ -497,6 +683,62 @@ class AsyncOptimizationsClient:
         _response = await self._raw_client.delete_optimizations_by_id(ids=ids, request_options=request_options)
         return _response.data
 
+    async def find_studio_optimizations(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
+        dataset_id: typing.Optional[str] = None,
+        name: typing.Optional[str] = None,
+        dataset_deleted: typing.Optional[bool] = None,
+        filters: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> OptimizationPagePublic:
+        """
+        Find Studio optimizations
+
+        Parameters
+        ----------
+        page : typing.Optional[int]
+
+        size : typing.Optional[int]
+
+        dataset_id : typing.Optional[str]
+
+        name : typing.Optional[str]
+
+        dataset_deleted : typing.Optional[bool]
+
+        filters : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OptimizationPagePublic
+            Studio optimizations 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.optimizations.find_studio_optimizations()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.find_studio_optimizations(
+            page=page,
+            size=size,
+            dataset_id=dataset_id,
+            name=name,
+            dataset_deleted=dataset_deleted,
+            filters=filters,
+            request_options=request_options,
+        )
+        return _response.data
+
     async def get_optimization_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> OptimizationPublic:
@@ -566,3 +808,63 @@ class AsyncOptimizationsClient:
             id, name=name, status=status, request_options=request_options
         )
         return _response.data
+
+    async def get_studio_optimization_by_id(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OptimizationPublic:
+        """
+        Get Studio optimization with config included
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OptimizationPublic
+            Studio optimization 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.optimizations.get_studio_optimization_by_id(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_studio_optimization_by_id(id, request_options=request_options)
+        return _response.data
+
+    async def get_studio_optimization_logs(
+        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> OptimizationStudioLog:
+        """
+        Get presigned S3 URL for downloading optimization logs
+
+        Parameters
+        ----------
+        id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        OptimizationStudioLog
+            Logs response
+
+        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.optimizations.get_studio_optimization_logs(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_studio_optimization_logs(id, request_options=request_options)
+        return _response.data