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

agenta/client/backend/types/invite_request.py

@@ -13,6 +13,7 @@ except ImportError:
 
 class InviteRequest(pydantic.BaseModel):
     email: str
+    roles: typing.List[str]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {

agenta/client/backend/types/{list_api_keys_output.py → list_api_keys_response.py}

@@ -11,7 +11,7 @@ except ImportError:
     import pydantic  # type: ignore
 
 
-class ListApiKeysOutput(pydantic.BaseModel):
+class ListApiKeysResponse(pydantic.BaseModel):
     prefix: str
     created_at: dt.datetime
     last_used_at: typing.Optional[dt.datetime]

agenta/client/backend/types/llm_tokens.py

@@ -0,0 +1,38 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class LlmTokens(pydantic.BaseModel):
+    prompt_tokens: typing.Optional[int]
+    completion_tokens: typing.Optional[int]
+    total_tokens: typing.Optional[int]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/new_human_evaluation.py

@@ -0,0 +1,42 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .evaluation_type import EvaluationType
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class NewHumanEvaluation(pydantic.BaseModel):
+    app_id: str
+    variant_ids: typing.List[str]
+    evaluation_type: EvaluationType
+    inputs: typing.List[str]
+    testset_id: str
+    status: str
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/organization.py

@@ -19,6 +19,7 @@ class Organization(pydantic.BaseModel):
     owner: str
     members: typing.Optional[typing.List[str]]
     invitations: typing.Optional[typing.List[typing.Any]]
+    is_paying: typing.Optional[bool]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {

agenta/client/backend/types/permission.py

@@ -0,0 +1,141 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class Permission(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    READ_SYSTEM = "read_system"
+    VIEW_APPLICATION = "view_application"
+    EDIT_APPLICATION = "edit_application"
+    CREATE_APPLICATION = "create_application"
+    DELETE_APPLICATION = "delete_application"
+    CREATE_APP_VARIANT = "create_app_variant"
+    DELETE_APP_VARIANT = "delete_app_variant"
+    MODIFY_VARIANT_CONFIGURATIONS = "modify_variant_configurations"
+    DELETE_APPLICATION_VARIANT = "delete_application_variant"
+    VIEW_APP_ENVIRONMENT_DEPLOYMENT = "view_app_environment_deployment"
+    EDIT_APP_ENVIRONMENT_DEPLOYMENT = "edit_app_environment_deployment"
+    CREATE_APP_ENVIRONMENT_DEPLOYMENT = "create_app_environment_deployment"
+    VIEW_TESTSET = "view_testset"
+    EDIT_TESTSET = "edit_testset"
+    CREATE_TESTSET = "create_testset"
+    DELETE_TESTSET = "delete_testset"
+    VIEW_EVALUATION = "view_evaluation"
+    RUN_EVALUATIONS = "run_evaluations"
+    EDIT_EVALUATION = "edit_evaluation"
+    CREATE_EVALUATION = "create_evaluation"
+    DELETE_EVALUATION = "delete_evaluation"
+    DEPLOY_APPLICATION = "deploy_application"
+    VIEW_WORKSPACE = "view_workspace"
+    EDIT_WORKSPACE = "edit_workspace"
+    CREATE_WORKSPACE = "create_workspace"
+    DELETE_WORKSPACE = "delete_workspace"
+    MODIFY_USER_ROLES = "modify_user_roles"
+    ADD_NEW_USER_TO_WORKSPACE = "add_new_user_to_workspace"
+    EDIT_ORGANIZATION = "edit_organization"
+    DELETE_ORGANIZATION = "delete_organization"
+    ADD_NEW_USER_TO_ORGANIZATION = "add_new_user_to_organization"
+
+    def visit(
+        self,
+        read_system: typing.Callable[[], T_Result],
+        view_application: typing.Callable[[], T_Result],
+        edit_application: typing.Callable[[], T_Result],
+        create_application: typing.Callable[[], T_Result],
+        delete_application: typing.Callable[[], T_Result],
+        create_app_variant: typing.Callable[[], T_Result],
+        delete_app_variant: typing.Callable[[], T_Result],
+        modify_variant_configurations: typing.Callable[[], T_Result],
+        delete_application_variant: typing.Callable[[], T_Result],
+        view_app_environment_deployment: typing.Callable[[], T_Result],
+        edit_app_environment_deployment: typing.Callable[[], T_Result],
+        create_app_environment_deployment: typing.Callable[[], T_Result],
+        view_testset: typing.Callable[[], T_Result],
+        edit_testset: typing.Callable[[], T_Result],
+        create_testset: typing.Callable[[], T_Result],
+        delete_testset: typing.Callable[[], T_Result],
+        view_evaluation: typing.Callable[[], T_Result],
+        run_evaluations: typing.Callable[[], T_Result],
+        edit_evaluation: typing.Callable[[], T_Result],
+        create_evaluation: typing.Callable[[], T_Result],
+        delete_evaluation: typing.Callable[[], T_Result],
+        deploy_application: typing.Callable[[], T_Result],
+        view_workspace: typing.Callable[[], T_Result],
+        edit_workspace: typing.Callable[[], T_Result],
+        create_workspace: typing.Callable[[], T_Result],
+        delete_workspace: typing.Callable[[], T_Result],
+        modify_user_roles: typing.Callable[[], T_Result],
+        add_new_user_to_workspace: typing.Callable[[], T_Result],
+        edit_organization: typing.Callable[[], T_Result],
+        delete_organization: typing.Callable[[], T_Result],
+        add_new_user_to_organization: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is Permission.READ_SYSTEM:
+            return read_system()
+        if self is Permission.VIEW_APPLICATION:
+            return view_application()
+        if self is Permission.EDIT_APPLICATION:
+            return edit_application()
+        if self is Permission.CREATE_APPLICATION:
+            return create_application()
+        if self is Permission.DELETE_APPLICATION:
+            return delete_application()
+        if self is Permission.CREATE_APP_VARIANT:
+            return create_app_variant()
+        if self is Permission.DELETE_APP_VARIANT:
+            return delete_app_variant()
+        if self is Permission.MODIFY_VARIANT_CONFIGURATIONS:
+            return modify_variant_configurations()
+        if self is Permission.DELETE_APPLICATION_VARIANT:
+            return delete_application_variant()
+        if self is Permission.VIEW_APP_ENVIRONMENT_DEPLOYMENT:
+            return view_app_environment_deployment()
+        if self is Permission.EDIT_APP_ENVIRONMENT_DEPLOYMENT:
+            return edit_app_environment_deployment()
+        if self is Permission.CREATE_APP_ENVIRONMENT_DEPLOYMENT:
+            return create_app_environment_deployment()
+        if self is Permission.VIEW_TESTSET:
+            return view_testset()
+        if self is Permission.EDIT_TESTSET:
+            return edit_testset()
+        if self is Permission.CREATE_TESTSET:
+            return create_testset()
+        if self is Permission.DELETE_TESTSET:
+            return delete_testset()
+        if self is Permission.VIEW_EVALUATION:
+            return view_evaluation()
+        if self is Permission.RUN_EVALUATIONS:
+            return run_evaluations()
+        if self is Permission.EDIT_EVALUATION:
+            return edit_evaluation()
+        if self is Permission.CREATE_EVALUATION:
+            return create_evaluation()
+        if self is Permission.DELETE_EVALUATION:
+            return delete_evaluation()
+        if self is Permission.DEPLOY_APPLICATION:
+            return deploy_application()
+        if self is Permission.VIEW_WORKSPACE:
+            return view_workspace()
+        if self is Permission.EDIT_WORKSPACE:
+            return edit_workspace()
+        if self is Permission.CREATE_WORKSPACE:
+            return create_workspace()
+        if self is Permission.DELETE_WORKSPACE:
+            return delete_workspace()
+        if self is Permission.MODIFY_USER_ROLES:
+            return modify_user_roles()
+        if self is Permission.ADD_NEW_USER_TO_WORKSPACE:
+            return add_new_user_to_workspace()
+        if self is Permission.EDIT_ORGANIZATION:
+            return edit_organization()
+        if self is Permission.DELETE_ORGANIZATION:
+            return delete_organization()
+        if self is Permission.ADD_NEW_USER_TO_ORGANIZATION:
+            return add_new_user_to_organization()

agenta/client/backend/types/result.py

@@ -4,6 +4,7 @@ import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .error import Error
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -14,6 +15,7 @@ except ImportError:
 class Result(pydantic.BaseModel):
     type: str
     value: typing.Optional[typing.Any]
+    error: typing.Optional[Error]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {

agenta/client/backend/types/{human_evaluation_scenario_score.py → score.py}

@@ -2,4 +2,4 @@
 
 import typing
 
-HumanEvaluationScenarioScore = typing.Union[str, int]
+Score = typing.Union[str, int]

agenta/client/backend/types/span.py

@@ -1,9 +1,14 @@
 # This file was auto-generated by Fern from our API Definition.
 
+from __future__ import annotations
+
 import datetime as dt
 import typing
 
 from ..core.datetime_utils import serialize_datetime
+from .span_kind import SpanKind
+from .span_status_code import SpanStatusCode
+from .span_variant import SpanVariant
 
 try:
     import pydantic.v1 as pydantic  # type: ignore
@@ -12,23 +17,17 @@ except ImportError:
 
 
 class Span(pydantic.BaseModel):
+    id: str
+    name: str
     parent_span_id: typing.Optional[str]
-    meta: typing.Optional[typing.Dict[str, typing.Any]]
-    event_name: str
-    event_type: typing.Optional[str]
-    start_time: dt.datetime
-    duration: typing.Optional[int]
-    status: str
-    end_time: dt.datetime
-    inputs: typing.Optional[typing.List[str]]
-    outputs: typing.Optional[typing.List[str]]
-    prompt_template: typing.Optional[str]
-    tokens_input: typing.Optional[int]
-    tokens_output: typing.Optional[int]
-    token_total: typing.Optional[int]
-    cost: typing.Optional[float]
-    tags: typing.Optional[typing.List[str]]
-    span_id: str
+    created_at: dt.datetime
+    variant: SpanVariant
+    environment: typing.Optional[str]
+    spankind: SpanKind
+    status: SpanStatusCode
+    metadata: typing.Dict[str, typing.Any]
+    user_id: typing.Optional[str]
+    children: typing.Optional[typing.List[Span]]
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {
@@ -50,3 +49,6 @@ class Span(pydantic.BaseModel):
         frozen = True
         smart_union = True
         json_encoders = {dt.datetime: serialize_datetime}
+
+
+Span.update_forward_refs()

agenta/client/backend/types/span_detail.py

@@ -0,0 +1,52 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .span import Span
+from .span_kind import SpanKind
+from .span_status_code import SpanStatusCode
+from .span_variant import SpanVariant
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class SpanDetail(pydantic.BaseModel):
+    id: str
+    name: str
+    parent_span_id: typing.Optional[str]
+    created_at: dt.datetime
+    variant: SpanVariant
+    environment: typing.Optional[str]
+    spankind: SpanKind
+    status: SpanStatusCode
+    metadata: typing.Dict[str, typing.Any]
+    user_id: typing.Optional[str]
+    children: typing.Optional[typing.List[Span]]
+    content: typing.Dict[str, typing.Any]
+    config: typing.Optional[typing.Dict[str, typing.Any]]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/span_kind.py

@@ -0,0 +1,49 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class SpanKind(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    TOOL = "TOOL"
+    CHAIN = "CHAIN"
+    LLM = "LLM"
+    WORKFLOW = "WORKFLOW"
+    RETRIEVER = "RETRIEVER"
+    EMBEDDING = "EMBEDDING"
+    AGENT = "AGENT"
+    UNKNOWN = "UNKNOWN"
+
+    def visit(
+        self,
+        tool: typing.Callable[[], T_Result],
+        chain: typing.Callable[[], T_Result],
+        llm: typing.Callable[[], T_Result],
+        workflow: typing.Callable[[], T_Result],
+        retriever: typing.Callable[[], T_Result],
+        embedding: typing.Callable[[], T_Result],
+        agent: typing.Callable[[], T_Result],
+        unknown: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is SpanKind.TOOL:
+            return tool()
+        if self is SpanKind.CHAIN:
+            return chain()
+        if self is SpanKind.LLM:
+            return llm()
+        if self is SpanKind.WORKFLOW:
+            return workflow()
+        if self is SpanKind.RETRIEVER:
+            return retriever()
+        if self is SpanKind.EMBEDDING:
+            return embedding()
+        if self is SpanKind.AGENT:
+            return agent()
+        if self is SpanKind.UNKNOWN:
+            return unknown()

agenta/client/backend/types/span_status_code.py

@@ -0,0 +1,29 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+import typing
+
+T_Result = typing.TypeVar("T_Result")
+
+
+class SpanStatusCode(str, enum.Enum):
+    """
+    An enumeration.
+    """
+
+    UNSET = "UNSET"
+    OK = "OK"
+    ERROR = "ERROR"
+
+    def visit(
+        self,
+        unset: typing.Callable[[], T_Result],
+        ok: typing.Callable[[], T_Result],
+        error: typing.Callable[[], T_Result],
+    ) -> T_Result:
+        if self is SpanStatusCode.UNSET:
+            return unset()
+        if self is SpanStatusCode.OK:
+            return ok()
+        if self is SpanStatusCode.ERROR:
+            return error()

agenta/client/backend/types/span_variant.py

@@ -0,0 +1,38 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class SpanVariant(pydantic.BaseModel):
+    variant_id: str
+    variant_name: str
+    revision: int
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/trace_detail.py

@@ -0,0 +1,52 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .span import Span
+from .span_kind import SpanKind
+from .span_status_code import SpanStatusCode
+from .span_variant import SpanVariant
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class TraceDetail(pydantic.BaseModel):
+    id: str
+    name: str
+    parent_span_id: typing.Optional[str]
+    created_at: dt.datetime
+    variant: SpanVariant
+    environment: typing.Optional[str]
+    spankind: SpanKind
+    status: SpanStatusCode
+    metadata: typing.Dict[str, typing.Any]
+    user_id: typing.Optional[str]
+    children: typing.Optional[typing.List[Span]]
+    content: typing.Dict[str, typing.Any]
+    config: typing.Dict[str, typing.Any]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/with_pagination.py

@@ -0,0 +1,40 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class WithPagination(pydantic.BaseModel):
+    data: typing.List[typing.Any]
+    total: int
+    page: int
+    page_size: int = pydantic.Field(alias="pageSize")
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        allow_population_by_field_name = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/workspace_member_response.py

@@ -0,0 +1,38 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .workspace_permission import WorkspacePermission
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class WorkspaceMemberResponse(pydantic.BaseModel):
+    user: typing.Dict[str, typing.Any]
+    roles: typing.List[WorkspacePermission]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}

agenta/client/backend/types/workspace_permission.py

@@ -0,0 +1,40 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from .permission import Permission
+from .workspace_role import WorkspaceRole
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class WorkspacePermission(pydantic.BaseModel):
+    role_name: WorkspaceRole
+    role_description: typing.Optional[str]
+    permissions: typing.Optional[typing.List[Permission]]
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {
+            "by_alias": True,
+            "exclude_unset": True,
+            **kwargs,
+        }
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        json_encoders = {dt.datetime: serialize_datetime}