dbagent-cli 0.8.8__tar.gz → 0.8.9__tar.gz

{dbagent_cli-0.8.8 → dbagent_cli-0.8.9}/PKG-INFO

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

{dbagent_cli-0.8.8 → dbagent_cli-0.8.9}/dbagent/__init__.py

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

{dbagent_cli-0.8.8 → dbagent_cli-0.8.9}/dbagent/agent/pipeline.py

@@ -205,16 +205,17 @@ class QueryPipeline:
         # --- 1.2. Architectural / Linkage Analysis Path ---
         if result.intent == "analysis":
             try:
-                cached = self.cache_mgr.load_cached_schema(self.db_id)
-                if cached:
-                    schema = cached
-                else:
-                    schema = self.connector.inspect_schema(include_samples=True, max_samples=2)
+                try:
+                    exact, fuzzy = self.connector.resolve_tables(user_prompt)
+                except Exception:
+                    exact, fuzzy = [], []
 
-                result.schema = schema
+                sub_schema, join_steps = self._get_fast_schema_context(user_prompt, exact, fuzzy)
+                result.schema = sub_schema
+                result.join_steps = join_steps
                 conv_history = self.conversation.format_context_for_prompt()
                 result.message = self.generator.generate_analysis_response(
-                    schema=schema,
+                    schema=sub_schema,
                     user_prompt=user_prompt,
                     conversation_history=conv_history,
                     model=self.model,

{dbagent_cli-0.8.8 → dbagent_cli-0.8.9}/dbagent/connectors/relational.py

@@ -200,16 +200,37 @@ class RelationalConnector(BaseConnector):
             lower_targets = {t.lower() for t in table_names}
             target_tables = [t for t in all_table_names if t.lower() in lower_targets]
 
-        for t_name in target_tables:
-            table_model = self._inspect_single_table(
-                insp=insp,
-                table_name=t_name,
-                is_view=False,
-                include_samples=include_samples,
-                max_samples=max_samples,
-                include_row_counts=include_row_counts,
-            )
-            tables.append(table_model)
+        if len(target_tables) > 1:
+            from concurrent.futures import ThreadPoolExecutor
+            with ThreadPoolExecutor(max_workers=min(len(target_tables), 8)) as executor:
+                futures = [
+                    executor.submit(
+                        self._inspect_single_table,
+                        insp,
+                        t_name,
+                        False,
+                        include_samples,
+                        max_samples,
+                        include_row_counts,
+                    )
+                    for t_name in target_tables
+                ]
+                for f in futures:
+                    try:
+                        tables.append(f.result())
+                    except Exception:
+                        pass
+        else:
+            for t_name in target_tables:
+                table_model = self._inspect_single_table(
+                    insp=insp,
+                    table_name=t_name,
+                    is_view=False,
+                    include_samples=include_samples,
+                    max_samples=max_samples,
+                    include_row_counts=include_row_counts,
+                )
+                tables.append(table_model)
 
         # 2. Inspect views
         if include_views:
@@ -220,20 +241,41 @@ class RelationalConnector(BaseConnector):
                 lower_targets = {t.lower() for t in table_names}
                 target_views = [v for v in all_view_names if v.lower() in lower_targets]
 
-            for v_name in target_views:
-                view_model = self._inspect_single_table(
-                    insp=insp,
-                    table_name=v_name,
-                    is_view=True,
-                    include_samples=include_samples,
-                    max_samples=max_samples,
-                    include_row_counts=include_row_counts,
-                )
-                try:
-                    view_model.view_definition = insp.get_view_definition(v_name)
-                except Exception:
-                    pass
-                tables.append(view_model)
+            if len(target_views) > 1:
+                from concurrent.futures import ThreadPoolExecutor
+                with ThreadPoolExecutor(max_workers=min(len(target_views), 6)) as executor:
+                    view_futures = [
+                        executor.submit(
+                            self._inspect_single_table,
+                            insp,
+                            v_name,
+                            True,
+                            include_samples,
+                            max_samples,
+                            include_row_counts,
+                        )
+                        for v_name in target_views
+                    ]
+                    for f in view_futures:
+                        try:
+                            tables.append(f.result())
+                        except Exception:
+                            pass
+            else:
+                for v_name in target_views:
+                    view_model = self._inspect_single_table(
+                        insp=insp,
+                        table_name=v_name,
+                        is_view=True,
+                        include_samples=include_samples,
+                        max_samples=max_samples,
+                        include_row_counts=include_row_counts,
+                    )
+                    try:
+                        view_model.view_definition = insp.get_view_definition(v_name)
+                    except Exception:
+                        pass
+                    tables.append(view_model)
 
         return DatabaseSchema(
             dialect_name=dialect_name,

{dbagent_cli-0.8.8 → dbagent_cli-0.8.9}/dbagent_cli.egg-info/PKG-INFO

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

{dbagent_cli-0.8.8 → dbagent_cli-0.8.9}/pyproject.toml

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