dbagent-cli 0.9.4__tar.gz → 0.9.6__tar.gz

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/PKG-INFO

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

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/dbagent/__init__.py

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

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/dbagent/agent/generator.py

@@ -166,11 +166,27 @@ class ScriptGenerator:
             "related tables", "related tebales", "table mappings", "related tables mappings",
             "related tebales mappings", "not correct", "not right", "check all the tables",
             "mappings also", "find all mappings", "check links", "check mappings",
+            "is fk_", "are fk_", "is pk_", "are pk_", "difference between", "compare",
+            "tell me", "check it and tell me", "check and tell me", "same or different",
+            "flow diagram", "flow chart", "give the flow diagram", "give the diagram",
+            "show flow diagram", "give diagram", "show diagram", "architecture diagram",
         ]
         for kw in analysis_keywords:
             if kw in prompt_lower:
                 return "analysis"
 
+        # Regex for "how <X> and <Y> were/are linked/connected/related"
+        if re.search(r'\bhow\b.*\b(?:linked|connected|related|mapped|joined)\b', prompt_lower):
+            return "analysis"
+
+        # Regex for diagram / blueprint requests
+        if re.search(r'\b(?:flow\s+diagram|diagram|flow\s+chart|blueprint|er\s+diagram)\b', prompt_lower):
+            return "analysis"
+
+        # Comparative analysis questions (e.g. "is fk_site_group_id and fk_group_id same", "are table_a and table_b related")
+        if re.search(r'\b(?:is|are)\b.*\b(?:same|different|difference|related|equal|linked)\b', prompt_lower):
+            return "analysis"
+
         # Contextual clarification for analysis (e.g. "i mean ivis_scope", "i mean table_x")
         if re.search(r'\bi mean\b', prompt_lower) and not re.search(r'\b(insert|update|delete|drop)\b', prompt_lower):
             return "analysis"

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/dbagent/agent/pipeline.py

@@ -241,7 +241,7 @@ class QueryPipeline:
                 if not exact and not fuzzy:
                     inherited = self.conversation.get_active_tables()
                     if inherited:
-                        exact = inherited[:4]
+                        exact = inherited[:8]
 
                 sub_schema, join_steps = self._get_fast_schema_context(user_prompt, exact, fuzzy)
                 result.schema = sub_schema
@@ -594,10 +594,16 @@ class QueryPipeline:
         )
 
         if result.success:
+            sql_tables = set()
+            if result.sql:
+                import re as _re
+                for m in _re.findall(r'\b(?:FROM|JOIN)\s+([a-zA-Z0-9_]+)', result.sql, _re.IGNORECASE):
+                    sql_tables.add(m.lower())
+            turn_tables = list(set((exact or []) + (fuzzy or []) + list(sql_tables)))
             self.conversation.add_turn(
                 user_prompt=user_prompt,
                 sql=result.sql,
-                tables=exact or fuzzy,
+                tables=turn_tables,
                 is_chat=False,
             )
 

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/dbagent/cli.py

@@ -98,10 +98,13 @@ audit_logger = AuditLogger()
 # Dynamic Animated Spinner with Rotating Messages
 # ---------------------------------------------------------------------------
 
+_active_spinner: Optional["DynamicSpinner"] = None
+
+
 class DynamicSpinner:
     """
     Context manager that shows an animated spinner with rotating contextual messages.
-    Messages cycle every ~1.5s so the user always sees progress.
+    Messages cycle every ~1.2s so the user always sees progress.
     Displays elapsed time suffix like (2.3s).
     """
 
@@ -193,7 +196,7 @@ class DynamicSpinner:
         ]
         return cls(messages=messages, interval=1.2)
 
-    def __init__(self, messages: Optional[List[str]] = None, interval: float = 1.5):
+    def __init__(self, messages: Optional[List[str]] = None, interval: float = 1.2):
         self._messages = messages or self.DEFAULT_MESSAGES
         self._interval = interval
         self._stop_event = threading.Event()
@@ -202,6 +205,8 @@ class DynamicSpinner:
         self._status = None
 
     def __enter__(self):
+        global _active_spinner
+        _active_spinner = self
         self._start_time = time.time()
         self._stop_event.clear()
         self._status = console.status(f"[bold cyan]{self._messages[0]}[/bold cyan]")
@@ -211,11 +216,36 @@ class DynamicSpinner:
         return self
 
     def __exit__(self, *args):
+        global _active_spinner
+        _active_spinner = None
         self._stop_event.set()
         if self._thread:
-            self._thread.join(timeout=2)
+            self._thread.join(timeout=1)
+        if self._status:
+            try:
+                self._status.__exit__(*args)
+            except Exception:
+                pass
+
+    def pause(self):
+        """Temporarily pause the spinner so interactive prompts don't clash."""
+        self._stop_event.set()
         if self._status:
-            self._status.__exit__(*args)
+            try:
+                self._status.stop()
+            except Exception:
+                pass
+
+    def resume(self):
+        """Resume the spinner after interactive prompt finishes."""
+        self._stop_event.clear()
+        if self._status:
+            try:
+                self._status.start()
+            except Exception:
+                pass
+        self._thread = threading.Thread(target=self._rotate, daemon=True)
+        self._thread.start()
 
     @property
     def elapsed_ms(self) -> float:
@@ -265,28 +295,42 @@ def resolve_db_url(db_arg: Optional[str] = None) -> str:
 
 def _confirm_callback(sql: str, warnings: List[str]) -> bool:
     """Prompt user to confirm a write or DDL operation."""
-    console.print("")
-    print_code(sql, "sql", "Action Requires Confirmation")
-    for w in warnings:
-        print_warning(w)
-    return Confirm.ask("Execute this statement on the database?", default=False)
+    global _active_spinner
+    if _active_spinner:
+        _active_spinner.pause()
+    try:
+        console.print("")
+        print_code(sql, "sql", "Action Requires Confirmation")
+        for w in warnings:
+            print_warning(w)
+        return Confirm.ask("Execute this statement on the database?", default=False)
+    finally:
+        if _active_spinner:
+            _active_spinner.resume()
 
 
 def _choice_callback(prompt_text: str, choices: List[str]) -> Optional[str]:
     """Prompt user to disambiguate table choices."""
-    console.print(f"\n[bold yellow]?[/bold yellow] {prompt_text}")
-    for i, c in enumerate(choices, 1):
-        console.print(f"  [{i}] {c}")
-    choice = Prompt.ask("Select number or table name", default="1")
+    global _active_spinner
+    if _active_spinner:
+        _active_spinner.pause()
     try:
-        idx = int(choice) - 1
-        if 0 <= idx < len(choices):
-            return choices[idx]
-    except ValueError:
-        pass
-    if choice in choices:
-        return choice
-    return choices[0]
+        console.print(f"\n[bold yellow]?[/bold yellow] {prompt_text}")
+        for i, c in enumerate(choices, 1):
+            console.print(f"  [{i}] {c}")
+        choice = Prompt.ask("Select number or table name", default="1")
+        try:
+            idx = int(choice) - 1
+            if 0 <= idx < len(choices):
+                return choices[idx]
+        except ValueError:
+            pass
+        if choice in choices:
+            return choice
+        return choices[0]
+    finally:
+        if _active_spinner:
+            _active_spinner.resume()
 
 
 def _display_pipeline_result(result: PipelineResult, show_sql: bool = True) -> None:

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/dbagent/connectors/base.py

@@ -53,7 +53,32 @@ class BaseConnector(ABC):
         exact = []
         fuzzy = []
 
-        tokens = set(re.findall(r'\b[a-zA-Z0-9_]+\b', prompt_lower))
+        stop_words = {
+            "check", "tell", "same", "show", "list", "find", "get", "is", "are", "does", "will", "can", "what",
+            "which", "how", "why", "where", "with", "and", "from", "that", "this", "these", "those", "have",
+            "has", "had", "not", "all", "any", "few", "some", "many", "each", "every", "table", "tables",
+            "column", "columns", "field", "fields", "row", "rows", "record", "records", "data", "info",
+            "information", "name", "names", "type", "types", "key", "keys", "id", "ids", "status", "true",
+            "false", "null", "seriously", "please", "mean", "like", "about", "between", "among", "into",
+            "onto", "over", "under", "before", "after", "during", "since", "until", "across", "along",
+            "throughout", "outside", "inside", "above", "below", "behind", "beyond", "plus", "except",
+            "but", "against", "towards", "upon", "within", "without", "give", "also", "explain", "look",
+            "know", "make", "take", "help", "view", "count", "done", "good", "nice", "okay", "fine",
+            "flow", "diagram", "chart", "graph", "link", "links", "structure", "design", "model", "blueprint",
+            "mapping", "mappings", "overview", "query", "code", "script", "report", "summary", "relation",
+            "relations", "relationship", "relationships", "architecture"
+        }
+
+        raw_tokens = set(re.findall(r'\b[a-zA-Z0-9_]+\b', prompt_lower))
+        tokens = set(raw_tokens)
+
+        # Extract entity stems from foreign key column tokens (e.g. fk_site_group_id -> site_group, fk_group_id -> group)
+        for tok in raw_tokens:
+            if tok.startswith("fk_") or tok.startswith("pk_"):
+                entity = re.sub(r'^(?:fk_|pk_)', '', tok)
+                entity = re.sub(r'_(?:id|code|key|no|num)$', '', entity)
+                if entity and len(entity) >= 3 and entity not in stop_words:
+                    tokens.add(entity)
 
         for t in all_tables:
             t_lower = t.lower()
@@ -64,7 +89,7 @@ class BaseConnector(ABC):
 
             # 2. Match without common prefixes (e.g. "ivis_scope" -> "scope", "scopes")
             unprefixed = re.sub(r'^(?:ivis_|tbl_|table_|app_)', '', t_lower)
-            if unprefixed and len(unprefixed) >= 3:
+            if unprefixed and len(unprefixed) >= 3 and unprefixed not in stop_words:
                 if (unprefixed in tokens or 
                     f"{unprefixed}s" in tokens or 
                     f"{unprefixed}es" in tokens or 
@@ -73,8 +98,10 @@ class BaseConnector(ABC):
                         exact.append(t)
                         continue
 
-            # 3. Fuzzy partial match
+            # 3. Fuzzy partial match (strictly exclude English stop words)
             for tok in tokens:
+                if tok in stop_words:
+                    continue
                 if len(tok) >= 4 and (tok in t_lower or t_lower in tok):
                     if t not in exact and t not in fuzzy:
                         fuzzy.append(t)

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/dbagent_cli.egg-info/PKG-INFO

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

{dbagent_cli-0.9.4 → dbagent_cli-0.9.6}/pyproject.toml

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