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

opik/rest_api/projects/client.py

@@ -1,30 +1,23 @@
 # This file was auto-generated by Fern from our API Definition.
 
+import datetime as dt
 import typing
-from ..core.client_wrapper import SyncClientWrapper
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..types.project_page_public import ProjectPagePublic
-from ..core.pydantic_utilities import parse_obj_as
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-from ..errors.bad_request_error import BadRequestError
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.project_public import ProjectPublic
-from ..core.jsonable_encoder import jsonable_encoder
-from ..errors.conflict_error import ConflictError
 from ..types.feedback_score_names import FeedbackScoreNames
-from .types.project_metric_request_public_metric_type import (
-    ProjectMetricRequestPublicMetricType,
-)
-from .types.project_metric_request_public_interval import (
-    ProjectMetricRequestPublicInterval,
-)
-import datetime as dt
+from ..types.project_detailed import ProjectDetailed
 from ..types.project_metric_response_public import ProjectMetricResponsePublic
-from ..errors.not_found_error import NotFoundError
+from ..types.project_page_public import ProjectPagePublic
+from ..types.project_public import ProjectPublic
 from ..types.project_stats_summary import ProjectStatsSummary
-from ..types.project_detailed import ProjectDetailed
-from ..core.client_wrapper import AsyncClientWrapper
+from ..types.trace_filter_public import TraceFilterPublic
+from ..types.trace_thread_filter_public import TraceThreadFilterPublic
+from .raw_client import AsyncRawProjectsClient, RawProjectsClient
+from .types.project_metric_request_public_interval import ProjectMetricRequestPublicInterval
+from .types.project_metric_request_public_metric_type import ProjectMetricRequestPublicMetricType
+from .types.project_update_visibility import ProjectUpdateVisibility
+from .types.project_write_visibility import ProjectWriteVisibility
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -32,7 +25,18 @@ OMIT = typing.cast(typing.Any, ...)
 
 class ProjectsClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawProjectsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> RawProjectsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawProjectsClient
+        """
+        return self._raw_client
 
     def find_projects(
         self,
@@ -67,42 +71,19 @@ class ProjectsClient:
         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.projects.find_projects()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/projects",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "name": name,
-                "sorting": sorting,
-            },
-            request_options=request_options,
+        _response = self._raw_client.find_projects(
+            page=page, size=size, name=name, sorting=sorting, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectPagePublic,
-                    parse_obj_as(
-                        type_=ProjectPagePublic,  # 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_project(
         self,
         *,
         name: str,
+        visibility: typing.Optional[ProjectWriteVisibility] = OMIT,
         description: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -113,6 +94,8 @@ class ProjectsClient:
         ----------
         name : str
 
+        visibility : typing.Optional[ProjectWriteVisibility]
+
         description : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
@@ -125,59 +108,15 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.create_project(
-            name="name",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.create_project(name='name', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/projects",
-            method="POST",
-            json={
-                "name": name,
-                "description": description,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.create_project(
+            name=name, visibility=visibility, description=description, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    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_project_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> ProjectPublic:
+        return _response.data
+
+    def get_project_by_id(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> ProjectPublic:
         """
         Get project by id
 
@@ -196,37 +135,13 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.get_project_by_id(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.get_project_by_id(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectPublic,
-                    parse_obj_as(
-                        type_=ProjectPublic,  # 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 delete_project_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> None:
+        _response = self._raw_client.get_project_by_id(id, request_options=request_options)
+        return _response.data
+
+    def delete_project_by_id(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
         Delete project by id
 
@@ -244,37 +159,11 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.delete_project_by_id(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.delete_project_by_id(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 409:
-                raise ConflictError(
-                    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.delete_project_by_id(id, request_options=request_options)
+        return _response.data
 
     def update_project(
         self,
@@ -282,6 +171,7 @@ class ProjectsClient:
         *,
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
+        visibility: typing.Optional[ProjectUpdateVisibility] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -295,6 +185,8 @@ class ProjectsClient:
 
         description : typing.Optional[str]
 
+        visibility : typing.Optional[ProjectUpdateVisibility]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -305,61 +197,16 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.update_project(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.update_project(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}",
-            method="PATCH",
-            json={
-                "name": name,
-                "description": description,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.update_project(
+            id, name=name, description=description, visibility=visibility, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    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 delete_projects_batch(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete projects batch
@@ -378,37 +225,14 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.delete_projects_batch(
-            ids=["ids"],
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.delete_projects_batch(ids=['ids'], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/projects/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
-            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_projects_batch(ids=ids, request_options=request_options)
+        return _response.data
 
     def find_feedback_score_names_by_project_ids(
-        self,
-        *,
-        project_ids: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, project_ids: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> FeedbackScoreNames:
         """
         Find Feedback Score names By Project Ids
@@ -428,34 +252,13 @@ class ProjectsClient:
         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.projects.find_feedback_score_names_by_project_ids()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/projects/feedback-scores/names",
-            method="GET",
-            params={
-                "project_ids": project_ids,
-            },
-            request_options=request_options,
+        _response = self._raw_client.find_feedback_score_names_by_project_ids(
+            project_ids=project_ids, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    FeedbackScoreNames,
-                    parse_obj_as(
-                        type_=FeedbackScoreNames,  # 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_project_metrics(
         self,
@@ -465,6 +268,8 @@ class ProjectsClient:
         interval: typing.Optional[ProjectMetricRequestPublicInterval] = OMIT,
         interval_start: typing.Optional[dt.datetime] = OMIT,
         interval_end: typing.Optional[dt.datetime] = OMIT,
+        trace_filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
+        thread_filters: typing.Optional[typing.Sequence[TraceThreadFilterPublic]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ProjectMetricResponsePublic:
         """
@@ -482,6 +287,10 @@ class ProjectsClient:
 
         interval_end : typing.Optional[dt.datetime]
 
+        trace_filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
+
+        thread_filters : typing.Optional[typing.Sequence[TraceThreadFilterPublic]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -493,63 +302,20 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.get_project_metrics(
-            id="id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}/metrics",
-            method="POST",
-            json={
-                "metric_type": metric_type,
-                "interval": interval,
-                "interval_start": interval_start,
-                "interval_end": interval_end,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.get_project_metrics(id='id', )
+        """
+        _response = self._raw_client.get_project_metrics(
+            id,
+            metric_type=metric_type,
+            interval=interval,
+            interval_start=interval_start,
+            interval_end=interval_end,
+            trace_filters=trace_filters,
+            thread_filters=thread_filters,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectMetricResponsePublic,
-                    parse_obj_as(
-                        type_=ProjectMetricResponsePublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # 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_project_stats(
         self,
@@ -584,37 +350,13 @@ class ProjectsClient:
         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.projects.get_project_stats()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/projects/stats",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "name": name,
-                "sorting": sorting,
-            },
-            request_options=request_options,
+        _response = self._raw_client.get_project_stats(
+            page=page, size=size, name=name, sorting=sorting, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectStatsSummary,
-                    parse_obj_as(
-                        type_=ProjectStatsSummary,  # 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 retrieve_project(
         self, *, name: str, request_options: typing.Optional[RequestOptions] = None
@@ -637,75 +379,27 @@ class ProjectsClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.projects.retrieve_project(
-            name="name",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.projects.retrieve_project(name='name', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/projects/retrieve",
-            method="POST",
-            json={
-                "name": name,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectDetailed,
-                    parse_obj_as(
-                        type_=ProjectDetailed,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # 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(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    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.retrieve_project(name=name, request_options=request_options)
+        return _response.data
 
 
 class AsyncProjectsClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = AsyncRawProjectsClient(client_wrapper=client_wrapper)
+
+    @property
+    def with_raw_response(self) -> AsyncRawProjectsClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        AsyncRawProjectsClient
+        """
+        return self._raw_client
 
     async def find_projects(
         self,
@@ -739,51 +433,23 @@ class AsyncProjectsClient:
 
         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.projects.find_projects()
-
-
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/projects",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "name": name,
-                "sorting": sorting,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.find_projects(
+            page=page, size=size, name=name, sorting=sorting, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectPagePublic,
-                    parse_obj_as(
-                        type_=ProjectPagePublic,  # 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_project(
         self,
         *,
         name: str,
+        visibility: typing.Optional[ProjectWriteVisibility] = OMIT,
         description: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -794,6 +460,8 @@ class AsyncProjectsClient:
         ----------
         name : str
 
+        visibility : typing.Optional[ProjectWriteVisibility]
+
         description : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
@@ -805,64 +473,17 @@ class AsyncProjectsClient:
 
         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.projects.create_project(
-                name="name",
-            )
-
-
+            await client.projects.create_project(name='name', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/projects",
-            method="POST",
-            json={
-                "name": name,
-                "description": description,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.create_project(
+            name=name, visibility=visibility, description=description, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    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_project_by_id(
         self, id: str, *, request_options: typing.Optional[RequestOptions] = None
@@ -884,46 +505,17 @@ class AsyncProjectsClient:
 
         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.projects.get_project_by_id(
-                id="id",
-            )
-
-
+            await client.projects.get_project_by_id(id='id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectPublic,
-                    parse_obj_as(
-                        type_=ProjectPublic,  # 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 delete_project_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> None:
+        _response = await self._raw_client.get_project_by_id(id, request_options=request_options)
+        return _response.data
+
+    async def delete_project_by_id(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
         Delete project by id
 
@@ -940,46 +532,15 @@ class AsyncProjectsClient:
 
         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.projects.delete_project_by_id(
-                id="id",
-            )
-
-
+            await client.projects.delete_project_by_id(id='id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 409:
-                raise ConflictError(
-                    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.delete_project_by_id(id, request_options=request_options)
+        return _response.data
 
     async def update_project(
         self,
@@ -987,6 +548,7 @@ class AsyncProjectsClient:
         *,
         name: typing.Optional[str] = OMIT,
         description: typing.Optional[str] = OMIT,
+        visibility: typing.Optional[ProjectUpdateVisibility] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -1000,6 +562,8 @@ class AsyncProjectsClient:
 
         description : typing.Optional[str]
 
+        visibility : typing.Optional[ProjectUpdateVisibility]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1009,70 +573,20 @@ class AsyncProjectsClient:
 
         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.projects.update_project(
-                id="id",
-            )
-
-
+            await client.projects.update_project(id='id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}",
-            method="PATCH",
-            json={
-                "name": name,
-                "description": description,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.update_project(
+            id, name=name, description=description, visibility=visibility, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    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 delete_projects_batch(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete projects batch
@@ -1090,46 +604,18 @@ class AsyncProjectsClient:
 
         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.projects.delete_projects_batch(
-                ids=["ids"],
-            )
-
-
+            await client.projects.delete_projects_batch(ids=['ids'], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/projects/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
-            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_projects_batch(ids=ids, request_options=request_options)
+        return _response.data
 
     async def find_feedback_score_names_by_project_ids(
-        self,
-        *,
-        project_ids: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, project_ids: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> FeedbackScoreNames:
         """
         Find Feedback Score names By Project Ids
@@ -1148,43 +634,17 @@ class AsyncProjectsClient:
 
         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.projects.find_feedback_score_names_by_project_ids()
-
-
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/projects/feedback-scores/names",
-            method="GET",
-            params={
-                "project_ids": project_ids,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.find_feedback_score_names_by_project_ids(
+            project_ids=project_ids, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    FeedbackScoreNames,
-                    parse_obj_as(
-                        type_=FeedbackScoreNames,  # 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_project_metrics(
         self,
@@ -1194,6 +654,8 @@ class AsyncProjectsClient:
         interval: typing.Optional[ProjectMetricRequestPublicInterval] = OMIT,
         interval_start: typing.Optional[dt.datetime] = OMIT,
         interval_end: typing.Optional[dt.datetime] = OMIT,
+        trace_filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
+        thread_filters: typing.Optional[typing.Sequence[TraceThreadFilterPublic]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ProjectMetricResponsePublic:
         """
@@ -1211,6 +673,10 @@ class AsyncProjectsClient:
 
         interval_end : typing.Optional[dt.datetime]
 
+        trace_filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
+
+        thread_filters : typing.Optional[typing.Sequence[TraceThreadFilterPublic]]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1221,72 +687,24 @@ class AsyncProjectsClient:
 
         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.projects.get_project_metrics(
-                id="id",
-            )
-
-
+            await client.projects.get_project_metrics(id='id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/projects/{jsonable_encoder(id)}/metrics",
-            method="POST",
-            json={
-                "metric_type": metric_type,
-                "interval": interval,
-                "interval_start": interval_start,
-                "interval_end": interval_end,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.get_project_metrics(
+            id,
+            metric_type=metric_type,
+            interval=interval,
+            interval_start=interval_start,
+            interval_end=interval_end,
+            trace_filters=trace_filters,
+            thread_filters=thread_filters,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectMetricResponsePublic,
-                    parse_obj_as(
-                        type_=ProjectMetricResponsePublic,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # 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_project_stats(
         self,
@@ -1320,46 +738,17 @@ class AsyncProjectsClient:
 
         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.projects.get_project_stats()
-
-
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/projects/stats",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "name": name,
-                "sorting": sorting,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.get_project_stats(
+            page=page, size=size, name=name, sorting=sorting, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectStatsSummary,
-                    parse_obj_as(
-                        type_=ProjectStatsSummary,  # 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 retrieve_project(
         self, *, name: str, request_options: typing.Optional[RequestOptions] = None
@@ -1381,76 +770,12 @@ class AsyncProjectsClient:
 
         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.projects.retrieve_project(
-                name="name",
-            )
-
-
+            await client.projects.retrieve_project(name='name', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/projects/retrieve",
-            method="POST",
-            json={
-                "name": name,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectDetailed,
-                    parse_obj_as(
-                        type_=ProjectDetailed,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 400:
-                raise BadRequestError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # 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(),
-                        ),
-                    )
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    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.retrieve_project(name=name, request_options=request_options)
+        return _response.data