dbagent-cli 0.1.6__tar.gz → 0.1.7__tar.gz

{dbagent_cli-0.1.6 → dbagent_cli-0.1.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.1.6
+Version: 0.1.7
 Summary: Universal Database Introspection and Script Generation AI Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.1.6 → dbagent_cli-0.1.7}/dbagent/__init__.py

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

{dbagent_cli-0.1.6 → dbagent_cli-0.1.7}/dbagent/cli.py

@@ -509,7 +509,7 @@ def chat_command(
     console.print(f"[bold green]Connected:[/bold green] {db_name} ({dialect}) - {len(table_names)} tables")
     console.print(f"[bold cyan]AI:[/bold cyan] {llm.name}")
     console.print("[dim]Type your question in plain English. Read queries auto-execute. Write queries ask for confirmation.[/dim]")
-    console.print("[dim]Commands: [bold]:tables[/bold] | [bold]:table <name>[/bold] | [bold]:fk <name>[/bold] | [bold]:explain[/bold] | [bold]:history[/bold] | [bold]:auto[/bold]/[bold]:noauto[/bold] | [bold]:run[/bold] | [bold]:export <file>[/bold] | [bold]:exit[/bold][/dim]\n")
+    console.print("[dim]Commands: [bold]:tables[/bold] | [bold]:table <name>[/bold] | [bold]:fk <name>[/bold] | [bold]:model <name>[/bold] | [bold]:explain[/bold] | [bold]:history[/bold] | [bold]:auto[/bold]/[bold]:noauto[/bold] | [bold]:run[/bold] | [bold]:export <file>[/bold] | [bold]:exit[/bold][/dim]\n")
 
     history_file = config_mgr.config_dir / "chat_history.txt"
     session = PromptSession(history=FileHistory(str(history_file)))
@@ -578,6 +578,28 @@ def chat_command(
                 console.print("")
                 continue
 
+            if user_input.startswith(":model ") or user_input.startswith(":provider ") or user_input.startswith(":use "):
+                parts = user_input.split(maxsplit=2)
+                if len(parts) > 1:
+                    new_target = parts[1].strip()
+                    if new_target.lower() in ["ollama", "gemini", "groq", "openrouter", "mock"]:
+                        llm = get_llm_provider(provider_name=new_target.lower(), config_mgr=config_mgr)
+                        model = None
+                        print_success(f"Switched AI provider to [bold cyan]{llm.name}[/bold cyan]")
+                    else:
+                        model = new_target
+                        print_success(f"Switched active model to [bold cyan]{model}[/bold cyan] (Provider: {llm.name})")
+                else:
+                    print_info(f"Active AI Provider: [bold cyan]{llm.name}[/bold cyan] | Model: [bold green]{model or getattr(llm, 'default_model', 'default')}[/bold green]")
+                continue
+
+            if user_input.lower() 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 on the fly: [bold]:provider <ollama|gemini|groq|openrouter>[/bold] or [bold]:model <model_name>[/bold][/dim]\n")
+                continue
+
             if user_input.lower() == ":explain":
                 if last_result and last_result.sql:
                     print_code(last_result.sql, "sql", "Last Executed SQL")

{dbagent_cli-0.1.6 → dbagent_cli-0.1.7}/dbagent_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.1.6
+Version: 0.1.7
 Summary: Universal Database Introspection and Script Generation AI Agent (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9

{dbagent_cli-0.1.6 → dbagent_cli-0.1.7}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.1.6"
+version = "0.1.7"
 description = "Universal Database Introspection and Script Generation AI Agent (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"