agenta 0.27.4a1__tar.gz → 0.27.5a1__tar.gz

{agenta-0.27.4a1 → agenta-0.27.5a1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.27.4a1
+Version: 0.27.5a1
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering
@@ -22,16 +22,13 @@ Requires-Dist: docker (>=6.1.1,<8.0.0)
 Requires-Dist: fastapi (>=0.100.0)
 Requires-Dist: httpx (>=0.24,<0.28)
 Requires-Dist: importlib-metadata (>=8.0.0,<9.0)
-Requires-Dist: ipdb (>=0.13)
 Requires-Dist: litellm (>=1.48.0,<2.0.0)
 Requires-Dist: opentelemetry-api (>=1.27.0,<2.0.0)
 Requires-Dist: opentelemetry-exporter-otlp (>=1.27.0,<2.0.0)
 Requires-Dist: opentelemetry-sdk (>=1.27.0,<2.0.0)
 Requires-Dist: posthog (>=3.1.0,<4.0.0)
 Requires-Dist: pydantic (>=2)
-Requires-Dist: pymongo (>=4.6.3,<5.0.0)
 Requires-Dist: python-dotenv (>=1.0.0,<2.0.0)
-Requires-Dist: python-multipart (>=0.0.6,<0.0.10)
 Requires-Dist: pyyaml (>=6.0.2,<7.0.0)
 Requires-Dist: questionary (>=1.10,<3.0)
 Requires-Dist: toml (>=0.10.2,<0.11.0)

{agenta-0.27.4a1 → agenta-0.27.5a1}/agenta/__init__.py

@@ -1,3 +1,5 @@
+from typing import Any, Callable, Optional
+
 from .sdk.utils.preinit import PreInitObject
 
 import agenta.client.backend.types as client_types  # pylint: disable=wrong-import-order
@@ -18,7 +20,7 @@ from .sdk.types import (
 )
 
 from .sdk.utils.logging import log as logging
-from .sdk.tracing import Tracing
+from .sdk.tracing import Tracing, get_tracer
 from .sdk.decorators.tracing import instrument
 from .sdk.tracing.conventions import Reference
 from .sdk.decorators.routing import entrypoint, app, route
@@ -36,15 +38,36 @@ config = PreInitObject("agenta.config", Config)
 DEFAULT_AGENTA_SINGLETON_INSTANCE = AgentaSingleton()
 
 types = client_types
-tracing = None
+
 api = None
 async_api = None
 
+tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore
+tracer = get_tracer(tracing)
 
-def init(*args, **kwargs):
-    global api, async_api, tracing, config
-    _init(*args, **kwargs)
 
-    tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore
+def init(
+    host: Optional[str] = None,
+    api_key: Optional[str] = None,
+    config_fname: Optional[str] = None,
+    redact: Optional[Callable[..., Any]] = None,
+    redact_on_error: Optional[bool] = True,
+    # DEPRECATING
+    app_id: Optional[str] = None,
+):
+    global api, async_api, tracing, tracer  # pylint: disable=global-statement
+
+    _init(
+        host=host,
+        api_key=api_key,
+        config_fname=config_fname,
+        redact=redact,
+        redact_on_error=redact_on_error,
+        app_id=app_id,
+    )
+
     api = DEFAULT_AGENTA_SINGLETON_INSTANCE.api  # type: ignore
     async_api = DEFAULT_AGENTA_SINGLETON_INSTANCE.async_api  # type: ignore
+
+    tracing = DEFAULT_AGENTA_SINGLETON_INSTANCE.tracing  # type: ignore
+    tracer = get_tracer(tracing)

{agenta-0.27.4a1 → agenta-0.27.5a1}/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-0.27.4a1 → agenta-0.27.5a1}/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-0.27.4a1 → agenta-0.27.5a1}/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-0.27.4a1 → agenta-0.27.5a1}/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-0.27.4a1 → agenta-0.27.5a1}/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",