dbagent-cli 0.1.9__tar.gz → 0.2.1__tar.gz

{dbagent_cli-0.1.9 → dbagent_cli-0.2.1}/PKG-INFO

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

{dbagent_cli-0.1.9 → dbagent_cli-0.2.1}/dbagent/__init__.py

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

{dbagent_cli-0.1.9 → dbagent_cli-0.2.1}/dbagent/cli.py

@@ -787,30 +787,30 @@ def config_command():
         config_mgr.set_setting("default_provider", provider_choice)
         print_success(f"Default provider updated to [bold]{provider_choice}[/bold]")
 
-    if provider_choice == "gemini" or Prompt.ask("Configure Google Gemini API key (Free Tier)?", choices=["y", "n"], default="n") == "y":
-        key = Prompt.ask("Enter GEMINI_API_KEY (leave blank to skip)", default="")
+    if provider_choice == "gemini":
+        console.print("[dim]Get your free Gemini API key at: [bold]https://aistudio.google.com/apikey[/bold][/dim]")
+        key = Prompt.ask("Enter GEMINI_API_KEY", default=config_mgr.get_setting("gemini_api_key", "")).strip()
         if key:
             config_mgr.set_setting("gemini_api_key", key)
             print_success("Gemini API key saved.")
-
-    if provider_choice == "groq" or Prompt.ask("Configure Groq API key (Free Tier)?", choices=["y", "n"], default="n") == "y":
-        key = Prompt.ask("Enter GROQ_API_KEY (leave blank to skip)", default="")
+    elif provider_choice == "groq":
+        console.print("[dim]Get your free Groq API key at: [bold]https://console.groq.com[/bold][/dim]")
+        key = Prompt.ask("Enter GROQ_API_KEY", default=config_mgr.get_setting("groq_api_key", "")).strip()
         if key:
             config_mgr.set_setting("groq_api_key", key)
             print_success("Groq API key saved.")
-
-    if provider_choice == "openrouter" or Prompt.ask("Configure OpenRouter API key (Free Tier)?", choices=["y", "n"], default="n") == "y":
-        key = Prompt.ask("Enter OPENROUTER_API_KEY (leave blank to skip)", default="")
+    elif provider_choice == "openrouter":
+        console.print("[dim]Get your OpenRouter API key at: [bold]https://openrouter.ai[/bold][/dim]")
+        key = Prompt.ask("Enter OPENROUTER_API_KEY", default=config_mgr.get_setting("openrouter_api_key", "")).strip()
         if key:
             config_mgr.set_setting("openrouter_api_key", key)
             print_success("OpenRouter API key saved.")
+    elif provider_choice == "ollama":
+        base_url = Prompt.ask("Ollama Base URL", default=config_mgr.get_setting("ollama_base_url", "http://localhost:11434")).strip()
+        config_mgr.set_setting("ollama_base_url", base_url)
+        print_success("Ollama base URL configured.")
 
-    if Prompt.ask("\nAdd a new database connection profile?", choices=["y", "n"], default="n") == "y":
-        p_name = Prompt.ask("Profile Name (e.g. prod_pg, staging, local_dev)")
-        p_url = Prompt.ask("Connection URL (e.g. postgresql://user:pass@localhost:5432/dbname)")
-        p_desc = Prompt.ask("Description", default=f"{p_name} database")
-        config_mgr.save_profile(name=p_name, connection_url=p_url, description=p_desc)
-        print_success(f"Profile '[bold]{p_name}[/bold]' saved successfully!")
+    print_success("Configuration updated successfully!\n")
 
 
 # ---- profiles ----
@@ -1013,5 +1013,17 @@ def use_command(
     print_success(f"Default AI provider switched to [bold cyan]{p}[/bold cyan]!")
 
 
+def main():
+    """Top-level entry point with graceful Ctrl+C / interrupt handling."""
+    try:
+        app()
+    except (KeyboardInterrupt, EOFError):
+        console.print("\n[dim]Operation cancelled.[/dim]")
+        sys.exit(0)
+    except Exception as e:
+        console.print(f"\n[red]Error:[/red] {e}")
+        sys.exit(1)
+
+
 if __name__ == "__main__":
-    app()
+    main()

{dbagent_cli-0.1.9 → dbagent_cli-0.2.1}/dbagent_cli.egg-info/PKG-INFO

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

dbagent_cli-0.2.1/dbagent_cli.egg-info/entry_points.txt

@@ -0,0 +1,3 @@
+[console_scripts]
+db-agent = dbagent.cli:main
+dbagent = dbagent.cli:main

{dbagent_cli-0.1.9 → dbagent_cli-0.2.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.1.9"
+version = "0.2.1"
 description = "Universal Database Introspection and Script Generation AI Agent (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"
@@ -35,8 +35,8 @@ dev = [
 ]
 
 [project.scripts]
-db-agent = "dbagent.cli:app"
-dbagent = "dbagent.cli:app"
+db-agent = "dbagent.cli:main"
+dbagent = "dbagent.cli:main"
 
 [tool.setuptools.packages.find]
 where = ["."]

dbagent_cli-0.1.9/dbagent_cli.egg-info/entry_points.txt

@@ -1,3 +0,0 @@
-[console_scripts]
-db-agent = dbagent.cli:app
-dbagent = dbagent.cli:app