This repository was archived by the owner on Jul 17, 2026. It is now read-only.
textual v0 + CLI cleanups #1195
There are no files selected for viewing
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1 @@ | ||
| """Agent command module.""" |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,159 @@ | ||
| """Agent command for creating remote agent runs.""" | ||
|
|
||
| import requests | ||
| import typer | ||
| from rich import box | ||
| from rich.console import Console | ||
| from rich.panel import Panel | ||
|
|
||
| from codegen.cli.api.endpoints import API_ENDPOINT | ||
| from codegen.cli.auth.token_manager import get_current_token | ||
| from codegen.cli.rich.spinners import create_spinner | ||
| from codegen.cli.utils.org import resolve_org_id | ||
|
|
||
| console = Console() | ||
|
|
||
| # Create the agent app | ||
| agent_app = typer.Typer(help="Create and manage individual agent runs") | ||
|
|
||
|
|
||
| @agent_app.command() | ||
| def create( | ||
| prompt: str = typer.Option(..., "--prompt", "-p", help="The prompt to send to the agent"), | ||
| org_id: int | None = typer.Option(None, help="Organization ID (defaults to CODEGEN_ORG_ID/REPOSITORY_ORG_ID or auto-detect)"), | ||
| model: str | None = typer.Option(None, help="Model to use for this agent run (optional)"), | ||
| repo_id: int | None = typer.Option(None, help="Repository ID to use for this agent run (optional)"), | ||
| ): | ||
| """Create a new agent run with the given prompt.""" | ||
| # Get the current token | ||
| token = get_current_token() | ||
| if not token: | ||
| console.print("[red]Error:[/red] Not authenticated. Please run 'codegen login' first.") | ||
| raise typer.Exit(1) | ||
|
|
||
| try: | ||
| # Resolve org id | ||
| resolved_org_id = resolve_org_id(org_id) | ||
| if resolved_org_id is None: | ||
| console.print("[red]Error:[/red] Organization ID not provided. Pass --org-id, set CODEGEN_ORG_ID, or REPOSITORY_ORG_ID.") | ||
| raise typer.Exit(1) | ||
|
|
||
| # Prepare the request payload | ||
| payload = { | ||
| "prompt": prompt, | ||
| } | ||
|
|
||
| if model: | ||
| payload["model"] = model | ||
| if repo_id: | ||
| payload["repo_id"] = repo_id | ||
|
|
||
| # Make API request to create agent run with spinner | ||
| spinner = create_spinner("Creating agent run...") | ||
| spinner.start() | ||
|
|
||
| try: | ||
| headers = {"Authorization": f"Bearer {token}", "Content-Type": "application/json"} | ||
| url = f"{API_ENDPOINT.rstrip('/')}/v1/organizations/{resolved_org_id}/agent/run" | ||
| response = requests.post(url, headers=headers, json=payload) | ||
| response.raise_for_status() | ||
| agent_run_data = response.json() | ||
| finally: | ||
| spinner.stop() | ||
|
|
||
| # Extract agent run information | ||
| run_id = agent_run_data.get("id", "Unknown") | ||
| status = agent_run_data.get("status", "Unknown") | ||
| web_url = agent_run_data.get("web_url", "") | ||
| created_at = agent_run_data.get("created_at", "") | ||
|
|
||
| # Format created date | ||
| if created_at: | ||
| try: | ||
| from datetime import datetime | ||
|
|
||
| dt = datetime.fromisoformat(created_at.replace("Z", "+00:00")) | ||
| created_display = dt.strftime("%B %d, %Y at %H:%M") | ||
| except (ValueError, TypeError): | ||
| created_display = created_at | ||
| else: | ||
| created_display = "Unknown" | ||
|
|
||
| # Status with emoji | ||
| status_display = status | ||
| if status == "COMPLETE": | ||
| status_display = "✅ Complete" | ||
| elif status == "RUNNING": | ||
| status_display = "🏃 Running" | ||
| elif status == "FAILED": | ||
| status_display = "❌ Failed" | ||
| elif status == "STOPPED": | ||
| status_display = "⏹️ Stopped" | ||
| elif status == "PENDING": | ||
| status_display = "⏳ Pending" | ||
|
|
||
| # Create result display | ||
| result_info = [] | ||
| result_info.append(f"[cyan]Agent Run ID:[/cyan] {run_id}") | ||
| result_info.append(f"[cyan]Status:[/cyan] {status_display}") | ||
| result_info.append(f"[cyan]Created:[/cyan] {created_display}") | ||
| if web_url: | ||
| result_info.append(f"[cyan]Web URL:[/cyan] {web_url}") | ||
|
|
||
| result_text = "\n".join(result_info) | ||
|
|
||
| console.print( | ||
| Panel( | ||
| result_text, | ||
| title="🤖 [bold]Agent Run Created[/bold]", | ||
| border_style="green", | ||
| box=box.ROUNDED, | ||
| padding=(1, 2), | ||
| ) | ||
| ) | ||
|
|
||
| # Show next steps | ||
| console.print("\n[dim]💡 Track progress with:[/dim] [cyan]codegen agents[/cyan]") | ||
| if web_url: | ||
| console.print(f"[dim]🌐 View in browser:[/dim] [link]{web_url}[/link]") | ||
|
|
||
| except requests.RequestException as e: | ||
| console.print(f"[red]Error creating agent run:[/red] {e}", style="bold red") | ||
| if hasattr(e, "response") and e.response is not None: | ||
| try: | ||
| error_detail = e.response.json().get("detail", "Unknown error") | ||
| console.print(f"[red]Details:[/red] {error_detail}") | ||
| except (ValueError, KeyError): | ||
| pass | ||
| raise typer.Exit(1) | ||
| except Exception as e: | ||
| console.print(f"[red]Unexpected error:[/red] {e}", style="bold red") | ||
| raise typer.Exit(1) | ||
|
|
||
|
|
||
| # Default callback for the agent app | ||
| @agent_app.callback(invoke_without_command=True) | ||
| def agent_callback(ctx: typer.Context): | ||
| """Create and manage individual agent runs.""" | ||
| if ctx.invoked_subcommand is None: | ||
| # If no subcommand is provided, show help | ||
| print(ctx.get_help()) | ||
| raise typer.Exit() | ||
|
|
||
|
|
||
| # For backward compatibility, also allow `codegen agent --prompt "..."` | ||
| def agent( | ||
| prompt: str = typer.Option(None, "--prompt", "-p", help="The prompt to send to the agent"), | ||
| org_id: int | None = typer.Option(None, help="Organization ID (defaults to CODEGEN_ORG_ID/REPOSITORY_ORG_ID or auto-detect)"), | ||
| model: str | None = typer.Option(None, help="Model to use for this agent run (optional)"), | ||
| repo_id: int | None = typer.Option(None, help="Repository ID to use for this agent run (optional)"), | ||
| ): | ||
| """Create a new agent run with the given prompt.""" | ||
| if prompt: | ||
| # If prompt is provided, create the agent run | ||
| create(prompt=prompt, org_id=org_id, model=model, repo_id=repo_id) | ||
| else: | ||
| # If no prompt, show help | ||
| console.print("[red]Error:[/red] --prompt is required") | ||
| console.print("Usage: [cyan]codegen agent --prompt 'Your prompt here'[/cyan]") | ||
| raise typer.Exit(1) | ||
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Oops, something went wrong.
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
ImportError:
resolve_org_idhelper is used but not present in repoMultiple new modules call
from codegen.cli.utils.org import resolve_org_id, yet the packagecodegen.cli.utils.orgdoes not exist in the repository. Importing these modules will crash immediately.