agenta 0.27.6a2__py3-none-any.whl → 0.27.7__py3-none-any.whl

agenta/client/backend/types/node_type.py

@@ -0,0 +1,19 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+NodeType = typing.Union[
+    typing.Literal[
+        "agent",
+        "workflow",
+        "chain",
+        "task",
+        "tool",
+        "embedding",
+        "query",
+        "completion",
+        "chat",
+        "rerank",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/o_tel_context_dto.py

@@ -0,0 +1,22 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class OTelContextDto(UniversalBaseModel):
+    trace_id: str
+    span_id: str
+
+    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/o_tel_event_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelEventDto(UniversalBaseModel):
+    name: str
+    timestamp: str
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = 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/o_tel_extra_dto.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .o_tel_event_dto import OTelEventDto
+from .o_tel_link_dto import OTelLinkDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelExtraDto(UniversalBaseModel):
+    kind: typing.Optional[str] = None
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    events: typing.Optional[typing.List[OTelEventDto]] = None
+    links: typing.Optional[typing.List[OTelLinkDto]] = 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/o_tel_link_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .o_tel_context_dto import OTelContextDto
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelLinkDto(UniversalBaseModel):
+    context: OTelContextDto
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = 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/o_tel_span_dto.py

@@ -0,0 +1,37 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .o_tel_context_dto import OTelContextDto
+import typing
+from .o_tel_span_kind import OTelSpanKind
+import datetime as dt
+from .o_tel_status_code import OTelStatusCode
+from .o_tel_event_dto import OTelEventDto
+from .o_tel_link_dto import OTelLinkDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelSpanDto(UniversalBaseModel):
+    context: OTelContextDto
+    name: str
+    kind: typing.Optional[OTelSpanKind] = None
+    start_time: dt.datetime
+    end_time: dt.datetime
+    status_code: typing.Optional[OTelStatusCode] = None
+    status_message: typing.Optional[str] = None
+    attributes: typing.Optional[typing.Dict[str, typing.Optional[typing.Any]]] = None
+    events: typing.Optional[typing.List[OTelEventDto]] = None
+    parent: typing.Optional[OTelContextDto] = None
+    links: typing.Optional[typing.List[OTelLinkDto]] = 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/o_tel_span_kind.py

@@ -0,0 +1,15 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+OTelSpanKind = typing.Union[
+    typing.Literal[
+        "SPAN_KIND_UNSPECIFIED",
+        "SPAN_KIND_INTERNAL",
+        "SPAN_KIND_SERVER",
+        "SPAN_KIND_CLIENT",
+        "SPAN_KIND_PRODUCER",
+        "SPAN_KIND_CONSUMER",
+    ],
+    typing.Any,
+]

agenta/client/backend/types/o_tel_spans_response.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .o_tel_span_dto import OTelSpanDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class OTelSpansResponse(UniversalBaseModel):
+    version: str
+    count: typing.Optional[int] = None
+    spans: typing.List[OTelSpanDto]
+
+    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/o_tel_status_code.py

@@ -0,0 +1,8 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+OTelStatusCode = typing.Union[
+    typing.Literal["STATUS_CODE_OK", "STATUS_CODE_ERROR", "STATUS_CODE_UNSET"],
+    typing.Any,
+]

agenta/client/backend/types/parent_dto.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class ParentDto(UniversalBaseModel):
+    id: str
+
+    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/root_dto.py

@@ -0,0 +1,21 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class RootDto(UniversalBaseModel):
+    id: str
+
+    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/span_dto.py

@@ -0,0 +1,54 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+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 ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class SpanDto(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["SpanDtoNodesValue"]]
+    ] = 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
+
+
+from .span_dto_nodes_value import SpanDtoNodesValue  # noqa: E402
+
+update_forward_refs(SpanDto)

agenta/client/backend/types/span_dto_nodes_value.py

@@ -0,0 +1,9 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from __future__ import annotations
+import typing
+import typing
+
+if typing.TYPE_CHECKING:
+    from .span_dto import SpanDto
+SpanDtoNodesValue = typing.Union["SpanDto", typing.List["SpanDto"]]

agenta/client/backend/types/status_code.py

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

agenta/client/backend/types/status_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .status_code import StatusCode
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class StatusDto(UniversalBaseModel):
+    code: StatusCode
+    message: typing.Optional[str] = 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/time_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import typing
+import pydantic
+
+
+class TimeDto(UniversalBaseModel):
+    start: dt.datetime
+    end: dt.datetime
+
+    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/tree_dto.py

@@ -0,0 +1,23 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import typing
+from .tree_type import TreeType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class TreeDto(UniversalBaseModel):
+    id: str
+    type: typing.Optional[TreeType] = 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/tree_type.py

@@ -0,0 +1,5 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import typing
+
+TreeType = typing.Literal["invocation"]

agenta/client/backend/variants/client.py

@@ -1104,14 +1104,14 @@ class VariantsClient:
     def configs_delete(
         self,
         *,
-        variant_ref: typing.Optional[ReferenceRequestModel] = OMIT,
+        variant_ref: ReferenceRequestModel,
         application_ref: typing.Optional[ReferenceRequestModel] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> int:
         """
         Parameters
         ----------
-        variant_ref : typing.Optional[ReferenceRequestModel]
+        variant_ref : ReferenceRequestModel
 
         application_ref : typing.Optional[ReferenceRequestModel]
 
@@ -1125,13 +1125,15 @@ class VariantsClient:
 
         Examples
         --------
-        from agenta import AgentaApi
+        from agenta import AgentaApi, ReferenceRequestModel
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.variants.configs_delete()
+        client.variants.configs_delete(
+            variant_ref=ReferenceRequestModel(),
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "variants/configs/delete",
@@ -1244,14 +1246,14 @@ class VariantsClient:
     def configs_history(
         self,
         *,
-        variant_ref: typing.Optional[ReferenceRequestModel] = OMIT,
+        variant_ref: ReferenceRequestModel,
         application_ref: typing.Optional[ReferenceRequestModel] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[ConfigResponseModel]:
         """
         Parameters
         ----------
-        variant_ref : typing.Optional[ReferenceRequestModel]
+        variant_ref : ReferenceRequestModel
 
         application_ref : typing.Optional[ReferenceRequestModel]
 
@@ -1265,13 +1267,15 @@ class VariantsClient:
 
         Examples
         --------
-        from agenta import AgentaApi
+        from agenta import AgentaApi, ReferenceRequestModel
 
         client = AgentaApi(
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.variants.configs_history()
+        client.variants.configs_history(
+            variant_ref=ReferenceRequestModel(),
+        )
         """
         _response = self._client_wrapper.httpx_client.request(
             "variants/configs/history",
@@ -2504,14 +2508,14 @@ class AsyncVariantsClient:
     async def configs_delete(
         self,
         *,
-        variant_ref: typing.Optional[ReferenceRequestModel] = OMIT,
+        variant_ref: ReferenceRequestModel,
         application_ref: typing.Optional[ReferenceRequestModel] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> int:
         """
         Parameters
         ----------
-        variant_ref : typing.Optional[ReferenceRequestModel]
+        variant_ref : ReferenceRequestModel
 
         application_ref : typing.Optional[ReferenceRequestModel]
 
@@ -2527,7 +2531,7 @@ class AsyncVariantsClient:
         --------
         import asyncio
 
-        from agenta import AsyncAgentaApi
+        from agenta import AsyncAgentaApi, ReferenceRequestModel
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
@@ -2536,7 +2540,9 @@ class AsyncVariantsClient:
 
 
         async def main() -> None:
-            await client.variants.configs_delete()
+            await client.variants.configs_delete(
+                variant_ref=ReferenceRequestModel(),
+            )
 
 
         asyncio.run(main())
@@ -2660,14 +2666,14 @@ class AsyncVariantsClient:
     async def configs_history(
         self,
         *,
-        variant_ref: typing.Optional[ReferenceRequestModel] = OMIT,
+        variant_ref: ReferenceRequestModel,
         application_ref: typing.Optional[ReferenceRequestModel] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> typing.List[ConfigResponseModel]:
         """
         Parameters
         ----------
-        variant_ref : typing.Optional[ReferenceRequestModel]
+        variant_ref : ReferenceRequestModel
 
         application_ref : typing.Optional[ReferenceRequestModel]
 
@@ -2683,7 +2689,7 @@ class AsyncVariantsClient:
         --------
         import asyncio
 
-        from agenta import AsyncAgentaApi
+        from agenta import AsyncAgentaApi, ReferenceRequestModel
 
         client = AsyncAgentaApi(
             api_key="YOUR_API_KEY",
@@ -2692,7 +2698,9 @@ class AsyncVariantsClient:
 
 
         async def main() -> None:
-            await client.variants.configs_history()
+            await client.variants.configs_history(
+                variant_ref=ReferenceRequestModel(),
+            )
 
 
         asyncio.run(main())

agenta/sdk/__init__.py

@@ -17,6 +17,8 @@ from .types import (
     FileInputURL,
     BinaryParam,
     Prompt,
+    AgentaNodeDto,
+    AgentaNodesResponse,
 )
 
 from .tracing import Tracing, get_tracer

agenta/sdk/agenta_init.py

@@ -1,4 +1,3 @@
-import logging
 import toml
 from os import getenv
 from typing import Optional, Callable, Any
@@ -11,10 +10,6 @@ from agenta.sdk.tracing import Tracing
 from agenta.client.exceptions import APIRequestError
 
 
-logger = logging.getLogger(__name__)
-logger.setLevel(logging.DEBUG)
-
-
 class AgentaSingleton:
     """Singleton class to save all the "global variables" for the sdk."""
 
@@ -164,7 +159,7 @@ class Config:
         try:
             self.push(config_name="default", overwrite=overwrite, **kwargs)
         except Exception as ex:
-            logger.warning(
+            log.warning(
                 "Unable to push the default configuration to the server. %s", str(ex)
             )
 
@@ -185,7 +180,7 @@ class Config:
                 overwrite=overwrite,
             )
         except Exception as ex:
-            logger.warning(
+            log.warning(
                 "Failed to push the configuration to the server with error: %s", ex
             )
 
@@ -212,14 +207,14 @@ class Config:
                         config_name=config_name,
                     )
             except Exception as ex:
-                logger.warning(
+                log.warning(
                     "Failed to pull the configuration from the server with error: %s",
                     str(ex),
                 )
         try:
             self.set(**{"current_version": config.current_version, **config.parameters})
         except Exception as ex:
-            logger.warning("Failed to set the configuration with error: %s", str(ex))
+            log.warning("Failed to set the configuration with error: %s", str(ex))
 
     def all(self):
         """Returns all the parameters for the app variant"""