#!/usr/bin/env python3 # /// script # dependencies = [ # "pinecone>=8.0.0", # "typer>=0.15.0", # "rich>=13.0.0", # ] # /// """ List all Pinecone Assistants in the account. Usage: uv run list.py [--json] [--files] Environment Variables: PINECONE_API_KEY: Required Pinecone API key Output: Formatted table or JSON list of assistants with name, region, status, and host Optionally include files for each assistant with --files flag """ import os import sys import json import typer from rich.console import Console from rich.table import Table from rich.panel import Panel from pinecone import Pinecone app = typer.Typer() console = Console() @app.command() def main( json_output: bool = typer.Option(False, "--json", help="Output in JSON format"), files: bool = typer.Option(False, "--files", "-f", help="Include file listing for each assistant"), ): """List all Pinecone Assistants in your account.""" # Check for API key api_key = os.environ.get('PINECONE_API_KEY') if not api_key: console.print("[red]Error: PINECONE_API_KEY environment variable not set[/red]") console.print("\nGet your API key from: https://app.pinecone.io/?sessionType=signup") raise typer.Exit(1) try: # Initialize Pinecone client pc = Pinecone(api_key=api_key, source_tag="claude_code_plugin:assistant") # List assistants assistants = pc.assistant.list_assistants() if not assistants: if json_output: print(json.dumps({"assistants": [], "count": 0})) else: console.print("[yellow]No assistants found.[/yellow]\n") console.print("Create your first assistant with:") console.print(" [cyan]/pinecone:assistant[/cyan] — \"create a new assistant called [name]\"") return if json_output: # JSON output assistants_data = [] for asst in assistants: asst_data = { "name": asst.name, "region": getattr(asst, 'region', 'unknown'), "status": asst.status, "host": getattr(asst, 'host', ''), } if files: # Get files for this assistant try: assistant_instance = pc.assistant.Assistant(assistant_name=asst.name) file_list = assistant_instance.list_files() asst_data["files"] = [ { "name": f.name, "id": f.id, "status": f.status, "metadata": getattr(f, 'metadata', {}), } for f in file_list ] asst_data["file_count"] = len(file_list) except Exception as e: asst_data["files"] = [] asst_data["file_count"] = 0 asst_data["file_error"] = str(e) assistants_data.append(asst_data) result = { "assistants": assistants_data, "count": len(assistants) } print(json.dumps(result, indent=2)) else: # Rich table output console.print(f"\n[bold]Found {len(assistants)} assistant(s):[/bold]\n") # Assistants table table = Table(show_header=True, header_style="bold cyan") table.add_column("Name", style="green", width=30) table.add_column("Region", style="blue", width=10) table.add_column("Status", style="yellow", width=15) if files: table.add_column("Files", style="magenta", width=10) table.add_column("Host", style="dim", width=40 if files else 50) for asst in assistants: name = asst.name region = getattr(asst, 'region', 'unknown') status = asst.status host = getattr(asst, 'host', '') # Color code status if status == 'ready': status_display = f"[green]{status}[/green]" elif status == 'indexing': status_display = f"[yellow]{status}[/yellow]" else: status_display = status if files: # Get file count for this assistant try: assistant_instance = pc.assistant.Assistant(assistant_name=asst.name) file_list = assistant_instance.list_files() file_count = str(len(file_list)) except Exception: file_count = "?" table.add_row(name, region, status_display, file_count, host) else: table.add_row(name, region, status_display, host) console.print(table) console.print() # If --files flag is set, show detailed file listing for each assistant if files: console.print("[bold]File Details:[/bold]\n") for asst in assistants: try: assistant_instance = pc.assistant.Assistant(assistant_name=asst.name) file_list = assistant_instance.list_files() if file_list: # Create a table for this assistant's files file_table = Table(show_header=True, header_style="bold blue", title=f"[cyan]{asst.name}[/cyan]") file_table.add_column("#", style="dim", width=4) file_table.add_column("File Name", style="green", width=50) file_table.add_column("Status", style="yellow", width=15) file_table.add_column("ID", style="dim", width=30) for idx, file_obj in enumerate(file_list, 1): file_name = file_obj.name file_id = file_obj.id file_status = file_obj.status # Color code file status if file_status == 'available': file_status_display = f"[green]{file_status}[/green]" elif file_status == 'processing': file_status_display = f"[yellow]{file_status}[/yellow]" else: file_status_display = file_status file_table.add_row(str(idx), file_name, file_status_display, file_id) console.print(file_table) console.print() else: console.print(f"[dim]{asst.name}: No files uploaded[/dim]\n") except Exception as e: console.print(f"[red]Error listing files for {asst.name}: {e}[/red]\n") # Next steps panel next_steps = """[bold]Next steps:[/bold] \u2022 List with files: [cyan]/pinecone:assistant[/cyan] \u2014 list my assistants with their files \u2022 Chat: [cyan]/pinecone:assistant[/cyan] \u2014 ask [name] about [your question] \u2022 Upload: [cyan]/pinecone:assistant[/cyan] \u2014 upload files from [path] to [name] \u2022 Context: [cyan]/pinecone:assistant[/cyan] \u2014 search [name] for context about [topic]""" console.print(Panel(next_steps, title="Available Commands", border_style="blue")) except Exception as e: console.print(f"[red]Error listing assistants: {e}[/red]") raise typer.Exit(1) if __name__ == "__main__": app()