dbagent-cli 0.8.0__tar.gz → 0.8.1__tar.gz

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (81) hide show
  1. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/PKG-INFO +1 -1
  2. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/__init__.py +1 -1
  3. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/cli.py +156 -66
  4. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/ui/chat_screen.py +25 -21
  5. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent_cli.egg-info/PKG-INFO +1 -1
  6. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/pyproject.toml +1 -1
  7. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/README.md +0 -0
  8. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/confidence.py +0 -0
  9. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/conversation.py +0 -0
  10. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/doctor.py +0 -0
  11. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/error_classifier.py +0 -0
  12. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/generator.py +0 -0
  13. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/pipeline.py +0 -0
  14. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/recommender.py +0 -0
  15. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/agent/validator.py +0 -0
  16. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/config.py +0 -0
  17. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/connectors/base.py +0 -0
  18. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/connectors/factory.py +0 -0
  19. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/connectors/mongo.py +0 -0
  20. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/connectors/relational.py +0 -0
  21. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/__init__.py +0 -0
  22. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/cache.py +0 -0
  23. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/changeset.py +0 -0
  24. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/cost.py +0 -0
  25. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/recovery.py +0 -0
  26. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/sandbox.py +0 -0
  27. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/execution/streaming.py +0 -0
  28. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/anthropic_provider.py +0 -0
  29. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/auto_setup.py +0 -0
  30. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/base.py +0 -0
  31. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/deepseek_provider.py +0 -0
  32. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/factory.py +0 -0
  33. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/gemini_provider.py +0 -0
  34. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/groq_provider.py +0 -0
  35. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/mistral_provider.py +0 -0
  36. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/mock_provider.py +0 -0
  37. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/ollama_provider.py +0 -0
  38. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/openai_provider.py +0 -0
  39. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/llm/openrouter_provider.py +0 -0
  40. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/schema/cache.py +0 -0
  41. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/schema/formatter.py +0 -0
  42. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/schema/graph.py +0 -0
  43. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/schema/models.py +0 -0
  44. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/schema/retriever.py +0 -0
  45. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/schema/selector.py +0 -0
  46. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/security/__init__.py +0 -0
  47. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/security/audit.py +0 -0
  48. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/security/environment.py +0 -0
  49. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/security/masking.py +0 -0
  50. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/security/modes.py +0 -0
  51. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/security/rbac.py +0 -0
  52. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/semantic/__init__.py +0 -0
  53. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/semantic/dictionary.py +0 -0
  54. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/semantic/templates.py +0 -0
  55. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/sql/__init__.py +0 -0
  56. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/sql/optimizer.py +0 -0
  57. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/sql/validator.py +0 -0
  58. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/ui/console.py +0 -0
  59. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/ui/grid_window.py +0 -0
  60. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/ui/viewer.py +0 -0
  61. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent/ui/visualizer.py +0 -0
  62. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent_cli.egg-info/SOURCES.txt +0 -0
  63. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent_cli.egg-info/dependency_links.txt +0 -0
  64. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent_cli.egg-info/entry_points.txt +0 -0
  65. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent_cli.egg-info/requires.txt +0 -0
  66. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/dbagent_cli.egg-info/top_level.txt +0 -0
  67. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/setup.cfg +0 -0
  68. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_cache.py +0 -0
  69. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_cli.py +0 -0
  70. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_connectors.py +0 -0
  71. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_direct_sql.py +0 -0
  72. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_doctor_and_viz.py +0 -0
  73. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_generator.py +0 -0
  74. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_graph.py +0 -0
  75. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_providers.py +0 -0
  76. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_recovery.py +0 -0
  77. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_safety.py +0 -0
  78. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_sandbox.py +0 -0
  79. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_schema.py +0 -0
  80. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_security.py +0 -0
  81. {dbagent_cli-0.8.0 → dbagent_cli-0.8.1}/tests/test_semantic.py +0 -0
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dbagent-cli
3
- Version: 0.8.0
3
+ Version: 0.8.1
4
4
  Summary: Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)
5
5
  Author: Santhosh Gupta
6
6
  Requires-Python: >=3.9
@@ -2,5 +2,5 @@
2
2
  DB-Agent: Universal Database Introspection and Script Generation AI Agent (CLI).
3
3
  """
4
4
 
5
- __version__ = "0.8.0"
5
+ __version__ = "0.8.1"
6
6
  __author__ = "Santhosh Gupta"
@@ -439,7 +439,6 @@ def chat_command(
439
439
  mode: str = typer.Option("CONFIRM", "--mode", help="Operation mode (SAFE, READ_ONLY, CONFIRM, SANDBOX, ADMIN)"),
440
440
  ):
441
441
  """Interactive multi-turn AI database copilot session."""
442
- print_banner()
443
442
  db_url = resolve_db_url(db)
444
443
 
445
444
  with DynamicSpinner(DynamicSpinner.CONNECT_MESSAGES):
@@ -542,31 +541,30 @@ def chat_command(
542
541
  break
543
542
 
544
543
  if cmd_name in [":help", ":h", ":?"]:
545
- console.print("\n[bold cyan]Interactive Chat Commands:[/bold cyan]")
546
- console.print(" [bold]:tables[/bold] -> List all tables in this database")
547
- console.print(" [bold]:table <name>[/bold] -> Inspect columns, PKs, types & sample data")
548
- console.print(" [bold]:fk <name>[/bold] -> View mapped foreign keys & related tables")
549
- console.print(" [bold]:path <t1> <t2>[/bold] -> Discover shortest join path between tables")
550
- console.print(" [bold]:join <t1> <t2>[/bold] -> Explain join cardinality & syntax")
551
- console.print(" [bold]:doctor[/bold] -> Run database health diagnostics")
552
- console.print(" [bold]:mode <name>[/bold] -> Switch mode (SAFE, READ_ONLY, CONFIRM, SANDBOX, ADMIN)")
553
- console.print(" [bold]:changes[/bold] -> List recent write ChangeSets")
554
- console.print(" [bold]:provider <name>[/bold] -> Switch AI provider (gemini, ollama, groq, openrouter)")
555
- console.print(" [bold]:models[/bold] -> View list of available AI models & speed benchmarks")
556
- console.print(" [bold]:model <num|name>[/bold] -> Switch active model (:model 1, :model 3.7, :model 3.6)")
557
- console.print(" [bold]:page <n>[/bold] -> View column page <n> for wide multi-column tables")
558
- console.print(" [bold]:card[/bold] -> View last result in expanded vertical card format (all columns)")
559
- console.print(" [bold]:wide[/bold] -> Force-show all columns in terminal (zoom out)")
560
- console.print(" [bold]:scroll[/bold] -> Open full-screen interactive horizontal/vertical scroller")
561
- console.print(" [bold]:window[/bold] -> Open desktop GUI window with scrollbars & search")
562
- console.print(" [bold]:web[/bold] -> Open browser-based interactive data grid")
563
- console.print(" [bold]:explain[/bold] -> Show generated SQL & query analysis for last turn")
564
- console.print(" [bold]:history[/bold] -> View history of queries in this session")
565
- console.print(" [bold]:auto[/bold] / [bold]:noauto[/bold] -> Toggle automatic execution of read queries")
566
- console.print(" [bold]:run[/bold] -> Re-execute the last generated SQL query")
567
- console.print(" [bold]:clear[/bold] -> Clear conversation context")
568
- console.print(" [bold]:export <file>[/bold] -> Save the last generated SQL to a file")
569
- console.print(" [bold]:exit[/bold] -> Exit chat session\n")
544
+ console.print("\n[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
545
+ console.print("[bold cyan]⌨️ DB-Agent Interactive Commands:[/bold cyan]\n")
546
+ cmd_table = Table(box=None, padding=(0, 2), show_header=False)
547
+ cmd_table.add_column("Command", style="bold cyan", width=22)
548
+ cmd_table.add_column("Description", style="dim")
549
+ cmd_table.add_row(":models, :model", "Switch active AI model (Claude 3.7, GPT-4o, DeepSeek, Gemini...)")
550
+ cmd_table.add_row(":provider <name>", "Switch AI provider (claude, openai, deepseek, mistral, gemini)")
551
+ cmd_table.add_row(":tables, :dt", "List all tables & schemas in this database")
552
+ cmd_table.add_row(":table <name>", "Inspect columns, primary keys, data types & sample data")
553
+ cmd_table.add_row(":fk <name>", "View foreign key relationships & dependency graph")
554
+ cmd_table.add_row(":path <t1> <t2>", "Discover shortest SQL join path between two tables")
555
+ cmd_table.add_row(":doctor", "Run database health, index, and query performance diagnostics")
556
+ cmd_table.add_row(":mode <name>", "Switch safety mode (SAFE, CONFIRM, ADMIN, READ_ONLY)")
557
+ cmd_table.add_row(":history", "View session query history, execution timing & row counts")
558
+ cmd_table.add_row(":page <n>, :card", "Inspect wide multi-column query results")
559
+ cmd_table.add_row(":window, :web", "Open interactive desktop or browser data grid")
560
+ cmd_table.add_row(":explain", "Explain last generated SQL query with execution plan")
561
+ cmd_table.add_row(":run", "Re-execute the last generated SQL query")
562
+ cmd_table.add_row(":clear", "Clear conversation memory & redraw screen")
563
+ cmd_table.add_row(":export <file>", "Save the last generated SQL to a file")
564
+ cmd_table.add_row(":exit, :q", "Exit DB-Agent session")
565
+ console.print(cmd_table)
566
+ console.print("[dim]──────────────────────────────────────────────────────────────────────────────[/dim]")
567
+ console.print("[dim]Enter to continue · Type any natural language question or direct SQL[/dim]\n")
570
568
  continue
571
569
 
572
570
  if cmd_name in [":tables", ":schema", ":dt"]:
@@ -727,46 +725,46 @@ def chat_command(
727
725
 
728
726
  # If no arg passed or :models/:providers called, show rich model selection menu
729
727
  if not cmd_arg or cmd_name in [":models", ":providers", ":ai"]:
730
- console.print(f"\n[bold cyan]Active AI Provider:[/bold cyan] [bold white]{llm.name}[/bold white]")
731
- console.print(f"[bold cyan]Active Model:[/bold cyan] [bold green]{curr_model}[/bold green]\n")
732
- if avail_models:
733
- console.print(f"[bold]Available Models for {llm.name}:[/bold]")
734
- model_descriptions = {
735
- # Gemini
736
- "gemini-flash-lite-latest": " Ultra-Fast (~0.8s) Instant chat & queries",
737
- "gemini-3.6-flash": "⚡ High Quality (~3.5s) Strong SQL reasoning & accuracy",
738
- "gemini-3.7-flash": "🧠 Deep Reasoning (~13s) • Latest flagship model",
739
- "gemini-3.1-flash-lite": "⚡ Fast Lite (~4.0s) • Low latency",
740
- "gemini-flash-latest": " Standard Flash (~7.0s)",
741
- # Claude
742
- "claude-3-7-sonnet-20250219": "🧠 Hybrid Reasoning Flagship Deepest architecture understanding",
743
- "claude-3-5-sonnet-20241022": " High Intelligence • Expert SQL engineer & schema specialist",
744
- "claude-3-5-haiku-20241022": "⚡ Ultra-Fast (~1s) • High speed and cost efficient",
745
- "claude-3-opus-20240229": "Deep complex schema reasoning",
746
- # OpenAI
747
- "gpt-4o": " Omni Flagship • High speed multimodal & complex SQL reasoning",
748
- "gpt-4o-mini": " Ultra-Fast (~1s) High speed and cost efficient",
749
- "gpt-4.5-preview": "🧠 Next-Generation Reasoning model",
750
- "o3-mini": "🧠 Deep STEM & Advanced Coding Reasoning",
751
- "o1": "🧠 Maximum Reasoning & Math Architecture",
752
- # DeepSeek
753
- "deepseek-chat": "⚡ DeepSeek-V3 Ultra-Fast • Strong code & SQL",
754
- "deepseek-reasoner": "🧠 DeepSeek-R1 Full Chain-of-Thought Reasoning",
755
- # Mistral
756
- "codestral-latest": "⚡ Specialized Code & SQL Architect (22B)",
757
- "mistral-large-latest": "🧠 Flagship Reasoning & Multi-language",
758
- "mistral-small-latest": "⚡ Fast & lightweight",
759
- # Groq
760
- "llama-3.3-70b-versatile": "⚡ Llama 3.3 70B (500+ tokens/sec on Groq LPUs)",
761
- "llama-3.1-8b-instant": "⚡ Ultra-Fast Llama 3.1 8B (800+ tokens/sec)",
762
- "qwen-2.5-coder-32b": "⚡ Qwen 2.5 Coder 32B specialized SQL engineer",
763
- }
764
- for idx, m_name in enumerate(avail_models, 1):
765
- is_active = " [bold green]✓ Active[/bold green]" if m_name == curr_model else ""
766
- desc = model_descriptions.get(m_name, "")
767
- desc_str = f" - [dim]{desc}[/dim]" if desc else ""
768
- console.print(f" [bold cyan]{idx}[/bold cyan]. [bold white]{m_name}[/bold white]{desc_str}{is_active}")
769
- console.print("\n[dim]Quick Switch: [bold]:model <number|name>[/bold] (e.g. [bold]:model 1[/bold] or [bold]:model 3.7[/bold] or [bold]:model 4o[/bold] or [bold]:model r1[/bold])[/dim]\n")
728
+ console.print(f"\n[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
729
+ console.print(f"[bold cyan]Available Models for {llm.name}:[/bold cyan]\n")
730
+ model_descriptions = {
731
+ # Gemini
732
+ "gemini-flash-lite-latest": "⚡ Ultra-Fast (~0.8s) · Instant chat & queries",
733
+ "gemini-3.6-flash": "⚡ High Quality (~3.5s) · Strong SQL reasoning & accuracy",
734
+ "gemini-3.7-flash": "🧠 Deep Reasoning (~13s) · Latest flagship model",
735
+ "gemini-3.1-flash-lite": "⚡ Fast Lite (~4.0s) · Low latency",
736
+ "gemini-flash-latest": " Standard Flash (~7.0s)",
737
+ # Claude
738
+ "claude-3-7-sonnet-20250219": "🧠 Hybrid Reasoning Flagship · Deepest architecture understanding",
739
+ "claude-3-5-sonnet-20241022": "⚡ High Intelligence · Expert SQL engineer & schema specialist",
740
+ "claude-3-5-haiku-20241022": " Ultra-Fast (~1s) · High speed and cost efficient",
741
+ "claude-3-opus-20240229": "Deep complex schema reasoning",
742
+ # OpenAI
743
+ "gpt-4o": " Omni Flagship · High speed multimodal & complex SQL reasoning",
744
+ "gpt-4o-mini": "⚡ Ultra-Fast (~1s) · High speed and cost efficient",
745
+ "gpt-4.5-preview": "🧠 Next-Generation Reasoning model",
746
+ "o3-mini": "🧠 Deep STEM & Advanced Coding Reasoning",
747
+ "o1": "🧠 Maximum Reasoning & Math Architecture",
748
+ # DeepSeek
749
+ "deepseek-chat": " DeepSeek-V3 Ultra-Fast · Strong code & SQL",
750
+ "deepseek-reasoner": "🧠 DeepSeek-R1 Full Chain-of-Thought Reasoning",
751
+ # Mistral
752
+ "codestral-latest": " Specialized Code & SQL Architect (22B)",
753
+ "mistral-large-latest": "🧠 Flagship Reasoning & Multi-language",
754
+ "mistral-small-latest": "⚡ Fast & lightweight",
755
+ # Groq
756
+ "llama-3.3-70b-versatile": "⚡ Llama 3.3 70B (500+ tokens/sec on Groq LPUs)",
757
+ "llama-3.1-8b-instant": "⚡ Ultra-Fast Llama 3.1 8B (800+ tokens/sec)",
758
+ "qwen-2.5-coder-32b": "⚡ Qwen 2.5 Coder 32B specialized SQL engineer",
759
+ }
760
+ for idx, m_name in enumerate(avail_models, 1):
761
+ marker = "[bold cyan]❯[/bold cyan]" if m_name == curr_model else " "
762
+ active_str = " [bold green]✓ Active[/bold green]" if m_name == curr_model else ""
763
+ desc = model_descriptions.get(m_name, "")
764
+ desc_str = f" · [dim]{desc}[/dim]" if desc else ""
765
+ console.print(f" {marker} [bold cyan]{idx}.[/bold cyan] [bold white]{m_name}[/bold white]{active_str}{desc_str}")
766
+ console.print("\n[dim]──────────────────────────────────────────────────────────────────────────────[/dim]")
767
+ console.print("[dim]Quick Switch: [bold]:model <1-5|name>[/bold] (e.g. [bold]:model 1[/bold] or [bold]:model 3.7[/bold] or [bold]:model 4o[/bold])[/dim]\n")
770
768
  continue
771
769
 
772
770
  # Handle numeric shortcut or name shortcut
@@ -1867,6 +1865,98 @@ def use_command(
1867
1865
  print_success(f"Default AI provider switched to [bold cyan]{p}[/bold cyan]!")
1868
1866
 
1869
1867
 
1868
+ def interactive_entrypoint(
1869
+ db: Optional[str] = None,
1870
+ provider: Optional[str] = None,
1871
+ model: Optional[str] = None,
1872
+ mode: Optional[str] = None,
1873
+ ):
1874
+ """
1875
+ Direct root interactive launcher for DB-Agent.
1876
+ Prompts user to select or configure a database connection and transitions into chat.
1877
+ """
1878
+ if db:
1879
+ chat_command(db=db, alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
1880
+ return
1881
+
1882
+ saved_profiles = config_mgr.list_profiles()
1883
+
1884
+ console.print("")
1885
+ console.print("[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
1886
+ console.print("[bold cyan]⚡ DB-AGENT — Universal AI Database Copilot[/bold cyan]\n")
1887
+ console.print(f"[dim white]Working Directory: {os.getcwd()}[/dim white]\n")
1888
+ console.print("[bold white]Select a database connection to launch:[/bold white]\n")
1889
+
1890
+ options = []
1891
+ # If saved connections exist
1892
+ if saved_profiles:
1893
+ for p_name, prof in saved_profiles.items():
1894
+ p_url = prof.get("url", "")
1895
+ dialect = p_url.split("://")[0] if "://" in p_url else "database"
1896
+ options.append((p_name, f"[bold white]{p_name}[/bold white] [dim]({dialect})[/dim]"))
1897
+
1898
+ # Also add quick options
1899
+ options.append(("__new__", "[bold cyan]+ Add a new database connection URL[/bold cyan]"))
1900
+ options.append(("__sqlite__", "[bold white]Connect to local SQLite (:memory:)[/bold white]"))
1901
+ options.append(("__exit__", "[dim]Exit[/dim]"))
1902
+
1903
+ for idx, (opt_id, opt_label) in enumerate(options, 1):
1904
+ marker = "[bold cyan]❯[/bold cyan]" if idx == 1 else " "
1905
+ console.print(f" {marker} [bold cyan]{idx}.[/bold cyan] {opt_label}")
1906
+
1907
+ console.print(f"\n[dim]Enter to confirm · Type number [1-{len(options)}] to select · Ctrl+C to exit[/dim]\n")
1908
+
1909
+ try:
1910
+ raw_choice = Prompt.ask("[bold cyan]Select database[/bold cyan]", default="1")
1911
+ raw_choice = raw_choice.strip()
1912
+ if not raw_choice.isdigit() or int(raw_choice) < 1 or int(raw_choice) > len(options):
1913
+ print_error("Invalid selection.")
1914
+ return
1915
+
1916
+ selected_idx = int(raw_choice) - 1
1917
+ sel_id, _ = options[selected_idx]
1918
+
1919
+ if sel_id == "__exit__":
1920
+ return
1921
+ elif sel_id == "__sqlite__":
1922
+ chat_command(db="sqlite:///:memory:", alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
1923
+ elif sel_id == "__new__":
1924
+ new_url = Prompt.ask("Enter database connection URL (e.g. postgresql://user:pass@host:5432/db)")
1925
+ if not new_url.strip():
1926
+ return
1927
+ new_name = Prompt.ask("Save profile name as (leave empty for temporary)", default="")
1928
+ if new_name.strip():
1929
+ config_mgr.save_profile(name=new_name.strip(), connection_url=new_url.strip(), description=f"Connection {new_name}")
1930
+ print_success(f"Connection profile '[bold cyan]{new_name}[/bold cyan]' saved!")
1931
+ chat_command(db=new_name.strip(), alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
1932
+ else:
1933
+ chat_command(db=new_url.strip(), alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
1934
+ else:
1935
+ # Saved connection name
1936
+ chat_command(db=sel_id, alias=None, provider=provider, model=model, mode=mode or "CONFIRM")
1937
+ except (KeyboardInterrupt, EOFError):
1938
+ console.print("\n[dim]Operation cancelled.[/dim]")
1939
+ return
1940
+
1941
+
1942
+ @app.callback(invoke_without_command=True)
1943
+ def main_callback(
1944
+ ctx: typer.Context,
1945
+ db: Optional[str] = typer.Option(None, "--db", "-d", help="Database connection profile or URL"),
1946
+ provider: Optional[str] = typer.Option(None, "--provider", "-p", help="AI provider name"),
1947
+ model: Optional[str] = typer.Option(None, "--model", "-m", help="AI model name"),
1948
+ mode: Optional[str] = typer.Option("CONFIRM", "--mode", help="Operation mode (SAFE, CONFIRM, ADMIN)"),
1949
+ version: Optional[bool] = typer.Option(None, "--version", "-v", help="Show DB-Agent version"),
1950
+ ):
1951
+ """DB-Agent: Enterprise AI Database Copilot CLI."""
1952
+ if version:
1953
+ console.print(f"DB-Agent version [bold green]{__version__}[/bold green] by {__author__}")
1954
+ raise typer.Exit(0)
1955
+
1956
+ if ctx.invoked_subcommand is None:
1957
+ interactive_entrypoint(db=db, provider=provider, model=model, mode=mode)
1958
+
1959
+
1870
1960
  def main():
1871
1961
  """Top-level entry point with graceful Ctrl+C / interrupt handling."""
1872
1962
  try:
@@ -44,28 +44,28 @@ def render_chat_header(
44
44
  mode: str = "CONFIRM",
45
45
  recommendations: Optional[List[str]] = None,
46
46
  ) -> None:
47
- """Render the exact Claude Code CLI split-card header panel."""
47
+ """Render DB-Agent's split-card welcome header in signature cyan & green theme."""
48
48
  model_str = ai_model or "gemini-3.6-flash"
49
49
 
50
50
  # Left Column: Greeting, Mascot, Connected Model & DB Path
51
51
  left_grid = Table.grid(padding=(0, 1))
52
52
  left_grid.add_column(justify="center", min_width=28)
53
- left_grid.add_row("[bold white]Welcome back![/bold white]")
54
- left_grid.add_row(f"[bold #D97757]{MASCOT_ART.strip()}[/bold #D97757]")
55
- left_grid.add_row(f"[dim white]{model_str} · {ai_provider}[/dim white]")
56
- left_grid.add_row(f"[dim]{db_name} ({dialect} • {table_count} tables)[/dim]")
53
+ left_grid.add_row("[bold white]Welcome to DB-Agent![/bold white]")
54
+ left_grid.add_row(f"[bold cyan]{MASCOT_ART.strip()}[/bold cyan]")
55
+ left_grid.add_row(f"[bold cyan]{model_str}[/bold cyan] [dim]({ai_provider})[/dim]")
56
+ left_grid.add_row(f"[bold green]🟢 Connected:[/bold green] [bold white]{db_name}[/bold white] [dim]({dialect} • {table_count} tables)[/dim]")
57
57
 
58
- # Right Column: Tips for getting started + What's new in v0.8.0
58
+ # Right Column: Quick Tips + Powerful Features
59
59
  right_grid = Table.grid(padding=(0, 1))
60
60
  right_grid.add_column()
61
- right_grid.add_row("[bold #D97757]Tips for getting started[/bold #D97757]")
61
+ right_grid.add_row("[bold cyan]💡 Quick Database Tips[/bold cyan]")
62
62
  right_grid.add_row("[dim]Ask in plain English or paste direct SQL for 0ms execution[/dim]")
63
63
  right_grid.add_row("[dim]Type :help to inspect tables (:tables), relations (:fk), or diagnostics (:doctor)[/dim]")
64
64
  right_grid.add_row("[dim]────────────────────────────────────────────────────────────────────────[/dim]")
65
- right_grid.add_row("[bold #D97757]What's new in v0.8.0[/bold #D97757]")
66
- right_grid.add_row("[dim]Added Universal AI: Claude 3.7 Sonnet, OpenAI GPT-4o, DeepSeek R1, Mistral, Gemini[/dim]")
67
- right_grid.add_row("[dim]Added Zero-Latency Direct SQL Fast-Path with autonomous AI auto-repair on error[/dim]")
68
- right_grid.add_row("[dim]Added Multi-turn conversational session memory graph & persistent connection pooling[/dim]")
65
+ right_grid.add_row(f"[bold green]🚀 Features in v{__version__}[/bold green]")
66
+ right_grid.add_row("[dim]Universal AI: Claude 3.7 Sonnet, OpenAI GPT-4o, DeepSeek R1, Mistral, Gemini[/dim]")
67
+ right_grid.add_row("[dim]Zero-Latency Direct SQL Fast-Path with autonomous AI auto-repair on error[/dim]")
68
+ right_grid.add_row("[dim]Multi-turn conversational session memory graph & persistent connection pooling[/dim]")
69
69
 
70
70
  layout_table = Table.grid(expand=True, padding=(0, 2))
71
71
  layout_table.add_column(ratio=2)
@@ -74,14 +74,14 @@ def render_chat_header(
74
74
 
75
75
  header_panel = Panel(
76
76
  layout_table,
77
- title=f"[bold #D97757] DB-Agent v{__version__} [/bold #D97757]",
77
+ title=f"[bold cyan] DB-AGENT [/bold cyan][bold green]v{__version__}[/bold green]",
78
78
  title_align="left",
79
- border_style="#D97757",
79
+ border_style="cyan",
80
80
  padding=(1, 1),
81
81
  )
82
82
 
83
83
  console.print(header_panel)
84
- console.print(f"[bold #D97757]│[/bold #D97757] [bold #D97757]⚡ Claude 3.7 & Gemini 3.7 are live![/bold #D97757] [dim]Type [bold white]:models[/bold white] to switch models · [bold white]:doctor[/bold white] for diagnostics[/dim]\n")
84
+ console.print(f"[bold cyan]│[/bold cyan] [bold cyan]⚡ Multi-AI Engine Live![/bold cyan] [dim]Type [bold white]:models[/bold white] to switch AI models · [bold white]:doctor[/bold white] for diagnostics · Mode: [bold yellow]{mode}[/bold yellow][/dim]\n")
85
85
 
86
86
  if recommendations:
87
87
  console.print("[dim italic]💡 Try asking:[/dim italic]")
@@ -200,16 +200,20 @@ def render_assistant_turn(
200
200
 
201
201
 
202
202
  def render_session_history(query_history: List[PipelineResult]) -> None:
203
- """Render a clean visual session history table."""
203
+ """Render a clean visual session history table in DB-Agent cyan/green style."""
204
204
  if not query_history:
205
205
  console.print("[dim]No query history in this session yet.[/dim]\n")
206
206
  return
207
207
 
208
- table = Table(title="📜 Session Query History", border_style="cyan", show_lines=True)
208
+ console.print("")
209
+ console.print("[bold cyan]──────────────────────────────────────────────────────────────────────────────[/bold cyan]")
210
+ console.print("[bold cyan]📜 Session Query History:[/bold cyan]\n")
211
+
212
+ table = Table(box=None, padding=(0, 2), show_header=True, header_style="bold cyan")
209
213
  table.add_column("#", style="dim", width=4, justify="right")
210
- table.add_column("Status", width=8, justify="center")
214
+ table.add_column("Status", width=10, justify="center")
211
215
  table.add_column("Type", width=8, justify="center")
212
- table.add_column("Query / Message Preview", style="white", min_width=30)
216
+ table.add_column("Query / Message Preview", style="white", min_width=32)
213
217
  table.add_column("Rows", style="cyan", width=6, justify="right")
214
218
  table.add_column("Time", style="green", width=8, justify="right")
215
219
 
@@ -231,9 +235,9 @@ def render_session_history(query_history: List[PipelineResult]) -> None:
231
235
  preview = (r.error or r.sql or "")[:60]
232
236
  rows = "0"
233
237
 
234
- time_str = f"{r.execution_time_ms / 1000:.1f}s" if r.execution_time_ms > 0 else "-"
238
+ time_str = f"{r.execution_time_ms / 1000:.2f}s" if r.execution_time_ms > 0 else "-"
235
239
  table.add_row(str(i), status, q_type, preview, rows, time_str)
236
240
 
237
- console.print("")
238
241
  console.print(table)
239
- console.print("")
242
+ console.print("[dim]──────────────────────────────────────────────────────────────────────────────[/dim]")
243
+ console.print("[dim]Enter to continue · :run to re-execute last query[/dim]\n")
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: dbagent-cli
3
- Version: 0.8.0
3
+ Version: 0.8.1
4
4
  Summary: Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)
5
5
  Author: Santhosh Gupta
6
6
  Requires-Python: >=3.9
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
4
4
 
5
5
  [project]
6
6
  name = "dbagent-cli"
7
- version = "0.8.0"
7
+ version = "0.8.1"
8
8
  description = "Enterprise AI Database Copilot, Schema Graph Introspector & Autonomous Recovery Agent (CLI)"
9
9
  readme = "README.md"
10
10
  requires-python = ">=3.9"
File without changes
File without changes