Skyvern/scripts/create_organization.py

import asyncio
from typing import Annotated, Optional

import typer

from scripts.create_api_key import create_org_api_token
from skyvern.forge import app
from skyvern.forge.forge_app_initializer import start_forge_app


async def create_org(org_name: str, webhook_callback_url: str | None = None) -> None:
    organization = await app.DATABASE.organizations.create_organization(org_name, webhook_callback_url)
    print(f"Created organization: {organization}")
    await create_org_api_token(organization.organization_id)


def main(
    org_name: str,
    webhook_callback_url: Annotated[Optional[str], typer.Argument()] = None,
) -> None:
    start_forge_app()
    asyncio.run(create_org(org_name, webhook_callback_url))


if __name__ == "__main__":
    typer.run(main)