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

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

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

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

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

{dbagent_cli-0.8.6 → dbagent_cli-0.8.8}/dbagent/agent/generator.py

@@ -47,17 +47,12 @@ Generate ONLY the raw executable SQL query for the given database. No markdown f
 {schema_context}
 
 ### Rules:
-1. Output ONLY the SQL query — nothing else. No ```sql fences, no explanations.
+1. Output ONLY the executable SQL query — nothing else. No ```sql fences, no explanations.
 2. Use the exact table and column names from the schema above.
 3. Use dialect-specific syntax for `{dialect_name}`.
 4. Write clean, optimized SQL with uppercase keywords.
-5. If the user asks for recent/time-based data, use native datetime functions (e.g. for PostgreSQL: `NOW() - INTERVAL '1 hour'`, for SQLite: `datetime('now', '-1 hour')`, for MySQL: `NOW() - INTERVAL 1 HOUR`).
-6. For queries inspecting PostgreSQL foreign key mappings or table relationships via system catalog, use standard joins:
-   SELECT tc.table_name, kcu.column_name, ccu.table_name AS foreign_table_name, ccu.column_name AS foreign_column_name
-   FROM information_schema.table_constraints tc
-   JOIN information_schema.key_column_usage kcu ON tc.constraint_name = kcu.constraint_name
-   JOIN information_schema.constraint_column_usage ccu ON tc.constraint_name = ccu.constraint_name
-   WHERE tc.constraint_type = 'FOREIGN KEY';
+5. Always query the user's actual database tables and junction/mapping tables from the schema context, using foreign key relationships for JOINs.
+6. If the user asks for recent/time-based data, use native datetime functions (e.g. for PostgreSQL: `NOW() - INTERVAL '1 hour'`, for SQLite: `datetime('now', '-1 hour')`, for MySQL: `NOW() - INTERVAL 1 HOUR`).
 7. Always include appropriate WHERE clauses, JOINs, and ORDER BY as needed.
 8. Default to LIMIT 50 unless the user specifies a count.
 9. For status, active, or flag columns (e.g. `active`, `is_active`, `status`, `enabled`, `deleted`):
@@ -84,6 +79,36 @@ You are chatting with the user in the interactive database shell.
 """
 
 
+# --- Architectural / Linkage analysis prompt ---
+
+ARCHITECTURE_SYSTEM_PROMPT = """You are DB-Agent, an elite Principal Database Architect, Data Modeler, and Systems Engineer.
+Target Database Context:
+- **Dialect**: {dialect_name}
+- **Database Name**: {database_name}
+
+### Detailed Schema Sub-Graph Context (Tables, Columns, Primary Keys, Foreign Keys, Sample Data):
+{schema_context}
+
+The user is asking a question regarding database architecture, entity relationships, linkages, mappings, data flows, or structural design.
+
+Analyze the provided schema and the user's question, and produce a clear, authoritative, and comprehensive architectural response structured as follows:
+
+### 1. 🗺️ Entity-Relationship & Linkage Architecture Diagram
+Provide a clean, readable ASCII / Unicode box diagram showing the entities, relationship paths, cardinality (1:1, 1:N, N:M junction tables), and the exact connecting foreign keys / join keys.
+
+### 2. 📐 Architectural & Theoretical Breakdown
+- Explain in clear theoretical and domain terms how the entities map to each other in this database.
+- Highlight the mapping mechanisms (direct foreign keys, junction/mapping tables, hierarchical ownership).
+- Detail the cardinality, tenant isolation, and user access models represented in the schema.
+
+### 3. 🔍 How to Query & Verify Mappings (SQL Examples)
+Provide ready-to-run, dialect-accurate SQL JOIN queries answering the user's specific questions (e.g. how to find all sites mapped to a user, how to get tenant membership for a user). Format all SQL in ```sql blocks.
+
+### 4. 💡 Architectural Insights & Best Practices
+Mention indexing recommendations, cascade rules, or performance considerations for these join paths.
+"""
+
+
 # --- Retry prompt when SQL fails ---
 
 RETRY_PROMPT_TEMPLATE = """The previous SQL query failed with this error:
@@ -111,7 +136,7 @@ class ScriptGenerator:
     @staticmethod
     def classify_intent(user_prompt: str) -> str:
         """
-        Classify user's natural language intent as 'chat', 'read', 'write', or 'ddl'.
+        Classify user's natural language intent as 'chat', 'analysis', 'read', 'write', or 'ddl'.
         Used before SQL generation to predict the query type.
         """
         prompt_lower = user_prompt.lower().strip()
@@ -132,6 +157,21 @@ class ScriptGenerator:
             if clean_prompt == trig or prompt_lower.startswith(f"{trig} ") or prompt_lower.startswith(f"{trig},") or prompt_lower.startswith(f"{trig}!"):
                 return "chat"
 
+        # Architecture / Relationship / Linkage / Analysis indicators
+        analysis_keywords = [
+            "linkage architecture", "linkage", "architecture", "relationship between",
+            "how are", "how is", "how were", "how they were related", "how were they related",
+            "how they were mapped", "how to map", "mapped to", "er diagram", "diagram of",
+            "data model", "explain relationship", "explain schema", "schema analysis",
+            "analyze the database", "analyze database", "table relationship",
+            "foreign key relationship", "how are tables related", "which sites in",
+            "how to know that which", "how to know which", "mapping architecture",
+            "entity relationship", "theoritical information", "theoretical information",
+        ]
+        for kw in analysis_keywords:
+            if kw in prompt_lower:
+                return "analysis"
+
         # DDL indicators
         ddl_keywords = [
             "create table", "alter table", "drop table", "add column",
@@ -194,6 +234,39 @@ class ScriptGenerator:
             model=model,
         )
 
+    def generate_analysis_response(
+        self,
+        schema: DatabaseSchema,
+        user_prompt: str,
+        conversation_history: Optional[str] = None,
+        model: Optional[str] = None,
+    ) -> str:
+        """Generate a comprehensive architectural analysis report with ASCII diagrams, theory, and SQL queries."""
+        # 1. Filter relevant tables including foreign keys and junction tables
+        relevant_tables = SchemaSelector.select_relevant_tables(schema, user_prompt, max_tables=20)
+        schema_markdown = SchemaFormatter.to_markdown(
+            schema,
+            selected_tables=relevant_tables,
+            include_samples=True,
+            max_sample_rows=2,
+        )
+
+        system_prompt = ARCHITECTURE_SYSTEM_PROMPT.format(
+            dialect_name=schema.dialect_name,
+            database_name=schema.database_name,
+            schema_context=schema_markdown,
+        )
+
+        full_prompt = user_prompt
+        if conversation_history:
+            full_prompt = f"{conversation_history}\n\nUser: {user_prompt}"
+
+        return self.llm.generate(
+            prompt=full_prompt,
+            system_prompt=system_prompt,
+            model=model,
+        )
+
     # --- Full generation (for `generate` command) ---
 
     def generate_script(

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

@@ -202,6 +202,34 @@ class QueryPipeline:
                 result.execution_time_ms = (time.time() - t_start) * 1000
                 return result
 
+        # --- 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)
+
+                result.schema = schema
+                conv_history = self.conversation.format_context_for_prompt()
+                result.message = self.generator.generate_analysis_response(
+                    schema=schema,
+                    user_prompt=user_prompt,
+                    conversation_history=conv_history,
+                    model=self.model,
+                )
+                result.is_chat = True
+                result.was_executed = True
+                result.query_type = "analysis"
+                result.execution_time_ms = (time.time() - t_start) * 1000
+                self.conversation.add_turn(user_prompt=user_prompt, sql="", assistant_response=result.message, is_chat=True)
+                return result
+            except Exception as e:
+                result.error = f"Architecture analysis failed: {str(e)}"
+                result.execution_time_ms = (time.time() - t_start) * 1000
+                return result
+
         # --- 1.5. Direct Raw SQL Fast-Path (0ms AI latency, 0 tokens) ---
         if SQLValidator.is_direct_sql(user_prompt):
             raw_sql = user_prompt.strip().rstrip(";") + ";"

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

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

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

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

{dbagent_cli-0.8.6 → dbagent_cli-0.8.8}/tests/test_generator.py

@@ -71,3 +71,23 @@ def test_validator_destructive_delete_without_where():
     is_safe, warnings = ScriptValidator.analyze_safety("DELETE FROM orders")
     assert is_safe is False
     assert any("without WHERE" in w for w in warnings)
+
+
+def test_classify_intent_analysis():
+    prompt = "analyze the database and give me the linkage architecture that how ivis_user and ivis_tenant was related and how they were mapped and how to know that which sites in ivis_site were mapped to which user"
+    assert ScriptGenerator.classify_intent(prompt) == "analysis"
+
+    assert ScriptGenerator.classify_intent("how are users and orders related?") == "analysis"
+    assert ScriptGenerator.classify_intent("explain the relationship between customers and payments") == "analysis"
+    assert ScriptGenerator.classify_intent("give me the ER diagram and linkage architecture") == "analysis"
+
+
+def test_generate_analysis_response(mock_schema):
+    provider = MockProvider()
+    generator = ScriptGenerator(provider)
+    result = generator.generate_analysis_response(
+        mock_schema,
+        "analyze the linkage architecture between events and users",
+    )
+    assert result is not None
+    assert len(result) > 0