Skyvern/scripts/create_organization.py
Kerem Yilmaz 59fa78e642
Move all cloud imports after skyvern imports (#371)
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
2024-05-27 00:17:41 -07:00

25 lines
679 B
Python

import asyncio
from typing import Annotated, Optional
import typer
from skyvern.forge.app import DATABASE
from scripts.create_api_key import create_org_api_token
async def create_org(org_name: str, webhook_callback_url: str | None = None) -> None:
organization = await DATABASE.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:
asyncio.run(create_org(org_name, webhook_callback_url))
if __name__ == "__main__":
typer.run(main)