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

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

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

{dbagent_cli-0.9.6 → 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[: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,

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

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