agenta 0.30.0a2__py3-none-any.whl → 0.30.0a4__py3-none-any.whl

agenta/client/backend/observability/client.py

@@ -66,7 +66,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/dashboard/",
+            "observability/dashboard",
             method="GET",
             params={
                 "app_id": app_id,
@@ -152,7 +152,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/trace/",
+            "observability/trace",
             method="POST",
             json={
                 "trace": trace,
@@ -162,6 +162,9 @@ class ObservabilityClient:
                     direction="write",
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -242,7 +245,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/traces/",
+            "observability/traces",
             method="GET",
             params={
                 "app_id": app_id,
@@ -312,7 +315,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/traces/",
+            "observability/traces",
             method="DELETE",
             json=request,
             request_options=request_options,
@@ -371,7 +374,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"observability/traces/{jsonable_encoder(trace_id)}/",
+            f"observability/traces/{jsonable_encoder(trace_id)}",
             method="GET",
             request_options=request_options,
         )
@@ -452,7 +455,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/spans/",
+            "observability/spans",
             method="GET",
             params={
                 "app_id": app_id,
@@ -522,7 +525,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            "observability/spans/",
+            "observability/spans",
             method="DELETE",
             json=request,
             request_options=request_options,
@@ -587,7 +590,7 @@ class ObservabilityClient:
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"observability/spans/{jsonable_encoder(span_id)}/",
+            f"observability/spans/{jsonable_encoder(span_id)}",
             method="GET",
             params={
                 "type": type,
@@ -672,7 +675,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/dashboard/",
+            "observability/dashboard",
             method="GET",
             params={
                 "app_id": app_id,
@@ -765,7 +768,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/trace/",
+            "observability/trace",
             method="POST",
             json={
                 "trace": trace,
@@ -775,6 +778,9 @@ class AsyncObservabilityClient:
                     direction="write",
                 ),
             },
+            headers={
+                "content-type": "application/json",
+            },
             request_options=request_options,
             omit=OMIT,
         )
@@ -863,7 +869,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/traces/",
+            "observability/traces",
             method="GET",
             params={
                 "app_id": app_id,
@@ -941,7 +947,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/traces/",
+            "observability/traces",
             method="DELETE",
             json=request,
             request_options=request_options,
@@ -1008,7 +1014,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"observability/traces/{jsonable_encoder(trace_id)}/",
+            f"observability/traces/{jsonable_encoder(trace_id)}",
             method="GET",
             request_options=request_options,
         )
@@ -1097,7 +1103,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/spans/",
+            "observability/spans",
             method="GET",
             params={
                 "app_id": app_id,
@@ -1175,7 +1181,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            "observability/spans/",
+            "observability/spans",
             method="DELETE",
             json=request,
             request_options=request_options,
@@ -1248,7 +1254,7 @@ class AsyncObservabilityClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"observability/spans/{jsonable_encoder(span_id)}/",
+            f"observability/spans/{jsonable_encoder(span_id)}",
             method="GET",
             params={
                 "type": type,

agenta/client/backend/observability_v_1/__init__.py

@@ -1,5 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .types import Format, QueryTracesResponse
+from .types import Format, QueryAnalyticsResponse, QueryTracesResponse
 
-__all__ = ["Format", "QueryTracesResponse"]
+__all__ = ["Format", "QueryAnalyticsResponse", "QueryTracesResponse"]

agenta/client/backend/observability_v_1/client.py

@@ -11,6 +11,7 @@ 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
 
 
@@ -270,6 +271,103 @@ class ObservabilityV1Client:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
+    def query_analytics(
+        self,
+        *,
+        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,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> QueryAnalyticsResponse:
+        """
+        Parameters
+        ----------
+        format : typing.Optional[Format]
+
+        focus : typing.Optional[str]
+
+        oldest : typing.Optional[str]
+
+        newest : typing.Optional[str]
+
+        window : typing.Optional[int]
+
+        filtering : typing.Optional[str]
+
+        time_range : typing.Optional[str]
+
+        app_id : typing.Optional[str]
+
+        environment : typing.Optional[str]
+
+        variant : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        QueryAnalyticsResponse
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.observability_v_1.query_analytics()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "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,
+                "environment": environment,
+                "variant": variant,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    QueryAnalyticsResponse,
+                    parse_obj_as(
+                        type_=QueryAnalyticsResponse,  # 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)
+
 
 class AsyncObservabilityV1Client:
     def __init__(self, *, client_wrapper: AsyncClientWrapper):
@@ -558,3 +656,108 @@ class AsyncObservabilityV1Client:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+    async def query_analytics(
+        self,
+        *,
+        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,
+        request_options: typing.Optional[RequestOptions] = None,
+    ) -> QueryAnalyticsResponse:
+        """
+        Parameters
+        ----------
+        format : typing.Optional[Format]
+
+        focus : typing.Optional[str]
+
+        oldest : typing.Optional[str]
+
+        newest : typing.Optional[str]
+
+        window : typing.Optional[int]
+
+        filtering : typing.Optional[str]
+
+        time_range : typing.Optional[str]
+
+        app_id : typing.Optional[str]
+
+        environment : typing.Optional[str]
+
+        variant : typing.Optional[str]
+
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        QueryAnalyticsResponse
+            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_v_1.query_analytics()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "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,
+                "environment": environment,
+                "variant": variant,
+            },
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    QueryAnalyticsResponse,
+                    parse_obj_as(
+                        type_=QueryAnalyticsResponse,  # 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)

agenta/client/backend/observability_v_1/types/__init__.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .format import Format
+from .query_analytics_response import QueryAnalyticsResponse
 from .query_traces_response import QueryTracesResponse
 
-__all__ = ["Format", "QueryTracesResponse"]
+__all__ = ["Format", "QueryAnalyticsResponse", "QueryTracesResponse"]

agenta/client/backend/observability_v_1/types/format.py

@@ -2,4 +2,4 @@
 
 import typing
 
-Format = typing.Union[typing.Literal["opentelemetry", "agenta"], typing.Any]
+Format = typing.Union[typing.Literal["legacy", "agenta"], typing.Any]

agenta/client/backend/observability_v_1/types/query_analytics_response.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+from ...types.legacy_analytics_response import LegacyAnalyticsResponse
+from ...types.analytics_response import AnalyticsResponse
+
+QueryAnalyticsResponse = typing.Union[LegacyAnalyticsResponse, AnalyticsResponse]

agenta/client/backend/scopes/__init__.py

@@ -0,0 +1 @@
+# This file was auto-generated by Fern from our API Definition.

agenta/client/backend/scopes/client.py

@@ -0,0 +1,114 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.client_wrapper import SyncClientWrapper
+import typing
+from ..core.request_options import RequestOptions
+from ..types.projects_response import ProjectsResponse
+from ..core.pydantic_utilities import parse_obj_as
+from json.decoder import JSONDecodeError
+from ..core.api_error import ApiError
+from ..core.client_wrapper import AsyncClientWrapper
+
+
+class ScopesClient:
+    def __init__(self, *, client_wrapper: SyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    def get_projects(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[ProjectsResponse]:
+        """
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[ProjectsResponse]
+            Successful Response
+
+        Examples
+        --------
+        from agenta import AgentaApi
+
+        client = AgentaApi(
+            api_key="YOUR_API_KEY",
+            base_url="https://yourhost.com/path/to/api",
+        )
+        client.scopes.get_projects()
+        """
+        _response = self._client_wrapper.httpx_client.request(
+            "projects",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[ProjectsResponse],
+                    parse_obj_as(
+                        type_=typing.List[ProjectsResponse],  # type: ignore
+                        object_=_response.json(),
+                    ),
+                )
+            _response_json = _response.json()
+        except JSONDecodeError:
+            raise ApiError(status_code=_response.status_code, body=_response.text)
+        raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+class AsyncScopesClient:
+    def __init__(self, *, client_wrapper: AsyncClientWrapper):
+        self._client_wrapper = client_wrapper
+
+    async def get_projects(
+        self, *, request_options: typing.Optional[RequestOptions] = None
+    ) -> typing.List[ProjectsResponse]:
+        """
+        Parameters
+        ----------
+        request_options : typing.Optional[RequestOptions]
+            Request-specific configuration.
+
+        Returns
+        -------
+        typing.List[ProjectsResponse]
+            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.scopes.get_projects()
+
+
+        asyncio.run(main())
+        """
+        _response = await self._client_wrapper.httpx_client.request(
+            "projects",
+            method="GET",
+            request_options=request_options,
+        )
+        try:
+            if 200 <= _response.status_code < 300:
+                return typing.cast(
+                    typing.List[ProjectsResponse],
+                    parse_obj_as(
+                        type_=typing.List[ProjectsResponse],  # 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)