tinet-agent-cli 0.1.0.dev36__py3-none-any.whl

taco/commands/profile.py

@@ -0,0 +1,124 @@
+from __future__ import annotations
+
+import os
+from typing import Any
+
+import typer
+
+from taco.core.context import get_cli_context
+from taco.core.errors import TacoError
+from taco.core.output import emit_success, exit_with_error
+from taco.core.profile_manager import ProfileManager
+
+
+profile_app = typer.Typer(help="查看 Zenava/TACO profile，并管理 TACO profile")
+
+
+@profile_app.command("set", help="创建或修改 profile，并切换为当前 profile。")
+def set_profile(
+    ctx: typer.Context,
+    name: str = typer.Option(..., "--name", help="唯一 profile 名称。"),
+    endpoint: str = typer.Option(..., "--endpoint", help="开发者中心平台根地址。"),
+    access_key_id: str = typer.Option(..., "--access-key-id", help="AccessKeyId。"),
+    access_key_secret: str | None = typer.Option(
+        None,
+        "--access-key-secret",
+        help="AccessKeySecret；自动化环境推荐使用 TACO_ACCESS_KEY_SECRET。",
+    ),
+) -> None:
+    cli_context = get_cli_context(ctx)
+    secret = access_key_secret or os.getenv("TACO_ACCESS_KEY_SECRET")
+    if not secret and not cli_context.no_input:
+        secret = typer.prompt("AccessKeySecret", hide_input=True)
+    if not secret:
+        exit_with_error(
+            ctx,
+            TacoError(
+                code="ACCESS_KEY_SECRET_REQUIRED",
+                message="缺少 AccessKeySecret。",
+                exit_code=2,
+                hint="请传入 --access-key-secret，或设置 TACO_ACCESS_KEY_SECRET。",
+            ),
+        )
+    try:
+        result = ProfileManager().set_profile(
+            name=name,
+            endpoint=endpoint,
+            access_key_id=access_key_id,
+            access_key_secret=secret,
+        )
+    except TacoError as error:
+        exit_with_error(ctx, error)
+    emit_success(ctx, result, human_message=f"已保存并切换到 profile `{name}`。")
+
+
+@profile_app.command("list", help="列出 Zenava 和 TACO 的全部 profile 及当前状态。")
+def list_profiles(ctx: typer.Context) -> None:
+    try:
+        result = ProfileManager().list_profiles()
+    except TacoError as error:
+        exit_with_error(ctx, error)
+    emit_success(ctx, result, human_message=_format_list(result))
+
+
+@profile_app.command("use", help="切换当前 TACO profile。")
+def use(ctx: typer.Context, profile: str = typer.Argument(...)) -> None:
+    try:
+        result = ProfileManager().use(profile)
+    except TacoError as error:
+        exit_with_error(ctx, error)
+    emit_success(ctx, result, human_message=f"当前 profile 已切换为 `{profile}`。")
+
+
+@profile_app.command("delete", help="删除指定 TACO profile。")
+def delete(
+    ctx: typer.Context,
+    profile: str = typer.Argument(...),
+    yes: bool = typer.Option(False, "--yes", help="确认删除 profile。"),
+) -> None:
+    cli_context = get_cli_context(ctx)
+    if not yes:
+        if cli_context.no_input:
+            exit_with_error(
+                ctx,
+                TacoError(
+                    code="PROFILE_DELETE_CONFIRM_REQUIRED",
+                    message="删除 profile 前必须显式确认。",
+                    exit_code=2,
+                    hint=f"确认后执行 taco profile delete {profile} --yes。",
+                ),
+            )
+        if not typer.confirm(f"确认删除 profile `{profile}`？"):
+            raise typer.Abort()
+    try:
+        result = ProfileManager().delete(profile)
+    except TacoError as error:
+        exit_with_error(ctx, error)
+    emit_success(ctx, result, human_message=f"已删除 profile `{profile}`。")
+
+
+def _format_list(profiles: list[dict[str, Any]]) -> str:
+    if not profiles:
+        return "尚未找到 Zenava 或 TACO profile。"
+    columns = (
+        ("CURRENT", "current"),
+        ("SOURCE", "source"),
+        ("KEY", "key"),
+        ("ENDPOINT", "endpoint"),
+        ("ACCESS-KEY-ID", "access_key_id"),
+        ("ACCESS-KEY-SECRET", "access_key_secret"),
+        ("ACCESS-TOKEN", "access_token"),
+        ("ACCESS-TOKEN-UPDATED-AT", "access_token_updated_at"),
+    )
+    rows = [
+        ["*" if item["current"] else "", *(str(item[key]) or "-" for _, key in columns[1:])]
+        for item in profiles
+    ]
+    widths = [
+        max(len(title), *(len(row[index]) for row in rows))
+        for index, (title, _) in enumerate(columns)
+    ]
+    header = "  ".join(title.ljust(widths[index]) for index, (title, _) in enumerate(columns))
+    separator = "  ".join("-" * width for width in widths)
+    body = ["  ".join(value.ljust(widths[index]) for index, value in enumerate(row)) for row in rows]
+    return "\n".join([header, separator, *body])

taco/commands/tool.py

@@ -0,0 +1,61 @@
+from __future__ import annotations
+
+import typer
+
+from taco.core.context import get_cli_context
+from taco.core.errors import TacoError
+from taco.core.output import emit_success, exit_with_error
+from taco.services.tool_service import ToolService
+
+
+tool_app = typer.Typer(help="内置工具发现")
+tool_provider_app = typer.Typer(help="工具 Provider 发现")
+tool_app.add_typer(tool_provider_app, name="provider")
+
+
+@tool_provider_app.command("list", help="列出 builtin/plugin 工具 Provider。")
+def list_tool_providers(ctx: typer.Context) -> None:
+    cli_context = get_cli_context(ctx)
+    try:
+        result = ToolService().list_providers(profile=cli_context.profile)
+    except TacoError as error:
+        exit_with_error(ctx, error, json_default=True)
+    emit_success(ctx, result, json_default=True)
+
+
+@tool_app.command("list", help="列出内置工具，可按 Provider 过滤。")
+def list_tools(
+    ctx: typer.Context,
+    provider: str | None = typer.Option(
+        None,
+        "--provider",
+        help="完整 Provider ID。",
+    ),
+) -> None:
+    cli_context = get_cli_context(ctx)
+    try:
+        result = ToolService().list_tools(
+            provider=provider,
+            profile=cli_context.profile,
+        )
+    except TacoError as error:
+        exit_with_error(ctx, error, json_default=True)
+    emit_success(ctx, result, json_default=True)
+
+
+@tool_app.command("describe", help="读取工具参数与 Provider 凭据 schema。")
+def describe_tool(
+    ctx: typer.Context,
+    provider: str = typer.Option(..., "--provider", help="完整 Provider ID。"),
+    tool: str = typer.Option(..., "--tool", help="工具 action 名称。"),
+) -> None:
+    cli_context = get_cli_context(ctx)
+    try:
+        result = ToolService().describe_tool(
+            provider=provider,
+            tool=tool,
+            profile=cli_context.profile,
+        )
+    except TacoError as error:
+        exit_with_error(ctx, error, json_default=True)
+    emit_success(ctx, result, json_default=True)