agenta 0.33.0a1__py3-none-any.whl → 0.33.0a3__py3-none-any.whl

agenta/client/backend/observability/client.py

@@ -1,56 +1,38 @@
 # This file was auto-generated by Fern from our API Definition.
 
-import typing
 from ..core.client_wrapper import SyncClientWrapper
+import typing
 from ..core.request_options import RequestOptions
+from ..types.collect_status_response import CollectStatusResponse
 from ..core.pydantic_utilities import parse_obj_as
-from ..errors.unprocessable_entity_error import UnprocessableEntityError
-from ..types.http_validation_error import HttpValidationError
 from json.decoder import JSONDecodeError
 from ..core.api_error import ApiError
-from ..types.create_span import CreateSpan
-from ..types.create_trace_response import CreateTraceResponse
-from ..core.serialization import convert_and_respect_annotation_metadata
-from ..types.with_pagination import WithPagination
-from ..types.trace_detail import TraceDetail
-from ..core.jsonable_encoder import jsonable_encoder
-from ..types.span_detail import SpanDetail
+from .types.format import Format
+from .types.query_traces_response import QueryTracesResponse
+from ..errors.unprocessable_entity_error import UnprocessableEntityError
+from ..types.http_validation_error import HttpValidationError
+from .types.query_analytics_response import QueryAnalyticsResponse
 from ..core.client_wrapper import AsyncClientWrapper
 
-# this is used as the default value for optional parameters
-OMIT = typing.cast(typing.Any, ...)
-
 
 class ObservabilityClient:
     def __init__(self, *, client_wrapper: SyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    def dashboard(
-        self,
-        *,
-        app_id: str,
-        time_range: typing.Optional[str] = None,
-        environment: typing.Optional[str] = None,
-        variant: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    def otlp_status(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> CollectStatusResponse:
         """
+        Status of OTLP endpoint.
+
         Parameters
         ----------
-        app_id : str
-
-        time_range : typing.Optional[str]
-
-        environment : typing.Optional[str]
-
-        variant : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        CollectStatusResponse
             Successful Response
 
         Examples
@@ -61,175 +43,115 @@ class ObservabilityClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.dashboard(
-            app_id="app_id",
-        )
+        client.observability.otlp_status()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/dashboard",
+            "observability/v1/otlp/traces",
             method="GET",
-            params={
-                "app_id": app_id,
-                "timeRange": time_range,
-                "environment": environment,
-                "variant": variant,
-            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.Optional[typing.Any],
+                    CollectStatusResponse,
                     parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
+                        type_=CollectStatusResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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 create_traces(
-        self,
-        *,
-        trace: str,
-        spans: typing.Sequence[CreateSpan],
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> CreateTraceResponse:
+    def otlp_receiver(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> CollectStatusResponse:
         """
+        Receive traces via OTLP.
+
         Parameters
         ----------
-        trace : str
-
-        spans : typing.Sequence[CreateSpan]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        CreateTraceResponse
+        CollectStatusResponse
             Successful Response
 
         Examples
         --------
-        import datetime
-
-        from agenta import AgentaApi, CreateSpan
+        from agenta import AgentaApi
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.create_traces(
-            trace="trace",
-            spans=[
-                CreateSpan(
-                    id="id",
-                    app_id="app_id",
-                    name="name",
-                    spankind="spankind",
-                    status="status",
-                    start_time=datetime.datetime.fromisoformat(
-                        "2024-01-15 09:30:00+00:00",
-                    ),
-                    end_time=datetime.datetime.fromisoformat(
-                        "2024-01-15 09:30:00+00:00",
-                    ),
-                )
-            ],
-        )
+        client.observability.otlp_receiver()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/trace",
+            "observability/v1/otlp/traces",
             method="POST",
-            json={
-                "trace": trace,
-                "spans": convert_and_respect_annotation_metadata(
-                    object_=spans,
-                    annotation=typing.Sequence[CreateSpan],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    CreateTraceResponse,
+                    CollectStatusResponse,
                     parse_obj_as(
-                        type_=CreateTraceResponse,  # type: ignore
+                        type_=CollectStatusResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
             _response_json = _response.json()
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_traces(
+    def query_traces(
         self,
         *,
-        app_id: str,
+        format: typing.Optional[Format] = None,
+        focus: typing.Optional[str] = None,
+        oldest: typing.Optional[str] = None,
+        newest: typing.Optional[str] = None,
+        filtering: typing.Optional[str] = None,
         page: typing.Optional[int] = None,
-        page_size: typing.Optional[int] = None,
-        type: typing.Optional[str] = None,
-        trace_id: typing.Optional[str] = None,
-        environment: typing.Optional[str] = None,
-        variant: typing.Optional[str] = None,
-        created_at: typing.Optional[str] = None,
+        size: typing.Optional[int] = None,
+        next: typing.Optional[str] = None,
+        stop: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> WithPagination:
+    ) -> QueryTracesResponse:
         """
+        Query traces, with optional grouping, windowing, filtering, and pagination.
+
         Parameters
         ----------
-        app_id : str
+        format : typing.Optional[Format]
 
-        page : typing.Optional[int]
+        focus : typing.Optional[str]
 
-        page_size : typing.Optional[int]
+        oldest : typing.Optional[str]
 
-        type : typing.Optional[str]
+        newest : typing.Optional[str]
 
-        trace_id : typing.Optional[str]
+        filtering : typing.Optional[str]
 
-        environment : typing.Optional[str]
+        page : typing.Optional[int]
 
-        variant : typing.Optional[str]
+        size : typing.Optional[int]
 
-        created_at : typing.Optional[str]
+        next : typing.Optional[str]
+
+        stop : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        WithPagination
+        QueryTracesResponse
             Successful Response
 
         Examples
@@ -240,31 +162,30 @@ class ObservabilityClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.get_traces(
-            app_id="app_id",
-        )
+        client.observability.query_traces()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/traces",
+            "observability/v1/traces",
             method="GET",
             params={
-                "app_id": app_id,
+                "format": format,
+                "focus": focus,
+                "oldest": oldest,
+                "newest": newest,
+                "filtering": filtering,
                 "page": page,
-                "pageSize": page_size,
-                "type": type,
-                "trace_id": trace_id,
-                "environment": environment,
-                "variant": variant,
-                "created_at": created_at,
+                "size": size,
+                "next": next,
+                "stop": stop,
             },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    WithPagination,
+                    QueryTracesResponse,
                     parse_obj_as(
-                        type_=WithPagination,  # type: ignore
+                        type_=QueryTracesResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -283,82 +204,28 @@ class ObservabilityClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def delete_traces_legacy(
+    def delete_traces(
         self,
         *,
-        request: typing.Sequence[str],
+        node_id: typing.Optional[str] = None,
+        node_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> bool:
+    ) -> CollectStatusResponse:
         """
-        Parameters
-        ----------
-        request : typing.Sequence[str]
+        Delete trace.
 
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        bool
-            Successful Response
-
-        Examples
-        --------
-        from agenta import AgentaApi
-
-        client = AgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.observability.delete_traces_legacy(
-            request=["string"],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "observability/traces",
-            method="DELETE",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    bool,
-                    parse_obj_as(
-                        type_=bool,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def get_trace_detail(
-        self, trace_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> TraceDetail:
-        """
         Parameters
         ----------
-        trace_id : str
+        node_id : typing.Optional[str]
+
+        node_ids : typing.Optional[typing.Union[str, typing.Sequence[str]]]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        TraceDetail
+        CollectStatusResponse
             Successful Response
 
         Examples
@@ -369,21 +236,23 @@ class ObservabilityClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.get_trace_detail(
-            trace_id="trace_id",
-        )
+        client.observability.delete_traces()
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"observability/traces/{jsonable_encoder(trace_id)}",
-            method="GET",
+            "observability/v1/traces",
+            method="DELETE",
+            params={
+                "node_id": node_id,
+                "node_ids": node_ids,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    TraceDetail,
+                    CollectStatusResponse,
                     parse_obj_as(
-                        type_=TraceDetail,  # type: ignore
+                        type_=CollectStatusResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -402,44 +271,50 @@ class ObservabilityClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_spans_of_generation(
+    def query_analytics(
         self,
         *,
-        app_id: str,
-        page: typing.Optional[int] = None,
-        page_size: typing.Optional[int] = None,
-        type: typing.Optional[str] = None,
-        trace_id: typing.Optional[str] = None,
+        format: typing.Optional[Format] = None,
+        focus: typing.Optional[str] = None,
+        oldest: typing.Optional[str] = None,
+        newest: typing.Optional[str] = None,
+        window: typing.Optional[int] = None,
+        filtering: typing.Optional[str] = None,
+        time_range: typing.Optional[str] = None,
+        app_id: typing.Optional[str] = None,
         environment: typing.Optional[str] = None,
         variant: typing.Optional[str] = None,
-        created_at: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> QueryAnalyticsResponse:
         """
         Parameters
         ----------
-        app_id : str
+        format : typing.Optional[Format]
 
-        page : typing.Optional[int]
+        focus : typing.Optional[str]
+
+        oldest : typing.Optional[str]
 
-        page_size : typing.Optional[int]
+        newest : typing.Optional[str]
 
-        type : typing.Optional[str]
+        window : typing.Optional[int]
 
-        trace_id : typing.Optional[str]
+        filtering : typing.Optional[str]
+
+        time_range : typing.Optional[str]
+
+        app_id : typing.Optional[str]
 
         environment : typing.Optional[str]
 
         variant : typing.Optional[str]
 
-        created_at : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        QueryAnalyticsResponse
             Successful Response
 
         Examples
@@ -450,159 +325,31 @@ class ObservabilityClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.get_spans_of_generation(
-            app_id="app_id",
-        )
+        client.observability.query_analytics()
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/spans",
+            "observability/v1/analytics",
             method="GET",
             params={
+                "format": format,
+                "focus": focus,
+                "oldest": oldest,
+                "newest": newest,
+                "window": window,
+                "filtering": filtering,
+                "timeRange": time_range,
                 "app_id": app_id,
-                "page": page,
-                "pageSize": page_size,
-                "type": type,
-                "trace_id": trace_id,
                 "environment": environment,
                 "variant": variant,
-                "created_at": created_at,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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_spans_of_trace(
-        self,
-        *,
-        request: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> bool:
-        """
-        Parameters
-        ----------
-        request : typing.Sequence[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        bool
-            Successful Response
-
-        Examples
-        --------
-        from agenta import AgentaApi
-
-        client = AgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.observability.delete_spans_of_trace(
-            request=["string"],
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "observability/spans",
-            method="DELETE",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    bool,
-                    parse_obj_as(
-                        type_=bool,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # type: ignore
-                            object_=_response.json(),
-                        ),
-                    )
-                )
-            _response_json = _response.json()
-        except JSONDecodeError:
-            raise ApiError(status_code=_response.status_code, body=_response.text)
-        raise ApiError(status_code=_response.status_code, body=_response_json)
-
-    def get_span_of_generation(
-        self,
-        span_id: str,
-        *,
-        type: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> SpanDetail:
-        """
-        Parameters
-        ----------
-        span_id : str
-
-        type : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        SpanDetail
-            Successful Response
-
-        Examples
-        --------
-        from agenta import AgentaApi
-
-        client = AgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-        client.observability.get_span_of_generation(
-            span_id="span_id",
-        )
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            f"observability/spans/{jsonable_encoder(span_id)}",
-            method="GET",
-            params={
-                "type": type,
             },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    SpanDetail,
+                    QueryAnalyticsResponse,
                     parse_obj_as(
-                        type_=SpanDetail,  # type: ignore
+                        type_=QueryAnalyticsResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -626,32 +373,20 @@ class AsyncObservabilityClient:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
         self._client_wrapper = client_wrapper
 
-    async def dashboard(
-        self,
-        *,
-        app_id: str,
-        time_range: typing.Optional[str] = None,
-        environment: typing.Optional[str] = None,
-        variant: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    async def otlp_status(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> CollectStatusResponse:
         """
+        Status of OTLP endpoint.
+
         Parameters
         ----------
-        app_id : str
-
-        time_range : typing.Optional[str]
-
-        environment : typing.Optional[str]
-
-        variant : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        CollectStatusResponse
             Successful Response
 
         Examples
@@ -667,76 +402,51 @@ class AsyncObservabilityClient:
 
 
         async def main() -> None:
-            await client.observability.dashboard(
-                app_id="app_id",
-            )
+            await client.observability.otlp_status()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/dashboard",
+            "observability/v1/otlp/traces",
             method="GET",
-            params={
-                "app_id": app_id,
-                "timeRange": time_range,
-                "environment": environment,
-                "variant": variant,
-            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    typing.Optional[typing.Any],
+                    CollectStatusResponse,
                     parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
+                        type_=CollectStatusResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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 create_traces(
-        self,
-        *,
-        trace: str,
-        spans: typing.Sequence[CreateSpan],
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> CreateTraceResponse:
+    async def otlp_receiver(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> CollectStatusResponse:
         """
+        Receive traces via OTLP.
+
         Parameters
         ----------
-        trace : str
-
-        spans : typing.Sequence[CreateSpan]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        CreateTraceResponse
+        CollectStatusResponse
             Successful Response
 
         Examples
         --------
         import asyncio
-        import datetime
 
-        from agenta import AsyncAgentaApi, CreateSpan
+        from agenta import AsyncAgentaApi
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
@@ -745,107 +455,73 @@ class AsyncObservabilityClient:
 
 
         async def main() -> None:
-            await client.observability.create_traces(
-                trace="trace",
-                spans=[
-                    CreateSpan(
-                        id="id",
-                        app_id="app_id",
-                        name="name",
-                        spankind="spankind",
-                        status="status",
-                        start_time=datetime.datetime.fromisoformat(
-                            "2024-01-15 09:30:00+00:00",
-                        ),
-                        end_time=datetime.datetime.fromisoformat(
-                            "2024-01-15 09:30:00+00:00",
-                        ),
-                    )
-                ],
-            )
+            await client.observability.otlp_receiver()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/trace",
+            "observability/v1/otlp/traces",
             method="POST",
-            json={
-                "trace": trace,
-                "spans": convert_and_respect_annotation_metadata(
-                    object_=spans,
-                    annotation=typing.Sequence[CreateSpan],
-                    direction="write",
-                ),
-            },
-            headers={
-                "content-type": "application/json",
-            },
             request_options=request_options,
-            omit=OMIT,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    CreateTraceResponse,
+                    CollectStatusResponse,
                     parse_obj_as(
-                        type_=CreateTraceResponse,  # type: ignore
+                        type_=CollectStatusResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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 get_traces(
+    async def query_traces(
         self,
         *,
-        app_id: str,
+        format: typing.Optional[Format] = None,
+        focus: typing.Optional[str] = None,
+        oldest: typing.Optional[str] = None,
+        newest: typing.Optional[str] = None,
+        filtering: typing.Optional[str] = None,
         page: typing.Optional[int] = None,
-        page_size: typing.Optional[int] = None,
-        type: typing.Optional[str] = None,
-        trace_id: typing.Optional[str] = None,
-        environment: typing.Optional[str] = None,
-        variant: typing.Optional[str] = None,
-        created_at: typing.Optional[str] = None,
+        size: typing.Optional[int] = None,
+        next: typing.Optional[str] = None,
+        stop: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> WithPagination:
+    ) -> QueryTracesResponse:
         """
+        Query traces, with optional grouping, windowing, filtering, and pagination.
+
         Parameters
         ----------
-        app_id : str
+        format : typing.Optional[Format]
 
-        page : typing.Optional[int]
+        focus : typing.Optional[str]
 
-        page_size : typing.Optional[int]
+        oldest : typing.Optional[str]
 
-        type : typing.Optional[str]
+        newest : typing.Optional[str]
 
-        trace_id : typing.Optional[str]
+        filtering : typing.Optional[str]
 
-        environment : typing.Optional[str]
+        page : typing.Optional[int]
 
-        variant : typing.Optional[str]
+        size : typing.Optional[int]
+
+        next : typing.Optional[str]
 
-        created_at : typing.Optional[str]
+        stop : typing.Optional[str]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        WithPagination
+        QueryTracesResponse
             Successful Response
 
         Examples
@@ -861,34 +537,33 @@ class AsyncObservabilityClient:
 
 
         async def main() -> None:
-            await client.observability.get_traces(
-                app_id="app_id",
-            )
+            await client.observability.query_traces()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/traces",
+            "observability/v1/traces",
             method="GET",
             params={
-                "app_id": app_id,
+                "format": format,
+                "focus": focus,
+                "oldest": oldest,
+                "newest": newest,
+                "filtering": filtering,
                 "page": page,
-                "pageSize": page_size,
-                "type": type,
-                "trace_id": trace_id,
-                "environment": environment,
-                "variant": variant,
-                "created_at": created_at,
+                "size": size,
+                "next": next,
+                "stop": stop,
             },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    WithPagination,
+                    QueryTracesResponse,
                     parse_obj_as(
-                        type_=WithPagination,  # type: ignore
+                        type_=QueryTracesResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -907,90 +582,28 @@ class AsyncObservabilityClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def delete_traces_legacy(
+    async def delete_traces(
         self,
         *,
-        request: typing.Sequence[str],
+        node_id: typing.Optional[str] = None,
+        node_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> bool:
-        """
-        Parameters
-        ----------
-        request : typing.Sequence[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        bool
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from agenta import AsyncAgentaApi
-
-        client = AsyncAgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-
-
-        async def main() -> None:
-            await client.observability.delete_traces_legacy(
-                request=["string"],
-            )
-
-
-        asyncio.run(main())
+    ) -> CollectStatusResponse:
         """
-        _response = await self._client_wrapper.httpx_client.request(
-            "observability/traces",
-            method="DELETE",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    bool,
-                    parse_obj_as(
-                        type_=bool,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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)
+        Delete trace.
 
-    async def get_trace_detail(
-        self, trace_id: str, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> TraceDetail:
-        """
         Parameters
         ----------
-        trace_id : str
+        node_id : typing.Optional[str]
+
+        node_ids : typing.Optional[typing.Union[str, typing.Sequence[str]]]
 
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        TraceDetail
+        CollectStatusResponse
             Successful Response
 
         Examples
@@ -1006,24 +619,26 @@ class AsyncObservabilityClient:
 
 
         async def main() -> None:
-            await client.observability.get_trace_detail(
-                trace_id="trace_id",
-            )
+            await client.observability.delete_traces()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"observability/traces/{jsonable_encoder(trace_id)}",
-            method="GET",
+            "observability/v1/traces",
+            method="DELETE",
+            params={
+                "node_id": node_id,
+                "node_ids": node_ids,
+            },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    TraceDetail,
+                    CollectStatusResponse,
                     parse_obj_as(
-                        type_=TraceDetail,  # type: ignore
+                        type_=CollectStatusResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )
@@ -1042,44 +657,50 @@ class AsyncObservabilityClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_spans_of_generation(
+    async def query_analytics(
         self,
         *,
-        app_id: str,
-        page: typing.Optional[int] = None,
-        page_size: typing.Optional[int] = None,
-        type: typing.Optional[str] = None,
-        trace_id: typing.Optional[str] = None,
+        format: typing.Optional[Format] = None,
+        focus: typing.Optional[str] = None,
+        oldest: typing.Optional[str] = None,
+        newest: typing.Optional[str] = None,
+        window: typing.Optional[int] = None,
+        filtering: typing.Optional[str] = None,
+        time_range: typing.Optional[str] = None,
+        app_id: typing.Optional[str] = None,
         environment: typing.Optional[str] = None,
         variant: typing.Optional[str] = None,
-        created_at: typing.Optional[str] = None,
         request_options: typing.Optional[RequestOptions] = None,
-    ) -> typing.Optional[typing.Any]:
+    ) -> QueryAnalyticsResponse:
         """
         Parameters
         ----------
-        app_id : str
+        format : typing.Optional[Format]
 
-        page : typing.Optional[int]
+        focus : typing.Optional[str]
+
+        oldest : typing.Optional[str]
+
+        newest : typing.Optional[str]
 
-        page_size : typing.Optional[int]
+        window : typing.Optional[int]
 
-        type : typing.Optional[str]
+        filtering : typing.Optional[str]
 
-        trace_id : typing.Optional[str]
+        time_range : typing.Optional[str]
+
+        app_id : typing.Optional[str]
 
         environment : typing.Optional[str]
 
         variant : typing.Optional[str]
 
-        created_at : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
         Returns
         -------
-        typing.Optional[typing.Any]
+        QueryAnalyticsResponse
             Successful Response
 
         Examples
@@ -1095,178 +716,34 @@ class AsyncObservabilityClient:
 
 
         async def main() -> None:
-            await client.observability.get_spans_of_generation(
-                app_id="app_id",
-            )
+            await client.observability.query_analytics()
 
 
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/spans",
+            "observability/v1/analytics",
             method="GET",
             params={
+                "format": format,
+                "focus": focus,
+                "oldest": oldest,
+                "newest": newest,
+                "window": window,
+                "filtering": filtering,
+                "timeRange": time_range,
                 "app_id": app_id,
-                "page": page,
-                "pageSize": page_size,
-                "type": type,
-                "trace_id": trace_id,
                 "environment": environment,
                 "variant": variant,
-                "created_at": created_at,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    typing.Optional[typing.Any],
-                    parse_obj_as(
-                        type_=typing.Optional[typing.Any],  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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_spans_of_trace(
-        self,
-        *,
-        request: typing.Sequence[str],
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> bool:
-        """
-        Parameters
-        ----------
-        request : typing.Sequence[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        bool
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from agenta import AsyncAgentaApi
-
-        client = AsyncAgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-
-
-        async def main() -> None:
-            await client.observability.delete_spans_of_trace(
-                request=["string"],
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "observability/spans",
-            method="DELETE",
-            json=request,
-            request_options=request_options,
-            omit=OMIT,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    bool,
-                    parse_obj_as(
-                        type_=bool,  # type: ignore
-                        object_=_response.json(),
-                    ),
-                )
-            if _response.status_code == 422:
-                raise UnprocessableEntityError(
-                    typing.cast(
-                        HttpValidationError,
-                        parse_obj_as(
-                            type_=HttpValidationError,  # 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 get_span_of_generation(
-        self,
-        span_id: str,
-        *,
-        type: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> SpanDetail:
-        """
-        Parameters
-        ----------
-        span_id : str
-
-        type : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        SpanDetail
-            Successful Response
-
-        Examples
-        --------
-        import asyncio
-
-        from agenta import AsyncAgentaApi
-
-        client = AsyncAgentaApi(
-            api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
-        )
-
-
-        async def main() -> None:
-            await client.observability.get_span_of_generation(
-                span_id="span_id",
-            )
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            f"observability/spans/{jsonable_encoder(span_id)}",
-            method="GET",
-            params={
-                "type": type,
             },
             request_options=request_options,
         )
         try:
             if 200 <= _response.status_code < 300:
                 return typing.cast(
-                    SpanDetail,
+                    QueryAnalyticsResponse,
                     parse_obj_as(
-                        type_=SpanDetail,  # type: ignore
+                        type_=QueryAnalyticsResponse,  # type: ignore
                         object_=_response.json(),
                     ),
                 )