@app.command()
def main(
task: str = typer.Option(..., "-t", "--task", help="Task/problem statement", show_default=False, prompt=True),
model_name: str = typer.Option(
os.getenv("MSWEA_MODEL_NAME"),
"-m",
"--model",
help="Model name (defaults to MSWEA_MODEL_NAME env var)",
prompt="What model do you want to use?",
),
) -> DefaultAgent:
agent = DefaultAgent(
LitellmModel(model_name=model_name),
LocalEnvironment(),
**yaml.safe_load(Path(package_dir / "config" / "default.yaml").read_text())["agent"],
)
agent.run(task)
return agent