agenta 0.12.7__py3-none-any.whl → 0.13.0__py3-none-any.whl

agenta/__init__.py

@@ -13,8 +13,10 @@ from .sdk.types import (
     FileInputURL,
     BinaryParam,
 )
+from .sdk.tracing.decorators import span
 from .sdk.utils.preinit import PreInitObject
-from .sdk.agenta_init import Config, init
+from .sdk.agenta_init import Config, init, llm_tracing
 from .sdk.utils.helper.openai_cost import calculate_token_usage
 
+
 config = PreInitObject("agenta.config", Config)

agenta/cli/helper.py

@@ -137,7 +137,7 @@ def update_variants_from_backend(
     )
 
     try:
-        variants: List[AppVariant] = client.list_app_variants(app_id=app_id)
+        variants: List[AppVariant] = client.apps.list_app_variants(app_id=app_id)
     except Exception as ex:
         raise ex
 

agenta/cli/main.py

@@ -188,7 +188,7 @@ def init(app_name: str, backend_host: str):
 
         # Get app_id after creating new app in the backend server
         try:
-            app_id = client.create_app(
+            app_id = client.apps.create_app(
                 app_name=app_name,
                 organization_id=filtered_org.id if filtered_org else None,
             ).app_id

agenta/cli/variant_commands.py

@@ -126,7 +126,7 @@ def add_variant(
             )
         )
         with tar_path.open("rb") as tar_file:
-            built_image: Image = client.build_image(
+            built_image: Image = client.containers.build_image(
                 app_id=app_id,
                 base_name=base_name,
                 tar_file=tar_file,
@@ -147,7 +147,7 @@ def add_variant(
                 )
             )
             variant_id = config["variant_ids"][config["variants"].index(variant_name)]
-            client.update_variant_image(
+            client.variants.update_variant_image(
                 variant_id=variant_id,
                 request=image,  # because Fern code uses "request: Image" instead of "image: Image"
             )  # this automatically restarts
@@ -259,7 +259,9 @@ def start_variant(variant_id: str, app_folder: str, host: str):
         api_key=api_key,
     )
 
-    endpoint = client.start_variant(variant_id=variant_id, action={"action": "START"})
+    endpoint = client.variants.start_variant(
+        variant_id=variant_id, action={"action": "START"}
+    )
     click.echo("\n" + click.style("Congratulations! 🎉", bold=True, fg="green"))
     click.echo(
         click.style("Your app has been deployed locally as an API. 🚀", fg="cyan")
@@ -327,7 +329,7 @@ def remove_variant(variant_name: str, app_folder: str, host: str):
     )
 
     try:
-        client.remove_variant(variant_id=variant_id)
+        client.variants.remove_variant(variant_id=variant_id)
     except Exception as ex:
         click.echo(
             click.style(
@@ -365,7 +367,7 @@ def list_variants(app_folder: str, host: str):
     )
 
     try:
-        variants: List[AppVariant] = client.list_app_variants(app_id=app_id)
+        variants: List[AppVariant] = client.apps.list_app_variants(app_id=app_id)
     except Exception as ex:
         raise ex
 

agenta/client/api.py

@@ -47,7 +47,7 @@ def add_variant_to_server(
     )
     for attempt in range(retries):
         try:
-            response = client.add_variant_from_image(
+            response = client.apps.add_variant_from_image(
                 app_id=app_id,
                 variant_name=f"{base_name.lower()}.default",
                 base_name=base_name,

agenta/client/backend/__init__.py

@@ -1,115 +1,175 @@
 # This file was auto-generated by Fern from our API Definition.
 
 from .types import (
-    AddVariantFromBaseAndConfigResponse,
     AggregatedResult,
+    AggregatedResultEvaluatorConfig,
     App,
-    AppVariantOutput,
-    AppVariantOutputExtended,
+    AppVariantResponse,
     AppVariantRevision,
     BaseOutput,
     BodyImportTestset,
     ConfigDb,
-    ContainerTemplatesResponse,
     CreateAppOutput,
-    DeleteEvaluation,
+    CreateSpan,
+    CreateTraceResponse,
     DockerEnvVars,
     EnvironmentOutput,
+    EnvironmentOutputExtended,
+    EnvironmentRevision,
+    Error,
     Evaluation,
     EvaluationScenario,
     EvaluationScenarioInput,
     EvaluationScenarioOutput,
     EvaluationScenarioResult,
+    EvaluationScenarioScoreUpdate,
     EvaluationStatusEnum,
     EvaluationType,
     EvaluationWebhook,
     Evaluator,
     EvaluatorConfig,
     Feedback,
-    GetConfigReponse,
+    GetConfigResponse,
     HttpValidationError,
     HumanEvaluation,
     HumanEvaluationScenario,
     HumanEvaluationScenarioInput,
     HumanEvaluationScenarioOutput,
-    HumanEvaluationScenarioScore,
-    HumanEvaluationScenarioUpdateScore,
+    HumanEvaluationScenarioUpdate,
+    HumanEvaluationUpdate,
     Image,
     InviteRequest,
-    ListApiKeysOutput,
+    ListApiKeysResponse,
     LlmRunRateLimit,
+    LlmTokens,
+    NewHumanEvaluation,
     NewTestset,
     Organization,
     OrganizationOutput,
+    Permission,
     Result,
+    Score,
     SimpleEvaluationOutput,
     Span,
+    SpanDetail,
+    SpanKind,
+    SpanStatusCode,
+    SpanVariant,
     Template,
     TemplateImageInfo,
     TestSetOutputResponse,
     TestSetSimpleResponse,
-    Trace,
+    TraceDetail,
     Uri,
     ValidationError,
     ValidationErrorLocItem,
     VariantAction,
     VariantActionEnum,
+    WithPagination,
+    WorkspaceMemberResponse,
+    WorkspacePermission,
+    WorkspaceResponse,
+    WorkspaceRole,
+    WorkspaceRoleResponse,
 )
 from .errors import UnprocessableEntityError
+from .resources import (
+    AddVariantFromBaseAndConfigResponse,
+    ContainerTemplatesResponse,
+    apps,
+    bases,
+    configs,
+    containers,
+    environments,
+    evaluations,
+    evaluators,
+    observability,
+    testsets,
+    variants,
+)
 
 __all__ = [
     "AddVariantFromBaseAndConfigResponse",
     "AggregatedResult",
+    "AggregatedResultEvaluatorConfig",
     "App",
-    "AppVariantOutput",
-    "AppVariantOutputExtended",
+    "AppVariantResponse",
     "AppVariantRevision",
     "BaseOutput",
     "BodyImportTestset",
     "ConfigDb",
     "ContainerTemplatesResponse",
     "CreateAppOutput",
-    "DeleteEvaluation",
+    "CreateSpan",
+    "CreateTraceResponse",
     "DockerEnvVars",
     "EnvironmentOutput",
+    "EnvironmentOutputExtended",
+    "EnvironmentRevision",
+    "Error",
     "Evaluation",
     "EvaluationScenario",
     "EvaluationScenarioInput",
     "EvaluationScenarioOutput",
     "EvaluationScenarioResult",
+    "EvaluationScenarioScoreUpdate",
     "EvaluationStatusEnum",
     "EvaluationType",
     "EvaluationWebhook",
     "Evaluator",
     "EvaluatorConfig",
     "Feedback",
-    "GetConfigReponse",
+    "GetConfigResponse",
     "HttpValidationError",
     "HumanEvaluation",
     "HumanEvaluationScenario",
     "HumanEvaluationScenarioInput",
     "HumanEvaluationScenarioOutput",
-    "HumanEvaluationScenarioScore",
-    "HumanEvaluationScenarioUpdateScore",
+    "HumanEvaluationScenarioUpdate",
+    "HumanEvaluationUpdate",
     "Image",
     "InviteRequest",
-    "ListApiKeysOutput",
+    "ListApiKeysResponse",
     "LlmRunRateLimit",
+    "LlmTokens",
+    "NewHumanEvaluation",
     "NewTestset",
     "Organization",
     "OrganizationOutput",
+    "Permission",
     "Result",
+    "Score",
     "SimpleEvaluationOutput",
     "Span",
+    "SpanDetail",
+    "SpanKind",
+    "SpanStatusCode",
+    "SpanVariant",
     "Template",
     "TemplateImageInfo",
     "TestSetOutputResponse",
     "TestSetSimpleResponse",
-    "Trace",
+    "TraceDetail",
     "UnprocessableEntityError",
     "Uri",
     "ValidationError",
     "ValidationErrorLocItem",
     "VariantAction",
     "VariantActionEnum",
+    "WithPagination",
+    "WorkspaceMemberResponse",
+    "WorkspacePermission",
+    "WorkspaceResponse",
+    "WorkspaceRole",
+    "WorkspaceRoleResponse",
+    "apps",
+    "bases",
+    "configs",
+    "containers",
+    "environments",
+    "evaluations",
+    "evaluators",
+    "observability",
+    "testsets",
+    "variants",
 ]