Skyvern/scripts/create_organization.py
2025-04-30 13:21:10 -04:00

24 lines
678 B
Python
Executable file

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)