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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.9.5
+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.5 → 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.5"
+__version__ = "0.9.6"
 __author__ = "Santhosh Gupta"

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

@@ -168,11 +168,21 @@ class ScriptGenerator:
             "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"

{dbagent_cli-0.9.5 → 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.5 → dbagent_cli-0.9.6}/dbagent/connectors/base.py

@@ -63,7 +63,10 @@ class BaseConnector(ABC):
             "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"
+            "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))

{dbagent_cli-0.9.5 → 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.5
+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.5 → dbagent_cli-0.9.6}/pyproject.toml

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