agenta 0.27.7__py3-none-any.whl → 0.27.7a0__py3-none-any.whl

agenta/client/backend/observability_v_1/client.py

@@ -1,560 +0,0 @@
-# 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.collect_status_response import CollectStatusResponse
-from ..core.pydantic_utilities import parse_obj_as
-from json.decoder import JSONDecodeError
-from ..core.api_error import ApiError
-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 ..core.client_wrapper import AsyncClientWrapper
-
-
-class ObservabilityV1Client:
-    def __init__(self, *, client_wrapper: SyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    def otlp_status(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> CollectStatusResponse:
-        """
-        Status of OTLP endpoint.
-
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        CollectStatusResponse
-            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.otlp_status()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "observability/v1/otlp/traces",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    CollectStatusResponse,
-                    parse_obj_as(
-                        type_=CollectStatusResponse,  # 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 otlp_receiver(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> CollectStatusResponse:
-        """
-        Receive traces via OTLP.
-
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        CollectStatusResponse
-            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.otlp_receiver()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "observability/v1/otlp/traces",
-            method="POST",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    CollectStatusResponse,
-                    parse_obj_as(
-                        type_=CollectStatusResponse,  # 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 query_traces(
-        self,
-        *,
-        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,
-        size: typing.Optional[int] = None,
-        next: typing.Optional[str] = None,
-        stop: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> QueryTracesResponse:
-        """
-        Query traces, with optional grouping, windowing, filtering, and pagination.
-
-        Parameters
-        ----------
-        format : typing.Optional[Format]
-
-        focus : typing.Optional[str]
-
-        oldest : typing.Optional[str]
-
-        newest : typing.Optional[str]
-
-        filtering : typing.Optional[str]
-
-        page : typing.Optional[int]
-
-        size : typing.Optional[int]
-
-        next : typing.Optional[str]
-
-        stop : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        QueryTracesResponse
-            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_traces()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "observability/v1/traces",
-            method="GET",
-            params={
-                "format": format,
-                "focus": focus,
-                "oldest": oldest,
-                "newest": newest,
-                "filtering": filtering,
-                "page": page,
-                "size": size,
-                "next": next,
-                "stop": stop,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    QueryTracesResponse,
-                    parse_obj_as(
-                        type_=QueryTracesResponse,  # 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_traces(
-        self,
-        *,
-        node_id: typing.Optional[str] = None,
-        node_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> CollectStatusResponse:
-        """
-        Delete trace.
-
-        Parameters
-        ----------
-        node_id : typing.Optional[str]
-
-        node_ids : typing.Optional[typing.Union[str, typing.Sequence[str]]]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        CollectStatusResponse
-            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.delete_traces()
-        """
-        _response = self._client_wrapper.httpx_client.request(
-            "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(
-                    CollectStatusResponse,
-                    parse_obj_as(
-                        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)
-
-
-class AsyncObservabilityV1Client:
-    def __init__(self, *, client_wrapper: AsyncClientWrapper):
-        self._client_wrapper = client_wrapper
-
-    async def otlp_status(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> CollectStatusResponse:
-        """
-        Status of OTLP endpoint.
-
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        CollectStatusResponse
-            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.otlp_status()
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "observability/v1/otlp/traces",
-            method="GET",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    CollectStatusResponse,
-                    parse_obj_as(
-                        type_=CollectStatusResponse,  # 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 otlp_receiver(
-        self, *, request_options: typing.Optional[RequestOptions] = None
-    ) -> CollectStatusResponse:
-        """
-        Receive traces via OTLP.
-
-        Parameters
-        ----------
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        CollectStatusResponse
-            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.otlp_receiver()
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "observability/v1/otlp/traces",
-            method="POST",
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    CollectStatusResponse,
-                    parse_obj_as(
-                        type_=CollectStatusResponse,  # 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 query_traces(
-        self,
-        *,
-        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,
-        size: typing.Optional[int] = None,
-        next: typing.Optional[str] = None,
-        stop: typing.Optional[str] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> QueryTracesResponse:
-        """
-        Query traces, with optional grouping, windowing, filtering, and pagination.
-
-        Parameters
-        ----------
-        format : typing.Optional[Format]
-
-        focus : typing.Optional[str]
-
-        oldest : typing.Optional[str]
-
-        newest : typing.Optional[str]
-
-        filtering : typing.Optional[str]
-
-        page : typing.Optional[int]
-
-        size : typing.Optional[int]
-
-        next : typing.Optional[str]
-
-        stop : typing.Optional[str]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        QueryTracesResponse
-            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_traces()
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "observability/v1/traces",
-            method="GET",
-            params={
-                "format": format,
-                "focus": focus,
-                "oldest": oldest,
-                "newest": newest,
-                "filtering": filtering,
-                "page": page,
-                "size": size,
-                "next": next,
-                "stop": stop,
-            },
-            request_options=request_options,
-        )
-        try:
-            if 200 <= _response.status_code < 300:
-                return typing.cast(
-                    QueryTracesResponse,
-                    parse_obj_as(
-                        type_=QueryTracesResponse,  # 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_traces(
-        self,
-        *,
-        node_id: typing.Optional[str] = None,
-        node_ids: typing.Optional[typing.Union[str, typing.Sequence[str]]] = None,
-        request_options: typing.Optional[RequestOptions] = None,
-    ) -> CollectStatusResponse:
-        """
-        Delete trace.
-
-        Parameters
-        ----------
-        node_id : typing.Optional[str]
-
-        node_ids : typing.Optional[typing.Union[str, typing.Sequence[str]]]
-
-        request_options : typing.Optional[RequestOptions]
-            Request-specific configuration.
-
-        Returns
-        -------
-        CollectStatusResponse
-            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.delete_traces()
-
-
-        asyncio.run(main())
-        """
-        _response = await self._client_wrapper.httpx_client.request(
-            "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(
-                    CollectStatusResponse,
-                    parse_obj_as(
-                        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)

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

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

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

@@ -1,5 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-
-Format = typing.Union[typing.Literal["opentelemetry", "agenta"], typing.Any]

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

@@ -1,11 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from ...types.o_tel_spans_response import OTelSpansResponse
-from ...types.agenta_nodes_response import AgentaNodesResponse
-from ...types.agenta_trees_response import AgentaTreesResponse
-from ...types.agenta_roots_response import AgentaRootsResponse
-
-QueryTracesResponse = typing.Union[
-    OTelSpansResponse, AgentaNodesResponse, AgentaTreesResponse, AgentaRootsResponse
-]

agenta/client/backend/types/agenta_node_dto.py

@@ -1,48 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from ..core.pydantic_utilities import UniversalBaseModel
-import typing
-from .lifecycle_dto import LifecycleDto
-from .root_dto import RootDto
-from .tree_dto import TreeDto
-from .node_dto import NodeDto
-from .parent_dto import ParentDto
-from .time_dto import TimeDto
-from .status_dto import StatusDto
-from .exception_dto import ExceptionDto
-from .link_dto import LinkDto
-from .o_tel_extra_dto import OTelExtraDto
-from .agenta_node_dto_nodes_value import AgentaNodeDtoNodesValue
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-
-
-class AgentaNodeDto(UniversalBaseModel):
-    lifecycle: typing.Optional[LifecycleDto] = None
-    root: RootDto
-    tree: TreeDto
-    node: NodeDto
-    parent: typing.Optional[ParentDto] = None
-    time: TimeDto
-    status: StatusDto
-    exception: typing.Optional[ExceptionDto] = None
-    data: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
-    metrics: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
-    meta: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
-    refs: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
-    links: typing.Optional[typing.List[LinkDto]] = None
-    otel: typing.Optional[OTelExtraDto] = None
-    nodes: typing.Optional[
-        typing.Dict[str, typing.Optional[AgentaNodeDtoNodesValue]]
-    ] = None
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow

agenta/client/backend/types/agenta_node_dto_nodes_value.py

@@ -1,6 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-import typing
-from .span_dto import SpanDto
-
-AgentaNodeDtoNodesValue = typing.Union[SpanDto, typing.List[SpanDto]]

agenta/client/backend/types/agenta_nodes_response.py

@@ -1,30 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_dto import SpanDto
-import typing
-from .agenta_node_dto import AgentaNodeDto
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class AgentaNodesResponse(UniversalBaseModel):
-    nodes: typing.List[AgentaNodeDto]
-    version: str
-    count: typing.Optional[int] = None
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaNodesResponse=AgentaNodesResponse)

agenta/client/backend/types/agenta_root_dto.py

@@ -1,30 +0,0 @@
-# This file was auto-generated by Fern from our API Definition.
-
-from __future__ import annotations
-from ..core.pydantic_utilities import UniversalBaseModel
-from .span_dto import SpanDto
-from .root_dto import RootDto
-import typing
-from .agenta_tree_dto import AgentaTreeDto
-from ..core.pydantic_utilities import IS_PYDANTIC_V2
-import pydantic
-from ..core.pydantic_utilities import update_forward_refs
-
-
-class AgentaRootDto(UniversalBaseModel):
-    root: RootDto
-    trees: typing.List[AgentaTreeDto]
-
-    if IS_PYDANTIC_V2:
-        model_config: typing.ClassVar[pydantic.ConfigDict] = pydantic.ConfigDict(
-            extra="allow", frozen=True
-        )  # type: ignore # Pydantic v2
-    else:
-
-        class Config:
-            frozen = True
-            smart_union = True
-            extra = pydantic.Extra.allow
-
-
-update_forward_refs(SpanDto, AgentaRootDto=AgentaRootDto)