[Enhancement] Update SDK to return BaseResponse version 3.0 and new trace tree format

agenta-cli/agenta/client/backend/__init__.py

@@ -1,13 +1,21 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
+    AgentaNodeDto,
+    AgentaNodeDtoNodesValue,
+    AgentaNodesResponse,
+    AgentaRootDto,
+    AgentaRootsResponse,
+    AgentaTreeDto,
+    AgentaTreesResponse,
     AggregatedResult,
     AggregatedResultEvaluatorConfig,
     App,
     AppVariantResponse,
     AppVariantRevision,
     BaseOutput,
     BodyImportTestset,
+    CollectStatusResponse,
     ConfigDb,
     ConfigDto,
     ConfigResponseModel,
@@ -32,6 +40,7 @@
     EvaluatorConfig,
     EvaluatorMappingOutputInterface,
     EvaluatorOutputInterface,
+    ExceptionDto,
     GetConfigResponse,
     HttpValidationError,
     HumanEvaluation,
@@ -43,30 +52,50 @@
     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,
@@ -90,16 +119,25 @@
     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",
@@ -108,6 +146,7 @@
     "AsyncAgentaApi",
     "BaseOutput",
     "BodyImportTestset",
+    "CollectStatusResponse",
     "ConfigDb",
     "ConfigDto",
     "ConfigResponseModel",
@@ -133,6 +172,8 @@
     "EvaluatorConfig",
     "EvaluatorMappingOutputInterface",
     "EvaluatorOutputInterface",
+    "ExceptionDto",
+    "Format",
     "GetConfigResponse",
     "HttpValidationError",
     "HumanEvaluation",
@@ -144,30 +185,51 @@
     "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",
@@ -189,6 +251,7 @@
     "evaluations",
     "evaluators",
     "observability",
+    "observability_v_1",
     "testsets",
     "variants",
 ]

agenta-cli/agenta/client/backend/client.py

@@ -13,6 +13,7 @@
 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
@@ -40,6 +41,7 @@
 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, ...)
@@ -89,17 +91,13 @@ def __init__(
         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)
@@ -112,6 +110,9 @@ def __init__(
         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
@@ -1619,17 +1620,13 @@ def __init__(
         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(
@@ -1644,6 +1641,9 @@ def __init__(
         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-cli/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,7 +250,9 @@ def request(
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
+                convert_file_dict_to_httpx_tuples(
+                    remove_omit_from_dict(remove_none_from_dict(files), omit)
+                )
                 if (files is not None and files is not omit)
                 else None
             ),
@@ -351,7 +353,9 @@ def stream(
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
+                convert_file_dict_to_httpx_tuples(
+                    remove_omit_from_dict(remove_none_from_dict(files), omit)
+                )
                 if (files is not None and files is not omit)
                 else None
             ),
@@ -458,7 +462,9 @@ async def request(
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
+                convert_file_dict_to_httpx_tuples(
+                    remove_omit_from_dict(remove_none_from_dict(files), omit)
+                )
                 if files is not None
                 else None
             ),
@@ -558,7 +564,9 @@ async def stream(
             data=data_body,
             content=content,
             files=(
-                convert_file_dict_to_httpx_tuples(remove_none_from_dict(files))
+                convert_file_dict_to_httpx_tuples(
+                    remove_omit_from_dict(remove_none_from_dict(files), omit)
+                )
                 if files is not None
                 else None
             ),

agenta-cli/agenta/client/backend/observability/client.py

@@ -280,7 +280,7 @@ def get_traces(
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def delete_traces(
+    def delete_traces_legacy(
         self,
         *,
         request: typing.Sequence[str],
@@ -307,7 +307,7 @@ def delete_traces(
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.observability.delete_traces(
+        client.observability.delete_traces_legacy(
             request=["string"],
         )
         """
@@ -901,7 +901,7 @@ async def main() -> None:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def delete_traces(
+    async def delete_traces_legacy(
         self,
         *,
         request: typing.Sequence[str],
@@ -933,7 +933,7 @@ async def delete_traces(
 
 
         async def main() -> None:
-            await client.observability.delete_traces(
+            await client.observability.delete_traces_legacy(
                 request=["string"],
             )
 

agenta-cli/agenta/client/backend/observability_v_1/__init__.py

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