mirascope 2.0.0a6__py3-none-any.whl → 2.0.1__py3-none-any.whl

mirascope/api/_generated/traces/raw_client.py

@@ -15,11 +15,15 @@ from ..errors.conflict_error import ConflictError
 from ..errors.forbidden_error import ForbiddenError
 from ..errors.internal_server_error import InternalServerError
 from ..errors.not_found_error import NotFoundError
+from ..errors.payment_required_error import PaymentRequiredError
+from ..errors.service_unavailable_error import ServiceUnavailableError
+from ..errors.too_many_requests_error import TooManyRequestsError
 from ..errors.unauthorized_error import UnauthorizedError
-from ..types.already_exists_error import AlreadyExistsError
-from ..types.http_api_decode_error import HttpApiDecodeError
 from ..types.not_found_error_body import NotFoundErrorBody
+from ..types.number_from_string import NumberFromString
 from ..types.permission_denied_error import PermissionDeniedError
+from ..types.plan_limit_exceeded_error import PlanLimitExceededError
+from ..types.rate_limit_error import RateLimitError
 from ..types.unauthorized_error_body import UnauthorizedErrorBody
 from .types.traces_create_request_resource_spans_item import (
     TracesCreateRequestResourceSpansItem,
@@ -28,7 +32,21 @@ from .types.traces_create_response import TracesCreateResponse
 from .types.traces_get_analytics_summary_response import (
     TracesGetAnalyticsSummaryResponse,
 )
+from .types.traces_get_trace_detail_by_env_response import (
+    TracesGetTraceDetailByEnvResponse,
+)
 from .types.traces_get_trace_detail_response import TracesGetTraceDetailResponse
+from .types.traces_list_by_function_hash_response import (
+    TracesListByFunctionHashResponse,
+)
+from .types.traces_search_by_env_request_attribute_filters_item import (
+    TracesSearchByEnvRequestAttributeFiltersItem,
+)
+from .types.traces_search_by_env_request_sort_by import TracesSearchByEnvRequestSortBy
+from .types.traces_search_by_env_request_sort_order import (
+    TracesSearchByEnvRequestSortOrder,
+)
+from .types.traces_search_by_env_response import TracesSearchByEnvResponse
 from .types.traces_search_request_attribute_filters_item import (
     TracesSearchRequestAttributeFiltersItem,
 )
@@ -93,9 +111,9 @@ class RawTracesClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -111,6 +129,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 402:
+                raise PaymentRequiredError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PlanLimitExceededError,
+                        parse_obj_as(
+                            type_=PlanLimitExceededError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 403:
                 raise ForbiddenError(
                     headers=dict(_response.headers),
@@ -137,9 +166,20 @@ class RawTracesClient:
                 raise ConflictError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        AlreadyExistsError,
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
                         parse_obj_as(
-                            type_=AlreadyExistsError,  # type: ignore
+                            type_=RateLimitError,  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -155,6 +195,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -174,12 +225,16 @@ class RawTracesClient:
         start_time: str,
         end_time: str,
         query: typing.Optional[str] = OMIT,
+        input_messages_query: typing.Optional[str] = OMIT,
+        output_messages_query: typing.Optional[str] = OMIT,
+        fuzzy_search: typing.Optional[bool] = OMIT,
         trace_id: typing.Optional[str] = OMIT,
         span_id: typing.Optional[str] = OMIT,
         model: typing.Optional[typing.Sequence[str]] = OMIT,
         provider: typing.Optional[typing.Sequence[str]] = OMIT,
         function_id: typing.Optional[str] = OMIT,
         function_name: typing.Optional[str] = OMIT,
+        span_name_prefix: typing.Optional[str] = OMIT,
         has_error: typing.Optional[bool] = OMIT,
         min_tokens: typing.Optional[float] = OMIT,
         max_tokens: typing.Optional[float] = OMIT,
@@ -192,6 +247,7 @@ class RawTracesClient:
         offset: typing.Optional[float] = OMIT,
         sort_by: typing.Optional[TracesSearchRequestSortBy] = OMIT,
         sort_order: typing.Optional[TracesSearchRequestSortOrder] = OMIT,
+        root_spans_only: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> HttpResponse[TracesSearchResponse]:
         """
@@ -203,6 +259,12 @@ class RawTracesClient:
 
         query : typing.Optional[str]
 
+        input_messages_query : typing.Optional[str]
+
+        output_messages_query : typing.Optional[str]
+
+        fuzzy_search : typing.Optional[bool]
+
         trace_id : typing.Optional[str]
 
         span_id : typing.Optional[str]
@@ -215,6 +277,8 @@ class RawTracesClient:
 
         function_name : typing.Optional[str]
 
+        span_name_prefix : typing.Optional[str]
+
         has_error : typing.Optional[bool]
 
         min_tokens : typing.Optional[float]
@@ -235,6 +299,8 @@ class RawTracesClient:
 
         sort_order : typing.Optional[TracesSearchRequestSortOrder]
 
+        root_spans_only : typing.Optional[bool]
+
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -250,12 +316,16 @@ class RawTracesClient:
                 "startTime": start_time,
                 "endTime": end_time,
                 "query": query,
+                "inputMessagesQuery": input_messages_query,
+                "outputMessagesQuery": output_messages_query,
+                "fuzzySearch": fuzzy_search,
                 "traceId": trace_id,
                 "spanId": span_id,
                 "model": model,
                 "provider": provider,
                 "functionId": function_id,
                 "functionName": function_name,
+                "spanNamePrefix": span_name_prefix,
                 "hasError": has_error,
                 "minTokens": min_tokens,
                 "maxTokens": max_tokens,
@@ -270,6 +340,7 @@ class RawTracesClient:
                 "offset": offset,
                 "sortBy": sort_by,
                 "sortOrder": sort_order,
+                "rootSpansOnly": root_spans_only,
             },
             headers={
                 "content-type": "application/json",
@@ -291,9 +362,9 @@ class RawTracesClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -320,6 +391,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 500:
                 raise InternalServerError(
                     headers=dict(_response.headers),
@@ -331,6 +413,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -379,9 +472,9 @@ class RawTracesClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -419,6 +512,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 500:
                 raise InternalServerError(
                     headers=dict(_response.headers),
@@ -430,6 +534,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -492,9 +607,9 @@ class RawTracesClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -521,6 +636,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 500:
                 raise InternalServerError(
                     headers=dict(_response.headers),
@@ -532,6 +658,17 @@ class RawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -545,63 +682,57 @@ class RawTracesClient:
             body=_response_json,
         )
 
-
-class AsyncRawTracesClient:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def create(
+    def listbyfunctionhash(
         self,
+        hash: str,
         *,
-        resource_spans: typing.Sequence[TracesCreateRequestResourceSpansItem],
+        limit: typing.Optional[NumberFromString] = None,
+        offset: typing.Optional[NumberFromString] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TracesCreateResponse]:
+    ) -> HttpResponse[TracesListByFunctionHashResponse]:
         """
         Parameters
         ----------
-        resource_spans : typing.Sequence[TracesCreateRequestResourceSpansItem]
+        hash : str
+
+        limit : typing.Optional[NumberFromString]
+
+        offset : typing.Optional[NumberFromString]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[TracesCreateResponse]
+        HttpResponse[TracesListByFunctionHashResponse]
             Success
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "traces",
-            method="POST",
-            json={
-                "resourceSpans": convert_and_respect_annotation_metadata(
-                    object_=resource_spans,
-                    annotation=typing.Sequence[TracesCreateRequestResourceSpansItem],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
+        _response = self._client_wrapper.httpx_client.request(
+            f"traces/function/hash/{jsonable_encoder(hash)}",
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
             },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TracesCreateResponse,
+                    TracesListByFunctionHashResponse,
                     parse_obj_as(
-                        type_=TracesCreateResponse,  # type: ignore
+                        type_=TracesListByFunctionHashResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-                return AsyncHttpResponse(response=_response, data=_data)
+                return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -639,13 +770,13 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
-            if _response.status_code == 409:
-                raise ConflictError(
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        AlreadyExistsError,
+                        RateLimitError,
                         parse_obj_as(
-                            type_=AlreadyExistsError,  # type: ignore
+                            type_=RateLimitError,  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -661,6 +792,17 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -674,41 +816,61 @@ class AsyncRawTracesClient:
             body=_response_json,
         )
 
-    async def search(
+    def searchbyenv(
         self,
+        organization_id: str,
+        project_id: str,
+        environment_id: str,
         *,
         start_time: str,
         end_time: str,
         query: typing.Optional[str] = OMIT,
+        input_messages_query: typing.Optional[str] = OMIT,
+        output_messages_query: typing.Optional[str] = OMIT,
+        fuzzy_search: typing.Optional[bool] = OMIT,
         trace_id: typing.Optional[str] = OMIT,
         span_id: typing.Optional[str] = OMIT,
         model: typing.Optional[typing.Sequence[str]] = OMIT,
         provider: typing.Optional[typing.Sequence[str]] = OMIT,
         function_id: typing.Optional[str] = OMIT,
         function_name: typing.Optional[str] = OMIT,
+        span_name_prefix: typing.Optional[str] = OMIT,
         has_error: typing.Optional[bool] = OMIT,
         min_tokens: typing.Optional[float] = OMIT,
         max_tokens: typing.Optional[float] = OMIT,
         min_duration: typing.Optional[float] = OMIT,
         max_duration: typing.Optional[float] = OMIT,
         attribute_filters: typing.Optional[
-            typing.Sequence[TracesSearchRequestAttributeFiltersItem]
+            typing.Sequence[TracesSearchByEnvRequestAttributeFiltersItem]
         ] = OMIT,
         limit: typing.Optional[float] = OMIT,
         offset: typing.Optional[float] = OMIT,
-        sort_by: typing.Optional[TracesSearchRequestSortBy] = OMIT,
-        sort_order: typing.Optional[TracesSearchRequestSortOrder] = OMIT,
+        sort_by: typing.Optional[TracesSearchByEnvRequestSortBy] = OMIT,
+        sort_order: typing.Optional[TracesSearchByEnvRequestSortOrder] = OMIT,
+        root_spans_only: typing.Optional[bool] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TracesSearchResponse]:
+    ) -> HttpResponse[TracesSearchByEnvResponse]:
         """
         Parameters
         ----------
+        organization_id : str
+
+        project_id : str
+
+        environment_id : str
+
         start_time : str
 
         end_time : str
 
         query : typing.Optional[str]
 
+        input_messages_query : typing.Optional[str]
+
+        output_messages_query : typing.Optional[str]
+
+        fuzzy_search : typing.Optional[bool]
+
         trace_id : typing.Optional[str]
 
         span_id : typing.Optional[str]
@@ -721,6 +883,8 @@ class AsyncRawTracesClient:
 
         function_name : typing.Optional[str]
 
+        span_name_prefix : typing.Optional[str]
+
         has_error : typing.Optional[bool]
 
         min_tokens : typing.Optional[float]
@@ -731,37 +895,43 @@ class AsyncRawTracesClient:
 
         max_duration : typing.Optional[float]
 
-        attribute_filters : typing.Optional[typing.Sequence[TracesSearchRequestAttributeFiltersItem]]
+        attribute_filters : typing.Optional[typing.Sequence[TracesSearchByEnvRequestAttributeFiltersItem]]
 
         limit : typing.Optional[float]
 
         offset : typing.Optional[float]
 
-        sort_by : typing.Optional[TracesSearchRequestSortBy]
+        sort_by : typing.Optional[TracesSearchByEnvRequestSortBy]
 
-        sort_order : typing.Optional[TracesSearchRequestSortOrder]
+        sort_order : typing.Optional[TracesSearchByEnvRequestSortOrder]
+
+        root_spans_only : typing.Optional[bool]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[TracesSearchResponse]
+        HttpResponse[TracesSearchByEnvResponse]
             Success
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "traces/search",
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(organization_id)}/projects/{jsonable_encoder(project_id)}/environments/{jsonable_encoder(environment_id)}/traces/search",
             method="POST",
             json={
                 "startTime": start_time,
                 "endTime": end_time,
                 "query": query,
+                "inputMessagesQuery": input_messages_query,
+                "outputMessagesQuery": output_messages_query,
+                "fuzzySearch": fuzzy_search,
                 "traceId": trace_id,
                 "spanId": span_id,
                 "model": model,
                 "provider": provider,
                 "functionId": function_id,
                 "functionName": function_name,
+                "spanNamePrefix": span_name_prefix,
                 "hasError": has_error,
                 "minTokens": min_tokens,
                 "maxTokens": max_tokens,
@@ -769,13 +939,16 @@ class AsyncRawTracesClient:
                 "maxDuration": max_duration,
                 "attributeFilters": convert_and_respect_annotation_metadata(
                     object_=attribute_filters,
-                    annotation=typing.Sequence[TracesSearchRequestAttributeFiltersItem],
+                    annotation=typing.Sequence[
+                        TracesSearchByEnvRequestAttributeFiltersItem
+                    ],
                     direction="write",
                 ),
                 "limit": limit,
                 "offset": offset,
                 "sortBy": sort_by,
                 "sortOrder": sort_order,
+                "rootSpansOnly": root_spans_only,
             },
             headers={
                 "content-type": "application/json",
@@ -786,20 +959,20 @@ class AsyncRawTracesClient:
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TracesSearchResponse,
+                    TracesSearchByEnvResponse,
                     parse_obj_as(
-                        type_=TracesSearchResponse,  # type: ignore
+                        type_=TracesSearchByEnvResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-                return AsyncHttpResponse(response=_response, data=_data)
+                return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -826,6 +999,28 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 500:
                 raise InternalServerError(
                     headers=dict(_response.headers),
@@ -837,6 +1032,17 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -850,12 +1056,24 @@ class AsyncRawTracesClient:
             body=_response_json,
         )
 
-    async def gettracedetail(
-        self, trace_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> AsyncHttpResponse[TracesGetTraceDetailResponse]:
+    def gettracedetailbyenv(
+        self,
+        organization_id: str,
+        project_id: str,
+        environment_id: str,
+        trace_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> HttpResponse[TracesGetTraceDetailByEnvResponse]:
         """
         Parameters
         ----------
+        organization_id : str
+
+        project_id : str
+
+        environment_id : str
+
         trace_id : str
 
         request_options : typing.Optional[RequestOptions]
@@ -863,31 +1081,31 @@ class AsyncRawTracesClient:
 
         Returns
         -------
-        AsyncHttpResponse[TracesGetTraceDetailResponse]
+        HttpResponse[TracesGetTraceDetailByEnvResponse]
             Success
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"traces/{jsonable_encoder(trace_id)}",
+        _response = self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(organization_id)}/projects/{jsonable_encoder(project_id)}/environments/{jsonable_encoder(environment_id)}/traces/{jsonable_encoder(trace_id)}",
             method="GET",
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TracesGetTraceDetailResponse,
+                    TracesGetTraceDetailByEnvResponse,
                     parse_obj_as(
-                        type_=TracesGetTraceDetailResponse,  # type: ignore
+                        type_=TracesGetTraceDetailByEnvResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-                return AsyncHttpResponse(response=_response, data=_data)
+                return HttpResponse(response=_response, data=_data)
             if _response.status_code == 400:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -925,6 +1143,17 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 500:
                 raise InternalServerError(
                     headers=dict(_response.headers),
@@ -936,6 +1165,17 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(
@@ -949,47 +1189,1044 @@ class AsyncRawTracesClient:
             body=_response_json,
         )
 
-    async def getanalyticssummary(
+
+class AsyncRawTracesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def create(
         self,
         *,
-        start_time: str,
-        end_time: str,
-        function_id: typing.Optional[str] = None,
+        resource_spans: typing.Sequence[TracesCreateRequestResourceSpansItem],
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> AsyncHttpResponse[TracesGetAnalyticsSummaryResponse]:
+    ) -> AsyncHttpResponse[TracesCreateResponse]:
         """
         Parameters
         ----------
-        start_time : str
-
-        end_time : str
-
-        function_id : typing.Optional[str]
+        resource_spans : typing.Sequence[TracesCreateRequestResourceSpansItem]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        AsyncHttpResponse[TracesGetAnalyticsSummaryResponse]
+        AsyncHttpResponse[TracesCreateResponse]
             Success
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "traces/analytics",
-            method="GET",
-            params={
-                "startTime": start_time,
-                "endTime": end_time,
-                "functionId": function_id,
-            },
-            request_options=request_options,
-        )
+            "traces",
+            method="POST",
+            json={
+                "resourceSpans": convert_and_respect_annotation_metadata(
+                    object_=resource_spans,
+                    annotation=typing.Sequence[TracesCreateRequestResourceSpansItem],
+                    direction="write",
+                ),
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TracesCreateResponse,
+                    parse_obj_as(
+                        type_=TracesCreateResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 402:
+                raise PaymentRequiredError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PlanLimitExceededError,
+                        parse_obj_as(
+                            type_=PlanLimitExceededError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 409:
+                raise ConflictError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+    async def search(
+        self,
+        *,
+        start_time: str,
+        end_time: str,
+        query: typing.Optional[str] = OMIT,
+        input_messages_query: typing.Optional[str] = OMIT,
+        output_messages_query: typing.Optional[str] = OMIT,
+        fuzzy_search: typing.Optional[bool] = OMIT,
+        trace_id: typing.Optional[str] = OMIT,
+        span_id: typing.Optional[str] = OMIT,
+        model: typing.Optional[typing.Sequence[str]] = OMIT,
+        provider: typing.Optional[typing.Sequence[str]] = OMIT,
+        function_id: typing.Optional[str] = OMIT,
+        function_name: typing.Optional[str] = OMIT,
+        span_name_prefix: typing.Optional[str] = OMIT,
+        has_error: typing.Optional[bool] = OMIT,
+        min_tokens: typing.Optional[float] = OMIT,
+        max_tokens: typing.Optional[float] = OMIT,
+        min_duration: typing.Optional[float] = OMIT,
+        max_duration: typing.Optional[float] = OMIT,
+        attribute_filters: typing.Optional[
+            typing.Sequence[TracesSearchRequestAttributeFiltersItem]
+        ] = OMIT,
+        limit: typing.Optional[float] = OMIT,
+        offset: typing.Optional[float] = OMIT,
+        sort_by: typing.Optional[TracesSearchRequestSortBy] = OMIT,
+        sort_order: typing.Optional[TracesSearchRequestSortOrder] = OMIT,
+        root_spans_only: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TracesSearchResponse]:
+        """
+        Parameters
+        ----------
+        start_time : str
+
+        end_time : str
+
+        query : typing.Optional[str]
+
+        input_messages_query : typing.Optional[str]
+
+        output_messages_query : typing.Optional[str]
+
+        fuzzy_search : typing.Optional[bool]
+
+        trace_id : typing.Optional[str]
+
+        span_id : typing.Optional[str]
+
+        model : typing.Optional[typing.Sequence[str]]
+
+        provider : typing.Optional[typing.Sequence[str]]
+
+        function_id : typing.Optional[str]
+
+        function_name : typing.Optional[str]
+
+        span_name_prefix : typing.Optional[str]
+
+        has_error : typing.Optional[bool]
+
+        min_tokens : typing.Optional[float]
+
+        max_tokens : typing.Optional[float]
+
+        min_duration : typing.Optional[float]
+
+        max_duration : typing.Optional[float]
+
+        attribute_filters : typing.Optional[typing.Sequence[TracesSearchRequestAttributeFiltersItem]]
+
+        limit : typing.Optional[float]
+
+        offset : typing.Optional[float]
+
+        sort_by : typing.Optional[TracesSearchRequestSortBy]
+
+        sort_order : typing.Optional[TracesSearchRequestSortOrder]
+
+        root_spans_only : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TracesSearchResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "traces/search",
+            method="POST",
+            json={
+                "startTime": start_time,
+                "endTime": end_time,
+                "query": query,
+                "inputMessagesQuery": input_messages_query,
+                "outputMessagesQuery": output_messages_query,
+                "fuzzySearch": fuzzy_search,
+                "traceId": trace_id,
+                "spanId": span_id,
+                "model": model,
+                "provider": provider,
+                "functionId": function_id,
+                "functionName": function_name,
+                "spanNamePrefix": span_name_prefix,
+                "hasError": has_error,
+                "minTokens": min_tokens,
+                "maxTokens": max_tokens,
+                "minDuration": min_duration,
+                "maxDuration": max_duration,
+                "attributeFilters": convert_and_respect_annotation_metadata(
+                    object_=attribute_filters,
+                    annotation=typing.Sequence[TracesSearchRequestAttributeFiltersItem],
+                    direction="write",
+                ),
+                "limit": limit,
+                "offset": offset,
+                "sortBy": sort_by,
+                "sortOrder": sort_order,
+                "rootSpansOnly": root_spans_only,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TracesSearchResponse,
+                    parse_obj_as(
+                        type_=TracesSearchResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+    async def gettracedetail(
+        self, trace_id: str, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> AsyncHttpResponse[TracesGetTraceDetailResponse]:
+        """
+        Parameters
+        ----------
+        trace_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TracesGetTraceDetailResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"traces/{jsonable_encoder(trace_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TracesGetTraceDetailResponse,
+                    parse_obj_as(
+                        type_=TracesGetTraceDetailResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+    async def getanalyticssummary(
+        self,
+        *,
+        start_time: str,
+        end_time: str,
+        function_id: typing.Optional[str] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TracesGetAnalyticsSummaryResponse]:
+        """
+        Parameters
+        ----------
+        start_time : str
+
+        end_time : str
+
+        function_id : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TracesGetAnalyticsSummaryResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "traces/analytics",
+            method="GET",
+            params={
+                "startTime": start_time,
+                "endTime": end_time,
+                "functionId": function_id,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TracesGetAnalyticsSummaryResponse,
+                    parse_obj_as(
+                        type_=TracesGetAnalyticsSummaryResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+    async def listbyfunctionhash(
+        self,
+        hash: str,
+        *,
+        limit: typing.Optional[NumberFromString] = None,
+        offset: typing.Optional[NumberFromString] = None,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TracesListByFunctionHashResponse]:
+        """
+        Parameters
+        ----------
+        hash : str
+
+        limit : typing.Optional[NumberFromString]
+
+        offset : typing.Optional[NumberFromString]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TracesListByFunctionHashResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"traces/function/hash/{jsonable_encoder(hash)}",
+            method="GET",
+            params={
+                "limit": limit,
+                "offset": offset,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TracesListByFunctionHashResponse,
+                    parse_obj_as(
+                        type_=TracesListByFunctionHashResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+    async def searchbyenv(
+        self,
+        organization_id: str,
+        project_id: str,
+        environment_id: str,
+        *,
+        start_time: str,
+        end_time: str,
+        query: typing.Optional[str] = OMIT,
+        input_messages_query: typing.Optional[str] = OMIT,
+        output_messages_query: typing.Optional[str] = OMIT,
+        fuzzy_search: typing.Optional[bool] = OMIT,
+        trace_id: typing.Optional[str] = OMIT,
+        span_id: typing.Optional[str] = OMIT,
+        model: typing.Optional[typing.Sequence[str]] = OMIT,
+        provider: typing.Optional[typing.Sequence[str]] = OMIT,
+        function_id: typing.Optional[str] = OMIT,
+        function_name: typing.Optional[str] = OMIT,
+        span_name_prefix: typing.Optional[str] = OMIT,
+        has_error: typing.Optional[bool] = OMIT,
+        min_tokens: typing.Optional[float] = OMIT,
+        max_tokens: typing.Optional[float] = OMIT,
+        min_duration: typing.Optional[float] = OMIT,
+        max_duration: typing.Optional[float] = OMIT,
+        attribute_filters: typing.Optional[
+            typing.Sequence[TracesSearchByEnvRequestAttributeFiltersItem]
+        ] = OMIT,
+        limit: typing.Optional[float] = OMIT,
+        offset: typing.Optional[float] = OMIT,
+        sort_by: typing.Optional[TracesSearchByEnvRequestSortBy] = OMIT,
+        sort_order: typing.Optional[TracesSearchByEnvRequestSortOrder] = OMIT,
+        root_spans_only: typing.Optional[bool] = OMIT,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TracesSearchByEnvResponse]:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        environment_id : str
+
+        start_time : str
+
+        end_time : str
+
+        query : typing.Optional[str]
+
+        input_messages_query : typing.Optional[str]
+
+        output_messages_query : typing.Optional[str]
+
+        fuzzy_search : typing.Optional[bool]
+
+        trace_id : typing.Optional[str]
+
+        span_id : typing.Optional[str]
+
+        model : typing.Optional[typing.Sequence[str]]
+
+        provider : typing.Optional[typing.Sequence[str]]
+
+        function_id : typing.Optional[str]
+
+        function_name : typing.Optional[str]
+
+        span_name_prefix : typing.Optional[str]
+
+        has_error : typing.Optional[bool]
+
+        min_tokens : typing.Optional[float]
+
+        max_tokens : typing.Optional[float]
+
+        min_duration : typing.Optional[float]
+
+        max_duration : typing.Optional[float]
+
+        attribute_filters : typing.Optional[typing.Sequence[TracesSearchByEnvRequestAttributeFiltersItem]]
+
+        limit : typing.Optional[float]
+
+        offset : typing.Optional[float]
+
+        sort_by : typing.Optional[TracesSearchByEnvRequestSortBy]
+
+        sort_order : typing.Optional[TracesSearchByEnvRequestSortOrder]
+
+        root_spans_only : typing.Optional[bool]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TracesSearchByEnvResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(organization_id)}/projects/{jsonable_encoder(project_id)}/environments/{jsonable_encoder(environment_id)}/traces/search",
+            method="POST",
+            json={
+                "startTime": start_time,
+                "endTime": end_time,
+                "query": query,
+                "inputMessagesQuery": input_messages_query,
+                "outputMessagesQuery": output_messages_query,
+                "fuzzySearch": fuzzy_search,
+                "traceId": trace_id,
+                "spanId": span_id,
+                "model": model,
+                "provider": provider,
+                "functionId": function_id,
+                "functionName": function_name,
+                "spanNamePrefix": span_name_prefix,
+                "hasError": has_error,
+                "minTokens": min_tokens,
+                "maxTokens": max_tokens,
+                "minDuration": min_duration,
+                "maxDuration": max_duration,
+                "attributeFilters": convert_and_respect_annotation_metadata(
+                    object_=attribute_filters,
+                    annotation=typing.Sequence[
+                        TracesSearchByEnvRequestAttributeFiltersItem
+                    ],
+                    direction="write",
+                ),
+                "limit": limit,
+                "offset": offset,
+                "sortBy": sort_by,
+                "sortOrder": sort_order,
+                "rootSpansOnly": root_spans_only,
+            },
+            headers={
+                "content-type": "application/json",
+            },
+            request_options=request_options,
+            omit=OMIT,
+        )
         try:
             if 200 <= _response.status_code < 300:
                 _data = typing.cast(
-                    TracesGetAnalyticsSummaryResponse,
+                    TracesSearchByEnvResponse,
                     parse_obj_as(
-                        type_=TracesGetAnalyticsSummaryResponse,  # type: ignore
+                        type_=TracesSearchByEnvResponse,  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+                return AsyncHttpResponse(response=_response, data=_data)
+            if _response.status_code == 400:
+                raise BadRequestError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 401:
+                raise UnauthorizedError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        UnauthorizedErrorBody,
+                        parse_obj_as(
+                            type_=UnauthorizedErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 403:
+                raise ForbiddenError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        PermissionDeniedError,
+                        parse_obj_as(
+                            type_=PermissionDeniedError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 500:
+                raise InternalServerError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(
+                status_code=_response.status_code,
+                headers=dict(_response.headers),
+                body=_response.text,
+            )
+        raise ApiError(
+            status_code=_response.status_code,
+            headers=dict(_response.headers),
+            body=_response_json,
+        )
+
+    async def gettracedetailbyenv(
+        self,
+        organization_id: str,
+        project_id: str,
+        environment_id: str,
+        trace_id: str,
+        *,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> AsyncHttpResponse[TracesGetTraceDetailByEnvResponse]:
+        """
+        Parameters
+        ----------
+        organization_id : str
+
+        project_id : str
+
+        environment_id : str
+
+        trace_id : str
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        AsyncHttpResponse[TracesGetTraceDetailByEnvResponse]
+            Success
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            f"organizations/{jsonable_encoder(organization_id)}/projects/{jsonable_encoder(project_id)}/environments/{jsonable_encoder(environment_id)}/traces/{jsonable_encoder(trace_id)}",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                _data = typing.cast(
+                    TracesGetTraceDetailByEnvResponse,
+                    parse_obj_as(
+                        type_=TracesGetTraceDetailByEnvResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -998,9 +2235,9 @@ class AsyncRawTracesClient:
                 raise BadRequestError(
                     headers=dict(_response.headers),
                     body=typing.cast(
-                        HttpApiDecodeError,
+                        typing.Optional[typing.Any],
                         parse_obj_as(
-                            type_=HttpApiDecodeError,  # type: ignore
+                            type_=typing.Optional[typing.Any],  # type: ignore
                             object_=_response.json(),
                         ),
                     ),
@@ -1027,6 +2264,28 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 404:
+                raise NotFoundError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        NotFoundErrorBody,
+                        parse_obj_as(
+                            type_=NotFoundErrorBody,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
+            if _response.status_code == 429:
+                raise TooManyRequestsError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        RateLimitError,
+                        parse_obj_as(
+                            type_=RateLimitError,  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             if _response.status_code == 500:
                 raise InternalServerError(
                     headers=dict(_response.headers),
@@ -1038,6 +2297,17 @@ class AsyncRawTracesClient:
                         ),
                     ),
                 )
+            if _response.status_code == 503:
+                raise ServiceUnavailableError(
+                    headers=dict(_response.headers),
+                    body=typing.cast(
+                        typing.Optional[typing.Any],
+                        parse_obj_as(
+                            type_=typing.Optional[typing.Any],  # type: ignore
+                            object_=_response.json(),
+                        ),
+                    ),
+                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(