videosdkagent-cli 0.0.1__py2.py3-none-any.whl → 0.0.5__py2.py3-none-any.whl

videosdk_cli/main.py

@@ -3,67 +3,98 @@ import asyncio
 import nest_asyncio
 import sys
 from dotenv import load_dotenv
+
 load_dotenv()
 from pathlib import Path
-from videosdk_cli.auth import handle_auth, handle_logout
-from videosdk_cli.projects import list_projects
+from videosdk_cli.auth import auth_cli
 from videosdk_cli.templates import template_cli
 from videosdk_cli.run_agent import run_agent
 from videosdk_cli.utils import analytics
 from videosdk_cli.build import agent_cli
+from videosdk_cli.utils.ui.theme import (
+    console,
+    print_banner,
+    print_error,
+    print_warning,
+    PRIMARY,
+)
 import atexit
-from dotenv import load_dotenv
+
 load_dotenv()
 nest_asyncio.apply()
 
 CURRENT_CMD = " ".join([Path(sys.argv[0]).name] + sys.argv[1:])
 
+
 def run_async(coro):
     loop = asyncio.get_event_loop()
     return loop.run_until_complete(coro)
 
-@click.group()
+
+class VideoSDKCLI(click.Group):
+    """Custom Click Group that shows banner on help."""
+
+    def format_help(self, ctx, formatter):
+        print_banner(mini=False)
+        super().format_help(ctx, formatter)
+
+
+@click.group(cls=VideoSDKCLI)
+@click.version_option(version="0.2.0", prog_name="VideoSDK CLI")
 def cli():
-    """VideoSDK CLI tool"""
+    """
+    Build and deploy AI agents with VideoSDK cloud infrastructure.
+
+    \b
+    GETTING STARTED
+      $ videosdk auth login              # Authenticate
+      $ videosdk agent init --name bot   # Create agent
+      $ videosdk agent build             # Build Docker image
+      $ videosdk agent push              # Push to registry
+      $ videosdk agent deploy            # Deploy to cloud
+
+    \b
+    COMMON COMMANDS
+      auth        Login/logout from VideoSDK
+      agent       Manage agents, deployments, secrets, sessions
+      template    Create projects from templates
+      run         Run agent locally
+
+    \b
+    RESOURCES
+      Docs:     https://docs.videosdk.live
+      Discord:  https://discord.gg/HUe45teT
+    """
     pass
 
-@cli.command()
-def auth():
-    """Authenticate and store token"""
-    try:
-        run_async(handle_auth())
-    except KeyboardInterrupt:
-        click.echo("\nAuthentication cancelled by user.")
-    except Exception as e:
-        raise click.ClickException(f"Authentication failed: Please try again")
 
-@cli.command()
-def logout():
-    """Logout and remove token"""
-    try:
-        run_async(handle_logout())
-    except KeyboardInterrupt:
-        click.echo("\nLogout cancelled by user.")
-    except Exception as e:
-        raise click.ClickException(f"Logout failed: Please try again")
-
-# @cli.command()
-# def projects():
-#     """List projects"""
-#     try:
-#         run_async(list_projects())
-#     except KeyboardInterrupt:
-#         raise
-#     except Exception as e:
-#         raise e
+# Add auth group (videosdk auth login / videosdk auth logout)
+cli.add_command(auth_cli, name="auth")
 
+# Add template group (videosdk template list / videosdk template get)
 cli.add_command(template_cli, name="template")
+
+# Add agent group with all subcommands
 cli.add_command(agent_cli, name="agent")
 
+
 @cli.command()
 @click.argument("project_name")
 def run(project_name):
-    """Run a VideoSDK project"""
+    """
+    Run a VideoSDK agent project locally.
+
+    Starts the agent in console mode for local development and testing.
+    Automatically creates a virtual environment and installs requirements.
+
+    \b
+    Arguments:
+      PROJECT_NAME  Name of the project directory to run
+
+    \b
+    Example:
+      $ videosdk run my-assistant
+    """
     try:
         run_agent(project_name)
     except KeyboardInterrupt:
@@ -71,6 +102,7 @@ def run(project_name):
     except Exception as e:
         raise e
 
+
 def log_command_at_exit():
     """Ensures analytics logging happens on exit (normal or KeyboardInterrupt)"""
     try:
@@ -78,12 +110,13 @@ def log_command_at_exit():
     except Exception as e:
         print(f"Failed to log analytics: {e}")
 
+
 # atexit.register(log_command_at_exit)
 
 if __name__ == "__main__":
     try:
         cli()
     except KeyboardInterrupt:
-        click.echo("\n[bold yellow]Command interrupted by user[/bold yellow]")
+        console.print(f"\n[{PRIMARY}]Command interrupted by user[/{PRIMARY}]")
     except Exception as e:
-        click.echo(f"[bold red]Error:[/bold red] {e}")
+        print_error(str(e))

videosdk_cli/projects.py

@@ -1,8 +1,16 @@
 import asyncio
 from rich.console import Console
 from rich.table import Table
-from videosdk_cli.utils.config_manager import get_config_value, set_config_value, save_config
-from videosdk_cli.utils.api_client import VideoSDKClient, VideoSDKTokenClient, AuthenticationError
+from videosdk_cli.utils.config_manager import (
+    get_config_value,
+    set_config_value,
+    save_config,
+)
+from videosdk_cli.utils.api_client import (
+    VideoSDKClient,
+    VideoSDKTokenClient,
+    AuthenticationError,
+)
 from InquirerPy import inquirer
 
 console = Console()
@@ -10,14 +18,8 @@ console = Console()
 
 async def list_projects():
     """Fetches and displays a list of your projects with arrow selection."""
-    auth_token = get_config_value("auth_token")
-
-    if not auth_token:
-        console.print("[red]Authentication details not found. Please run:[/red] videosdk auth")
-        return
-
     try:
-        client = VideoSDKClient(auth_token=auth_token)
+        client = VideoSDKClient()
         data = await client.fetch_projects()
 
         projects = data.get("data", [])
@@ -33,13 +35,17 @@ async def list_projects():
         table.add_column("Status")
 
         for i, p in enumerate(projects, start=1):
-            status = "[green]Active[/green]" if not p.get("isDeactived") else "[red]Inactive[/red]"
+            status = (
+                "[green]Active[/green]"
+                if not p.get("isDeactived")
+                else "[red]Inactive[/red]"
+            )
             table.add_row(
                 str(i),
                 p.get("name"),
                 p.get("createdAt"),
                 p.get("keyPair", {}).get("apiKey"),
-                status
+                status,
             )
 
         console.print(table)
@@ -56,7 +62,9 @@ async def list_projects():
         selected = projects[selected_index]
 
         set_config_value("selected_project_name", selected["name"])
-        set_config_value("selected_project_api_key", selected.get("keyPair", {}).get("apiKey"))
+        set_config_value(
+            "selected_project_api_key", selected.get("keyPair", {}).get("apiKey")
+        )
         console.print(f"[green]✅ Selected project:[/green] {selected['name']}")
 
         await gen_token()
@@ -76,10 +84,14 @@ async def gen_token():
     auth_token = get_config_value("auth_token")
 
     if not api_key or not project_name:
-        console.print("[red]Project details not found. Please run:[/red] videosdk projects")
+        console.print(
+            "[red]Project details not found. Please run:[/red] videosdk projects"
+        )
         return
     if not auth_token:
-        console.print("[red]Authentication details not found. Please run:[/red] videosdk auth")
+        console.print(
+            "[red]Authentication details not found. Please run:[/red] videosdk auth"
+        )
         return
 
     try:
@@ -93,7 +105,9 @@ async def gen_token():
             console.print(f"[green]✅ Generated token:[/green]")
             set_config_value("VIDEOSDK_AUTH_TOKEN", token)
         else:
-            console.print("[red]❌ Token not found. Check your API key or auth token.[/red]")
+            console.print(
+                "[red]❌ Token not found. Check your API key or auth token.[/red]"
+            )
 
     except AuthenticationError as e:
         console.print(f"[red]Authentication error: {e}[/red]")
@@ -106,12 +120,15 @@ async def gen_token():
 async def handle_auth_reset():
     """Ask user before resetting auth token."""
     reset = await inquirer.confirm(
-        message="Do you want to reset your authentication token?",
-        default=False
+        message="Do you want to reset your authentication token?", default=False
     ).execute_async()
 
     if reset:
         save_config(None)
-        console.print("[yellow]Auth token cleared. Please run 'videosdk auth' again.[/yellow]")
+        console.print(
+            "[yellow]Auth token cleared. Please run 'videosdk auth' again.[/yellow]"
+        )
     else:
-        console.print("[yellow]Auth token not cleared. You may experience errors until re-authenticated.[/yellow]")
+        console.print(
+            "[yellow]Auth token not cleared. You may experience errors until re-authenticated.[/yellow]"
+        )

videosdk_cli/templates.py

@@ -11,20 +11,39 @@ from videosdk_cli.utils.videosdk_yaml_helper import AgentConfig
 
 console = Console()
 
+
 @click.group()
 def template_cli():
-    """Manage, download, and create projects from official VideoSDK templates."""
+    """
+    Manage and create projects from VideoSDK templates.
+
+    Templates provide pre-configured AI agent setups with different
+    pipeline types (cascading, real-time) and provider combinations.
+
+    \b
+    Commands:
+      list  Show all available templates
+      get   Create a new project from a template
+    """
     pass
 
 
 @template_cli.command(name="list")
 @click.option(
-    "--refresh", "-r",
-    is_flag=True,
-    help="Force refresh template metadata from server"
+    "--refresh", "-r", is_flag=True, help="Force refresh template metadata from server"
 )
 def list_templates(refresh: bool):
-    """List all available templates."""
+    """
+    List all available templates.
+
+    Shows a table of all VideoSDK agent templates with their IDs,
+    filenames, and descriptions.
+
+    \b
+    Examples:
+      $ videosdk template list
+      $ videosdk template list --refresh  # force refresh from server
+    """
     asyncio.run(list_templates_async(refresh))
 
 
@@ -43,28 +62,37 @@ async def list_templates_async(force_refresh: bool = False):
     table.add_column("Template id", style="cyan", no_wrap=True)
     table.add_column("File Name", style="green", no_wrap=True)
     table.add_column("Description", style="white")
-    # table.add_column("Mode", style="magenta")
-    # table.add_column("Pipeline Type", style="blue")
 
     for name, tpl in templates.items():
-        table.add_row(
-            name,
-            tpl.file_name,
-            tpl.description
-            # tpl.mode,
-            # tpl.pipe_type
-        )
+        table.add_row(name, tpl.file_name, tpl.description)
 
     console.print(table)
+
+
 @template_cli.command(name="get")
 @click.option(
-    "--template", "-t",
+    "--template",
+    "-t",
     required=True,
-    help="Template name to create project from"
+    help="Template ID to create project from (see 'template list')",
 )
 @click.argument("app_name")
 def create_project(template: str, app_name: str):
-    """Create a new project from a template."""
+    """
+    Create a new project from a template.
+
+    Downloads the template code and sets up a complete project directory
+    with configuration files, requirements, and environment variables.
+
+    \b
+    Arguments:
+      APP_NAME  Name of the project directory to create
+
+    \b
+    Examples:
+      $ videosdk template get -t ai-agent-cascading-pipeline my-assistant
+      $ videosdk template get -t ai-agent-real-time-pipeline voice-bot
+    """
     asyncio.run(create_project_async(template, app_name))
 
 
@@ -76,12 +104,16 @@ async def create_project_async(template_id: str, app_name: str):
     tpl = kv.get_template(template_id)
 
     if not tpl:
-        console.print(f"[bold red]Error:[/bold red] Template '[cyan]{template_id}[/cyan]' not found in cache.")
+        console.print(
+            f"[bold red]Error:[/bold red] Template '[cyan]{template_id}[/cyan]' not found in cache."
+        )
         return
 
     app_dir = Path(app_name)
     if app_dir.exists():
-        console.print(f"[bold red]Error:[/bold red] Directory '[cyan]{app_name}[/cyan]' already exists.")
+        console.print(
+            f"[bold red]Error:[/bold red] Directory '[cyan]{app_name}[/cyan]' already exists."
+        )
         return
 
     console.print(f"[cyan]→ Mode:[/cyan] {tpl.mode}")
@@ -109,17 +141,27 @@ async def create_project_async(template_id: str, app_name: str):
                 provider_modules.append(module)
 
     elif tpl.mode == "custom" and tpl.pipe_type == "cascading":
-        STT = await template_helper.select_provider_arrow("Select STT provider:", tpl.STT_options)
-        LLM = await template_helper.select_provider_arrow("Select LLM provider:", tpl.LLM_options)
-        TTS = await template_helper.select_provider_arrow("Select TTS provider:", tpl.TTS_options)
+        STT = await template_helper.select_provider_arrow(
+            "Select STT provider:", tpl.STT_options
+        )
+        LLM = await template_helper.select_provider_arrow(
+            "Select LLM provider:", tpl.LLM_options
+        )
+        TTS = await template_helper.select_provider_arrow(
+            "Select TTS provider:", tpl.TTS_options
+        )
         providers_dict = {"STT": STT, "LLM": LLM, "TTS": TTS}
-        imports, provider_modules = template_helper.generate_cascading_providers(providers_dict)
+        imports, provider_modules = template_helper.generate_cascading_providers(
+            providers_dict
+        )
 
     api_keys = await template_helper.prompt_provider_api_keys(providers_dict)
 
     env_path = await template_helper.prompt_existing_env()
     if env_path and not env_path.exists():
-        console.print("[red]Provided environment path does not exist. A new venv will be created.[/red]")
+        console.print(
+            "[red]Provided environment path does not exist. A new venv will be created.[/red]"
+        )
         env_path = None
 
     with Progress(
@@ -127,23 +169,32 @@ async def create_project_async(template_id: str, app_name: str):
         TextColumn("[progress.description]{task.description}"),
         transient=True,
     ) as progress:
-        progress.add_task(description=f"Creating project [cyan]{app_name}[/cyan]...", total=None)
+        progress.add_task(
+            description=f"Creating project [cyan]{app_name}[/cyan]...", total=None
+        )
         try:
             app_dir.mkdir(parents=True, exist_ok=True)
             template_helper.write_env_file(app_dir, providers_dict, api_keys)
-            template_helper.write_requirements(app_dir, template_helper.STATIC_REQUIREMENTS + provider_modules)
-            videosdk_yaml_helper.update_agent_config(app_dir, AgentConfig(templateId=template_id))
+            template_helper.write_requirements(
+                app_dir, template_helper.STATIC_REQUIREMENTS + provider_modules
+            )
+            videosdk_yaml_helper.update_agent_config(
+                app_dir, AgentConfig(templateId=template_id)
+            )
         except Exception as e:
             console.print(f"[bold red]Error creating project files: {e}[/bold red]")
             return
 
     try:
         import aiohttp
+
         async with aiohttp.ClientSession() as session:
             url = f"{template_helper.CDN_BASE_URL}/{tpl.file_name}"
             async with session.get(url) as resp:
                 if resp.status == 404:
-                    console.print(f"[bold red]Template '{template_id}' not found on server[/bold red]")
+                    console.print(
+                        f"[bold red]Template '{template_id}' not found on server[/bold red]"
+                    )
                     return
                 resp.raise_for_status()
                 template_code = await resp.text()
@@ -152,7 +203,9 @@ async def create_project_async(template_id: str, app_name: str):
         return
 
     imports_str = "\n".join(imports) if imports else ""
-    template_helper.write_template_file(app_dir, tpl.file_name, imports_str, template_code)
+    template_helper.write_template_file(
+        app_dir, tpl.file_name, imports_str, template_code
+    )
 
     template_helper.write_config(
         app_dir=app_dir,
@@ -161,8 +214,10 @@ async def create_project_async(template_id: str, app_name: str):
         description=tpl.description,
         providers=providers_dict,
         env_path=str(env_path) if env_path else None,
-        auto_configured=(tpl.mode=="preconfig"),
-        requirements_installed=False
+        auto_configured=(tpl.mode == "preconfig"),
+        requirements_installed=False,
     )
 
-    console.print(f"[green]✅ Project '[bold cyan]{app_name}[/bold cyan]' created successfully![/green]")
+    console.print(
+        f"[green]✅ Project '[bold cyan]{app_name}[/bold cyan]' created successfully![/green]"
+    )

videosdk_cli/utils/analytics.py

@@ -10,6 +10,7 @@ ANALYTICS_FILE = CONFIG_DIR / "analytics.json"
 
 _pending_analytics = []
 
+
 async def log_command(command: str):
     """
     Log a CLI command for analytics.
@@ -24,7 +25,7 @@ async def log_command(command: str):
         "user_id": user_id,
         "username": username,
         "api_key": api_key,
-        "command": command
+        "command": command,
     }
 
     if ANALYTICS_FILE.exists():
@@ -47,9 +48,9 @@ async def send_analytics_to_server(data):
     Send analytics to your server endpoint.
     """
     import aiohttp
-    from videosdk_cli.utils.template_helper import CDN_BASE_URL
+    from videosdk_cli.utils.template_helper import API_SERVER_URL
 
-    url = f"{CDN_BASE_URL}/analytics/collect/"
+    url = f"{API_SERVER_URL}/analytics/collect/"
     try:
         async with aiohttp.ClientSession() as session:
             async with session.post(url, json=data) as resp:

videosdk_cli/utils/api_client.py

@@ -1,30 +1,35 @@
 import aiohttp
 import logging
-from videosdk_cli.utils.config_manager import load_config, save_config, set_config_value
+from videosdk_cli.utils.config_manager import (
+    load_config,
+    save_config,
+    set_config_value,
+    get_config_value,
+)
 from InquirerPy import inquirer
 
 logger = logging.getLogger(__name__)
 logger.setLevel(logging.INFO)
 
 
-
 class AuthenticationError(Exception):
     """Raised when the stored authentication is invalid or expired."""
-    pass
 
+    pass
 
 
 class VideoSDKClient:
-    def __init__(self, auth_token):
-        self.auth_token = auth_token
-        self.url = "https://api.videosdk.live/v1/apikeys"
+    def __init__(self):
+        from videosdk_cli.utils.template_helper import API_BASE_URL
+
+        self.auth_token = get_config_value("VIDEOSDK_AUTH_TOKEN") or ""
+        self.url = API_BASE_URL + "/v1/apikeys"
         self.headers = {
             "Authorization": self.auth_token,
-            "Content-Type": "application/json"
+            "Content-Type": "application/json",
         }
         self.res_data = None
 
-    
     async def fetch_projects(self):
         async with aiohttp.ClientSession() as session:
             async with session.get(self.url, headers=self.headers) as resp:
@@ -41,50 +46,49 @@ class VideoSDKClient:
                 self.res_data = await resp.json()
                 return self.res_data
 
-    
     def _reset_credentials(self):
         """Clear stored auth_token."""
         logger.info("Resetting stored credentials...")
         try:
             save_config(None)
-        
+
         except Exception:
             config = load_config()
             config = None
             save_config(config)
 
 
-
 class VideoSDKTokenClient:
     def __init__(self, auth_token, api_key, expiresIn):
-        self.url = f"https://api.videosdk.live/v1/apikeys/{api_key}/token"
-        self.headers = {
-            "Authorization": auth_token,
-            "Content-Type": "application/json"
-        }
+        from videosdk_cli.utils.template_helper import API_BASE_URL
+
+        self.url = API_BASE_URL + f"/v1/apikeys/{api_key}/token"
+        self.headers = {"Authorization": auth_token, "Content-Type": "application/json"}
         self.payload = {
             "apiKey": api_key,
             "permissions": ["allow_join"],
-            "expiresIn": expiresIn
+            "expiresIn": expiresIn,
         }
         self.res_data = None
 
-    
     async def fetch_token(self):
         async with aiohttp.ClientSession() as session:
-            async with session.post(self.url, headers=self.headers, json=self.payload) as resp:
+            async with session.post(
+                self.url, headers=self.headers, json=self.payload
+            ) as resp:
                 text = await resp.text()
                 if 400 <= resp.status < 500:
                     logger.warning("Session expired or invalid credentials.")
                     reset = ask_reset_credentials()
                     if reset:
                         self._reset_credentials()
-                    raise AuthenticationError(f"Authentication failed ({resp.status}): {text}")
+                    raise AuthenticationError(
+                        f"Authentication failed ({resp.status}): {text}"
+                    )
                 resp.raise_for_status()
                 self.res_data = await resp.json()
                 return self.res_data
 
-    
     def _reset_credentials(self):
         """Clear stored auth_token."""
         logger.info("Resetting stored credentials...")
@@ -96,12 +100,10 @@ class VideoSDKTokenClient:
             save_config(config)
 
 
-
 def ask_reset_credentials():
     """Prompt user before clearing saved auth token."""
     result = inquirer.confirm(
         message="Request failed. Do you want to reset your stored authentication token?",
-        default=False
+        default=False,
     ).execute()
     return result
-