Skyvern/skyvern/cli/credential.py

91 lines
2.9 KiB
Python

"""Credential CLI commands with MCP-parity output and validation."""
from __future__ import annotations
from typing import Any
import typer
from dotenv import load_dotenv
from skyvern.config import settings
from skyvern.utils.env_paths import resolve_backend_env_path
from .commands._output import run_tool
from .mcp_tools.credential import skyvern_credential_delete as tool_credential_delete
from .mcp_tools.credential import skyvern_credential_get as tool_credential_get
from .mcp_tools.credential import skyvern_credential_list as tool_credential_list
credential_app = typer.Typer(
help="MCP-parity credential commands (list/get/delete). Use `skyvern credentials add` for secure creation.",
no_args_is_help=True,
)
@credential_app.callback()
def credential_callback(
api_key: str | None = typer.Option(
None,
"--api-key",
help="Skyvern API key",
envvar="SKYVERN_API_KEY",
),
) -> None:
"""Load environment and optional API key override."""
load_dotenv(resolve_backend_env_path())
if api_key:
settings.SKYVERN_API_KEY = api_key
@credential_app.command("list")
def credential_list(
page: int = typer.Option(1, "--page", min=1, help="Page number (1-based)."),
page_size: int = typer.Option(10, "--page-size", min=1, max=100, help="Results per page."),
json_output: bool = typer.Option(False, "--json", help="Output as JSON."),
) -> None:
"""List stored credentials (metadata only)."""
async def _run() -> dict[str, Any]:
return await tool_credential_list(page=page, page_size=page_size)
run_tool(
_run,
json_output=json_output,
hint_on_exception="Check your API key and Skyvern connection.",
action="skyvern_credential_list",
)
@credential_app.command("get")
def credential_get(
credential_id: str = typer.Option(..., "--id", "--credential-id", help="Credential ID (starts with cred_)."),
json_output: bool = typer.Option(False, "--json", help="Output as JSON."),
) -> None:
"""Get credential metadata by ID."""
async def _run() -> dict[str, Any]:
return await tool_credential_get(credential_id=credential_id)
run_tool(
_run,
json_output=json_output,
hint_on_exception="Check your API key and credential ID.",
action="skyvern_credential_get",
)
@credential_app.command("delete")
def credential_delete(
credential_id: str = typer.Option(..., "--id", "--credential-id", help="Credential ID (starts with cred_)."),
json_output: bool = typer.Option(False, "--json", help="Output as JSON."),
) -> None:
"""Delete a credential by ID."""
async def _run() -> dict[str, Any]:
return await tool_credential_delete(credential_id=credential_id)
run_tool(
_run,
json_output=json_output,
hint_on_exception="Check your API key and credential ID.",
action="skyvern_credential_delete",
)