-
Notifications
You must be signed in to change notification settings - Fork 0
Open
Labels
featureNew featureNew featureintegrationsFramework integrationsFramework integrationsweek-1Week 1 tasksWeek 1 tasks
Description
Description
Implement command-line interface for AgentGym.
Acceptance Criteria
- CLI entry point in
src/agentgym/cli/main.py -
agentgym --versioncommand -
agentgym train --scenario X --framework Ycommand - Beautiful terminal output using Rich
- Help text and usage examples
- Integration test showing full CLI usage
Technical Details
import typer
from rich.console import Console
app = typer.Typer()
console = Console()
@app.command()
def train(
scenario: str = typer.Option(..., help="Scenario name"),
framework: str = typer.Option("langchain", help="Framework to use"),
episodes: int = typer.Option(10000, help="Number of episodes"),
):
"""Train an agent using reinforcement learning."""
console.print(f"[bold]Training {framework} agent...[/bold]")
# Create trainer
config = TrainingConfig(
scenario=scenario,
framework=framework,
episodes=episodes
)
trainer = Trainer(config)
# Train
result = trainer.train()
# Display results
console.print(f"[green]✓[/green] Training complete!")
console.print(f"Tool reliability: {result.metrics.tool_reliability:.1%}")References
- Typer docs: https://typer.tiangolo.com/
- Rich docs: https://rich.readthedocs.io/
Estimated Time
3-4 hours
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
featureNew featureNew featureintegrationsFramework integrationsFramework integrationsweek-1Week 1 tasksWeek 1 tasks