dbagent-cli 0.8.2__tar.gz → 0.8.3__tar.gz

{dbagent_cli-0.8.2 → dbagent_cli-0.8.3}/PKG-INFO

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

{dbagent_cli-0.8.2 → dbagent_cli-0.8.3}/dbagent/__init__.py

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

{dbagent_cli-0.8.2 → dbagent_cli-0.8.3}/dbagent/cli.py

@@ -523,28 +523,6 @@ def chat_command(
         conversation_context=conversation_ctx,
     )
 
-    # Launch Full-Screen TUI (Fixed top header, middle scroller, static bottom chat bar)
-    if not inline and not os.getenv("DBAGENT_INLINE_CHAT"):
-        try:
-            from dbagent.ui.fullscreen_chat import FullScreenChatApp
-            app_tui = FullScreenChatApp(
-                connector=connector,
-                llm=llm,
-                pipeline=pipeline,
-                db_name=db_name,
-                dialect=dialect,
-                table_count=len(table_names),
-                config_mgr=config_mgr,
-                cache_mgr=cache_mgr,
-                model=model,
-                mode=current_mode,
-            )
-            app_tui.run()
-            connector.close()
-            return
-        except Exception:
-            pass
-
     def _get_bottom_toolbar():
         curr_model = model or getattr(llm, "default_model", "default")
         return HTML(

{dbagent_cli-0.8.2 → dbagent_cli-0.8.3}/dbagent_cli.egg-info/PKG-INFO

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

{dbagent_cli-0.8.2 → dbagent_cli-0.8.3}/dbagent_cli.egg-info/SOURCES.txt

@@ -54,7 +54,6 @@ dbagent/sql/optimizer.py
 dbagent/sql/validator.py
 dbagent/ui/chat_screen.py
 dbagent/ui/console.py
-dbagent/ui/fullscreen_chat.py
 dbagent/ui/grid_window.py
 dbagent/ui/viewer.py
 dbagent/ui/visualizer.py

{dbagent_cli-0.8.2 → dbagent_cli-0.8.3}/pyproject.toml

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

dbagent_cli-0.8.2/dbagent/ui/fullscreen_chat.py

@@ -1,392 +0,0 @@
-"""
-Full-Screen TUI for DB-Agent.
-Features:
-- Static pinned top Header with live connection and AI status.
-- Middle scrollable conversation history with Rich-rendered tables, SQL syntax boxes, and markdown.
-- Static pinned bottom Chat Bar with prompt input and shortcut footer.
-"""
-
-import io
-import os
-import sys
-import time
-from typing import Optional, List, Dict, Any, Callable
-from prompt_toolkit.application import Application
-from prompt_toolkit.layout.containers import HSplit, VSplit, Window
-from prompt_toolkit.layout.controls import FormattedTextControl
-from prompt_toolkit.layout.layout import Layout
-from prompt_toolkit.widgets import TextArea, Frame
-from prompt_toolkit.key_binding import KeyBindings
-from prompt_toolkit.formatted_text import HTML, ANSI
-from prompt_toolkit.styles import Style
-from rich.console import Console
-from rich.panel import Panel
-from rich.table import Table
-from rich.syntax import Syntax
-from rich.markdown import Markdown
-
-from dbagent import __version__
-from dbagent.agent.pipeline import QueryPipeline, PipelineResult
-from dbagent.connectors.base import BaseConnector
-from dbagent.llm.base import BaseLLMProvider
-from dbagent.llm.factory import get_llm_provider
-from dbagent.security.modes import OperationMode
-from dbagent.config import ConfigManager
-from dbagent.schema.cache import SchemaCacheManager
-from dbagent.schema.formatter import SchemaFormatter
-from dbagent.agent.recommender import QueryRecommender
-from dbagent.agent.doctor import DatabaseDoctor
-from dbagent.ui.chat_screen import MASCOT_ART, render_session_history
-
-
-class FullScreenChatApp:
-    """
-    Full-Screen interactive TUI with fixed Top Header, middle Scrollable History,
-    and fixed Bottom Chat Bar.
-    """
-
-    def __init__(
-        self,
-        connector: BaseConnector,
-        llm: BaseLLMProvider,
-        pipeline: QueryPipeline,
-        db_name: str,
-        dialect: str,
-        table_count: int,
-        config_mgr: ConfigManager,
-        cache_mgr: SchemaCacheManager,
-        model: Optional[str] = None,
-        mode: OperationMode = OperationMode.CONFIRM,
-    ):
-        self.connector = connector
-        self.llm = llm
-        self.pipeline = pipeline
-        self.db_name = db_name
-        self.dialect = dialect
-        self.table_count = table_count
-        self.config_mgr = config_mgr
-        self.cache_mgr = cache_mgr
-        self.model = model or getattr(llm, "default_model", "default")
-        self.mode = mode
-        self.query_history: List[PipelineResult] = []
-        self.last_result: Optional[PipelineResult] = None
-        self.auto_execute = True
-
-        # Rich console for buffer rendering
-        self.rich_io = io.StringIO()
-        self.rich_console = Console(
-            file=self.rich_io,
-            force_terminal=True,
-            color_system="truecolor",
-            width=110,
-        )
-
-        # Build UI layout components
-        self._setup_ui()
-
-    def _render_rich_to_ansi(self, renderable: Any) -> str:
-        """Helper to render any Rich renderable (Table, Panel, etc.) to ANSI string."""
-        self.rich_io.seek(0)
-        self.rich_io.truncate(0)
-        self.rich_console.print(renderable)
-        return self.rich_io.getvalue()
-
-    def _setup_ui(self) -> None:
-        """Create the prompt_toolkit UI containers and keybindings."""
-        kb = KeyBindings()
-
-        @kb.add("c-c")
-        @kb.add("c-q")
-        def _exit(event):
-            event.app.exit()
-
-        @kb.add("pageup")
-        def _pageup(event):
-            self.history_area.buffer.cursor_up(count=10)
-
-        @kb.add("pagedown")
-        def _pagedown(event):
-            self.history_area.buffer.cursor_down(count=10)
-
-        @kb.add("enter")
-        def _submit(event):
-            text = self.input_area.text.strip()
-            if not text:
-                return
-            self.input_area.text = ""
-            self._handle_input(text)
-
-        # 1. Top Fixed Header Window
-        def get_header_text():
-            return HTML(
-                f"<style bg='#0f172a' fg='#00ffff'><b> ⚡ DB-AGENT v{__version__} </b></style> "
-                f"<style bg='#0f172a' fg='#00ff88'><b>🟢 Connected:</b> {self.db_name} <i>({self.dialect} • {self.table_count} tables)</i></style>  │  "
-                f"<style bg='#0f172a' fg='#00ffff'><b>🤖 AI:</b> {self.llm.name} ({self.model})</style>  │  "
-                f"<style bg='#0f172a' fg='#fbbf24'><b>🛡️ {self.mode.value}</b></style>"
-            )
-
-        self.header_window = Window(
-            content=FormattedTextControl(get_header_text),
-            height=1,
-            style="bg:#0f172a",
-        )
-
-        # 2. Middle Scrollable Conversation History
-        self.history_area = TextArea(
-            text="",
-            read_only=True,
-            scrollbar=True,
-            wrap_lines=True,
-            style="bg:#020617 fg:#f8fafc",
-        )
-
-        # 3. Bottom Static Input Bar
-        self.input_area = TextArea(
-            height=1,
-            prompt="❯ ",
-            style="bg:#0f172a fg:#00ffff bold",
-            multiline=False,
-            wrap_lines=False,
-        )
-
-        self.input_frame = Frame(
-            body=self.input_area,
-            title="💬 Ask plain English or paste SQL",
-            style="bg:#0f172a fg:#00ffff",
-        )
-
-        # 4. Bottom Static Toolbar
-        def get_footer_text():
-            return HTML(
-                "<style bg='#0f172a' fg='#94a3b8'> <b>Enter</b> Send  │  "
-                "<b>PgUp/PgDn</b> Scroll History  │  "
-                "<b>:help</b> Commands  │  "
-                "<b>:models</b> Switch Model  │  "
-                "<b>:doctor</b> Health  │  "
-                "<b>:clear</b> Clear  │  "
-                "<b>:exit</b> Quit </style>"
-            )
-
-        self.footer_window = Window(
-            content=FormattedTextControl(get_footer_text),
-            height=1,
-            style="bg:#0f172a",
-        )
-
-        # Assemble FullScreen Layout
-        self.root_container = HSplit([
-            self.header_window,
-            self.history_area,
-            self.input_frame,
-            self.footer_window,
-        ])
-
-        self.custom_style = Style.from_dict({
-            "frame.label": "#00ffff bold",
-            "frame.border": "#00ffff",
-        })
-
-        self.app = Application(
-            layout=Layout(self.root_container, focused_element=self.input_area),
-            key_bindings=kb,
-            style=self.custom_style,
-            full_screen=True,
-            mouse_support=True,
-        )
-
-        # Print initial welcome banner into history
-        self._print_welcome()
-
-    def _append_text(self, text: str) -> None:
-        """Append text to the scrollable middle conversation buffer and auto-scroll to end."""
-        current = self.history_area.text
-        if current:
-            self.history_area.text = current + "\n" + text
-        else:
-            self.history_area.text = text
-        self.history_area.buffer.cursor_position = len(self.history_area.text)
-
-    def _print_welcome(self) -> None:
-        """Render initial split-card welcome screen into history area."""
-        left_grid = Table.grid(padding=(0, 1))
-        left_grid.add_column(justify="center", min_width=28)
-        left_grid.add_row("[bold white]Welcome to DB-Agent![/bold white]")
-        left_grid.add_row(f"[bold cyan]{MASCOT_ART.strip()}[/bold cyan]")
-        left_grid.add_row(f"[bold cyan]{self.model}[/bold cyan] [dim]({self.llm.name})[/dim]")
-        left_grid.add_row(f"[bold green]🟢 Connected:[/bold green] [bold white]{self.db_name}[/bold white] [dim]({self.dialect} • {self.table_count} tables)[/dim]")
-
-        right_grid = Table.grid(padding=(0, 1))
-        right_grid.add_column()
-        right_grid.add_row("[bold cyan]💡 Quick Database Tips[/bold cyan]")
-        right_grid.add_row("[dim]Ask in plain English or paste direct SQL for 0ms execution[/dim]")
-        right_grid.add_row("[dim]Type :help to inspect tables (:tables), relations (:fk), or diagnostics (:doctor)[/dim]")
-        right_grid.add_row("[dim]────────────────────────────────────────────────────────────────────────[/dim]")
-        right_grid.add_row(f"[bold green]🚀 Features in v{__version__}[/bold green]")
-        right_grid.add_row("[dim]Universal AI: Claude 3.7 Sonnet, OpenAI GPT-4o, DeepSeek R1, Mistral, Gemini[/dim]")
-        right_grid.add_row("[dim]Zero-Latency Direct SQL Fast-Path with autonomous AI auto-repair on error[/dim]")
-        right_grid.add_row("[dim]Multi-turn conversational session memory graph & persistent connection pooling[/dim]")
-
-        layout_table = Table.grid(expand=True, padding=(0, 2))
-        layout_table.add_column(ratio=2)
-        layout_table.add_column(ratio=3)
-        layout_table.add_row(left_grid, right_grid)
-
-        panel = Panel(
-            layout_table,
-            title=f"[bold cyan]⚡ DB-AGENT [/bold cyan][bold green]v{__version__}[/bold green]",
-            title_align="left",
-            border_style="cyan",
-            padding=(1, 1),
-        )
-
-        ansi_banner = self._render_rich_to_ansi(panel)
-        self._append_text(ansi_banner)
-
-        # Add sample recommendation prompts
-        try:
-            cached = self.cache_mgr.load_cached_schema(self.connector.connection_url)
-            if cached:
-                recs = QueryRecommender.generate_recommendations(cached, max_suggestions=3)
-                if recs:
-                    rec_lines = ["[dim italic]💡 Try asking:[/dim italic]"]
-                    for r in recs:
-                        rec_lines.append(f"  [dim]→[/dim] [cyan]{r}[/cyan]")
-                    self._append_text(self._render_rich_to_ansi("\n".join(rec_lines)))
-        except Exception:
-            pass
-
-    def _handle_input(self, user_input: str) -> None:
-        """Handle incoming prompt or REPL command."""
-        # 1. REPL Commands
-        if user_input.startswith(":"):
-            cmd_parts = user_input.split(maxsplit=1)
-            cmd_name = cmd_parts[0].lower()
-            cmd_arg = cmd_parts[1].strip() if len(cmd_parts) > 1 else ""
-
-            if cmd_name in [":exit", ":quit", ":q"]:
-                self.app.exit()
-                return
-
-            if cmd_name == ":clear":
-                self.history_area.text = ""
-                self._print_welcome()
-                return
-
-            if cmd_name in [":help", ":h", ":?"]:
-                cmd_table = Table(box=None, padding=(0, 2), show_header=False)
-                cmd_table.add_column("Command", style="bold cyan", width=22)
-                cmd_table.add_column("Description", style="dim")
-                cmd_table.add_row(":models, :model", "Switch active AI model (Claude 3.7, GPT-4o, DeepSeek, Gemini...)")
-                cmd_table.add_row(":provider <name>", "Switch AI provider (claude, openai, deepseek, mistral, gemini)")
-                cmd_table.add_row(":tables, :dt", "List all tables & schemas in this database")
-                cmd_table.add_row(":table <name>", "Inspect columns, primary keys, data types & sample data")
-                cmd_table.add_row(":fk <name>", "View foreign key relationships & dependency graph")
-                cmd_table.add_row(":path <t1> <t2>", "Discover shortest SQL join path between two tables")
-                cmd_table.add_row(":doctor", "Run database health, index, and query performance diagnostics")
-                cmd_table.add_row(":mode <name>", "Switch safety mode (SAFE, CONFIRM, ADMIN, READ_ONLY)")
-                cmd_table.add_row(":history", "View session query history, execution timing & row counts")
-                cmd_table.add_row(":explain", "Explain last generated SQL query with execution plan")
-                cmd_table.add_row(":run", "Re-execute the last generated SQL query")
-                cmd_table.add_row(":clear", "Clear conversation memory & redraw screen")
-                cmd_table.add_row(":exit, :q", "Exit DB-Agent session")
-                ansi_help = self._render_rich_to_ansi(Panel(cmd_table, title="[bold cyan]⌨️ Interactive Commands[/bold cyan]", border_style="cyan"))
-                self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n" + ansi_help)
-                return
-
-            if cmd_name in [":models", ":model"]:
-                avail = self.llm.list_models() if hasattr(self.llm, "list_models") else []
-                if not cmd_arg or cmd_name == ":models":
-                    lines = [f"[bold cyan]Available Models for {self.llm.name}:[/bold cyan]\n"]
-                    for idx, m in enumerate(avail, 1):
-                        marker = "❯" if m == self.model else " "
-                        active = " [bold green](Active)[/bold green]" if m == self.model else ""
-                        lines.append(f" {marker} [bold cyan]{idx}.[/bold cyan] [bold white]{m}[/bold white]{active}")
-                    lines.append("\n[dim]Switch with: :model <number|name> (e.g. :model 1 or :model 3.7)[/dim]")
-                    self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n" + self._render_rich_to_ansi("\n".join(lines)))
-                    return
-                else:
-                    # Switch model
-                    target = cmd_arg
-                    if target.isdigit() and avail:
-                        idx = int(target) - 1
-                        if 0 <= idx < len(avail):
-                            target = avail[idx]
-                    elif target in ["3.7", "37"]:
-                        target = "claude-3-7-sonnet-20250219" if self.llm.name == "claude" else "gemini-3.7-flash"
-                    elif target in ["3.6", "36"]:
-                        target = "gemini-3.6-flash"
-                    self.model = target
-                    self.pipeline.model = target
-                    self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n[bold green]✓ Switched model to {self.model}[/bold green]\n")
-                    return
-
-            if cmd_name == ":doctor":
-                doc = DatabaseDoctor(self.connector, self.llm)
-                diag = doc.diagnose()
-                self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n" + self._render_rich_to_ansi(diag.to_markdown()))
-                return
-
-            if cmd_name == ":history":
-                self._append_text(f"\n[cyan]❯ {user_input}[/cyan]")
-                render_session_history(self.query_history)
-                return
-
-            # Other commands
-            self._append_text(f"\n[cyan]❯ {user_input}[/cyan]\n[dim]Command executed.[/dim]")
-            return
-
-        # 2. Natural Language or Direct SQL Query
-        self._append_text(f"\n[bold blue]💬 You:[/bold blue] [bold white]{user_input}[/bold white]")
-
-        # Run via pipeline
-        t0 = time.time()
-        self.pipeline.model = self.model
-        self.pipeline.mode = self.mode
-        result = self.pipeline.run(user_prompt=user_input)
-
-        self.last_result = result
-        self.query_history.append(result)
-
-        timing_str = f"⚡ {result.execution_time_ms / 1000:.2f}s" if result.execution_time_ms > 0 else ""
-
-        # Chat response
-        if result.is_chat and result.message:
-            self._append_text(f"\n[bold cyan]🤖 DB-Agent ({self.model})[/bold cyan] [dim]{timing_str}[/dim]\n" + result.message)
-            return
-
-        # Error
-        if result.error:
-            self._append_text(f"\n[bold red]❌ Error ({timing_str}):[/bold red] {result.friendly_error or result.error}")
-            return
-
-        # Executed SQL results
-        if result.was_executed:
-            blocks = []
-            if result.is_direct_sql and result.retries == 0:
-                header_line = f"[bold cyan]🤖 DB-Agent[/bold cyan] [dim](Direct SQL • 0 tokens · {timing_str})[/dim]"
-            else:
-                header_line = f"[bold cyan]🤖 DB-Agent ({self.model})[/bold cyan] [dim]{timing_str}[/dim]"
-            blocks.append(header_line)
-
-            if result.sql:
-                syntax = Syntax(result.sql.strip(), "sql", theme="monokai", line_numbers=False)
-                blocks.append(self._render_rich_to_ansi(Panel(syntax, title="[dim]SQL Statement[/dim]", border_style="dim")))
-
-            if result.retries > 0:
-                blocks.append(f"[bold yellow]✓ Direct SQL error was automatically fixed by AI ({result.retries} retry)[/bold yellow]")
-
-            if result.has_results:
-                t = Table(box=None, padding=(0, 2), show_header=True, header_style="bold cyan")
-                for col in result.columns:
-                    t.add_column(str(col), style="white")
-                for row in result.rows[:50]:
-                    t.add_row(*[str(row.get(col, "")) for col in result.columns])
-                blocks.append(self._render_rich_to_ansi(t))
-                if len(result.rows) > 50:
-                    blocks.append(f"[dim]... and {len(result.rows) - 50} more rows[/dim]")
-
-            self._append_text("\n".join(blocks))
-
-    def run(self) -> None:
-        """Start the full-screen interactive TUI."""
-        self.app.run()