agenta 0.27.4a1__py3-none-any.whl → 0.27.5__py3-none-any.whl

agenta/client/backend/__init__.py

@@ -1,13 +1,6 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
-    AgentaNodeDto,
-    AgentaNodeDtoNodesValue,
-    AgentaNodesResponse,
-    AgentaRootDto,
-    AgentaRootsResponse,
-    AgentaTreeDto,
-    AgentaTreesResponse,
     AggregatedResult,
     AggregatedResultEvaluatorConfig,
     App,
@@ -15,7 +8,6 @@ from .types import (
     AppVariantRevision,
     BaseOutput,
     BodyImportTestset,
-    CollectStatusResponse,
     ConfigDb,
     ConfigDto,
     ConfigResponseModel,
@@ -40,7 +32,6 @@ from .types import (
     EvaluatorConfig,
     EvaluatorMappingOutputInterface,
     EvaluatorOutputInterface,
-    ExceptionDto,
     GetConfigResponse,
     HttpValidationError,
     HumanEvaluation,
@@ -52,50 +43,30 @@ from .types import (
     Image,
     InviteRequest,
     LifecycleDto,
-    LinkDto,
     ListApiKeysResponse,
     LlmRunRateLimit,
     LlmTokens,
     LmProvidersEnum,
     NewHumanEvaluation,
     NewTestset,
-    NodeDto,
-    NodeType,
-    OTelContextDto,
-    OTelEventDto,
-    OTelExtraDto,
-    OTelLinkDto,
-    OTelSpanDto,
-    OTelSpanKind,
-    OTelSpansResponse,
-    OTelStatusCode,
     Organization,
     OrganizationOutput,
     Outputs,
-    ParentDto,
     Permission,
     ReferenceDto,
     ReferenceRequestModel,
     Result,
-    RootDto,
     Score,
     SimpleEvaluationOutput,
     Span,
     SpanDetail,
-    SpanDto,
-    SpanDtoNodesValue,
     SpanStatusCode,
     SpanVariant,
-    StatusCode,
-    StatusDto,
     Template,
     TemplateImageInfo,
     TestSetOutputResponse,
     TestSetSimpleResponse,
-    TimeDto,
     TraceDetail,
-    TreeDto,
-    TreeType,
     UpdateAppOutput,
     Uri,
     ValidationError,
@@ -119,25 +90,16 @@ from . import (
     evaluations,
     evaluators,
     observability,
-    observability_v_1,
     testsets,
     variants,
 )
 from .client import AgentaApi, AsyncAgentaApi
 from .containers import ContainerTemplatesResponse
-from .observability_v_1 import Format, QueryTracesResponse
 from .variants import AddVariantFromBaseAndConfigResponse
 
 __all__ = [
     "AddVariantFromBaseAndConfigResponse",
     "AgentaApi",
-    "AgentaNodeDto",
-    "AgentaNodeDtoNodesValue",
-    "AgentaNodesResponse",
-    "AgentaRootDto",
-    "AgentaRootsResponse",
-    "AgentaTreeDto",
-    "AgentaTreesResponse",
     "AggregatedResult",
     "AggregatedResultEvaluatorConfig",
     "App",
@@ -146,7 +108,6 @@ __all__ = [
     "AsyncAgentaApi",
     "BaseOutput",
     "BodyImportTestset",
-    "CollectStatusResponse",
     "ConfigDb",
     "ConfigDto",
     "ConfigResponseModel",
@@ -172,8 +133,6 @@ __all__ = [
     "EvaluatorConfig",
     "EvaluatorMappingOutputInterface",
     "EvaluatorOutputInterface",
-    "ExceptionDto",
-    "Format",
     "GetConfigResponse",
     "HttpValidationError",
     "HumanEvaluation",
@@ -185,51 +144,30 @@ __all__ = [
     "Image",
     "InviteRequest",
     "LifecycleDto",
-    "LinkDto",
     "ListApiKeysResponse",
     "LlmRunRateLimit",
     "LlmTokens",
     "LmProvidersEnum",
     "NewHumanEvaluation",
     "NewTestset",
-    "NodeDto",
-    "NodeType",
-    "OTelContextDto",
-    "OTelEventDto",
-    "OTelExtraDto",
-    "OTelLinkDto",
-    "OTelSpanDto",
-    "OTelSpanKind",
-    "OTelSpansResponse",
-    "OTelStatusCode",
     "Organization",
     "OrganizationOutput",
     "Outputs",
-    "ParentDto",
     "Permission",
-    "QueryTracesResponse",
     "ReferenceDto",
     "ReferenceRequestModel",
     "Result",
-    "RootDto",
     "Score",
     "SimpleEvaluationOutput",
     "Span",
     "SpanDetail",
-    "SpanDto",
-    "SpanDtoNodesValue",
     "SpanStatusCode",
     "SpanVariant",
-    "StatusCode",
-    "StatusDto",
     "Template",
     "TemplateImageInfo",
     "TestSetOutputResponse",
     "TestSetSimpleResponse",
-    "TimeDto",
     "TraceDetail",
-    "TreeDto",
-    "TreeType",
     "UnprocessableEntityError",
     "UpdateAppOutput",
     "Uri",
@@ -251,7 +189,6 @@ __all__ = [
     "evaluations",
     "evaluators",
     "observability",
-    "observability_v_1",
     "testsets",
     "variants",
 ]

agenta/client/backend/client.py

@@ -13,7 +13,6 @@ from .containers.client import ContainersClient
 from .environments.client import EnvironmentsClient
 from .bases.client import BasesClient
 from .configs.client import ConfigsClient
-from .observability_v_1.client import ObservabilityV1Client
 from .core.request_options import RequestOptions
 from .types.list_api_keys_response import ListApiKeysResponse
 from .core.pydantic_utilities import parse_obj_as
@@ -41,7 +40,6 @@ from .containers.client import AsyncContainersClient
 from .environments.client import AsyncEnvironmentsClient
 from .bases.client import AsyncBasesClient
 from .configs.client import AsyncConfigsClient
-from .observability_v_1.client import AsyncObservabilityV1Client
 
 # this is used as the default value for optional parameters
 OMIT = typing.cast(typing.Any, ...)
@@ -91,13 +89,17 @@ class AgentaApi:
         self._client_wrapper = SyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.Client(
-                timeout=_defaulted_timeout, follow_redirects=follow_redirects
-            )
-            if follow_redirects is not None
-            else httpx.Client(timeout=_defaulted_timeout),
+            httpx_client=(
+                httpx_client
+                if httpx_client is not None
+                else (
+                    httpx.Client(
+                        timeout=_defaulted_timeout, follow_redirects=follow_redirects
+                    )
+                    if follow_redirects is not None
+                    else httpx.Client(timeout=_defaulted_timeout)
+                )
+            ),
             timeout=_defaulted_timeout,
         )
         self.observability = ObservabilityClient(client_wrapper=self._client_wrapper)
@@ -110,9 +112,6 @@ class AgentaApi:
         self.environments = EnvironmentsClient(client_wrapper=self._client_wrapper)
         self.bases = BasesClient(client_wrapper=self._client_wrapper)
         self.configs = ConfigsClient(client_wrapper=self._client_wrapper)
-        self.observability_v_1 = ObservabilityV1Client(
-            client_wrapper=self._client_wrapper
-        )
 
     def list_api_keys(
         self, *, request_options: typing.Optional[RequestOptions] = None
@@ -1620,13 +1619,17 @@ class AsyncAgentaApi:
         self._client_wrapper = AsyncClientWrapper(
             base_url=base_url,
             api_key=api_key,
-            httpx_client=httpx_client
-            if httpx_client is not None
-            else httpx.AsyncClient(
-                timeout=_defaulted_timeout, follow_redirects=follow_redirects
-            )
-            if follow_redirects is not None
-            else httpx.AsyncClient(timeout=_defaulted_timeout),
+            httpx_client=(
+                httpx_client
+                if httpx_client is not None
+                else (
+                    httpx.AsyncClient(
+                        timeout=_defaulted_timeout, follow_redirects=follow_redirects
+                    )
+                    if follow_redirects is not None
+                    else httpx.AsyncClient(timeout=_defaulted_timeout)
+                )
+            ),
             timeout=_defaulted_timeout,
         )
         self.observability = AsyncObservabilityClient(
@@ -1641,9 +1644,6 @@ class AsyncAgentaApi:
         self.environments = AsyncEnvironmentsClient(client_wrapper=self._client_wrapper)
         self.bases = AsyncBasesClient(client_wrapper=self._client_wrapper)
         self.configs = AsyncConfigsClient(client_wrapper=self._client_wrapper)
-        self.observability_v_1 = AsyncObservabilityV1Client(
-            client_wrapper=self._client_wrapper
-        )
 
     async def list_api_keys(
         self, *, request_options: typing.Optional[RequestOptions] = None

agenta/client/backend/core/http_client.py

@@ -148,9 +148,9 @@ def get_request_body(
         json_body = maybe_filter_request_body(json, request_options, omit)
 
     # If you have an empty JSON body, you should just send None
-    return (
-        json_body if json_body != {} else None
-    ), data_body if data_body != {} else None
+    return (json_body if json_body != {} else None), (
+        data_body if data_body != {} else None
+    )
 
 
 class HttpClient:
@@ -250,9 +250,7 @@ class HttpClient:
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(
-                    remove_omit_from_dict(remove_none_from_dict(files), omit)
-                )
+                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
                 if (files is not None and files is not omit)
                 else None
             ),
@@ -353,9 +351,7 @@ class HttpClient:
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(
-                    remove_omit_from_dict(remove_none_from_dict(files), omit)
-                )
+                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
                 if (files is not None and files is not omit)
                 else None
             ),
@@ -462,9 +458,7 @@ class AsyncHttpClient:
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(
-                    remove_omit_from_dict(remove_none_from_dict(files), omit)
-                )
+                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
                 if files is not None
                 else None
             ),
@@ -564,9 +558,7 @@ class AsyncHttpClient:
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(
-                    remove_omit_from_dict(remove_none_from_dict(files), omit)
-                )
+                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
                 if files is not None
                 else None
             ),

agenta/client/backend/observability/client.py

@@ -280,7 +280,7 @@ 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],
@@ -307,7 +307,7 @@ class ObservabilityClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.delete_traces_legacy(
+        client.observability.delete_traces(
             request=["string"],
         )
         """
@@ -901,7 +901,7 @@ 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],
@@ -933,7 +933,7 @@ class AsyncObservabilityClient:
 
 
         async def main() -> None:
-            await client.observability.delete_traces_legacy(
+            await client.observability.delete_traces(
                 request=["string"],
             )
 

agenta/client/backend/types/__init__.py

@@ -1,12 +1,5 @@
 # This file was auto-generated by Fern from our API Definition.
 
-from .agenta_node_dto import AgentaNodeDto
-from .agenta_node_dto_nodes_value import AgentaNodeDtoNodesValue
-from .agenta_nodes_response import AgentaNodesResponse
-from .agenta_root_dto import AgentaRootDto
-from .agenta_roots_response import AgentaRootsResponse
-from .agenta_tree_dto import AgentaTreeDto
-from .agenta_trees_response import AgentaTreesResponse
 from .aggregated_result import AggregatedResult
 from .aggregated_result_evaluator_config import AggregatedResultEvaluatorConfig
 from .app import App
@@ -14,7 +7,6 @@ from .app_variant_response import AppVariantResponse
 from .app_variant_revision import AppVariantRevision
 from .base_output import BaseOutput
 from .body_import_testset import BodyImportTestset
-from .collect_status_response import CollectStatusResponse
 from .config_db import ConfigDb
 from .config_dto import ConfigDto
 from .config_response_model import ConfigResponseModel
@@ -39,7 +31,6 @@ from .evaluator import Evaluator
 from .evaluator_config import EvaluatorConfig
 from .evaluator_mapping_output_interface import EvaluatorMappingOutputInterface
 from .evaluator_output_interface import EvaluatorOutputInterface
-from .exception_dto import ExceptionDto
 from .get_config_response import GetConfigResponse
 from .http_validation_error import HttpValidationError
 from .human_evaluation import HumanEvaluation
@@ -51,50 +42,30 @@ from .human_evaluation_update import HumanEvaluationUpdate
 from .image import Image
 from .invite_request import InviteRequest
 from .lifecycle_dto import LifecycleDto
-from .link_dto import LinkDto
 from .list_api_keys_response import ListApiKeysResponse
 from .llm_run_rate_limit import LlmRunRateLimit
 from .llm_tokens import LlmTokens
 from .lm_providers_enum import LmProvidersEnum
 from .new_human_evaluation import NewHumanEvaluation
 from .new_testset import NewTestset
-from .node_dto import NodeDto
-from .node_type import NodeType
-from .o_tel_context_dto import OTelContextDto
-from .o_tel_event_dto import OTelEventDto
-from .o_tel_extra_dto import OTelExtraDto
-from .o_tel_link_dto import OTelLinkDto
-from .o_tel_span_dto import OTelSpanDto
-from .o_tel_span_kind import OTelSpanKind
-from .o_tel_spans_response import OTelSpansResponse
-from .o_tel_status_code import OTelStatusCode
 from .organization import Organization
 from .organization_output import OrganizationOutput
 from .outputs import Outputs
-from .parent_dto import ParentDto
 from .permission import Permission
 from .reference_dto import ReferenceDto
 from .reference_request_model import ReferenceRequestModel
 from .result import Result
-from .root_dto import RootDto
 from .score import Score
 from .simple_evaluation_output import SimpleEvaluationOutput
 from .span import Span
 from .span_detail import SpanDetail
-from .span_dto import SpanDto
-from .span_dto_nodes_value import SpanDtoNodesValue
 from .span_status_code import SpanStatusCode
 from .span_variant import SpanVariant
-from .status_code import StatusCode
-from .status_dto import StatusDto
 from .template import Template
 from .template_image_info import TemplateImageInfo
 from .test_set_output_response import TestSetOutputResponse
 from .test_set_simple_response import TestSetSimpleResponse
-from .time_dto import TimeDto
 from .trace_detail import TraceDetail
-from .tree_dto import TreeDto
-from .tree_type import TreeType
 from .update_app_output import UpdateAppOutput
 from .uri import Uri
 from .validation_error import ValidationError
@@ -109,13 +80,6 @@ from .workspace_role import WorkspaceRole
 from .workspace_role_response import WorkspaceRoleResponse
 
 __all__ = [
-    "AgentaNodeDto",
-    "AgentaNodeDtoNodesValue",
-    "AgentaNodesResponse",
-    "AgentaRootDto",
-    "AgentaRootsResponse",
-    "AgentaTreeDto",
-    "AgentaTreesResponse",
     "AggregatedResult",
     "AggregatedResultEvaluatorConfig",
     "App",
@@ -123,7 +87,6 @@ __all__ = [
     "AppVariantRevision",
     "BaseOutput",
     "BodyImportTestset",
-    "CollectStatusResponse",
     "ConfigDb",
     "ConfigDto",
     "ConfigResponseModel",
@@ -148,7 +111,6 @@ __all__ = [
     "EvaluatorConfig",
     "EvaluatorMappingOutputInterface",
     "EvaluatorOutputInterface",
-    "ExceptionDto",
     "GetConfigResponse",
     "HttpValidationError",
     "HumanEvaluation",
@@ -160,50 +122,30 @@ __all__ = [
     "Image",
     "InviteRequest",
     "LifecycleDto",
-    "LinkDto",
     "ListApiKeysResponse",
     "LlmRunRateLimit",
     "LlmTokens",
     "LmProvidersEnum",
     "NewHumanEvaluation",
     "NewTestset",
-    "NodeDto",
-    "NodeType",
-    "OTelContextDto",
-    "OTelEventDto",
-    "OTelExtraDto",
-    "OTelLinkDto",
-    "OTelSpanDto",
-    "OTelSpanKind",
-    "OTelSpansResponse",
-    "OTelStatusCode",
     "Organization",
     "OrganizationOutput",
     "Outputs",
-    "ParentDto",
     "Permission",
     "ReferenceDto",
     "ReferenceRequestModel",
     "Result",
-    "RootDto",
     "Score",
     "SimpleEvaluationOutput",
     "Span",
     "SpanDetail",
-    "SpanDto",
-    "SpanDtoNodesValue",
     "SpanStatusCode",
     "SpanVariant",
-    "StatusCode",
-    "StatusDto",
     "Template",
     "TemplateImageInfo",
     "TestSetOutputResponse",
     "TestSetSimpleResponse",
-    "TimeDto",
     "TraceDetail",
-    "TreeDto",
-    "TreeType",
     "UpdateAppOutput",
     "Uri",
     "ValidationError",

agenta/client/backend/variants/client.py

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

agenta/sdk/__init__.py

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

agenta/sdk/decorators/routing.py

@@ -386,7 +386,7 @@ class entrypoint:
         log.info(f"Agenta SDK - exiting with success: 200")
         log.info(f"----------------------------------")
 
-        return BaseResponse(data=data, tree=trace)
+        return BaseResponse(data=data, trace=trace)
 
     def handle_failure(self, error: Exception):
         log.error("--------------------------------------------------")