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

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

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

{dbagent_cli-0.9.3 → 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.3 → 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.3
+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.3 → dbagent_cli-0.9.4}/pyproject.toml

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