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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.9.5
+Version: 0.9.7
 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.7}/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.7"
 __author__ = "Santhosh Gupta"

{dbagent_cli-0.9.5 → dbagent_cli-0.9.7}/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.7}/dbagent/agent/pipeline.py

@@ -238,12 +238,18 @@ class QueryPipeline:
                     exact, fuzzy = [], []
 
                 # Inherit active tables from conversation context if follow-up prompt doesn't name tables explicitly
+                is_inherited = False
                 if not exact and not fuzzy:
                     inherited = self.conversation.get_active_tables()
                     if inherited:
-                        exact = inherited[:4]
+                        exact = inherited[:8]
+                        is_inherited = True
 
-                sub_schema, join_steps = self._get_fast_schema_context(user_prompt, exact, fuzzy)
+                effective_prompt = user_prompt
+                if is_inherited and exact:
+                    effective_prompt = f"Focus strictly on the architecture, linkages, and data flow between {', '.join(exact[:5])}. User request: {user_prompt}"
+
+                sub_schema, join_steps = self._get_fast_schema_context(effective_prompt, exact, fuzzy)
                 result.schema = sub_schema
                 result.join_steps = join_steps
 
@@ -258,9 +264,9 @@ class QueryPipeline:
                         source="user_guidance",
                     )
 
-                kb_context = self.knowledge_base.format_prompt_context(self.db_name, user_prompt, exact + fuzzy)
+                kb_context = self.knowledge_base.format_prompt_context(self.db_name, effective_prompt, exact + fuzzy)
                 conv_history = self.conversation.format_context_for_prompt()
-                analysis_prompt = f"{kb_context}\n\n{user_prompt}" if kb_context else user_prompt
+                analysis_prompt = f"{kb_context}\n\n{effective_prompt}" if kb_context else effective_prompt
 
                 result.message = self.generator.generate_analysis_response(
                     schema=sub_schema,
@@ -594,10 +600,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.7}/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.7}/dbagent/connectors/relational.py

@@ -47,7 +47,7 @@ def _fuzzy_match_tables(table_names: List[str], user_prompt: str) -> Tuple[List[
     exact: List[str] = []
     fuzzy: List[str] = []
 
-    # Extract meaningful words from the prompt (3+ chars, skip SQL keywords)
+    # Extract meaningful words from the prompt (3+ chars, skip SQL keywords and diagram meta-terms)
     sql_keywords = {
         "select", "from", "where", "join", "left", "right", "inner", "outer",
         "order", "group", "having", "limit", "offset", "insert", "update",
@@ -57,6 +57,10 @@ def _fuzzy_match_tables(table_names: List[str], user_prompt: str) -> Tuple[List[
         "drop", "index", "primary", "foreign", "key", "null", "default",
         "hour", "minute", "day", "week", "month", "year", "ago", "today",
         "give", "tell", "want", "need", "please", "display", "fetch",
+        "flow", "diagram", "chart", "graph", "link", "links", "structure",
+        "design", "model", "blueprint", "mapping", "mappings", "overview",
+        "query", "code", "script", "report", "summary", "relation",
+        "relations", "relationship", "relationships", "architecture", "check",
     }
     prompt_words = set()
     for word in re.findall(r"\b[a-z_][a-z0-9_]*\b", prompt_lower):

{dbagent_cli-0.9.5 → dbagent_cli-0.9.7}/dbagent_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.9.5
+Version: 0.9.7
 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.7}/pyproject.toml

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