Files
Dorod-Sky/skyvern/cli/commands.py
Prakash Maheshwaran c1f3a282f4 Added skyvern init mcp and init browser (#2407)
Co-authored-by: ellipsis-dev[bot] <65095814+ellipsis-dev[bot]@users.noreply.github.com>
2025-05-20 20:45:15 -04:00

49 lines
1.4 KiB
Python

import typer
from dotenv import load_dotenv
from .docs import docs_app
from .init_command import init, init_browser, init_mcp
from .run_commands import run_app
from .setup_commands import setup_mcp_command
from .tasks import tasks_app
from .workflow import workflow_app
cli_app = typer.Typer()
cli_app.add_typer(run_app, name="run")
cli_app.add_typer(workflow_app, name="workflow")
cli_app.add_typer(tasks_app, name="tasks")
cli_app.add_typer(docs_app, name="docs")
setup_app = typer.Typer()
cli_app.add_typer(setup_app, name="setup")
init_app = typer.Typer(invoke_without_command=True)
cli_app.add_typer(init_app, name="init")
setup_app.command(name="mcp")(setup_mcp_command)
@init_app.callback()
def init_callback(
ctx: typer.Context,
no_postgres: bool = typer.Option(False, "--no-postgres", help="Skip starting PostgreSQL container"),
) -> None:
"""Run full initialization when no subcommand is provided."""
if ctx.invoked_subcommand is None:
init(no_postgres=no_postgres)
@init_app.command(name="browser")
def init_browser_command() -> None:
"""Initialize only the browser configuration."""
init_browser()
@init_app.command(name="mcp")
def init_mcp_command() -> None:
"""Initialize only the MCP server configuration."""
init_mcp()
if __name__ == "__main__": # pragma: no cover - manual CLI invocation
load_dotenv()
cli_app()