dbagent-cli 0.9.8__tar.gz → 0.9.9__tar.gz

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/PKG-INFO

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

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/dbagent/__init__.py

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

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/dbagent/cli.py

@@ -530,6 +530,8 @@ def ask_command(
         choice_callback=_choice_callback,
     )
 
+    from dbagent.ui.live_streamer import LiveMarkdownStreamer
+    streamer = LiveMarkdownStreamer(console)
     spinner = DynamicSpinner.for_prompt(prompt, model_name=model or getattr(llm, "default_model", None), connector=connector)
     stream_started = [False]
 
@@ -541,15 +543,14 @@ def ask_command(
             active_model = model or getattr(llm, "default_model", None)
             model_display = f" ({active_model})" if active_model else ""
             console.print(f"\n[bold cyan]● DB-Agent{model_display}[/bold cyan] [dim]{timing_str}[/dim]")
-        try:
-            sys.stdout.write(chunk)
-            sys.stdout.flush()
-        except Exception:
-            pass
+        streamer.feed(chunk)
 
     with spinner:
         result = pipeline.run(user_prompt=prompt, force=force, bypass_cache=no_cache, stream_callback=_query_stream_callback)
 
+    if stream_started[0]:
+        streamer.finalize()
+
     _display_pipeline_result(result, show_sql=True)
     connector.close()
 
@@ -1119,6 +1120,8 @@ def chat_command(
             pipeline.mode = current_mode
             pipeline.auto_execute = auto_execute
 
+            from dbagent.ui.live_streamer import LiveMarkdownStreamer
+            streamer = LiveMarkdownStreamer(console)
             spinner = DynamicSpinner.for_prompt(user_input, model_name=model or getattr(llm, "default_model", None), connector=connector)
             stream_started = [False]
 
@@ -1130,15 +1133,14 @@ def chat_command(
                     active_model = model or getattr(llm, "default_model", None)
                     model_display = f" ({active_model})" if active_model else ""
                     console.print(f"\n[bold cyan]● DB-Agent{model_display}[/bold cyan] [dim]{timing_str}[/dim]")
-                try:
-                    sys.stdout.write(chunk)
-                    sys.stdout.flush()
-                except Exception:
-                    pass
+                streamer.feed(chunk)
 
             with spinner:
                 result = pipeline.run(user_prompt=user_input, stream_callback=_live_stream_callback)
 
+            if stream_started[0]:
+                streamer.finalize()
+
             render_assistant_turn(
                 result,
                 show_sql=True,

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/dbagent/schema/ground_truth.py

@@ -85,17 +85,19 @@ class SchemaGroundTruthValidator:
                         if fk_desc not in result.verified_fks:
                             result.verified_fks.append(fk_desc)
 
-        # 2. Extract and validate SQL queries against physical schema
-        sql_blocks = re.findall(r'```(?:sql)?\s*([\s\S]*?)\s*```', response_text, re.IGNORECASE)
+        # 2. Extract and validate only actual SQL queries against physical schema
+        sql_blocks = re.findall(r'```(?:sql)\s*([\s\S]*?)\s*```', response_text, re.IGNORECASE)
         result.sql_queries_checked = len(sql_blocks)
         sql_referenced_tables = set()
+        sql_ignore_keywords = {"via", "lateral", "on", "using", "where", "select", "table", "as", "and", "or", "values"}
 
         for block in sql_blocks:
             # Extract FROM and JOIN table references
             matches = re.findall(r'\b(?:FROM|JOIN)\s+([a-zA-Z0-9_]+)', block, re.IGNORECASE)
             for m in matches:
                 m_clean = m.strip().lower()
-                sql_referenced_tables.add(m_clean)
+                if m_clean not in sql_ignore_keywords:
+                    sql_referenced_tables.add(m_clean)
 
         for sql_tbl in sql_referenced_tables:
             if sql_tbl in schema_table_map:

dbagent_cli-0.9.9/dbagent/ui/live_streamer.py

@@ -0,0 +1,117 @@
+"""
+Real-Time Line-Buffered Rich Markdown Streamer for DB-Agent CLI.
+Renders markdown headers, formatted diagram panels, syntax-highlighted SQL blocks,
+and sleek bulleted lists dynamically as tokens stream in from LLM providers.
+"""
+
+import re
+import sys
+from typing import List, Optional
+from rich.console import Console
+from rich.panel import Panel
+from rich.syntax import Syntax
+from rich.rule import Rule
+
+
+class LiveMarkdownStreamer:
+    """
+    Line-buffered dynamic terminal renderer for streaming LLM responses.
+    Renders syntax-highlighted code blocks, bordered ASCII diagrams,
+    styled markdown headers, and clean bullet lists live as tokens arrive.
+    """
+
+    def __init__(self, console: Optional[Console] = None):
+        self.console = console or Console()
+        self.buffer = ""
+        self.in_code_block = False
+        self.code_lang = ""
+        self.code_lines: List[str] = []
+        self._has_started = False
+
+    def feed(self, chunk: str) -> None:
+        """Feed a token chunk from the LLM stream."""
+        self._has_started = True
+        self.buffer += chunk
+        while "\n" in self.buffer:
+            line, self.buffer = self.buffer.split("\n", 1)
+            self._process_line(line)
+
+    def _process_line(self, line: str) -> None:
+        line_strip = line.strip()
+
+        # Code block fence (```sql, ```text, ```)
+        if line_strip.startswith("```"):
+            if self.in_code_block:
+                # Close code block
+                self.in_code_block = False
+                code_text = "\n".join(self.code_lines)
+                lang = self.code_lang or "sql"
+                if lang in ["sql", "postgres", "postgresql", "mysql", "sqlite", "tsql"]:
+                    syntax = Syntax(code_text, "sql", theme="monokai", line_numbers=False)
+                    self.console.print(Panel(syntax, title="[dim]SQL Statement[/dim]", title_align="left", border_style="dim", padding=(0, 1)))
+                else:
+                    syntax = Syntax(code_text, "text", theme="monokai", line_numbers=False)
+                    self.console.print(Panel(syntax, title="[dim]Architecture Flow Diagram[/dim]", title_align="left", border_style="cyan", padding=(0, 1)))
+                self.code_lines = []
+                self.code_lang = ""
+            else:
+                # Open code block
+                self.in_code_block = True
+                self.code_lang = line_strip[3:].strip().lower()
+                self.code_lines = []
+            return
+
+        if self.in_code_block:
+            self.code_lines.append(line)
+            return
+
+        # Markdown Section Headers (#, ##, ###)
+        if line_strip.startswith("#"):
+            header_text = line_strip.lstrip("#").strip()
+            self.console.print("")
+            self.console.print(f"[bold cyan]{header_text}[/bold cyan]")
+            return
+
+        # Markdown Horizontal Rules (---, ***, ___)
+        if line_strip in ["---", "***", "___"]:
+            self.console.print(Rule(style="dim cyan"))
+            return
+
+        # Markdown Bullet Points (- , * , • )
+        if line_strip.startswith(("- ", "* ", "• ")):
+            content = line_strip[2:].strip()
+            styled = re.sub(r'\*\*(.*?)\*\*', r'[bold cyan]\1[/bold cyan]', content)
+            styled = re.sub(r'`(.*?)`', r'[bold white]\1[/bold white]', styled)
+            self.console.print(f"  [dim]•[/dim] {styled}")
+            return
+
+        # Numbered List items (1. , 2. )
+        num_match = re.match(r'^(\d+[\.\)])\s+(.*)', line_strip)
+        if num_match:
+            num_prefix, content = num_match.groups()
+            styled = re.sub(r'\*\*(.*?)\*\*', r'[bold cyan]\1[/bold cyan]', content)
+            styled = re.sub(r'`(.*?)`', r'[bold white]\1[/bold white]', styled)
+            self.console.print(f" [bold yellow]{num_prefix}[/bold yellow] {styled}")
+            return
+
+        # Blank Line
+        if not line_strip:
+            self.console.print("")
+            return
+
+        # Regular Markdown paragraph text with bold and inline code styling
+        styled_line = re.sub(r'\*\*(.*?)\*\*', r'[bold cyan]\1[/bold cyan]', line)
+        styled_line = re.sub(r'`(.*?)`', r'[bold white]\1[/bold white]', styled_line)
+        self.console.print(styled_line)
+
+    def finalize(self) -> None:
+        """Flush remaining buffer and close any dangling open blocks."""
+        if self.buffer.strip():
+            self._process_line(self.buffer)
+        if self.in_code_block and self.code_lines:
+            code_text = "\n".join(self.code_lines)
+            syntax = Syntax(code_text, self.code_lang or "sql", theme="monokai", line_numbers=False)
+            self.console.print(Panel(syntax, border_style="dim", padding=(0, 1)))
+        self.buffer = ""
+        self.code_lines = []
+        self.in_code_block = False

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/dbagent_cli.egg-info/PKG-INFO

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

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/dbagent_cli.egg-info/SOURCES.txt

@@ -57,6 +57,7 @@ dbagent/sql/validator.py
 dbagent/ui/chat_screen.py
 dbagent/ui/console.py
 dbagent/ui/grid_window.py
+dbagent/ui/live_streamer.py
 dbagent/ui/viewer.py
 dbagent/ui/visualizer.py
 dbagent_cli.egg-info/PKG-INFO

{dbagent_cli-0.9.8 → dbagent_cli-0.9.9}/pyproject.toml

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