diff --git a/devops_agent/cli.py b/devops_agent/cli.py index 816656d..a1fff74 100644 --- a/devops_agent/cli.py +++ b/devops_agent/cli.py @@ -1,4 +1,5 @@ from pathlib import Path +import os import click from rich.console import Console @@ -17,7 +18,17 @@ def cli(): def default_provider() -> str: - return "openai" + """Automatically detect which LLM provider to use based on available API keys.""" + if os.environ.get("OPENAI_API_KEY"): + return "openai" + elif os.environ.get("ANTHROPIC_API_KEY"): + return "anthropic" + elif os.environ.get("GEMINI_API_KEY"): + return "google" + else: + # If nothing found, fallback and warn + console.print("[bold yellow]⚠️ No API key found. Defaulting to OpenAI (if set later).[/bold yellow]") + return "openai" @cli.command() @@ -31,7 +42,7 @@ def run(log_file, provider, query, output, format, interactive): """Run the DevOps agent with specified options""" if not provider: - console.print("[yellow]No provider specified, defaulting to openai[/yellow]") + console.print("[yellow]No provider specified[/yellow]") provider = default_provider() # Interactive mode