tars-cli 0.1.0__tar.gz

tars_cli-0.1.0/PKG-INFO

@@ -0,0 +1,18 @@
+Metadata-Version: 2.4
+Name: tars-cli
+Version: 0.1.0
+Summary: tars 平台命令行工具
+Requires-Python: >=3.11
+Requires-Dist: typer>=0.12
+Requires-Dist: httpx>=0.27
+Requires-Dist: rich>=13.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: platformdirs>=4.0
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.12; extra == "dev"
+Requires-Dist: pytest-cov>=5.0; extra == "dev"
+Requires-Dist: respx>=0.21; extra == "dev"
+Requires-Dist: mypy>=1.8; extra == "dev"
+Requires-Dist: ruff>=0.3; extra == "dev"
+Requires-Dist: build>=1.0; extra == "dev"

tars_cli-0.1.0/pyproject.toml

@@ -0,0 +1,43 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "tars-cli"
+version = "0.1.0"
+description = "tars 平台命令行工具"
+requires-python = ">=3.11"
+dependencies = [
+    "typer>=0.12",
+    "httpx>=0.27",
+    "rich>=13.0",
+    "pyyaml>=6.0",
+    "platformdirs>=4.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8.0",
+    "pytest-mock>=3.12",
+    "pytest-cov>=5.0",
+    "respx>=0.21",
+    "mypy>=1.8",
+    "ruff>=0.3",
+    "build>=1.0",
+]
+
+[project.scripts]
+tars = "tars_cli.app:app"
+
+[tool.pytest.ini_options]
+testpaths = ["tests"]
+
+[tool.mypy]
+python_version = "3.11"
+strict = false
+warn_return_any = false
+disallow_untyped_defs = true
+
+[tool.ruff]
+target-version = "py311"
+line-length = 120

tars_cli-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

tars_cli-0.1.0/tars_cli/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.1.0"

tars_cli-0.1.0/tars_cli/__main__.py

@@ -0,0 +1,4 @@
+from tars_cli.app import app
+
+if __name__ == "__main__":
+    app()

tars_cli-0.1.0/tars_cli/app.py

@@ -0,0 +1,164 @@
+from __future__ import annotations
+
+from typing import Optional
+
+import typer
+from rich.console import Console
+
+from tars_cli import __version__
+
+console = Console()
+
+app = typer.Typer(
+    name="tars",
+    help=(
+        "tars CLI - 评估平台命令行工具\n\n"
+        "🚀 快速上手:\n"
+        "  路线 A（智能生成）: tars auth login → tars plugin import → tars dataset generate → tars eval run\n"
+        "  路线 B（手动上传）: tars auth login → tars plugin import → tars template download → tars dataset upload → tars eval run\n\n"
+        "运行 tars quickstart 查看完整示例"
+    ),
+)
+
+auth_app = typer.Typer(
+    name="auth",
+    help="认证管理\n\n典型流程: tars auth login --server <url>",
+    no_args_is_help=True,
+)
+plugin_app = typer.Typer(
+    name="plugin",
+    help="Plugin 管理\n\n典型流程: tars plugin import <git_url> → tars plugin export-report <plugin_id>",
+    no_args_is_help=True,
+)
+dataset_app = typer.Typer(
+    name="dataset",
+    help="数据集管理\n\n典型流程: tars dataset generate <plugin_id> → tars dataset publish <version_id>",
+    no_args_is_help=True,
+)
+eval_app = typer.Typer(
+    name="eval",
+    help="评估管理\n\n典型流程: tars eval run <plugin_id> → tars eval status <task_id>",
+    no_args_is_help=True,
+)
+template_app = typer.Typer(
+    name="template",
+    help="模板管理\n\n典型流程: tars template download plan → tars template download testcase-single",
+    no_args_is_help=True,
+)
+
+app.add_typer(auth_app, name="auth")
+app.add_typer(plugin_app, name="plugin")
+app.add_typer(dataset_app, name="dataset")
+app.add_typer(eval_app, name="eval")
+app.add_typer(template_app, name="template")
+
+# Global options stored in context
+_output_format: str = "table"
+_quiet: bool = False
+_verbose: bool = False
+_server: Optional[str] = None
+
+
+@app.callback(invoke_without_command=True)
+def main(
+    ctx: typer.Context,
+    output: str = typer.Option("table", "--output", "-o", help="输出格式: table, json, yaml"),
+    quiet: bool = typer.Option(False, "--quiet", "-q", help="静默模式，仅输出核心结果"),
+    verbose: bool = typer.Option(False, "--verbose", help="详细模式，输出 HTTP 请求日志"),
+    server: Optional[str] = typer.Option(None, "--server", help="服务端地址"),
+    version: bool = typer.Option(False, "--version", "-V", help="显示版本号"),
+) -> None:
+    global _output_format, _quiet, _verbose, _server
+    if version:
+        typer.echo(f"tars-cli {__version__}")
+        raise typer.Exit()
+    _output_format = output
+    _quiet = quiet
+    _verbose = verbose
+    _server = server
+    if ctx.invoked_subcommand is None:
+        typer.echo(ctx.get_help())
+        raise typer.Exit()
+
+
+def get_global_options() -> tuple[str, bool, bool, Optional[str]]:
+    return _output_format, _quiet, _verbose, _server
+
+
+QUICKSTART_TEXT = """\
+🚀 tars CLI 快速上手
+
+━━━ 路线 A: 智能生成 ━━━
+
+  # 1. 登录
+  tars auth login --server https://tars.example.com
+
+  # 2. 导入 Plugin
+  tars plugin import https://git.example.com/my-plugin.git
+
+  # 3. 智能生成数据集（Agent / Skill）
+  tars dataset generate <plugin_id> --agent-name <name> --publish
+  tars dataset generate <plugin_id> --skill-name <name> --publish
+
+  # 多轮场景
+  tars dataset generate <plugin_id> --agent-name <name> -t multi_turn --publish
+
+  # 4. 运行评估
+  tars eval run <plugin_id> --agent-name <name> --dataset-version <version_id>
+  tars eval run <plugin_id> --skill-name <name> --dataset-version <version_id>
+
+  # 5. 导出报告（可选）
+  tars plugin export-report <plugin_id>
+
+━━━ 路线 B: 手动上传 ━━━
+
+  # 1. 登录
+  tars auth login --server https://tars.example.com
+
+  # 2. 导入 Plugin
+  tars plugin import https://git.example.com/my-plugin.git
+
+  # 3. 下载模板
+  tars template download plan
+  tars template download testcase-single
+  tars template download testcase-multi
+
+  # 4. 填写模板后上传（Agent / Skill）
+  tars dataset upload <plugin_id> --agent-name <name> --plan plan.yaml --cases cases.yaml
+  tars dataset upload <plugin_id> --skill-name <name> --plan plan.yaml --cases cases.yaml
+
+  # 5. 运行评估
+  tars eval run <plugin_id> --agent-name <name> --dataset-version <version_id>
+
+━━━ 其他常用命令 ━━━
+
+  # 查看数据集版本列表
+  tars dataset list <plugin_id> --agent-name <name>
+  tars dataset list <plugin_id> --skill-name <name> -t multi_turn
+
+  # 发布草稿数据集
+  tars dataset publish <version_id>
+
+  # 查看评估任务状态
+  tars eval status <task_id>
+
+━━━ CI/CD 集成 ━━━
+
+  export TARS_ACCESS_TOKEN=eyJhbGc...
+  export TARS_SERVER_URL=https://tars.example.com
+  tars eval run <plugin_id> --agent-name <name> --dataset-version <id> -o json --threshold 0.8
+  # 退出码: 0=通过, 1=未通过, 2=错误
+"""
+
+
+@app.command()
+def quickstart() -> None:
+    """输出完整的端到端使用示例"""
+    typer.echo(QUICKSTART_TEXT)
+
+
+import tars_cli.commands.auth_cmd  # noqa: E402, F401
+import tars_cli.commands.plugin_cmd  # noqa: E402, F401
+import tars_cli.commands.dataset_cmd  # noqa: E402, F401
+import tars_cli.commands.eval_cmd  # noqa: E402, F401
+import tars_cli.commands.template_cmd  # noqa: E402, F401

tars_cli-0.1.0/tars_cli/auth.py

@@ -0,0 +1,58 @@
+from __future__ import annotations
+
+import json
+import os
+import stat
+from dataclasses import dataclass
+
+from tars_cli.config import get_credentials_path, get_env_token, resolve_server_url
+
+
+@dataclass
+class Credentials:
+    server_url: str
+    access_token: str
+    refresh_token: str
+    expires_at: str = ""
+
+
+def save_credentials(creds: Credentials) -> None:
+    path = get_credentials_path()
+    path.parent.mkdir(parents=True, exist_ok=True)
+    path.write_text(json.dumps({
+        "server_url": creds.server_url,
+        "access_token": creds.access_token,
+        "refresh_token": creds.refresh_token,
+        "expires_at": creds.expires_at,
+    }, indent=2))
+    os.chmod(path, stat.S_IRUSR | stat.S_IWUSR)
+
+
+def load_credentials(cli_server: str | None = None) -> Credentials | None:
+    env_token = get_env_token()
+    if env_token:
+        return Credentials(
+            server_url=resolve_server_url(cli_server),
+            access_token=env_token,
+            refresh_token="",
+        )
+
+    path = get_credentials_path()
+    if not path.exists():
+        return None
+    try:
+        data = json.loads(path.read_text())
+        return Credentials(
+            server_url=resolve_server_url(cli_server) if cli_server else data.get("server_url", ""),
+            access_token=data["access_token"],
+            refresh_token=data.get("refresh_token", ""),
+            expires_at=data.get("expires_at", ""),
+        )
+    except (json.JSONDecodeError, KeyError):
+        return None
+
+
+def clear_credentials() -> None:
+    path = get_credentials_path()
+    if path.exists():
+        path.unlink()

tars_cli-0.1.0/tars_cli/client.py

@@ -0,0 +1,117 @@
+from __future__ import annotations
+
+from typing import Any
+
+import httpx
+import typer
+
+from httpx import HTTPTransport
+
+from tars_cli.auth import Credentials, load_credentials, save_credentials
+
+
+class AuthExpiredError(Exception):
+    pass
+
+
+class TarsAPIClient:
+    def __init__(self, server_url: str, credentials: Credentials, verbose: bool = False) -> None:
+        self._server_url = server_url.rstrip("/")
+        self._credentials = credentials
+        self._verbose = verbose
+        self._client = httpx.Client(base_url=f"{self._server_url}/api/v1", timeout=30.0, transport=HTTPTransport(proxy=None))
+
+    @classmethod
+    def from_context(cls, cli_server: str | None = None, verbose: bool = False) -> TarsAPIClient:
+        creds = load_credentials(cli_server)
+        if not creds:
+            typer.echo("未登录，请先执行 tars auth login", err=True)
+            raise typer.Exit(code=2)
+        server_url = creds.server_url
+        return cls(server_url=server_url, credentials=creds, verbose=verbose)
+
+    @property
+    def server_url(self) -> str:
+        return self._server_url
+
+    def _headers(self) -> dict[str, str]:
+        return {"Authorization": f"Bearer {self._credentials.access_token}"}
+
+    def _log_request(self, method: str, url: str, **kwargs: Any) -> None:
+        if self._verbose:
+            typer.echo(f"  → {method} {url}", err=True)
+            if "json" in kwargs:
+                typer.echo(f"    Body: {kwargs['json']}", err=True)
+
+    def _log_response(self, resp: httpx.Response) -> None:
+        if self._verbose:
+            typer.echo(f"  ← {resp.status_code} ({len(resp.content)} bytes)", err=True)
+
+    def _refresh_token(self) -> bool:
+        if not self._credentials.refresh_token:
+            return False
+        try:
+            resp = self._client.post("/auth/refresh", json={"refresh_token": self._credentials.refresh_token})
+            if resp.status_code == 200:
+                data = resp.json()
+                self._credentials.access_token = data["access_token"]
+                self._credentials.refresh_token = data.get("refresh_token", self._credentials.refresh_token)
+                self._credentials.expires_at = data.get("expires_at", "")
+                save_credentials(self._credentials)
+                return True
+        except httpx.HTTPError:
+            pass
+        return False
+
+    def _request(self, method: str, path: str, **kwargs: Any) -> httpx.Response:
+        url = path
+        headers = kwargs.pop("headers", None) or {}
+        headers.update(self._headers())
+        kwargs["headers"] = headers
+        self._log_request(method, url, **kwargs)
+
+        try:
+            resp = self._client.request(method, url, **kwargs)
+        except httpx.ConnectError:
+            typer.echo(f"无法连接到服务端: {self._server_url}", err=True)
+            raise typer.Exit(code=2)
+        except httpx.TimeoutException:
+            typer.echo(f"请求超时: {method} {url}", err=True)
+            raise typer.Exit(code=2)
+
+        self._log_response(resp)
+
+        if resp.status_code == 401:
+            if self._refresh_token():
+                kwargs["headers"].update(self._headers())
+                self._log_request(method, url, **kwargs)
+                resp = self._client.request(method, url, **kwargs)
+                self._log_response(resp)
+            else:
+                raise AuthExpiredError("认证已过期，请重新执行 tars auth login")
+
+        return resp
+
+    def get(self, path: str, **kwargs: Any) -> httpx.Response:
+        try:
+            return self._request("GET", path, **kwargs)
+        except AuthExpiredError as e:
+            typer.echo(str(e), err=True)
+            raise typer.Exit(code=2)
+
+    def post(self, path: str, **kwargs: Any) -> httpx.Response:
+        try:
+            return self._request("POST", path, **kwargs)
+        except AuthExpiredError as e:
+            typer.echo(str(e), err=True)
+            raise typer.Exit(code=2)
+
+    def put(self, path: str, **kwargs: Any) -> httpx.Response:
+        try:
+            return self._request("PUT", path, **kwargs)
+        except AuthExpiredError as e:
+            typer.echo(str(e), err=True)
+            raise typer.Exit(code=2)
+
+    def close(self) -> None:
+        self._client.close()

tars_cli-0.1.0/tars_cli/commands/auth_cmd.py

@@ -0,0 +1,71 @@
+from __future__ import annotations
+
+import httpx
+import typer
+
+from httpx import HTTPTransport
+
+from tars_cli.app import auth_app, get_global_options
+from tars_cli.auth import Credentials, clear_credentials, load_credentials, save_credentials
+from tars_cli.client import TarsAPIClient
+from tars_cli.config import resolve_server_url
+from tars_cli.hints import print_hints
+from tars_cli.output import render
+
+
+@auth_app.command()
+def login(
+    server: str = typer.Option("", "--server", "-s", help="服务端地址"),
+) -> None:
+    """登录 tars 平台"""
+    output_fmt, quiet, verbose, global_server = get_global_options()
+    server_url = resolve_server_url(server or global_server)
+
+    username = typer.prompt("用户名")
+    password = typer.prompt("密码", hide_input=True)
+
+    client = httpx.Client(base_url=f"{server_url}/api/v1", timeout=30.0, transport=HTTPTransport(proxy=None))
+    try:
+        resp = client.post("/auth/login", json={"username": username, "password": password})
+        if resp.status_code != 200:
+            typer.echo(f"登录失败: {resp.json().get('detail', resp.text)}", err=True)
+            raise typer.Exit(code=2)
+
+        data = resp.json()
+        creds = Credentials(
+            server_url=server_url,
+            access_token=data["access_token"],
+            refresh_token=data.get("refresh_token", ""),
+            expires_at=data.get("expires_at", ""),
+        )
+        save_credentials(creds)
+
+        me_resp = client.get("/auth/me", headers={"Authorization": f"Bearer {creds.access_token}"})
+        user_info = me_resp.json() if me_resp.status_code == 200 else {}
+
+        result = {
+            "message": "登录成功",
+            "username": user_info.get("username", username),
+            "server": server_url,
+        }
+        web_url, next_step = print_hints(server_url, "auth_login", quiet=quiet)
+        render(result, output_format=output_fmt, quiet=quiet, web_url=web_url, next_step=next_step)
+    finally:
+        client.close()
+
+
+@auth_app.command()
+def logout() -> None:
+    """登出并清除本地凭证"""
+    output_fmt, quiet, verbose, global_server = get_global_options()
+    creds = load_credentials(global_server)
+    if creds and creds.refresh_token:
+        try:
+            client = TarsAPIClient(server_url=creds.server_url, credentials=creds, verbose=verbose)
+            client.post("/auth/logout", json={"refresh_token": creds.refresh_token})
+            client.close()
+        except Exception:
+            pass
+
+    clear_credentials()
+    typer.echo("已登出")