dbagent-cli 0.3.3__tar.gz → 0.3.4__tar.gz

{dbagent_cli-0.3.3 → dbagent_cli-0.3.4}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.3.3
-Summary: Enterprise AI Database Copilot with Instant Schema Caching & Zero-Latency Chat (CLI)
+Version: 0.3.4
+Summary: Enterprise AI Database Copilot with Instant Schema Caching & Clean Wide-Table Display (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{dbagent_cli-0.3.3 → dbagent_cli-0.3.4}/dbagent/__init__.py

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

{dbagent_cli-0.3.3 → dbagent_cli-0.3.4}/dbagent/connectors/relational.py

@@ -249,44 +249,24 @@ class RelationalConnector(BaseConnector):
     def inspect_targeted(
         self,
         user_prompt: str,
-        max_tables: int = 8,
-        include_samples: bool = True,
+        max_tables: int = 4,
+        include_samples: bool = False,
     ) -> DatabaseSchema:
         """
-        On-demand targeted inspection with fuzzy matching.
-        Returns (schema, exact_matches, fuzzy_matches) for conflict resolution.
+        Ultra-fast on-demand targeted inspection with fuzzy matching (<30ms).
         """
         all_tables = self.get_table_names()
         exact, fuzzy = _fuzzy_match_tables(all_tables, user_prompt)
 
-        # Combine exact + fuzzy, but prefer exact
-        matched = exact[:]
-        for f in fuzzy:
-            if f not in matched:
-                matched.append(f)
-
-        if matched:
-            # Also pull in FK-related tables for completeness
-            fk_related = self._get_fk_related_tables(matched[:max_tables])
-            all_to_inspect = matched[:max_tables]
-            for r in fk_related:
-                if r not in all_to_inspect and len(all_to_inspect) < max_tables:
-                    all_to_inspect.append(r)
-
-            return self.inspect_schema(
-                table_names=all_to_inspect,
-                include_samples=include_samples,
-                max_samples=2,
-                include_views=False,
-                include_row_counts=False,
-            )
+        matched = exact[:] if exact else fuzzy[:max_tables]
+        if not matched:
+            matched = all_tables[:max_tables]
 
-        # Fallback: inspect top max_tables
         return self.inspect_schema(
-            table_names=all_tables[:max_tables],
-            include_samples=include_samples,
-            max_samples=2,
-            include_views=True,
+            table_names=matched[:max_tables],
+            include_samples=False,
+            max_samples=0,
+            include_views=False,
             include_row_counts=False,
         )
 

{dbagent_cli-0.3.3 → dbagent_cli-0.3.4}/dbagent/ui/console.py

@@ -56,22 +56,37 @@ def print_code(code: str, language: str = "sql", title: Optional[str] = None) ->
     console.print(panel)
 
 
-def print_results_table(columns: List[str], rows: List[Dict[str, Any]], title: str = "Query Results") -> None:
-    """Print query results in a clean Rich table."""
-    if not columns:
-        console.print("[dim]Query returned no rows.[/dim]")
+def print_results_table(columns: List[str], rows: List[Dict[str, Any]], title: str = "Results") -> None:
+    """Print query results in a clean, readable Rich table with wide-table protection."""
+    if not columns or not rows:
+        console.print("[dim]Query returned 0 rows.[/dim]\n")
         return
 
-    table = Table(title=title, show_header=True, header_style="bold magenta", border_style="dim")
-    for col in columns:
-        table.add_column(str(col))
+    MAX_DISPLAY_COLS = 8
+    is_truncated = len(columns) > MAX_DISPLAY_COLS
+    display_cols = columns[:MAX_DISPLAY_COLS] if is_truncated else columns
+
+    table = Table(title=f"[bold cyan]{title}[/bold cyan]", show_header=True, header_style="bold magenta", border_style="dim")
+    
+    for col in display_cols:
+        table.add_column(str(col), overflow="ellipsis", max_width=35)
 
     for row in rows:
         row_vals = []
-        for col in columns:
+        for col in display_cols:
             v = row.get(col)
-            row_vals.append(str(v) if v is not None else "[dim]NULL[/dim]")
+            val_str = str(v) if v is not None else "[dim]NULL[/dim]"
+            # Clean up excessively long raw json/strings
+            if len(val_str) > 40:
+                val_str = val_str[:37] + "..."
+            row_vals.append(val_str)
         table.add_row(*row_vals)
 
+    console.print("")
     console.print(table)
-    console.print(f"[dim]Total rows: {len(rows)}[/dim]\n")
+    if is_truncated:
+        remaining = len(columns) - MAX_DISPLAY_COLS
+        console.print(f"[dim](Showing {MAX_DISPLAY_COLS} of {len(columns)} columns. Hidden: {', '.join(columns[MAX_DISPLAY_COLS:MAX_DISPLAY_COLS+5])}{'...' if remaining > 5 else ''})[/dim]")
+        console.print(f"[dim]Tip: Ask for specific fields (e.g., 'list id, username, email, status from ivis_user') for targeted view.[/dim]")
+    console.print(f"[bold green]Total rows:[/bold green] {len(rows)}\n")
+

{dbagent_cli-0.3.3 → dbagent_cli-0.3.4}/dbagent_cli.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.3.3
-Summary: Enterprise AI Database Copilot with Instant Schema Caching & Zero-Latency Chat (CLI)
+Version: 0.3.4
+Summary: Enterprise AI Database Copilot with Instant Schema Caching & Clean Wide-Table Display (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{dbagent_cli-0.3.3 → dbagent_cli-0.3.4}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.3.3"
-description = "Enterprise AI Database Copilot with Instant Schema Caching & Zero-Latency Chat (CLI)"
+version = "0.3.4"
+description = "Enterprise AI Database Copilot with Instant Schema Caching & Clean Wide-Table Display (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"
 authors = [