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

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

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

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

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

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

@@ -162,6 +162,10 @@ class ScriptGenerator:
             "check with my previous context", "check with previous context", "previous context",
             "complete architecture", "scope architecture", "tenant level to user level",
             "tenant level", "user level",
+            "tables that were linked", "tables linked with", "linked with", "links of",
+            "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",
         ]
         for kw in analysis_keywords:
             if kw in prompt_lower:

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

@@ -157,17 +157,40 @@ class QueryPipeline:
                 max_tables=8,
             )
 
-        # 2. Targeted on-demand table introspection (~50ms instead of 5000ms for full scan)
+        # 2. Targeted on-demand table introspection with automatic junction expansion
         target_names = list(set(exact + fuzzy))
         if target_names:
+            try:
+                all_table_names = self.connector.get_table_names()
+                stems = set()
+                for t in target_names:
+                    unprefixed = re.sub(r'^(?:ivis_|tbl_|table_|app_)', '', t.lower())
+                    unprefixed = re.sub(r'(?:s|es)$', '', unprefixed)
+                    if len(unprefixed) >= 3:
+                        stems.add(unprefixed)
+
+                for t in all_table_names:
+                    t_lower = t.lower()
+                    if t in target_names:
+                        continue
+                    is_map = any(m in t_lower for m in ["_map", "mapping", "_rel", "_junction", "_link", "_access", "_config"])
+                    matches_stem = any(s in t_lower for s in stems)
+                    if (is_map and matches_stem) or sum(1 for s in stems if s in t_lower) >= 2:
+                        if t not in target_names:
+                            target_names.append(t)
+                    if len(target_names) >= 12:
+                        break
+            except Exception:
+                pass
+
             targeted = self.connector.inspect_schema(table_names=target_names, include_samples=True, max_samples=2)
             retriever = ContextRetriever(targeted)
-            return retriever.retrieve_context(user_prompt, exact_tables=exact, fuzzy_tables=fuzzy, max_tables=8)
+            return retriever.retrieve_context(user_prompt, exact_tables=exact, fuzzy_tables=fuzzy, max_tables=12)
 
         # 3. Fallback: inspect targeted
-        targeted = self.connector.inspect_targeted(user_prompt, max_tables=8, include_samples=True)
+        targeted = self.connector.inspect_targeted(user_prompt, max_tables=12, include_samples=True)
         retriever = ContextRetriever(targeted)
-        return retriever.retrieve_context(user_prompt, exact_tables=exact, fuzzy_tables=fuzzy, max_tables=8)
+        return retriever.retrieve_context(user_prompt, exact_tables=exact, fuzzy_tables=fuzzy, max_tables=12)
 
     def run(self, user_prompt: str, force: bool = False, bypass_cache: bool = False) -> PipelineResult:
         """

{dbagent_cli-0.9.2 → dbagent_cli-0.9.4}/dbagent/schema/ground_truth.py

@@ -42,6 +42,8 @@ class GroundTruthValidationResult:
                 lines.append(f"    [dim]• ... and {len(self.verified_fks) - 4} more[/dim]")
         if self.sql_queries_checked > 0:
             lines.append(f"  [bold green]✓[/bold green] [bold white]SQL Queries Verified:[/bold white] {self.sql_queries_checked} dialect statement(s) validated against schema")
+        for missing in self.missing_tables:
+            lines.append(f"  [bold yellow]⚠ Note:[/bold yellow] Query referenced [dim]{missing}[/dim]")
         for tbl, cols in self.missing_columns.items():
             lines.append(f"  [bold red]⚠ Warning:[/bold red] Column(s) [yellow]{', '.join(cols)}[/yellow] not found in table [cyan]{tbl}[/cyan]")
         return "\n".join(lines)
@@ -83,14 +85,40 @@ class SchemaGroundTruthValidator:
                         if fk_desc not in result.verified_fks:
                             result.verified_fks.append(fk_desc)
 
-        # 2. Extract and count SQL queries
+        # 2. Extract and validate SQL queries against physical schema
         sql_blocks = re.findall(r'```(?:sql)?\s*([\s\S]*?)\s*```', response_text, re.IGNORECASE)
         result.sql_queries_checked = len(sql_blocks)
+        sql_referenced_tables = set()
+
+        for block in sql_blocks:
+            # Extract FROM and JOIN table references
+            matches = re.findall(r'\b(?:FROM|JOIN)\s+([a-zA-Z0-9_]+)', block, re.IGNORECASE)
+            for m in matches:
+                m_clean = m.strip().lower()
+                sql_referenced_tables.add(m_clean)
+
+        for sql_tbl in sql_referenced_tables:
+            if sql_tbl in schema_table_map:
+                if schema_table_map[sql_tbl].name not in result.verified_tables:
+                    result.verified_tables.append(schema_table_map[sql_tbl].name)
+            else:
+                # Check if it's a common alias or slight naming divergence
+                found_match = None
+                for real_tbl in schema_table_map:
+                    if sql_tbl in real_tbl or real_tbl in sql_tbl:
+                        found_match = schema_table_map[real_tbl].name
+                        break
+                if found_match:
+                    result.missing_tables.append(f"{sql_tbl} ➔ (Actual: {found_match})")
+                else:
+                    result.missing_tables.append(sql_tbl)
 
         total_checked = len(result.verified_tables)
-        if total_checked > 0:
+        if result.missing_tables:
+            result.score_pct = max(70, 100 - (len(result.missing_tables) * 15))
+        elif total_checked > 0:
             result.score_pct = 100
         else:
             result.score_pct = 95
-            
+
         return result

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

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

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

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