Skyvern/scripts/create_organization.py
Kerem Yilmaz 4386ba5012
shu/revert isort (#378)
Co-authored-by: Shuchang Zheng <wintonzheng0325@gmail.com>
2024-05-27 19:33:32 +03:00

24 lines
678 B
Python

import asyncio
from typing import Annotated, Optional
import typer
from scripts.create_api_key import create_org_api_token
from skyvern.forge.app import DATABASE
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)