dbagent-cli 0.6.0__tar.gz → 0.6.1__tar.gz

{dbagent_cli-0.6.0 → dbagent_cli-0.6.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.6.0
+Version: 0.6.1
 Summary: Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.6.0 → dbagent_cli-0.6.1}/dbagent/__init__.py

@@ -2,5 +2,5 @@
 DB-Agent: Universal Database Introspection and Script Generation AI Agent (CLI).
 """
 
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 __author__ = "Santhosh Gupta"

{dbagent_cli-0.6.0 → dbagent_cli-0.6.1}/dbagent/cli.py

@@ -531,9 +531,9 @@ def chat_command(
                     console.print("  [bold]:doctor[/bold]                    -> Run database health diagnostics")
                     console.print("  [bold]:mode <name>[/bold]               -> Switch mode (SAFE, READ_ONLY, CONFIRM, SANDBOX, ADMIN)")
                     console.print("  [bold]:changes[/bold]                   -> List recent write ChangeSets")
-                    console.print("  [bold]:provider <name>[/bold]           -> Switch AI provider (ollama, gemini, groq, openrouter)")
-                    console.print("  [bold]:model <name>[/bold]              -> Switch active model name (e.g. gemini-3.6-flash)")
-                    console.print("  [bold]:models[/bold]                     -> Show active AI provider and model status")
+                    console.print("  [bold]:provider <name>[/bold]           -> Switch AI provider (gemini, ollama, groq, openrouter)")
+                    console.print("  [bold]:models[/bold]                     -> View list of available AI models & speed benchmarks")
+                    console.print("  [bold]:model <num|name>[/bold]          -> Switch active model (:model 1, :model 3.7, :model 3.6)")
                     console.print("  [bold]:page <n>[/bold]                  -> View column page <n> for wide multi-column tables")
                     console.print("  [bold]:card[/bold]                      -> View last result in expanded vertical card format (all columns)")
                     console.print("  [bold]:wide[/bold]                      -> Force-show all columns in terminal (zoom out)")
@@ -699,21 +699,53 @@ def chat_command(
                         print_error(f"Unknown provider '{cmd_arg}'. Supported: ollama, gemini, groq, openrouter")
                     continue
 
-                if cmd_name == ":model":
-                    if not cmd_arg:
-                        curr_model = model or getattr(llm, "default_model", "default")
-                        console.print(f"Active Model: [bold green]{curr_model}[/bold green] (Provider: {llm.name})")
-                        console.print("[dim]Usage: [bold]:model <model_name>[/bold][/dim]")
+                if cmd_name in [":model", ":models", ":providers", ":ai"]:
+                    avail_models = llm.list_models() if hasattr(llm, "list_models") else []
+                    curr_model = model or getattr(llm, "default_model", "default")
+
+                    # If no arg passed or :models/:providers called, show rich model selection menu
+                    if not cmd_arg or cmd_name in [":models", ":providers", ":ai"]:
+                        console.print(f"\n[bold cyan]Active AI Provider:[/bold cyan] [bold white]{llm.name}[/bold white]")
+                        console.print(f"[bold cyan]Active Model:[/bold cyan] [bold green]{curr_model}[/bold green]\n")
+                        if avail_models:
+                            console.print(f"[bold]Available Models for {llm.name}:[/bold]")
+                            model_descriptions = {
+                                "gemini-flash-lite-latest": "⚡ Ultra-Fast (~0.8s) • Recommended for instant chat & queries",
+                                "gemini-3.6-flash": "⚡ High Quality (~3.5s) • Strong SQL reasoning & accuracy",
+                                "gemini-3.7-flash": "🧠 Deep Reasoning (~13s) • Latest flagship model",
+                                "gemini-3.1-flash-lite": "⚡ Fast Lite (~4.0s) • Low latency",
+                                "gemini-flash-latest": "⚡ Standard Flash (~7.0s)",
+                            }
+                            for idx, m_name in enumerate(avail_models, 1):
+                                is_active = " [bold green]✓ Active[/bold green]" if m_name == curr_model else ""
+                                desc = model_descriptions.get(m_name, "")
+                                desc_str = f" - [dim]{desc}[/dim]" if desc else ""
+                                console.print(f"  [bold cyan]{idx}[/bold cyan]. [bold white]{m_name}[/bold white]{desc_str}{is_active}")
+                            console.print("\n[dim]Quick Switch: [bold]:model 1[/bold] (ultra-fast) | [bold]:model 3.7[/bold] | [bold]:model 3.6[/bold] | [bold]:model <name>[/bold][/dim]\n")
                         continue
-                    model = cmd_arg
-                    print_success(f"Switched active model to [bold cyan]{model}[/bold cyan] (Provider: {llm.name})")
-                    continue
 
-                if cmd_name in [":models", ":providers", ":ai"]:
-                    curr_model = model or getattr(llm, "default_model", "default")
-                    console.print(f"\n[bold cyan]Active AI Provider:[/bold cyan] [bold white]{llm.name}[/bold white]")
-                    console.print(f"[bold cyan]Active Model:[/bold cyan] [bold green]{curr_model}[/bold green]")
-                    console.print("[dim]Switch anytime: [bold]:provider <ollama|gemini|groq|openrouter>[/bold] or [bold]:model <name>[/bold][/dim]\n")
+                    # Handle numeric shortcut or name shortcut
+                    target_model = cmd_arg.strip()
+                    if target_model.isdigit() and avail_models:
+                        choice_idx = int(target_model) - 1
+                        if 0 <= choice_idx < len(avail_models):
+                            target_model = avail_models[choice_idx]
+                    elif target_model in ["3.7", "37"]:
+                        target_model = "gemini-3.7-flash"
+                    elif target_model in ["3.6", "36"]:
+                        target_model = "gemini-3.6-flash"
+                    elif target_model in ["lite", "fast", "flash-lite", "1"]:
+                        target_model = "gemini-flash-lite-latest"
+                    elif target_model in ["3.1"]:
+                        target_model = "gemini-3.1-flash-lite"
+                    elif avail_models:
+                        matches = [m for m in avail_models if target_model.lower() in m.lower()]
+                        if matches:
+                            target_model = matches[0]
+
+                    model = target_model
+                    pipeline.model = model
+                    print_success(f"Switched active model to [bold cyan]{model}[/bold cyan] (Provider: {llm.name})")
                     continue
 
                 if cmd_name == ":explain":

{dbagent_cli-0.6.0 → dbagent_cli-0.6.1}/dbagent/llm/gemini_provider.py

@@ -14,19 +14,18 @@ class GeminiProvider(BaseLLMProvider):
     """Google Gemini LLM provider (Free Tier available)."""
 
     FALLBACK_MODELS = [
+        "gemini-flash-lite-latest",
         "gemini-3.6-flash",
         "gemini-3.7-flash",
-        "gemini-3.5-flash",
+        "gemini-3.1-flash-lite",
         "gemini-flash-latest",
-        "gemini-pro-latest",
-        "gemini-2.5-flash",
     ]
 
     def __init__(self, api_key: Optional[str] = None, default_model: str = "gemini-3.6-flash"):
         self.api_key = api_key or os.getenv("GEMINI_API_KEY")
         self.default_model = default_model
         self._session = requests.Session()
-        self._session.headers.update({"User-Agent": "DB-Agent/0.5.0"})
+        self._session.headers.update({"User-Agent": "DB-Agent/0.6.0"})
 
     @property
     def name(self) -> str:
@@ -117,7 +116,7 @@ class GeminiProvider(BaseLLMProvider):
                     else:
                         raise RuntimeError(f"Gemini API error ({response.status_code}): {response.text}")
                 else:
-                    response = self._session.post(endpoint, json=payload, timeout=30)
+                    response = self._session.post(endpoint, json=payload, timeout=15)
                     if response.status_code == 200:
                         data = response.json()
                         candidates = data.get("candidates", [])

{dbagent_cli-0.6.0 → dbagent_cli-0.6.1}/dbagent_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.6.0
+Version: 0.6.1
 Summary: Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.6.0 → dbagent_cli-0.6.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.6.0"
+version = "0.6.1"
 description = "Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"