agenta 0.27.7a1__py3-none-any.whl → 0.27.8a2__py3-none-any.whl

agenta/client/backend/testsets/client.py

@@ -28,7 +28,6 @@ class TestsetsClient:
         file: core.File,
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
-        app_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
@@ -51,8 +50,6 @@ class TestsetsClient:
 
         testset_name : typing.Optional[str]
 
-        app_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -77,7 +74,6 @@ class TestsetsClient:
             data={
                 "upload_type": upload_type,
                 "testset_name": testset_name,
-                "app_id": app_id,
             },
             files={
                 "file": file,
@@ -173,18 +169,16 @@ class TestsetsClient:
 
     def create_testset(
         self,
-        app_id: str,
         *,
         name: str,
         csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
-        Create a testset with given name and app_name, save the testset to MongoDB.
+        Create a testset with given name, save the testset to MongoDB.
 
         Args:
         name (str): name of the test set.
-        app_name (str): name of the application.
         testset (Dict[str, str]): test set data.
 
         Returns:
@@ -192,8 +186,6 @@ class TestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         name : str
 
         csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
@@ -215,13 +207,12 @@ class TestsetsClient:
             base_url="https://yourhost.com/path/to/api",
         )
         client.testsets.create_testset(
-            app_id="app_id",
             name="name",
             csvdata=[{"key": "value"}],
         )
         """
         _response = self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
+            "testsets",
             method="POST",
             json={
                 "name": name,
@@ -405,7 +396,7 @@ class TestsetsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     def get_testsets(
-        self, *, app_id: str, request_options: typing.Optional[RequestOptions] = None
+        self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[TestSetOutputResponse]:
         """
         Get all testsets.
@@ -420,8 +411,6 @@ class TestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -438,16 +427,11 @@ class TestsetsClient:
             api_key="YOUR_API_KEY",
             base_url="https://yourhost.com/path/to/api",
         )
-        client.testsets.get_testsets(
-            app_id="app_id",
-        )
+        client.testsets.get_testsets()
         """
         _response = self._client_wrapper.httpx_client.request(
             "testsets",
             method="GET",
-            params={
-                "app_id": app_id,
-            },
             request_options=request_options,
         )
         try:
@@ -557,7 +541,6 @@ class AsyncTestsetsClient:
         file: core.File,
         upload_type: typing.Optional[str] = OMIT,
         testset_name: typing.Optional[str] = OMIT,
-        app_id: typing.Optional[str] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
@@ -580,8 +563,6 @@ class AsyncTestsetsClient:
 
         testset_name : typing.Optional[str]
 
-        app_id : typing.Optional[str]
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -614,7 +595,6 @@ class AsyncTestsetsClient:
             data={
                 "upload_type": upload_type,
                 "testset_name": testset_name,
-                "app_id": app_id,
             },
             files={
                 "file": file,
@@ -718,18 +698,16 @@ class AsyncTestsetsClient:
 
     async def create_testset(
         self,
-        app_id: str,
         *,
         name: str,
         csvdata: typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]],
         request_options: typing.Optional[RequestOptions] = None,
     ) -> TestSetSimpleResponse:
         """
-        Create a testset with given name and app_name, save the testset to MongoDB.
+        Create a testset with given name, save the testset to MongoDB.
 
         Args:
         name (str): name of the test set.
-        app_name (str): name of the application.
         testset (Dict[str, str]): test set data.
 
         Returns:
@@ -737,8 +715,6 @@ class AsyncTestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         name : str
 
         csvdata : typing.Sequence[typing.Dict[str, typing.Optional[typing.Any]]]
@@ -765,7 +741,6 @@ class AsyncTestsetsClient:
 
         async def main() -> None:
             await client.testsets.create_testset(
-                app_id="app_id",
                 name="name",
                 csvdata=[{"key": "value"}],
             )
@@ -774,7 +749,7 @@ class AsyncTestsetsClient:
         asyncio.run(main())
         """
         _response = await self._client_wrapper.httpx_client.request(
-            f"testsets/{jsonable_encoder(app_id)}",
+            "testsets",
             method="POST",
             json={
                 "name": name,
@@ -974,7 +949,7 @@ class AsyncTestsetsClient:
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
     async def get_testsets(
-        self, *, app_id: str, request_options: typing.Optional[RequestOptions] = None
+        self, *, request_options: typing.Optional[RequestOptions] = None
     ) -> typing.List[TestSetOutputResponse]:
         """
         Get all testsets.
@@ -989,8 +964,6 @@ class AsyncTestsetsClient:
 
         Parameters
         ----------
-        app_id : str
-
         request_options : typing.Optional[RequestOptions]
             Request-specific configuration.
 
@@ -1012,9 +985,7 @@ class AsyncTestsetsClient:
 
 
         async def main() -> None:
-            await client.testsets.get_testsets(
-                app_id="app_id",
-            )
+            await client.testsets.get_testsets()
 
 
         asyncio.run(main())
@@ -1022,9 +993,6 @@ class AsyncTestsetsClient:
         _response = await self._client_wrapper.httpx_client.request(
             "testsets",
             method="GET",
-            params={
-                "app_id": app_id,
-            },
             request_options=request_options,
         )
         try:

agenta/client/backend/types/__init__.py

@@ -1,5 +1,12 @@
 # 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
@@ -7,6 +14,7 @@ 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
@@ -31,6 +39,7 @@ 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
@@ -42,30 +51,50 @@ 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
@@ -80,6 +109,13 @@ from .workspace_role import WorkspaceRole
 from .workspace_role_response import WorkspaceRoleResponse
 
 __all__ = [
+    "AgentaNodeDto",
+    "AgentaNodeDtoNodesValue",
+    "AgentaNodesResponse",
+    "AgentaRootDto",
+    "AgentaRootsResponse",
+    "AgentaTreeDto",
+    "AgentaTreesResponse",
     "AggregatedResult",
     "AggregatedResultEvaluatorConfig",
     "App",
@@ -87,6 +123,7 @@ __all__ = [
     "AppVariantRevision",
     "BaseOutput",
     "BodyImportTestset",
+    "CollectStatusResponse",
     "ConfigDb",
     "ConfigDto",
     "ConfigResponseModel",
@@ -111,6 +148,7 @@ __all__ = [
     "EvaluatorConfig",
     "EvaluatorMappingOutputInterface",
     "EvaluatorOutputInterface",
+    "ExceptionDto",
     "GetConfigResponse",
     "HttpValidationError",
     "HumanEvaluation",
@@ -122,30 +160,50 @@ __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/types/agenta_node_dto.py

@@ -0,0 +1,48 @@
+# 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

@@ -0,0 +1,6 @@
+# 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

@@ -0,0 +1,30 @@
+# 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

@@ -0,0 +1,30 @@
+# 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)

agenta/client/backend/types/agenta_roots_response.py

@@ -0,0 +1,30 @@
+# 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_root_dto import AgentaRootDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class AgentaRootsResponse(UniversalBaseModel):
+    roots: typing.List[AgentaRootDto]
+    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, AgentaRootsResponse=AgentaRootsResponse)

agenta/client/backend/types/agenta_tree_dto.py

@@ -0,0 +1,30 @@
+# 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 .tree_dto import TreeDto
+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 AgentaTreeDto(UniversalBaseModel):
+    tree: TreeDto
+    nodes: typing.List[AgentaNodeDto]
+
+    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, AgentaTreeDto=AgentaTreeDto)

agenta/client/backend/types/agenta_trees_response.py

@@ -0,0 +1,30 @@
+# 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_tree_dto import AgentaTreeDto
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+from ..core.pydantic_utilities import update_forward_refs
+
+
+class AgentaTreesResponse(UniversalBaseModel):
+    trees: typing.List[AgentaTreeDto]
+    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, AgentaTreesResponse=AgentaTreesResponse)

agenta/client/backend/types/collect_status_response.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 CollectStatusResponse(UniversalBaseModel):
+    version: str
+    status: 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/exception_dto.py

@@ -0,0 +1,26 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+import datetime as dt
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class ExceptionDto(UniversalBaseModel):
+    timestamp: dt.datetime
+    type: str
+    message: typing.Optional[str] = None
+    stacktrace: typing.Optional[str] = None
+    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/link_dto.py

@@ -0,0 +1,24 @@
+# This file was auto-generated by Fern from our API Definition.
+
+from ..core.pydantic_utilities import UniversalBaseModel
+from .tree_type import TreeType
+import typing
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class LinkDto(UniversalBaseModel):
+    type: TreeType = "invocation"
+    id: str
+    tree_id: 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/node_dto.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 .node_type import NodeType
+from ..core.pydantic_utilities import IS_PYDANTIC_V2
+import pydantic
+
+
+class NodeDto(UniversalBaseModel):
+    id: str
+    name: str
+    type: typing.Optional[NodeType] = 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