dbagent-cli 0.3.1__tar.gz → 0.3.2__tar.gz

{dbagent_cli-0.3.1 → dbagent_cli-0.3.2}/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.3.1
-Summary: Enterprise AI Database Copilot with Long-Term Change Recovery (CLI)
+Version: 0.3.2
+Summary: Enterprise AI Database Copilot with High-Speed Schema Caching & Long-Term Recovery (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{dbagent_cli-0.3.1 → dbagent_cli-0.3.2}/dbagent/__init__.py

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

{dbagent_cli-0.3.1 → dbagent_cli-0.3.2}/dbagent/agent/pipeline.py

@@ -129,6 +129,40 @@ class QueryPipeline:
         self.db_name = "database"
         if hasattr(connector, "engine") and connector.engine:
             self.db_name = connector.engine.url.database or "database"
+            self.db_id = str(connector.engine.url)
+        else:
+            self.db_id = self.db_name
+        self.cache_mgr = SchemaCacheManager()
+
+    def _get_fast_schema_context(
+        self,
+        user_prompt: str,
+        exact: List[str],
+        fuzzy: List[str],
+    ) -> Tuple[DatabaseSchema, List[JoinStep]]:
+        """Fast schema sub-graph retrieval using cache (<5ms) or targeted introspection (<50ms)."""
+        # 1. Try local schema cache (< 5ms)
+        cached = self.cache_mgr.load_cached_schema(self.db_id)
+        if cached:
+            retriever = ContextRetriever(cached)
+            return retriever.retrieve_context(
+                user_prompt=user_prompt,
+                exact_tables=exact,
+                fuzzy_tables=fuzzy,
+                max_tables=8,
+            )
+
+        # 2. Targeted on-demand table introspection (~50ms instead of 5000ms for full scan)
+        target_names = list(set(exact + fuzzy))
+        if target_names:
+            targeted = self.connector.inspect_schema(table_names=target_names, include_samples=False, max_samples=0)
+            retriever = ContextRetriever(targeted)
+            return retriever.retrieve_context(user_prompt, exact_tables=exact, fuzzy_tables=fuzzy, max_tables=8)
+
+        # 3. Fallback: inspect targeted
+        targeted = self.connector.inspect_targeted(user_prompt, max_tables=8)
+        retriever = ContextRetriever(targeted)
+        return retriever.retrieve_context(user_prompt, exact_tables=exact, fuzzy_tables=fuzzy, max_tables=8)
 
     def run(self, user_prompt: str, force: bool = False, bypass_cache: bool = False) -> PipelineResult:
         """
@@ -138,10 +172,16 @@ class QueryPipeline:
         result = PipelineResult()
         result.intent = ScriptGenerator.classify_intent(user_prompt)
 
-        # --- 1. Conversational Chat Fast-Path ---
+        # --- 1. Conversational Chat Fast-Path (<10ms schema load) ---
         if result.intent == "chat":
             try:
-                schema = self.connector.inspect_schema(include_samples=False, max_samples=0, include_row_counts=False)
+                cached = self.cache_mgr.load_cached_schema(self.db_id)
+                if cached:
+                    schema = cached
+                else:
+                    quick_tables = self.connector.get_table_names()[:25]
+                    schema = self.connector.inspect_schema(table_names=quick_tables, include_samples=False, max_samples=0)
+
                 result.schema = schema
                 result.message = self.generator.generate_chat_response(
                     schema=schema,
@@ -177,16 +217,9 @@ class QueryPipeline:
                     result.error = "Ambiguous table reference — please specify table name."
                     return result
 
-        # --- 3. Fast AI Context Retrieval (Minimal sub-graph) ---
+        # --- 3. Fast AI Context Retrieval (Minimal sub-graph < 5ms) ---
         try:
-            full_schema = self.connector.inspect_schema(include_samples=False, max_samples=0, include_row_counts=False)
-            retriever = ContextRetriever(full_schema)
-            sub_schema, join_steps = retriever.retrieve_context(
-                user_prompt=user_prompt,
-                exact_tables=exact,
-                fuzzy_tables=fuzzy,
-                max_tables=8,
-            )
+            sub_schema, join_steps = self._get_fast_schema_context(user_prompt, exact, fuzzy)
             result.schema = sub_schema
             result.join_steps = join_steps
         except Exception as e:

{dbagent_cli-0.3.1 → dbagent_cli-0.3.2}/dbagent/cli.py

@@ -357,11 +357,15 @@ def chat_command(
     console.print(f"[bold cyan]AI:[/bold cyan] {llm.name} | [bold yellow]Mode:[/bold yellow] {current_mode.value}")
     console.print("[dim]Type your question in plain English or use colon commands (e.g. :help, :tables, :doctor).[/dim]")
 
-    # Display AI recommendations on connect
+    # Display AI recommendations on connect (instant via cache or top tables)
     try:
-        with console.status("[dim]Analyzing schema recommendations...[/dim]"):
-            full_schema = connector.inspect_schema(include_samples=False, max_samples=0, include_row_counts=False)
-            recs = QueryRecommender.generate_recommendations(full_schema, max_suggestions=3)
+        cached_schema = cache_mgr.load_cached_schema(db_url)
+        if cached_schema:
+            recs = QueryRecommender.generate_recommendations(cached_schema, max_suggestions=3)
+        else:
+            quick_tables = table_names[:10]
+            quick_schema = connector.inspect_schema(table_names=quick_tables, include_samples=False, max_samples=0)
+            recs = QueryRecommender.generate_recommendations(quick_schema, max_suggestions=3)
         if recs:
             console.print("\n[bold cyan]💡 Try asking:[/bold cyan]")
             for r in recs:
@@ -443,8 +447,12 @@ def chat_command(
                         continue
                     exact, fuzzy = connector.resolve_tables(cmd_arg)
                     target = exact[0] if exact else (fuzzy[0] if fuzzy else cmd_arg)
-                    with console.status(f"[bold cyan]Finding relationships for {target}...[/bold cyan]"):
-                        full_schema = connector.inspect_schema(include_samples=False, max_samples=0)
+                    cached = cache_mgr.load_cached_schema(db_url)
+                    if cached:
+                        full_schema = cached
+                    else:
+                        with console.status(f"[bold cyan]Finding relationships for {target}...[/bold cyan]"):
+                            full_schema = connector.inspect_schema(include_samples=False, max_samples=0)
                     
                     tmodel = full_schema.get_table(target)
                     outgoing = tmodel.foreign_keys if tmodel else []
@@ -473,10 +481,14 @@ def chat_command(
                     if len(args) < 2:
                         print_warning("Usage: :path <start_table> <end_table> (e.g. :path users events)")
                         continue
-                    with console.status("[bold cyan]Traversing relationship graph...[/bold cyan]"):
-                        full_schema = connector.inspect_schema(include_samples=False, max_samples=0)
-                        graph = RelationshipGraph(full_schema)
-                        path = graph.find_shortest_path(args[0], args[1])
+                    cached = cache_mgr.load_cached_schema(db_url)
+                    if cached:
+                        full_schema = cached
+                    else:
+                        with console.status("[bold cyan]Traversing relationship graph...[/bold cyan]"):
+                            full_schema = connector.inspect_schema(include_samples=False, max_samples=0)
+                    graph = RelationshipGraph(full_schema)
+                    path = graph.find_shortest_path(args[0], args[1])
                     if path:
                         console.print(f"\n[bold cyan]Shortest Relationship Path:[/bold cyan]\n{path.format_visual_path()}\n")
                     else:

{dbagent_cli-0.3.1 → dbagent_cli-0.3.2}/dbagent_cli.egg-info/PKG-INFO

@@ -1,7 +1,7 @@
 Metadata-Version: 2.4
 Name: dbagent-cli
-Version: 0.3.1
-Summary: Enterprise AI Database Copilot with Long-Term Change Recovery (CLI)
+Version: 0.3.2
+Summary: Enterprise AI Database Copilot with High-Speed Schema Caching & Long-Term Recovery (CLI)
 Author: Santhosh Gupta
 Requires-Python: >=3.9
 Description-Content-Type: text/markdown

{dbagent_cli-0.3.1 → dbagent_cli-0.3.2}/pyproject.toml

@@ -4,8 +4,8 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "dbagent-cli"
-version = "0.3.1"
-description = "Enterprise AI Database Copilot with Long-Term Change Recovery (CLI)"
+version = "0.3.2"
+description = "Enterprise AI Database Copilot with High-Speed Schema Caching & Long-Term Recovery (CLI)"
 readme = "README.md"
 requires-python = ">=3.9"
 authors = [