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

opik/rest_api/optimizations/client.py

@@ -256,59 +256,6 @@ 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:
@@ -373,33 +320,6 @@ 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:
@@ -683,62 +603,6 @@ 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:
@@ -809,36 +673,6 @@ class AsyncOptimizationsClient:
         )
         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:

opik/rest_api/optimizations/raw_client.py

@@ -327,81 +327,6 @@ class RawOptimizationsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    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,
-    ) -> HttpResponse[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
-        -------
-        HttpResponse[OptimizationPagePublic]
-            Studio optimizations resource
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/optimizations/studio",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "dataset_id": dataset_id,
-                "name": name,
-                "dataset_deleted": dataset_deleted,
-                "filters": filters,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    OptimizationPagePublic,
-                    parse_obj_as(
-                        type_=OptimizationPagePublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=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, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
     def get_optimization_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[OptimizationPublic]:
@@ -498,55 +423,6 @@ class RawOptimizationsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    def get_studio_optimization_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> HttpResponse[OptimizationPublic]:
-        """
-        Get Studio optimization with config included
-
-        Parameters
-        ----------
-        id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        HttpResponse[OptimizationPublic]
-            Studio optimization resource
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/optimizations/studio/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    OptimizationPublic,
-                    parse_obj_as(
-                        type_=OptimizationPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return HttpResponse(response=_response, data=_data)
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=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, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
     def get_studio_optimization_logs(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> HttpResponse[OptimizationStudioLog]:
@@ -898,81 +774,6 @@ class AsyncRawOptimizationsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    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,
-    ) -> AsyncHttpResponse[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
-        -------
-        AsyncHttpResponse[OptimizationPagePublic]
-            Studio optimizations resource
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/optimizations/studio",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "dataset_id": dataset_id,
-                "name": name,
-                "dataset_deleted": dataset_deleted,
-                "filters": filters,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    OptimizationPagePublic,
-                    parse_obj_as(
-                        type_=OptimizationPagePublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    headers=dict(_response.headers),
-                    body=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, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
     async def get_optimization_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[OptimizationPublic]:
@@ -1069,55 +870,6 @@ class AsyncRawOptimizationsClient:
             raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response.text)
         raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
 
-    async def get_studio_optimization_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[OptimizationPublic]:
-        """
-        Get Studio optimization with config included
-
-        Parameters
-        ----------
-        id : str
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        AsyncHttpResponse[OptimizationPublic]
-            Studio optimization resource
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/optimizations/studio/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                _data = typing.cast(
-                    OptimizationPublic,
-                    parse_obj_as(
-                        type_=OptimizationPublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-                return AsyncHttpResponse(response=_response, data=_data)
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    headers=dict(_response.headers),
-                    body=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, headers=dict(_response.headers), body=_response.text)
-        raise ApiError(status_code=_response.status_code, headers=dict(_response.headers), body=_response_json)
-
     async def get_studio_optimization_logs(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> AsyncHttpResponse[OptimizationStudioLog]:

opik/rest_api/projects/client.py

@@ -11,6 +11,7 @@ from ..types.project_metric_response_public import ProjectMetricResponsePublic
 from ..types.project_page_public import ProjectPagePublic
 from ..types.project_public import ProjectPublic
 from ..types.project_stats_summary import ProjectStatsSummary
+from ..types.span_filter_public import SpanFilterPublic
 from ..types.trace_filter_public import TraceFilterPublic
 from ..types.trace_thread_filter_public import TraceThreadFilterPublic
 from .raw_client import AsyncRawProjectsClient, RawProjectsClient
@@ -268,6 +269,7 @@ class ProjectsClient:
         interval: typing.Optional[ProjectMetricRequestPublicInterval] = OMIT,
         interval_start: typing.Optional[dt.datetime] = OMIT,
         interval_end: typing.Optional[dt.datetime] = OMIT,
+        span_filters: typing.Optional[typing.Sequence[SpanFilterPublic]] = OMIT,
         trace_filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
         thread_filters: typing.Optional[typing.Sequence[TraceThreadFilterPublic]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -287,6 +289,8 @@ class ProjectsClient:
 
         interval_end : typing.Optional[dt.datetime]
 
+        span_filters : typing.Optional[typing.Sequence[SpanFilterPublic]]
+
         trace_filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
 
         thread_filters : typing.Optional[typing.Sequence[TraceThreadFilterPublic]]
@@ -311,6 +315,7 @@ class ProjectsClient:
             interval=interval,
             interval_start=interval_start,
             interval_end=interval_end,
+            span_filters=span_filters,
             trace_filters=trace_filters,
             thread_filters=thread_filters,
             request_options=request_options,
@@ -654,6 +659,7 @@ class AsyncProjectsClient:
         interval: typing.Optional[ProjectMetricRequestPublicInterval] = OMIT,
         interval_start: typing.Optional[dt.datetime] = OMIT,
         interval_end: typing.Optional[dt.datetime] = OMIT,
+        span_filters: typing.Optional[typing.Sequence[SpanFilterPublic]] = OMIT,
         trace_filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
         thread_filters: typing.Optional[typing.Sequence[TraceThreadFilterPublic]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -673,6 +679,8 @@ class AsyncProjectsClient:
 
         interval_end : typing.Optional[dt.datetime]
 
+        span_filters : typing.Optional[typing.Sequence[SpanFilterPublic]]
+
         trace_filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
 
         thread_filters : typing.Optional[typing.Sequence[TraceThreadFilterPublic]]
@@ -700,6 +708,7 @@ class AsyncProjectsClient:
             interval=interval,
             interval_start=interval_start,
             interval_end=interval_end,
+            span_filters=span_filters,
             trace_filters=trace_filters,
             thread_filters=thread_filters,
             request_options=request_options,

opik/rest_api/projects/raw_client.py

@@ -21,6 +21,7 @@ from ..types.project_metric_response_public import ProjectMetricResponsePublic
 from ..types.project_page_public import ProjectPagePublic
 from ..types.project_public import ProjectPublic
 from ..types.project_stats_summary import ProjectStatsSummary
+from ..types.span_filter_public import SpanFilterPublic
 from ..types.trace_filter_public import TraceFilterPublic
 from ..types.trace_thread_filter_public import TraceThreadFilterPublic
 from .types.project_metric_request_public_interval import ProjectMetricRequestPublicInterval
@@ -400,6 +401,7 @@ class RawProjectsClient:
         interval: typing.Optional[ProjectMetricRequestPublicInterval] = OMIT,
         interval_start: typing.Optional[dt.datetime] = OMIT,
         interval_end: typing.Optional[dt.datetime] = OMIT,
+        span_filters: typing.Optional[typing.Sequence[SpanFilterPublic]] = OMIT,
         trace_filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
         thread_filters: typing.Optional[typing.Sequence[TraceThreadFilterPublic]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -419,6 +421,8 @@ class RawProjectsClient:
 
         interval_end : typing.Optional[dt.datetime]
 
+        span_filters : typing.Optional[typing.Sequence[SpanFilterPublic]]
+
         trace_filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
 
         thread_filters : typing.Optional[typing.Sequence[TraceThreadFilterPublic]]
@@ -439,6 +443,9 @@ class RawProjectsClient:
                 "interval": interval,
                 "interval_start": interval_start,
                 "interval_end": interval_end,
+                "span_filters": convert_and_respect_annotation_metadata(
+                    object_=span_filters, annotation=typing.Sequence[SpanFilterPublic], direction="write"
+                ),
                 "trace_filters": convert_and_respect_annotation_metadata(
                     object_=trace_filters, annotation=typing.Sequence[TraceFilterPublic], direction="write"
                 ),
@@ -992,6 +999,7 @@ class AsyncRawProjectsClient:
         interval: typing.Optional[ProjectMetricRequestPublicInterval] = OMIT,
         interval_start: typing.Optional[dt.datetime] = OMIT,
         interval_end: typing.Optional[dt.datetime] = OMIT,
+        span_filters: typing.Optional[typing.Sequence[SpanFilterPublic]] = OMIT,
         trace_filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
         thread_filters: typing.Optional[typing.Sequence[TraceThreadFilterPublic]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
@@ -1011,6 +1019,8 @@ class AsyncRawProjectsClient:
 
         interval_end : typing.Optional[dt.datetime]
 
+        span_filters : typing.Optional[typing.Sequence[SpanFilterPublic]]
+
         trace_filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
 
         thread_filters : typing.Optional[typing.Sequence[TraceThreadFilterPublic]]
@@ -1031,6 +1041,9 @@ class AsyncRawProjectsClient:
                 "interval": interval,
                 "interval_start": interval_start,
                 "interval_end": interval_end,
+                "span_filters": convert_and_respect_annotation_metadata(
+                    object_=span_filters, annotation=typing.Sequence[SpanFilterPublic], direction="write"
+                ),
                 "trace_filters": convert_and_respect_annotation_metadata(
                     object_=trace_filters, annotation=typing.Sequence[TraceFilterPublic], direction="write"
                 ),

opik/rest_api/projects/types/project_metric_request_public_metric_type.py

@@ -13,6 +13,10 @@ ProjectMetricRequestPublicMetricType = typing.Union[
         "THREAD_COUNT",
         "THREAD_DURATION",
         "THREAD_FEEDBACK_SCORES",
+        "SPAN_FEEDBACK_SCORES",
+        "SPAN_COUNT",
+        "SPAN_DURATION",
+        "SPAN_TOKEN_USAGE",
     ],
     typing.Any,
 ]