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

opik/rest_api/traces/client.py

@@ -1,29 +1,30 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing
-from ..core.client_wrapper import SyncClientWrapper
 import datetime as dt
+import typing
+
+from ..core.client_wrapper import AsyncClientWrapper, SyncClientWrapper
 from ..core.request_options import RequestOptions
-from ..core.jsonable_encoder import jsonable_encoder
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
+from ..types.comment import Comment
+from ..types.error_info import ErrorInfo
+from ..types.error_info_write import ErrorInfoWrite
+from ..types.feedback_score_batch_item import FeedbackScoreBatchItem
+from ..types.feedback_score_batch_item_thread import FeedbackScoreBatchItemThread
 from ..types.feedback_score_source import FeedbackScoreSource
+from ..types.json_list_string import JsonListString
+from ..types.json_list_string_write import JsonListStringWrite
+from ..types.project_stats_public import ProjectStatsPublic
+from ..types.trace_filter_public import TraceFilterPublic
 from ..types.trace_page_public import TracePagePublic
-from ..core.pydantic_utilities import parse_obj_as
-from ..types.json_node_write import JsonNodeWrite
-from ..types.error_info_write import ErrorInfoWrite
-from ..core.serialization import convert_and_respect_annotation_metadata
-from ..types.trace_write import TraceWrite
 from ..types.trace_public import TracePublic
-from ..types.json_node import JsonNode
-from ..types.error_info import ErrorInfo
-from ..types.project_stats_public import ProjectStatsPublic
-from ..types.comment import Comment
-from ..errors.not_found_error import NotFoundError
 from ..types.trace_thread import TraceThread
+from ..types.trace_thread_filter import TraceThreadFilter
 from ..types.trace_thread_page import TraceThreadPage
-from ..types.feedback_score_batch_item import FeedbackScoreBatchItem
-from ..core.client_wrapper import AsyncClientWrapper
+from ..types.trace_thread_update import TraceThreadUpdate
+from ..types.trace_update import TraceUpdate
+from ..types.trace_write import TraceWrite
+from ..types.value_entry import ValueEntry
+from .raw_client import AsyncRawTracesClient, RawTracesClient
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -31,9 +32,20 @@ OMIT = typing.cast(typing.Any, ...)
 
 class TracesClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        self._raw_client = RawTracesClient(client_wrapper=client_wrapper)
 
-    def add_trace_comment(
+    @property
+    def with_raw_response(self) -> RawTracesClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
+
+        Returns
+        -------
+        RawTracesClient
+        """
+        return self._raw_client
+
+    def add_thread_comment(
         self,
         id_: str,
         *,
@@ -46,7 +58,7 @@ class TracesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Add trace comment
+        Add thread comment
 
         Parameters
         ----------
@@ -74,37 +86,76 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.add_trace_comment(
-            id_="id",
-            text="text",
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.add_thread_comment(id_='id', text='text', )
+        """
+        _response = self._raw_client.add_thread_comment(
+            id_,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
+            request_options=request_options,
         )
+        return _response.data
+
+    def add_trace_comment(
+        self,
+        id_: str,
+        *,
+        text: str,
+        id: typing.Optional[str] = OMIT,
+        created_at: typing.Optional[dt.datetime] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        created_by: typing.Optional[str] = OMIT,
+        last_updated_by: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id_)}/comments",
-            method="POST",
-            json={
-                "id": id,
-                "text": text,
-                "created_at": created_at,
-                "last_updated_at": last_updated_at,
-                "created_by": created_by,
-                "last_updated_by": last_updated_by,
-            },
+        Add trace comment
+
+        Parameters
+        ----------
+        id_ : str
+
+        text : str
+
+        id : typing.Optional[str]
+
+        created_at : typing.Optional[dt.datetime]
+
+        last_updated_at : typing.Optional[dt.datetime]
+
+        created_by : typing.Optional[str]
+
+        last_updated_by : typing.Optional[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.traces.add_trace_comment(id_='id', text='text', )
+        """
+        _response = self._raw_client.add_trace_comment(
+            id_,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def add_trace_feedback_score(
         self,
@@ -119,6 +170,7 @@ class TracesClient:
         last_updated_at: typing.Optional[dt.datetime] = OMIT,
         created_by: typing.Optional[str] = OMIT,
         last_updated_by: typing.Optional[str] = OMIT,
+        value_by_author: typing.Optional[typing.Dict[str, ValueEntry]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
@@ -146,6 +198,47 @@ class TracesClient:
 
         last_updated_by : typing.Optional[str]
 
+        value_by_author : typing.Optional[typing.Dict[str, ValueEntry]]
+
+        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.traces.add_trace_feedback_score(id='id', name='name', value=1.1, source="ui", )
+        """
+        _response = self._raw_client.add_trace_feedback_score(
+            id,
+            name=name,
+            value=value,
+            source=source,
+            category_name=category_name,
+            reason=reason,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
+            value_by_author=value_by_author,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def create_traces(
+        self, *, traces: typing.Sequence[TraceWrite], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Create traces
+
+        Parameters
+        ----------
+        traces : typing.Sequence[TraceWrite]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -156,42 +249,137 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
+        from Opik import TraceWrite
+        import datetime
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.create_traces(traces=[TraceWrite(start_time=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), )], )
+        """
+        _response = self._raw_client.create_traces(traces=traces, request_options=request_options)
+        return _response.data
+
+    def batch_update_traces(
+        self,
+        *,
+        ids: typing.Sequence[str],
+        update: TraceUpdate,
+        merge_tags: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update multiple traces
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+            List of trace IDs to update (max 1000)
+
+        update : TraceUpdate
+
+        merge_tags : typing.Optional[bool]
+            If true, merge tags with existing tags instead of replacing them. Default: false
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
 
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
+        Examples
+        --------
+        from Opik import OpikApi
+        from Opik import TraceUpdate
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.batch_update_traces(ids=['ids'], update=TraceUpdate(), )
+        """
+        _response = self._raw_client.batch_update_traces(
+            ids=ids, update=update, merge_tags=merge_tags, request_options=request_options
         )
-        client.traces.add_trace_feedback_score(
-            id="id",
-            name="name",
-            value=1.1,
-            source="ui",
+        return _response.data
+
+    def batch_update_threads(
+        self,
+        *,
+        ids: typing.Sequence[str],
+        update: TraceThreadUpdate,
+        merge_tags: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update multiple threads
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+            List of thread model IDs to update (max 1000)
+
+        update : TraceThreadUpdate
+
+        merge_tags : typing.Optional[bool]
+            If true, merge tags with existing tags instead of replacing them. Default: false
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        from Opik import TraceThreadUpdate
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.batch_update_threads(ids=['ids'], update=TraceThreadUpdate(), )
+        """
+        _response = self._raw_client.batch_update_threads(
+            ids=ids, update=update, merge_tags=merge_tags, request_options=request_options
         )
+        return _response.data
+
+    def close_trace_thread(
+        self,
+        *,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        thread_id: typing.Optional[str] = OMIT,
+        thread_ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}/feedback-scores",
-            method="PUT",
-            json={
-                "name": name,
-                "category_name": category_name,
-                "value": value,
-                "reason": reason,
-                "source": source,
-                "created_at": created_at,
-                "last_updated_at": last_updated_at,
-                "created_by": created_by,
-                "last_updated_by": last_updated_by,
-            },
+        Close one or multiple trace threads. Supports both single thread_id and multiple thread_ids for batch operations.
+
+        Parameters
+        ----------
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        thread_id : typing.Optional[str]
+
+        thread_ids : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.close_trace_thread()
+        """
+        _response = self._raw_client.close_trace_thread(
+            project_name=project_name,
+            project_id=project_id,
+            thread_id=thread_id,
+            thread_ids=thread_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)
+        return _response.data
 
     def get_traces_by_project(
         self,
@@ -202,7 +390,11 @@ class TracesClient:
         project_id: typing.Optional[str] = None,
         filters: typing.Optional[str] = None,
         truncate: typing.Optional[bool] = None,
+        strip_attachments: typing.Optional[bool] = None,
         sorting: typing.Optional[str] = None,
+        exclude: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TracePagePublic:
         """
@@ -222,8 +414,16 @@ class TracesClient:
 
         truncate : typing.Optional[bool]
 
+        strip_attachments : typing.Optional[bool]
+
         sorting : typing.Optional[str]
 
+        exclude : typing.Optional[str]
+
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -235,54 +435,39 @@ class TracesClient:
         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.traces.get_traces_by_project()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "project_name": project_name,
-                "project_id": project_id,
-                "filters": filters,
-                "truncate": truncate,
-                "sorting": sorting,
-            },
+        _response = self._raw_client.get_traces_by_project(
+            page=page,
+            size=size,
+            project_name=project_name,
+            project_id=project_id,
+            filters=filters,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            sorting=sorting,
+            exclude=exclude,
+            from_time=from_time,
+            to_time=to_time,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TracePagePublic,
-                    parse_obj_as(
-                        type_=TracePagePublic,  # 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_trace(
         self,
         *,
-        name: str,
         start_time: dt.datetime,
         id: typing.Optional[str] = OMIT,
         project_name: typing.Optional[str] = OMIT,
+        name: typing.Optional[str] = OMIT,
         end_time: typing.Optional[dt.datetime] = OMIT,
-        input: typing.Optional[JsonNodeWrite] = OMIT,
-        output: typing.Optional[JsonNodeWrite] = OMIT,
-        metadata: typing.Optional[JsonNodeWrite] = OMIT,
+        input: typing.Optional[JsonListStringWrite] = OMIT,
+        output: typing.Optional[JsonListStringWrite] = OMIT,
+        metadata: typing.Optional[JsonListStringWrite] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
         error_info: typing.Optional[ErrorInfoWrite] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
         thread_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
@@ -291,8 +476,6 @@ class TracesClient:
 
         Parameters
         ----------
-        name : str
-
         start_time : dt.datetime
 
         id : typing.Optional[str]
@@ -300,18 +483,22 @@ class TracesClient:
         project_name : typing.Optional[str]
             If null, the default project is used
 
+        name : typing.Optional[str]
+
         end_time : typing.Optional[dt.datetime]
 
-        input : typing.Optional[JsonNodeWrite]
+        input : typing.Optional[JsonListStringWrite]
 
-        output : typing.Optional[JsonNodeWrite]
+        output : typing.Optional[JsonListStringWrite]
 
-        metadata : typing.Optional[JsonNodeWrite]
+        metadata : typing.Optional[JsonListStringWrite]
 
         tags : typing.Optional[typing.Sequence[str]]
 
         error_info : typing.Optional[ErrorInfoWrite]
 
+        last_updated_at : typing.Optional[dt.datetime]
+
         thread_id : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
@@ -323,117 +510,34 @@ class TracesClient:
 
         Examples
         --------
-        import datetime
-
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.create_trace(
-            name="name",
-            start_time=datetime.datetime.fromisoformat(
-                "2024-01-15 09:30:00+00:00",
-            ),
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces",
-            method="POST",
-            json={
-                "id": id,
-                "project_name": project_name,
-                "name": name,
-                "start_time": start_time,
-                "end_time": end_time,
-                "input": input,
-                "output": output,
-                "metadata": metadata,
-                "tags": tags,
-                "error_info": convert_and_respect_annotation_metadata(
-                    object_=error_info, annotation=ErrorInfoWrite, direction="write"
-                ),
-                "thread_id": thread_id,
-            },
+        import datetime
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.create_trace(start_time=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), )
+        """
+        _response = self._raw_client.create_trace(
+            start_time=start_time,
+            id=id,
+            project_name=project_name,
+            name=name,
+            end_time=end_time,
+            input=input,
+            output=output,
+            metadata=metadata,
+            tags=tags,
+            error_info=error_info,
+            last_updated_at=last_updated_at,
+            thread_id=thread_id,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    def create_traces(
+    def get_trace_by_id(
         self,
+        id: str,
         *,
-        traces: typing.Sequence[TraceWrite],
+        strip_attachments: typing.Optional[bool] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
-        """
-        Create traces
-
-        Parameters
-        ----------
-        traces : typing.Sequence[TraceWrite]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        None
-
-        Examples
-        --------
-        import datetime
-
-        from Opik import OpikApi, TraceWrite
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.create_traces(
-            traces=[
-                TraceWrite(
-                    name="name",
-                    start_time=datetime.datetime.fromisoformat(
-                        "2024-01-15 09:30:00+00:00",
-                    ),
-                )
-            ],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/batch",
-            method="POST",
-            json={
-                "traces": convert_and_respect_annotation_metadata(
-                    object_=traces,
-                    annotation=typing.Sequence[TraceWrite],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def get_trace_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> TracePublic:
         """
         Get trace by id
@@ -442,6 +546,8 @@ class TracesClient:
         ----------
         id : str
 
+        strip_attachments : typing.Optional[bool]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -453,37 +559,15 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.get_trace_by_id(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.get_trace_by_id(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
+        _response = self._raw_client.get_trace_by_id(
+            id, strip_attachments=strip_attachments, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TracePublic,
-                    parse_obj_as(
-                        type_=TracePublic,  # 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_trace_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> None:
+        return _response.data
+
+    def delete_trace_by_id(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
         """
         Delete trace by id
 
@@ -501,27 +585,11 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.delete_trace_by_id(
-            id="id",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.delete_trace_by_id(id='id', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}",
-            method="DELETE",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.delete_trace_by_id(id, request_options=request_options)
+        return _response.data
 
     def update_trace(
         self,
@@ -529,10 +597,11 @@ class TracesClient:
         *,
         project_name: typing.Optional[str] = OMIT,
         project_id: typing.Optional[str] = OMIT,
+        name: typing.Optional[str] = OMIT,
         end_time: typing.Optional[dt.datetime] = OMIT,
-        input: typing.Optional[JsonNode] = OMIT,
-        output: typing.Optional[JsonNode] = OMIT,
-        metadata: typing.Optional[JsonNode] = OMIT,
+        input: typing.Optional[JsonListString] = OMIT,
+        output: typing.Optional[JsonListString] = OMIT,
+        metadata: typing.Optional[JsonListString] = OMIT,
         tags: typing.Optional[typing.Sequence[str]] = OMIT,
         error_info: typing.Optional[ErrorInfo] = OMIT,
         thread_id: typing.Optional[str] = OMIT,
@@ -551,13 +620,15 @@ class TracesClient:
         project_id : typing.Optional[str]
             If null and project_name not specified, Default Project is assumed
 
+        name : typing.Optional[str]
+
         end_time : typing.Optional[dt.datetime]
 
-        input : typing.Optional[JsonNode]
+        input : typing.Optional[JsonListString]
 
-        output : typing.Optional[JsonNode]
+        output : typing.Optional[JsonListString]
 
-        metadata : typing.Optional[JsonNode]
+        metadata : typing.Optional[JsonListString]
 
         tags : typing.Optional[typing.Sequence[str]]
 
@@ -575,53 +646,30 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.update_trace(
-            id="id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}",
-            method="PATCH",
-            json={
-                "project_name": project_name,
-                "project_id": project_id,
-                "end_time": end_time,
-                "input": input,
-                "output": output,
-                "metadata": metadata,
-                "tags": tags,
-                "error_info": convert_and_respect_annotation_metadata(
-                    object_=error_info, annotation=ErrorInfo, direction="write"
-                ),
-                "thread_id": thread_id,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.update_trace(id='id', )
+        """
+        _response = self._raw_client.update_trace(
+            id,
+            project_name=project_name,
+            project_id=project_id,
+            name=name,
+            end_time=end_time,
+            input=input,
+            output=output,
+            metadata=metadata,
+            tags=tags,
+            error_info=error_info,
+            thread_id=thread_id,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    def delete_trace_comments(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+    def delete_thread_comments(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
-        Delete trace comments
+        Delete thread comments
 
         Parameters
         ----------
@@ -637,49 +685,35 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.delete_trace_comments(
-            ids=["ids"],
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.delete_thread_comments(ids=['ids'], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/comments/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_thread_comments(ids=ids, request_options=request_options)
+        return _response.data
 
-    def delete_trace_feedback_score(
+    def delete_thread_feedback_scores(
         self,
-        id: str,
         *,
-        name: str,
+        project_name: str,
+        thread_id: str,
+        names: typing.Sequence[str],
+        author: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Delete trace feedback score
+        Delete thread feedback scores
 
         Parameters
         ----------
-        id : str
+        project_name : str
 
-        name : str
+        thread_id : str
 
-        request_options : typing.Optional[RequestOptions]
+        names : typing.Sequence[str]
+
+        author : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
@@ -689,32 +723,76 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.delete_trace_feedback_score(
-            id="id",
-            name="name",
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.delete_thread_feedback_scores(project_name='project_name', thread_id='thread_id', names=['names'], )
+        """
+        _response = self._raw_client.delete_thread_feedback_scores(
+            project_name=project_name, thread_id=thread_id, names=names, author=author, request_options=request_options
         )
+        return _response.data
+
+    def delete_trace_comments(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}/feedback-scores/delete",
-            method="POST",
-            json={
-                "name": name,
-            },
-            request_options=request_options,
-            omit=OMIT,
+        Delete trace comments
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.delete_trace_comments(ids=['ids'], )
+        """
+        _response = self._raw_client.delete_trace_comments(ids=ids, request_options=request_options)
+        return _response.data
+
+    def delete_trace_feedback_score(
+        self,
+        id: str,
+        *,
+        name: str,
+        author: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Delete trace feedback score
+
+        Parameters
+        ----------
+        id : str
+
+        name : str
+
+        author : typing.Optional[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.traces.delete_trace_feedback_score(id='id', name='name', )
+        """
+        _response = self._raw_client.delete_trace_feedback_score(
+            id, name=name, author=author, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def delete_trace_threads(
         self,
@@ -747,42 +825,16 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.delete_trace_threads(
-            thread_ids=["thread_ids"],
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.delete_trace_threads(thread_ids=['thread_ids'], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/threads/delete",
-            method="POST",
-            json={
-                "project_name": project_name,
-                "project_id": project_id,
-                "thread_ids": thread_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = self._raw_client.delete_trace_threads(
+            thread_ids=thread_ids, project_name=project_name, project_id=project_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def delete_traces(
-        self,
-        *,
-        ids: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
         Delete traces
@@ -801,37 +853,14 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.delete_traces(
-            ids=["ids"],
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.delete_traces(ids=['ids'], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/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_traces(ids=ids, request_options=request_options)
+        return _response.data
 
     def find_feedback_score_names_2(
-        self,
-        *,
-        project_id: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
+        self, *, project_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[str]:
         """
         Find Feedback Score names
@@ -851,34 +880,40 @@ class TracesClient:
         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.traces.find_feedback_score_names_2()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/feedback-scores/names",
-            method="GET",
-            params={
-                "project_id": project_id,
-            },
-            request_options=request_options,
+        _response = self._raw_client.find_feedback_score_names_2(project_id=project_id, request_options=request_options)
+        return _response.data
+
+    def find_trace_threads_feedback_score_names(
+        self, *, project_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[str]:
+        """
+        Find Trace Threads Feedback Score names
+
+        Parameters
+        ----------
+        project_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[str]
+            Find Trace Threads Feedback Score names
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.find_trace_threads_feedback_score_names()
+        """
+        _response = self._raw_client.find_trace_threads_feedback_score_names(
+            project_id=project_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    parse_obj_as(
-                        type_=typing.List[str],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     def get_trace_stats(
         self,
@@ -886,6 +921,8 @@ class TracesClient:
         project_id: typing.Optional[str] = None,
         project_name: typing.Optional[str] = None,
         filters: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ProjectStatsPublic:
         """
@@ -899,6 +936,10 @@ class TracesClient:
 
         filters : typing.Optional[str]
 
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -910,43 +951,99 @@ class TracesClient:
         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.traces.get_trace_stats()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/stats",
-            method="GET",
-            params={
-                "project_id": project_id,
-                "project_name": project_name,
-                "filters": filters,
-            },
+        _response = self._raw_client.get_trace_stats(
+            project_id=project_id,
+            project_name=project_name,
+            filters=filters,
+            from_time=from_time,
+            to_time=to_time,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectStatsPublic,
-                    parse_obj_as(
-                        type_=ProjectStatsPublic,  # 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_trace_comment(
+    def get_thread_comment(
+        self, comment_id: str, thread_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> Comment:
+        """
+        Get thread comment
+
+        Parameters
+        ----------
+        comment_id : str
+
+        thread_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        Comment
+            Comment resource
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.get_thread_comment(comment_id='commentId', thread_id='threadId', )
+        """
+        _response = self._raw_client.get_thread_comment(comment_id, thread_id, request_options=request_options)
+        return _response.data
+
+    def get_trace_thread_stats(
         self,
-        comment_id: str,
-        trace_id: str,
         *,
+        project_id: typing.Optional[str] = None,
+        project_name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
+    ) -> ProjectStatsPublic:
+        """
+        Get trace thread stats
+
+        Parameters
+        ----------
+        project_id : typing.Optional[str]
+
+        project_name : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        ProjectStatsPublic
+            Trace thread stats resource
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.get_trace_thread_stats()
+        """
+        _response = self._raw_client.get_trace_thread_stats(
+            project_id=project_id,
+            project_name=project_name,
+            filters=filters,
+            from_time=from_time,
+            to_time=to_time,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def get_trace_comment(
+        self, comment_id: str, trace_id: str, *, request_options: typing.Optional[RequestOptions] = None
     ) -> Comment:
         """
         Get trace comment
@@ -968,50 +1065,19 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.get_trace_comment(
-            comment_id="commentId",
-            trace_id="traceId",
-        )
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.get_trace_comment(comment_id='commentId', trace_id='traceId', )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(trace_id)}/comments/{jsonable_encoder(comment_id)}",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Comment,
-                    parse_obj_as(
-                        type_=Comment,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 404:
-                raise NotFoundError(
-                    typing.cast(
-                        typing.Optional[typing.Any],
-                        parse_obj_as(
-                            type_=typing.Optional[typing.Any],  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = self._raw_client.get_trace_comment(comment_id, trace_id, request_options=request_options)
+        return _response.data
 
     def get_trace_thread(
         self,
         *,
-        project_id: str,
         thread_id: str,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TraceThread:
         """
@@ -1019,10 +1085,14 @@ class TracesClient:
 
         Parameters
         ----------
-        project_id : str
-
         thread_id : str
 
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        truncate : typing.Optional[bool]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1034,52 +1104,17 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.get_trace_thread(
-            project_id="project_id",
-            thread_id="thread_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/threads/retrieve",
-            method="POST",
-            json={
-                "project_id": project_id,
-                "thread_id": thread_id,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.get_trace_thread(thread_id='thread_id', )
+        """
+        _response = self._raw_client.get_trace_thread(
+            thread_id=thread_id,
+            project_name=project_name,
+            project_id=project_id,
+            truncate=truncate,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TraceThread,
-                    parse_obj_as(
-                        type_=TraceThread,  # 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_trace_threads(
         self,
@@ -1089,7 +1124,11 @@ class TracesClient:
         project_name: typing.Optional[str] = None,
         project_id: typing.Optional[str] = None,
         truncate: typing.Optional[bool] = None,
+        strip_attachments: typing.Optional[bool] = None,
         filters: typing.Optional[str] = None,
+        sorting: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TraceThreadPage:
         """
@@ -1107,8 +1146,16 @@ class TracesClient:
 
         truncate : typing.Optional[bool]
 
+        strip_attachments : typing.Optional[bool]
+
         filters : typing.Optional[str]
 
+        sorting : typing.Optional[str]
+
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1120,52 +1167,45 @@ class TracesClient:
         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.traces.get_trace_threads()
         """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/threads",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "project_name": project_name,
-                "project_id": project_id,
-                "truncate": truncate,
-                "filters": filters,
-            },
+        _response = self._raw_client.get_trace_threads(
+            page=page,
+            size=size,
+            project_name=project_name,
+            project_id=project_id,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            filters=filters,
+            sorting=sorting,
+            from_time=from_time,
+            to_time=to_time,
             request_options=request_options,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TraceThreadPage,
-                    parse_obj_as(
-                        type_=TraceThreadPage,  # 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 score_batch_of_traces(
+    def open_trace_thread(
         self,
         *,
-        scores: typing.Sequence[FeedbackScoreBatchItem],
+        thread_id: str,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Batch feedback scoring for traces
+        Open trace thread
 
         Parameters
         ----------
-        scores : typing.Sequence[FeedbackScoreBatchItem]
+        thread_id : str
+
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1176,74 +1216,31 @@ class TracesClient:
 
         Examples
         --------
-        from Opik import FeedbackScoreBatchItem, OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.score_batch_of_traces(
-            scores=[
-                FeedbackScoreBatchItem(
-                    id="id",
-                    name="name",
-                    value=1.1,
-                    source="ui",
-                )
-            ],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "v1/private/traces/feedback-scores",
-            method="PUT",
-            json={
-                "scores": convert_and_respect_annotation_metadata(
-                    object_=scores,
-                    annotation=typing.Sequence[FeedbackScoreBatchItem],
-                    direction="write",
-                ),
-            },
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.open_trace_thread(thread_id='thread_id', )
+        """
+        _response = self._raw_client.open_trace_thread(
+            thread_id=thread_id,
+            project_name=project_name,
+            project_id=project_id,
+            truncate=truncate,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    def update_trace_comment(
+    def score_batch_of_threads(
         self,
-        comment_id: str,
         *,
-        text: str,
-        id: typing.Optional[str] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        last_updated_at: typing.Optional[dt.datetime] = OMIT,
-        created_by: typing.Optional[str] = OMIT,
-        last_updated_by: typing.Optional[str] = OMIT,
+        scores: typing.Sequence[FeedbackScoreBatchItemThread],
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Update trace comment by id
+        Batch feedback scoring for threads
 
         Parameters
         ----------
-        comment_id : str
-
-        text : str
-
-        id : typing.Optional[str]
-
-        created_at : typing.Optional[dt.datetime]
-
-        last_updated_at : typing.Optional[dt.datetime]
-
-        created_by : typing.Optional[str]
-
-        last_updated_by : typing.Optional[str]
+        scores : typing.Sequence[FeedbackScoreBatchItemThread]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1255,83 +1252,25 @@ class TracesClient:
         Examples
         --------
         from Opik import OpikApi
-
-        client = OpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-        client.traces.update_trace_comment(
-            comment_id="commentId",
-            text="text",
-        )
+        from Opik import FeedbackScoreBatchItemThread
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.score_batch_of_threads(scores=[FeedbackScoreBatchItemThread(name='name', value=1.1, source="ui", thread_id='thread_id', )], )
         """
-        _response = self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/comments/{jsonable_encoder(comment_id)}",
-            method="PATCH",
-            json={
-                "id": id,
-                "text": text,
-                "created_at": created_at,
-                "last_updated_at": last_updated_at,
-                "created_by": created_by,
-                "last_updated_by": last_updated_by,
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            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)
-
-
-class AsyncTracesClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
+        _response = self._raw_client.score_batch_of_threads(scores=scores, request_options=request_options)
+        return _response.data
 
-    async def add_trace_comment(
+    def score_batch_of_traces(
         self,
-        id_: str,
         *,
-        text: str,
-        id: typing.Optional[str] = OMIT,
-        created_at: typing.Optional[dt.datetime] = OMIT,
-        last_updated_at: typing.Optional[dt.datetime] = OMIT,
-        created_by: typing.Optional[str] = OMIT,
-        last_updated_by: typing.Optional[str] = OMIT,
+        scores: typing.Sequence[FeedbackScoreBatchItem],
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Add trace comment
+        Batch feedback scoring for traces
 
         Parameters
         ----------
-        id_ : str
-
-        text : str
-
-        id : typing.Optional[str]
-
-        created_at : typing.Optional[dt.datetime]
-
-        last_updated_at : typing.Optional[dt.datetime]
-
-        created_by : typing.Optional[str]
-
-        last_updated_by : typing.Optional[str]
+        scores : typing.Sequence[FeedbackScoreBatchItem]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1342,56 +1281,180 @@ class AsyncTracesClient:
 
         Examples
         --------
-        import asyncio
+        from Opik import OpikApi
+        from Opik import FeedbackScoreBatchItem
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.score_batch_of_traces(scores=[FeedbackScoreBatchItem(name='name', value=1.1, source="ui", id='id', )], )
+        """
+        _response = self._raw_client.score_batch_of_traces(scores=scores, request_options=request_options)
+        return _response.data
 
-        from Opik import AsyncOpikApi
+    def search_trace_threads(
+        self,
+        *,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        filters: typing.Optional[typing.Sequence[TraceThreadFilter]] = OMIT,
+        last_retrieved_thread_model_id: typing.Optional[str] = OMIT,
+        limit: typing.Optional[int] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
+        strip_attachments: typing.Optional[bool] = OMIT,
+        from_time: typing.Optional[dt.datetime] = OMIT,
+        to_time: typing.Optional[dt.datetime] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Iterator[bytes]:
+        """
+        Search trace threads
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        Parameters
+        ----------
+        project_name : typing.Optional[str]
 
+        project_id : typing.Optional[str]
 
-        async def main() -> None:
-            await client.traces.add_trace_comment(
-                id_="id",
-                text="text",
-            )
+        filters : typing.Optional[typing.Sequence[TraceThreadFilter]]
 
+        last_retrieved_thread_model_id : typing.Optional[str]
 
-        asyncio.run(main())
+        limit : typing.Optional[int]
+            Max number of trace thread to be streamed
+
+        truncate : typing.Optional[bool]
+            Truncate input, output and metadata to slim payloads
+
+        strip_attachments : typing.Optional[bool]
+            If true, returns attachment references like [file.png]; if false, downloads and reinjects stripped attachments
+
+        from_time : typing.Optional[dt.datetime]
+            Filter trace threads created from this time (ISO-8601 format).
+
+        to_time : typing.Optional[dt.datetime]
+            Filter trace threads created up to this time (ISO-8601 format). If not provided, defaults to current time. Must be after 'from_time'.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
+
+        Returns
+        -------
+        typing.Iterator[bytes]
+            Trace threads stream or error during process
+        """
+        with self._raw_client.search_trace_threads(
+            project_name=project_name,
+            project_id=project_id,
+            filters=filters,
+            last_retrieved_thread_model_id=last_retrieved_thread_model_id,
+            limit=limit,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            from_time=from_time,
+            to_time=to_time,
+            request_options=request_options,
+        ) as r:
+            yield from r.data
+
+    def search_traces(
+        self,
+        *,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
+        last_retrieved_id: typing.Optional[str] = OMIT,
+        limit: typing.Optional[int] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
+        strip_attachments: typing.Optional[bool] = OMIT,
+        from_time: typing.Optional[dt.datetime] = OMIT,
+        to_time: typing.Optional[dt.datetime] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> typing.Iterator[bytes]:
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id_)}/comments",
-            method="POST",
-            json={
-                "id": id,
-                "text": text,
-                "created_at": created_at,
-                "last_updated_at": last_updated_at,
-                "created_by": created_by,
-                "last_updated_by": last_updated_by,
-            },
+        Search traces
+
+        Parameters
+        ----------
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
+
+        last_retrieved_id : typing.Optional[str]
+
+        limit : typing.Optional[int]
+            Max number of traces to be streamed
+
+        truncate : typing.Optional[bool]
+            Truncate input, output and metadata to slim payloads
+
+        strip_attachments : typing.Optional[bool]
+            If true, returns attachment references like [file.png]; if false, downloads and reinjects stripped attachments
+
+        from_time : typing.Optional[dt.datetime]
+            Filter traces created from this time (ISO-8601 format).
+
+        to_time : typing.Optional[dt.datetime]
+            Filter traces created up to this time (ISO-8601 format). If not provided, defaults to current time. Must be after 'from_time'.
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
+
+        Returns
+        -------
+        typing.Iterator[bytes]
+            Traces stream or error during process
+        """
+        with self._raw_client.search_traces(
+            project_name=project_name,
+            project_id=project_id,
+            filters=filters,
+            last_retrieved_id=last_retrieved_id,
+            limit=limit,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            from_time=from_time,
+            to_time=to_time,
             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)
+        ) as r:
+            yield from r.data
 
-    async def add_trace_feedback_score(
+    def update_thread(
         self,
-        id: str,
+        thread_model_id: str,
         *,
-        name: str,
-        value: float,
-        source: FeedbackScoreSource,
-        category_name: typing.Optional[str] = OMIT,
-        reason: typing.Optional[str] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update thread
+
+        Parameters
+        ----------
+        thread_model_id : str
+
+        tags : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.update_thread(thread_model_id='threadModelId', )
+        """
+        _response = self._raw_client.update_thread(thread_model_id, tags=tags, request_options=request_options)
+        return _response.data
+
+    def update_thread_comment(
+        self,
+        comment_id: str,
+        *,
+        text: str,
+        id: typing.Optional[str] = OMIT,
         created_at: typing.Optional[dt.datetime] = OMIT,
         last_updated_at: typing.Optional[dt.datetime] = OMIT,
         created_by: typing.Optional[str] = OMIT,
@@ -1399,21 +1462,71 @@ class AsyncTracesClient:
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Add trace feedback score
+        Update thread comment by id
 
         Parameters
         ----------
-        id : str
+        comment_id : str
 
-        name : str
+        text : str
 
-        value : float
+        id : typing.Optional[str]
 
-        source : FeedbackScoreSource
+        created_at : typing.Optional[dt.datetime]
 
-        category_name : typing.Optional[str]
+        last_updated_at : typing.Optional[dt.datetime]
 
-        reason : typing.Optional[str]
+        created_by : typing.Optional[str]
+
+        last_updated_by : typing.Optional[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.traces.update_thread_comment(comment_id='commentId', text='text', )
+        """
+        _response = self._raw_client.update_thread_comment(
+            comment_id,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
+            request_options=request_options,
+        )
+        return _response.data
+
+    def update_trace_comment(
+        self,
+        comment_id: str,
+        *,
+        text: str,
+        id: typing.Optional[str] = OMIT,
+        created_at: typing.Optional[dt.datetime] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        created_by: typing.Optional[str] = OMIT,
+        last_updated_by: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update trace comment by id
+
+        Parameters
+        ----------
+        comment_id : str
+
+        text : str
+
+        id : typing.Optional[str]
 
         created_at : typing.Optional[dt.datetime]
 
@@ -1432,180 +1545,855 @@ class AsyncTracesClient:
 
         Examples
         --------
-        import asyncio
+        from Opik import OpikApi
+        client = OpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        client.traces.update_trace_comment(comment_id='commentId', text='text', )
+        """
+        _response = self._raw_client.update_trace_comment(
+            comment_id,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
+            request_options=request_options,
+        )
+        return _response.data
 
-        from Opik import AsyncOpikApi
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+class AsyncTracesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._raw_client = AsyncRawTracesClient(client_wrapper=client_wrapper)
 
+    @property
+    def with_raw_response(self) -> AsyncRawTracesClient:
+        """
+        Retrieves a raw implementation of this client that returns raw responses.
 
-        async def main() -> None:
-            await client.traces.add_trace_feedback_score(
-                id="id",
-                name="name",
-                value=1.1,
-                source="ui",
-            )
+        Returns
+        -------
+        AsyncRawTracesClient
+        """
+        return self._raw_client
+
+    async def add_thread_comment(
+        self,
+        id_: str,
+        *,
+        text: str,
+        id: typing.Optional[str] = OMIT,
+        created_at: typing.Optional[dt.datetime] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        created_by: typing.Optional[str] = OMIT,
+        last_updated_by: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Add thread comment
+
+        Parameters
+        ----------
+        id_ : str
+
+        text : str
+
+        id : typing.Optional[str]
+
+        created_at : typing.Optional[dt.datetime]
+
+        last_updated_at : typing.Optional[dt.datetime]
 
+        created_by : typing.Optional[str]
+
+        last_updated_by : typing.Optional[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.traces.add_thread_comment(id_='id', text='text', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}/feedback-scores",
-            method="PUT",
-            json={
-                "name": name,
-                "category_name": category_name,
-                "value": value,
-                "reason": reason,
-                "source": source,
-                "created_at": created_at,
-                "last_updated_at": last_updated_at,
-                "created_by": created_by,
-                "last_updated_by": last_updated_by,
-            },
+        _response = await self._raw_client.add_thread_comment(
+            id_,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def get_traces_by_project(
+    async def add_trace_comment(
         self,
+        id_: str,
         *,
-        page: typing.Optional[int] = None,
-        size: typing.Optional[int] = None,
-        project_name: typing.Optional[str] = None,
-        project_id: typing.Optional[str] = None,
-        filters: typing.Optional[str] = None,
-        truncate: typing.Optional[bool] = None,
-        sorting: typing.Optional[str] = None,
+        text: str,
+        id: typing.Optional[str] = OMIT,
+        created_at: typing.Optional[dt.datetime] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        created_by: typing.Optional[str] = OMIT,
+        last_updated_by: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TracePagePublic:
+    ) -> None:
         """
-        Get traces by project_name or project_id
+        Add trace comment
 
         Parameters
         ----------
-        page : typing.Optional[int]
+        id_ : str
 
-        size : typing.Optional[int]
+        text : str
 
-        project_name : typing.Optional[str]
+        id : typing.Optional[str]
 
-        project_id : typing.Optional[str]
+        created_at : typing.Optional[dt.datetime]
 
-        filters : typing.Optional[str]
+        last_updated_at : typing.Optional[dt.datetime]
 
-        truncate : typing.Optional[bool]
+        created_by : typing.Optional[str]
 
-        sorting : typing.Optional[str]
+        last_updated_by : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        TracePagePublic
-            Trace resource
+        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.traces.add_trace_comment(id_='id', text='text', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.add_trace_comment(
+            id_,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def add_trace_feedback_score(
+        self,
+        id: str,
+        *,
+        name: str,
+        value: float,
+        source: FeedbackScoreSource,
+        category_name: typing.Optional[str] = OMIT,
+        reason: typing.Optional[str] = OMIT,
+        created_at: typing.Optional[dt.datetime] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        created_by: typing.Optional[str] = OMIT,
+        last_updated_by: typing.Optional[str] = OMIT,
+        value_by_author: typing.Optional[typing.Dict[str, ValueEntry]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Add trace feedback score
+
+        Parameters
+        ----------
+        id : str
+
+        name : str
+
+        value : float
+
+        source : FeedbackScoreSource
+
+        category_name : typing.Optional[str]
+
+        reason : typing.Optional[str]
+
+        created_at : typing.Optional[dt.datetime]
+
+        last_updated_at : typing.Optional[dt.datetime]
+
+        created_by : typing.Optional[str]
+
+        last_updated_by : typing.Optional[str]
+
+        value_by_author : typing.Optional[typing.Dict[str, ValueEntry]]
+
+        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.traces.add_trace_feedback_score(id='id', name='name', value=1.1, source="ui", )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.add_trace_feedback_score(
+            id,
+            name=name,
+            value=value,
+            source=source,
+            category_name=category_name,
+            reason=reason,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
+            value_by_author=value_by_author,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def create_traces(
+        self, *, traces: typing.Sequence[TraceWrite], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Create traces
+
+        Parameters
+        ----------
+        traces : typing.Sequence[TraceWrite]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        from Opik import TraceWrite
+        import datetime
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.create_traces(traces=[TraceWrite(start_time=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), )], )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_traces(traces=traces, request_options=request_options)
+        return _response.data
+
+    async def batch_update_traces(
+        self,
+        *,
+        ids: typing.Sequence[str],
+        update: TraceUpdate,
+        merge_tags: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update multiple traces
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+            List of trace IDs to update (max 1000)
+
+        update : TraceUpdate
+
+        merge_tags : typing.Optional[bool]
+            If true, merge tags with existing tags instead of replacing them. Default: false
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        from Opik import TraceUpdate
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.batch_update_traces(ids=['ids'], update=TraceUpdate(), )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.batch_update_traces(
+            ids=ids, update=update, merge_tags=merge_tags, request_options=request_options
+        )
+        return _response.data
+
+    async def batch_update_threads(
+        self,
+        *,
+        ids: typing.Sequence[str],
+        update: TraceThreadUpdate,
+        merge_tags: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update multiple threads
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+            List of thread model IDs to update (max 1000)
+
+        update : TraceThreadUpdate
+
+        merge_tags : typing.Optional[bool]
+            If true, merge tags with existing tags instead of replacing them. Default: false
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        from Opik import TraceThreadUpdate
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.batch_update_threads(ids=['ids'], update=TraceThreadUpdate(), )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.batch_update_threads(
+            ids=ids, update=update, merge_tags=merge_tags, request_options=request_options
+        )
+        return _response.data
+
+    async def close_trace_thread(
+        self,
+        *,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        thread_id: typing.Optional[str] = OMIT,
+        thread_ids: typing.Optional[typing.Sequence[str]] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Close one or multiple trace threads. Supports both single thread_id and multiple thread_ids for batch operations.
+
+        Parameters
+        ----------
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        thread_id : typing.Optional[str]
+
+        thread_ids : typing.Optional[typing.Sequence[str]]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.close_trace_thread()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.close_trace_thread(
+            project_name=project_name,
+            project_id=project_id,
+            thread_id=thread_id,
+            thread_ids=thread_ids,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def get_traces_by_project(
+        self,
+        *,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
+        project_name: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        truncate: typing.Optional[bool] = None,
+        strip_attachments: typing.Optional[bool] = None,
+        sorting: typing.Optional[str] = None,
+        exclude: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> TracePagePublic:
+        """
+        Get traces by project_name or project_id
+
+        Parameters
+        ----------
+        page : typing.Optional[int]
+
+        size : typing.Optional[int]
+
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
+        truncate : typing.Optional[bool]
+
+        strip_attachments : typing.Optional[bool]
+
+        sorting : typing.Optional[str]
+
+        exclude : typing.Optional[str]
+
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TracePagePublic
+            Trace 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.traces.get_traces_by_project()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_traces_by_project(
+            page=page,
+            size=size,
+            project_name=project_name,
+            project_id=project_id,
+            filters=filters,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            sorting=sorting,
+            exclude=exclude,
+            from_time=from_time,
+            to_time=to_time,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def create_trace(
+        self,
+        *,
+        start_time: dt.datetime,
+        id: typing.Optional[str] = OMIT,
+        project_name: typing.Optional[str] = OMIT,
+        name: typing.Optional[str] = OMIT,
+        end_time: typing.Optional[dt.datetime] = OMIT,
+        input: typing.Optional[JsonListStringWrite] = OMIT,
+        output: typing.Optional[JsonListStringWrite] = OMIT,
+        metadata: typing.Optional[JsonListStringWrite] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        error_info: typing.Optional[ErrorInfoWrite] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        thread_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Get trace
+
+        Parameters
+        ----------
+        start_time : dt.datetime
+
+        id : typing.Optional[str]
+
+        project_name : typing.Optional[str]
+            If null, the default project is used
+
+        name : typing.Optional[str]
+
+        end_time : typing.Optional[dt.datetime]
+
+        input : typing.Optional[JsonListStringWrite]
+
+        output : typing.Optional[JsonListStringWrite]
+
+        metadata : typing.Optional[JsonListStringWrite]
+
+        tags : typing.Optional[typing.Sequence[str]]
+
+        error_info : typing.Optional[ErrorInfoWrite]
+
+        last_updated_at : typing.Optional[dt.datetime]
+
+        thread_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import datetime
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.create_trace(start_time=datetime.datetime.fromisoformat("2024-01-15 09:30:00+00:00", ), )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.create_trace(
+            start_time=start_time,
+            id=id,
+            project_name=project_name,
+            name=name,
+            end_time=end_time,
+            input=input,
+            output=output,
+            metadata=metadata,
+            tags=tags,
+            error_info=error_info,
+            last_updated_at=last_updated_at,
+            thread_id=thread_id,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def get_trace_by_id(
+        self,
+        id: str,
+        *,
+        strip_attachments: typing.Optional[bool] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> TracePublic:
+        """
+        Get trace by id
+
+        Parameters
+        ----------
+        id : str
+
+        strip_attachments : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        TracePublic
+            Trace 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.traces.get_trace_by_id(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_trace_by_id(
+            id, strip_attachments=strip_attachments, request_options=request_options
+        )
+        return _response.data
+
+    async def delete_trace_by_id(self, id: str, *, request_options: typing.Optional[RequestOptions] = None) -> None:
+        """
+        Delete trace 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.traces.delete_trace_by_id(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_trace_by_id(id, request_options=request_options)
+        return _response.data
+
+    async def update_trace(
+        self,
+        id: str,
+        *,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        name: typing.Optional[str] = OMIT,
+        end_time: typing.Optional[dt.datetime] = OMIT,
+        input: typing.Optional[JsonListString] = OMIT,
+        output: typing.Optional[JsonListString] = OMIT,
+        metadata: typing.Optional[JsonListString] = OMIT,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
+        error_info: typing.Optional[ErrorInfo] = OMIT,
+        thread_id: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Update trace by id
+
+        Parameters
+        ----------
+        id : str
+
+        project_name : typing.Optional[str]
+            If null and project_id not specified, Default Project is assumed
+
+        project_id : typing.Optional[str]
+            If null and project_name not specified, Default Project is assumed
+
+        name : typing.Optional[str]
+
+        end_time : typing.Optional[dt.datetime]
+
+        input : typing.Optional[JsonListString]
+
+        output : typing.Optional[JsonListString]
+
+        metadata : typing.Optional[JsonListString]
+
+        tags : typing.Optional[typing.Sequence[str]]
+
+        error_info : typing.Optional[ErrorInfo]
+
+        thread_id : typing.Optional[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.traces.update_trace(id='id', )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.update_trace(
+            id,
+            project_name=project_name,
+            project_id=project_id,
+            name=name,
+            end_time=end_time,
+            input=input,
+            output=output,
+            metadata=metadata,
+            tags=tags,
+            error_info=error_info,
+            thread_id=thread_id,
+            request_options=request_options,
+        )
+        return _response.data
+
+    async def delete_thread_comments(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Delete thread comments
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.delete_thread_comments(ids=['ids'], )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_thread_comments(ids=ids, request_options=request_options)
+        return _response.data
+
+    async def delete_thread_feedback_scores(
+        self,
+        *,
+        project_name: str,
+        thread_id: str,
+        names: typing.Sequence[str],
+        author: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Delete thread feedback scores
+
+        Parameters
+        ----------
+        project_name : str
+
+        thread_id : str
+
+        names : typing.Sequence[str]
+
+        author : typing.Optional[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.traces.delete_thread_feedback_scores(project_name='project_name', thread_id='thread_id', names=['names'], )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_thread_feedback_scores(
+            project_name=project_name, thread_id=thread_id, names=names, author=author, request_options=request_options
+        )
+        return _response.data
+
+    async def delete_trace_comments(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
+    ) -> None:
+        """
+        Delete trace comments
+
+        Parameters
+        ----------
+        ids : typing.Sequence[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
+
+        Examples
+        --------
+        from Opik import AsyncOpikApi
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
+        async def main() -> None:
+            await client.traces.delete_trace_comments(ids=['ids'], )
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.delete_trace_comments(ids=ids, request_options=request_options)
+        return _response.data
+
+    async def delete_trace_feedback_score(
+        self,
+        id: str,
+        *,
+        name: str,
+        author: typing.Optional[str] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> None:
+        """
+        Delete trace feedback score
+
+        Parameters
+        ----------
+        id : str
+
+        name : str
+
+        author : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        None
 
         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.traces.get_traces_by_project()
-
-
+            await client.traces.delete_trace_feedback_score(id='id', name='name', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "project_name": project_name,
-                "project_id": project_id,
-                "filters": filters,
-                "truncate": truncate,
-                "sorting": sorting,
-            },
-            request_options=request_options,
+        _response = await self._raw_client.delete_trace_feedback_score(
+            id, name=name, author=author, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TracePagePublic,
-                    parse_obj_as(
-                        type_=TracePagePublic,  # 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_trace(
+    async def delete_trace_threads(
         self,
         *,
-        name: str,
-        start_time: dt.datetime,
-        id: typing.Optional[str] = OMIT,
+        thread_ids: typing.Sequence[str],
         project_name: typing.Optional[str] = OMIT,
-        end_time: typing.Optional[dt.datetime] = OMIT,
-        input: typing.Optional[JsonNodeWrite] = OMIT,
-        output: typing.Optional[JsonNodeWrite] = OMIT,
-        metadata: typing.Optional[JsonNodeWrite] = OMIT,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        error_info: typing.Optional[ErrorInfoWrite] = OMIT,
-        thread_id: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Get trace
+        Delete trace threads
 
         Parameters
         ----------
-        name : str
-
-        start_time : dt.datetime
-
-        id : typing.Optional[str]
+        thread_ids : typing.Sequence[str]
 
         project_name : typing.Optional[str]
-            If null, the default project is used
-
-        end_time : typing.Optional[dt.datetime]
-
-        input : typing.Optional[JsonNodeWrite]
-
-        output : typing.Optional[JsonNodeWrite]
-
-        metadata : typing.Optional[JsonNodeWrite]
-
-        tags : typing.Optional[typing.Sequence[str]]
-
-        error_info : typing.Optional[ErrorInfoWrite]
+            If null, project_id must be provided
 
-        thread_id : typing.Optional[str]
+        project_id : typing.Optional[str]
+            If null, project_name must be provided
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1616,69 +2404,27 @@ class AsyncTracesClient:
 
         Examples
         --------
-        import asyncio
-        import datetime
-
         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.traces.create_trace(
-                name="name",
-                start_time=datetime.datetime.fromisoformat(
-                    "2024-01-15 09:30:00+00:00",
-                ),
-            )
-
-
+            await client.traces.delete_trace_threads(thread_ids=['thread_ids'], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces",
-            method="POST",
-            json={
-                "id": id,
-                "project_name": project_name,
-                "name": name,
-                "start_time": start_time,
-                "end_time": end_time,
-                "input": input,
-                "output": output,
-                "metadata": metadata,
-                "tags": tags,
-                "error_info": convert_and_respect_annotation_metadata(
-                    object_=error_info, annotation=ErrorInfoWrite, direction="write"
-                ),
-                "thread_id": thread_id,
-            },
-            request_options=request_options,
-            omit=OMIT,
+        _response = await self._raw_client.delete_trace_threads(
+            thread_ids=thread_ids, project_name=project_name, project_id=project_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def create_traces(
-        self,
-        *,
-        traces: typing.Sequence[TraceWrite],
-        request_options: typing.Optional[RequestOptions] = None,
+    async def delete_traces(
+        self, *, ids: typing.Sequence[str], request_options: typing.Optional[RequestOptions] = None
     ) -> None:
         """
-        Create traces
+        Delete traces
 
         Parameters
         ----------
-        traces : typing.Sequence[TraceWrite]
+        ids : typing.Sequence[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -1689,462 +2435,389 @@ class AsyncTracesClient:
 
         Examples
         --------
+        from Opik import AsyncOpikApi
         import asyncio
-        import datetime
-
-        from Opik import AsyncOpikApi, TraceWrite
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.traces.create_traces(
-                traces=[
-                    TraceWrite(
-                        name="name",
-                        start_time=datetime.datetime.fromisoformat(
-                            "2024-01-15 09:30:00+00:00",
-                        ),
-                    )
-                ],
-            )
-
-
+            await client.traces.delete_traces(ids=['ids'], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/batch",
-            method="POST",
-            json={
-                "traces": convert_and_respect_annotation_metadata(
-                    object_=traces,
-                    annotation=typing.Sequence[TraceWrite],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.delete_traces(ids=ids, request_options=request_options)
+        return _response.data
 
-    async def get_trace_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> TracePublic:
+    async def find_feedback_score_names_2(
+        self, *, project_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[str]:
         """
-        Get trace by id
+        Find Feedback Score names
 
         Parameters
         ----------
-        id : str
+        project_id : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        TracePublic
-            Trace resource
+        typing.List[str]
+            Feedback Scores resource
 
         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.traces.get_trace_by_id(
-                id="id",
-            )
-
-
+            await client.traces.find_feedback_score_names_2()
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}",
-            method="GET",
-            request_options=request_options,
+        _response = await self._raw_client.find_feedback_score_names_2(
+            project_id=project_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TracePublic,
-                    parse_obj_as(
-                        type_=TracePublic,  # 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_trace_by_id(
-        self, id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> None:
+        return _response.data
+
+    async def find_trace_threads_feedback_score_names(
+        self, *, project_id: typing.Optional[str] = None, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[str]:
         """
-        Delete trace by id
+        Find Trace Threads Feedback Score names
 
         Parameters
         ----------
-        id : str
+        project_id : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        typing.List[str]
+            Find Trace Threads Feedback Score names
 
         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.traces.delete_trace_by_id(
-                id="id",
-            )
-
-
+            await client.traces.find_trace_threads_feedback_score_names()
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}",
-            method="DELETE",
-            request_options=request_options,
+        _response = await self._raw_client.find_trace_threads_feedback_score_names(
+            project_id=project_id, request_options=request_options
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def update_trace(
+    async def get_trace_stats(
         self,
-        id: str,
         *,
-        project_name: typing.Optional[str] = OMIT,
-        project_id: typing.Optional[str] = OMIT,
-        end_time: typing.Optional[dt.datetime] = OMIT,
-        input: typing.Optional[JsonNode] = OMIT,
-        output: typing.Optional[JsonNode] = OMIT,
-        metadata: typing.Optional[JsonNode] = OMIT,
-        tags: typing.Optional[typing.Sequence[str]] = OMIT,
-        error_info: typing.Optional[ErrorInfo] = OMIT,
-        thread_id: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = None,
+        project_name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> ProjectStatsPublic:
         """
-        Update trace by id
+        Get trace stats
 
         Parameters
         ----------
-        id : str
-
-        project_name : typing.Optional[str]
-            If null and project_id not specified, Default Project is assumed
-
         project_id : typing.Optional[str]
-            If null and project_name not specified, Default Project is assumed
-
-        end_time : typing.Optional[dt.datetime]
 
-        input : typing.Optional[JsonNode]
-
-        output : typing.Optional[JsonNode]
-
-        metadata : typing.Optional[JsonNode]
+        project_name : typing.Optional[str]
 
-        tags : typing.Optional[typing.Sequence[str]]
+        filters : typing.Optional[str]
 
-        error_info : typing.Optional[ErrorInfo]
+        from_time : typing.Optional[dt.datetime]
 
-        thread_id : typing.Optional[str]
+        to_time : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        ProjectStatsPublic
+            Trace stats 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.traces.get_trace_stats()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_trace_stats(
+            project_id=project_id,
+            project_name=project_name,
+            filters=filters,
+            from_time=from_time,
+            to_time=to_time,
+            request_options=request_options,
+        )
+        return _response.data
 
-        from Opik import AsyncOpikApi
+    async def get_thread_comment(
+        self, comment_id: str, thread_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> Comment:
+        """
+        Get thread comment
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        Parameters
+        ----------
+        comment_id : str
 
+        thread_id : str
 
-        async def main() -> None:
-            await client.traces.update_trace(
-                id="id",
-            )
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
 
+        Returns
+        -------
+        Comment
+            Comment 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.traces.get_thread_comment(comment_id='commentId', thread_id='threadId', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}",
-            method="PATCH",
-            json={
-                "project_name": project_name,
-                "project_id": project_id,
-                "end_time": end_time,
-                "input": input,
-                "output": output,
-                "metadata": metadata,
-                "tags": tags,
-                "error_info": convert_and_respect_annotation_metadata(
-                    object_=error_info, annotation=ErrorInfo, direction="write"
-                ),
-                "thread_id": thread_id,
-            },
-            headers={
-                "content-type": "application/json",
-            },
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.get_thread_comment(comment_id, thread_id, request_options=request_options)
+        return _response.data
 
-    async def delete_trace_comments(
+    async def get_trace_thread_stats(
         self,
         *,
-        ids: typing.Sequence[str],
+        project_id: typing.Optional[str] = None,
+        project_name: typing.Optional[str] = None,
+        filters: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> ProjectStatsPublic:
         """
-        Delete trace comments
+        Get trace thread stats
 
         Parameters
         ----------
-        ids : typing.Sequence[str]
+        project_id : typing.Optional[str]
+
+        project_name : typing.Optional[str]
+
+        filters : typing.Optional[str]
+
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        ProjectStatsPublic
+            Trace thread stats 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.traces.get_trace_thread_stats()
+        asyncio.run(main())
+        """
+        _response = await self._raw_client.get_trace_thread_stats(
+            project_id=project_id,
+            project_name=project_name,
+            filters=filters,
+            from_time=from_time,
+            to_time=to_time,
+            request_options=request_options,
+        )
+        return _response.data
 
-        from Opik import AsyncOpikApi
+    async def get_trace_comment(
+        self, comment_id: str, trace_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> Comment:
+        """
+        Get trace comment
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        Parameters
+        ----------
+        comment_id : str
 
+        trace_id : str
 
-        async def main() -> None:
-            await client.traces.delete_trace_comments(
-                ids=["ids"],
-            )
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
 
+        Returns
+        -------
+        Comment
+            Comment 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.traces.get_trace_comment(comment_id='commentId', trace_id='traceId', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/comments/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.get_trace_comment(comment_id, trace_id, request_options=request_options)
+        return _response.data
 
-    async def delete_trace_feedback_score(
+    async def get_trace_thread(
         self,
-        id: str,
         *,
-        name: str,
+        thread_id: str,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> TraceThread:
         """
-        Delete trace feedback score
+        Get trace thread
 
         Parameters
         ----------
-        id : str
+        thread_id : str
 
-        name : str
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        TraceThread
+            Trace thread resource
 
         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.traces.delete_trace_feedback_score(
-                id="id",
-                name="name",
-            )
-
-
+            await client.traces.get_trace_thread(thread_id='thread_id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(id)}/feedback-scores/delete",
-            method="POST",
-            json={
-                "name": name,
-            },
+        _response = await self._raw_client.get_trace_thread(
+            thread_id=thread_id,
+            project_name=project_name,
+            project_id=project_id,
+            truncate=truncate,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def delete_trace_threads(
+    async def get_trace_threads(
         self,
         *,
-        thread_ids: typing.Sequence[str],
-        project_name: typing.Optional[str] = OMIT,
-        project_id: typing.Optional[str] = OMIT,
+        page: typing.Optional[int] = None,
+        size: typing.Optional[int] = None,
+        project_name: typing.Optional[str] = None,
+        project_id: typing.Optional[str] = None,
+        truncate: typing.Optional[bool] = None,
+        strip_attachments: typing.Optional[bool] = None,
+        filters: typing.Optional[str] = None,
+        sorting: typing.Optional[str] = None,
+        from_time: typing.Optional[dt.datetime] = None,
+        to_time: typing.Optional[dt.datetime] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> None:
+    ) -> TraceThreadPage:
         """
-        Delete trace threads
+        Get trace threads
 
         Parameters
         ----------
-        thread_ids : typing.Sequence[str]
+        page : typing.Optional[int]
+
+        size : typing.Optional[int]
 
         project_name : typing.Optional[str]
-            If null, project_id must be provided
 
         project_id : typing.Optional[str]
-            If null, project_name must be provided
+
+        truncate : typing.Optional[bool]
+
+        strip_attachments : typing.Optional[bool]
+
+        filters : typing.Optional[str]
+
+        sorting : typing.Optional[str]
+
+        from_time : typing.Optional[dt.datetime]
+
+        to_time : typing.Optional[dt.datetime]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        None
+        TraceThreadPage
+            Trace threads resource
 
         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.traces.delete_trace_threads(
-                thread_ids=["thread_ids"],
-            )
-
-
+            await client.traces.get_trace_threads()
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/threads/delete",
-            method="POST",
-            json={
-                "project_name": project_name,
-                "project_id": project_id,
-                "thread_ids": thread_ids,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        _response = await self._raw_client.get_trace_threads(
+            page=page,
+            size=size,
+            project_name=project_name,
+            project_id=project_id,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            filters=filters,
+            sorting=sorting,
+            from_time=from_time,
+            to_time=to_time,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def delete_traces(
+    async def open_trace_thread(
         self,
         *,
-        ids: typing.Sequence[str],
+        thread_id: str,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Delete traces
+        Open trace thread
 
         Parameters
         ----------
-        ids : typing.Sequence[str]
+        thread_id : str
+
+        project_name : typing.Optional[str]
+
+        project_id : typing.Optional[str]
+
+        truncate : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -2155,420 +2828,283 @@ class AsyncTracesClient:
 
         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.traces.delete_traces(
-                ids=["ids"],
-            )
-
-
+            await client.traces.open_trace_thread(thread_id='thread_id', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/delete",
-            method="POST",
-            json={
-                "ids": ids,
-            },
+        _response = await self._raw_client.open_trace_thread(
+            thread_id=thread_id,
+            project_name=project_name,
+            project_id=project_id,
+            truncate=truncate,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
-    async def find_feedback_score_names_2(
+    async def score_batch_of_threads(
         self,
         *,
-        project_id: typing.Optional[str] = None,
+        scores: typing.Sequence[FeedbackScoreBatchItemThread],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.List[str]:
+    ) -> None:
         """
-        Find Feedback Score names
+        Batch feedback scoring for threads
 
         Parameters
         ----------
-        project_id : typing.Optional[str]
+        scores : typing.Sequence[FeedbackScoreBatchItemThread]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.List[str]
-            Feedback Scores resource
+        None
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        from Opik import FeedbackScoreBatchItemThread
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.traces.find_feedback_score_names_2()
-
-
+            await client.traces.score_batch_of_threads(scores=[FeedbackScoreBatchItemThread(name='name', value=1.1, source="ui", thread_id='thread_id', )], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/feedback-scores/names",
-            method="GET",
-            params={
-                "project_id": project_id,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.List[str],
-                    parse_obj_as(
-                        type_=typing.List[str],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        _response = await self._raw_client.score_batch_of_threads(scores=scores, request_options=request_options)
+        return _response.data
 
-    async def get_trace_stats(
+    async def score_batch_of_traces(
         self,
         *,
-        project_id: typing.Optional[str] = None,
-        project_name: typing.Optional[str] = None,
-        filters: typing.Optional[str] = None,
+        scores: typing.Sequence[FeedbackScoreBatchItem],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> ProjectStatsPublic:
+    ) -> None:
         """
-        Get trace stats
+        Batch feedback scoring for traces
 
         Parameters
         ----------
-        project_id : typing.Optional[str]
-
-        project_name : typing.Optional[str]
-
-        filters : typing.Optional[str]
+        scores : typing.Sequence[FeedbackScoreBatchItem]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        ProjectStatsPublic
-            Trace stats resource
+        None
 
         Examples
         --------
-        import asyncio
-
         from Opik import AsyncOpikApi
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        from Opik import FeedbackScoreBatchItem
+        import asyncio
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.traces.get_trace_stats()
-
-
+            await client.traces.score_batch_of_traces(scores=[FeedbackScoreBatchItem(name='name', value=1.1, source="ui", id='id', )], )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/stats",
-            method="GET",
-            params={
-                "project_id": project_id,
-                "project_name": project_name,
-                "filters": filters,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    ProjectStatsPublic,
-                    parse_obj_as(
-                        type_=ProjectStatsPublic,  # 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.score_batch_of_traces(scores=scores, request_options=request_options)
+        return _response.data
 
-    async def get_trace_comment(
+    async def search_trace_threads(
         self,
-        comment_id: str,
-        trace_id: str,
         *,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        filters: typing.Optional[typing.Sequence[TraceThreadFilter]] = OMIT,
+        last_retrieved_thread_model_id: typing.Optional[str] = OMIT,
+        limit: typing.Optional[int] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
+        strip_attachments: typing.Optional[bool] = OMIT,
+        from_time: typing.Optional[dt.datetime] = OMIT,
+        to_time: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> Comment:
+    ) -> typing.AsyncIterator[bytes]:
         """
-        Get trace comment
+        Search trace threads
 
         Parameters
         ----------
-        comment_id : str
+        project_name : typing.Optional[str]
 
-        trace_id : str
+        project_id : typing.Optional[str]
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
+        filters : typing.Optional[typing.Sequence[TraceThreadFilter]]
 
-        Returns
-        -------
-        Comment
-            Comment resource
+        last_retrieved_thread_model_id : typing.Optional[str]
 
-        Examples
-        --------
-        import asyncio
+        limit : typing.Optional[int]
+            Max number of trace thread to be streamed
 
-        from Opik import AsyncOpikApi
+        truncate : typing.Optional[bool]
+            Truncate input, output and metadata to slim payloads
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        strip_attachments : typing.Optional[bool]
+            If true, returns attachment references like [file.png]; if false, downloads and reinjects stripped attachments
 
+        from_time : typing.Optional[dt.datetime]
+            Filter trace threads created from this time (ISO-8601 format).
 
-        async def main() -> None:
-            await client.traces.get_trace_comment(
-                comment_id="commentId",
-                trace_id="traceId",
-            )
+        to_time : typing.Optional[dt.datetime]
+            Filter trace threads created up to this time (ISO-8601 format). If not provided, defaults to current time. Must be after 'from_time'.
 
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/{jsonable_encoder(trace_id)}/comments/{jsonable_encoder(comment_id)}",
-            method="GET",
+        Returns
+        -------
+        typing.AsyncIterator[bytes]
+            Trace threads stream or error during process
+        """
+        async with self._raw_client.search_trace_threads(
+            project_name=project_name,
+            project_id=project_id,
+            filters=filters,
+            last_retrieved_thread_model_id=last_retrieved_thread_model_id,
+            limit=limit,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            from_time=from_time,
+            to_time=to_time,
             request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    Comment,
-                    parse_obj_as(
-                        type_=Comment,  # 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)
+        ) as r:
+            async for data in r.data:
+                yield data
 
-    async def get_trace_thread(
+    async def search_traces(
         self,
         *,
-        project_id: str,
-        thread_id: str,
+        project_name: typing.Optional[str] = OMIT,
+        project_id: typing.Optional[str] = OMIT,
+        filters: typing.Optional[typing.Sequence[TraceFilterPublic]] = OMIT,
+        last_retrieved_id: typing.Optional[str] = OMIT,
+        limit: typing.Optional[int] = OMIT,
+        truncate: typing.Optional[bool] = OMIT,
+        strip_attachments: typing.Optional[bool] = OMIT,
+        from_time: typing.Optional[dt.datetime] = OMIT,
+        to_time: typing.Optional[dt.datetime] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TraceThread:
+    ) -> typing.AsyncIterator[bytes]:
         """
-        Get trace thread
+        Search traces
 
         Parameters
         ----------
-        project_id : str
+        project_name : typing.Optional[str]
 
-        thread_id : str
+        project_id : typing.Optional[str]
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
+        filters : typing.Optional[typing.Sequence[TraceFilterPublic]]
 
-        Returns
-        -------
-        TraceThread
-            Trace thread resource
+        last_retrieved_id : typing.Optional[str]
 
-        Examples
-        --------
-        import asyncio
+        limit : typing.Optional[int]
+            Max number of traces to be streamed
 
-        from Opik import AsyncOpikApi
+        truncate : typing.Optional[bool]
+            Truncate input, output and metadata to slim payloads
 
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
+        strip_attachments : typing.Optional[bool]
+            If true, returns attachment references like [file.png]; if false, downloads and reinjects stripped attachments
 
+        from_time : typing.Optional[dt.datetime]
+            Filter traces created from this time (ISO-8601 format).
 
-        async def main() -> None:
-            await client.traces.get_trace_thread(
-                project_id="project_id",
-                thread_id="thread_id",
-            )
+        to_time : typing.Optional[dt.datetime]
+            Filter traces created up to this time (ISO-8601 format). If not provided, defaults to current time. Must be after 'from_time'.
 
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration. You can pass in configuration such as `chunk_size`, and more to customize the request and response.
 
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/threads/retrieve",
-            method="POST",
-            json={
-                "project_id": project_id,
-                "thread_id": thread_id,
-            },
-            headers={
-                "content-type": "application/json",
-            },
+        Returns
+        -------
+        typing.AsyncIterator[bytes]
+            Traces stream or error during process
+        """
+        async with self._raw_client.search_traces(
+            project_name=project_name,
+            project_id=project_id,
+            filters=filters,
+            last_retrieved_id=last_retrieved_id,
+            limit=limit,
+            truncate=truncate,
+            strip_attachments=strip_attachments,
+            from_time=from_time,
+            to_time=to_time,
             request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TraceThread,
-                    parse_obj_as(
-                        type_=TraceThread,  # 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)
+        ) as r:
+            async for data in r.data:
+                yield data
 
-    async def get_trace_threads(
+    async def update_thread(
         self,
+        thread_model_id: str,
         *,
-        page: typing.Optional[int] = None,
-        size: typing.Optional[int] = None,
-        project_name: typing.Optional[str] = None,
-        project_id: typing.Optional[str] = None,
-        truncate: typing.Optional[bool] = None,
-        filters: typing.Optional[str] = None,
+        tags: typing.Optional[typing.Sequence[str]] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> TraceThreadPage:
+    ) -> None:
         """
-        Get trace threads
+        Update thread
 
         Parameters
         ----------
-        page : typing.Optional[int]
-
-        size : typing.Optional[int]
+        thread_model_id : str
 
-        project_name : typing.Optional[str]
-
-        project_id : typing.Optional[str]
-
-        truncate : typing.Optional[bool]
-
-        filters : typing.Optional[str]
+        tags : typing.Optional[typing.Sequence[str]]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        TraceThreadPage
-            Trace threads resource
+        None
 
         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.traces.get_trace_threads()
-
-
+            await client.traces.update_thread(thread_model_id='threadModelId', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/threads",
-            method="GET",
-            params={
-                "page": page,
-                "size": size,
-                "project_name": project_name,
-                "project_id": project_id,
-                "truncate": truncate,
-                "filters": filters,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    TraceThreadPage,
-                    parse_obj_as(
-                        type_=TraceThreadPage,  # 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.update_thread(thread_model_id, tags=tags, request_options=request_options)
+        return _response.data
 
-    async def score_batch_of_traces(
+    async def update_thread_comment(
         self,
+        comment_id: str,
         *,
-        scores: typing.Sequence[FeedbackScoreBatchItem],
+        text: str,
+        id: typing.Optional[str] = OMIT,
+        created_at: typing.Optional[dt.datetime] = OMIT,
+        last_updated_at: typing.Optional[dt.datetime] = OMIT,
+        created_by: typing.Optional[str] = OMIT,
+        last_updated_by: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> None:
         """
-        Batch feedback scoring for traces
+        Update thread comment by id
 
         Parameters
         ----------
-        scores : typing.Sequence[FeedbackScoreBatchItem]
+        comment_id : str
+
+        text : str
+
+        id : typing.Optional[str]
+
+        created_at : typing.Optional[dt.datetime]
+
+        last_updated_at : typing.Optional[dt.datetime]
+
+        created_by : typing.Optional[str]
+
+        last_updated_by : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
@@ -2579,51 +3115,24 @@ class AsyncTracesClient:
 
         Examples
         --------
+        from Opik import AsyncOpikApi
         import asyncio
-
-        from Opik import AsyncOpikApi, FeedbackScoreBatchItem
-
-        client = AsyncOpikApi(
-            api_key="YOUR_API_KEY",
-            workspace_name="YOUR_WORKSPACE_NAME",
-        )
-
-
+        client = AsyncOpikApi(api_key="YOUR_API_KEY", workspace_name="YOUR_WORKSPACE_NAME", )
         async def main() -> None:
-            await client.traces.score_batch_of_traces(
-                scores=[
-                    FeedbackScoreBatchItem(
-                        id="id",
-                        name="name",
-                        value=1.1,
-                        source="ui",
-                    )
-                ],
-            )
-
-
+            await client.traces.update_thread_comment(comment_id='commentId', text='text', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "v1/private/traces/feedback-scores",
-            method="PUT",
-            json={
-                "scores": convert_and_respect_annotation_metadata(
-                    object_=scores,
-                    annotation=typing.Sequence[FeedbackScoreBatchItem],
-                    direction="write",
-                ),
-            },
+        _response = await self._raw_client.update_thread_comment(
+            comment_id,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
+        return _response.data
 
     async def update_trace_comment(
         self,
@@ -2665,53 +3174,21 @@ class AsyncTracesClient:
 
         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.traces.update_trace_comment(
-                comment_id="commentId",
-                text="text",
-            )
-
-
+            await client.traces.update_trace_comment(comment_id='commentId', text='text', )
         asyncio.run(main())
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"v1/private/traces/comments/{jsonable_encoder(comment_id)}",
-            method="PATCH",
-            json={
-                "id": id,
-                "text": text,
-                "created_at": created_at,
-                "last_updated_at": last_updated_at,
-                "created_by": created_by,
-                "last_updated_by": last_updated_by,
-            },
+        _response = await self._raw_client.update_trace_comment(
+            comment_id,
+            text=text,
+            id=id,
+            created_at=created_at,
+            last_updated_at=last_updated_at,
+            created_by=created_by,
+            last_updated_by=last_updated_by,
             request_options=request_options,
-            omit=OMIT,
         )
-        try:
-            if 200 <= _response.status_code < 300:
-                return
-            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