dbagent-cli 0.7.1__tar.gz → 0.7.2__tar.gz

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/PKG-INFO

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

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/dbagent/__init__.py

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

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/dbagent/agent/conversation.py

@@ -12,7 +12,8 @@ from dataclasses import dataclass, field
 class ConversationTurn:
     """A single turn in the conversation."""
     user_prompt: str
-    sql: str
+    sql: str = ""
+    assistant_response: str = ""
     tables: List[str] = field(default_factory=list)
     filters: List[str] = field(default_factory=list)
     entities: Dict[str, str] = field(default_factory=dict)
@@ -22,10 +23,10 @@ class ConversationTurn:
 
 class ConversationContext:
     """
-    Tracks state across interactive chat turns.
+    Tracks state and multi-turn conversational history across interactive chat turns.
     """
 
-    def __init__(self, max_history: int = 10):
+    def __init__(self, max_history: int = 15):
         self.max_history = max_history
         self.turns: List[ConversationTurn] = []
         self.active_tables: Set[str] = set()
@@ -34,7 +35,8 @@ class ConversationContext:
     def add_turn(
         self,
         user_prompt: str,
-        sql: str,
+        sql: str = "",
+        assistant_response: str = "",
         tables: Optional[List[str]] = None,
         filters: Optional[List[str]] = None,
         is_chat: bool = False,
@@ -43,6 +45,7 @@ class ConversationContext:
         turn = ConversationTurn(
             user_prompt=user_prompt,
             sql=sql,
+            assistant_response=assistant_response,
             tables=tables or [],
             filters=filters or [],
             is_chat=is_chat,
@@ -59,19 +62,23 @@ class ConversationContext:
             return self.turns[-1]
         return None
 
-    def format_context_for_prompt(self) -> str:
-        """Formats recent context for inclusion in LLM prompt."""
+    def format_context_for_prompt(self, max_turns: int = 5) -> str:
+        """Formats multi-turn conversation memory for inclusion in LLM prompt."""
         if not self.turns:
             return ""
 
-        recent_valid = [t for t in self.turns[-3:] if t.sql and not t.is_chat]
-        if not recent_valid:
-            return ""
-
-        lines = ["Previous Conversation Context (Use for follow-up refinements):"]
-        for i, t in enumerate(recent_valid, 1):
-            lines.append(f"  {i}. User: \"{t.user_prompt}\"")
-            lines.append(f"     Last SQL: `{t.sql.strip()}`")
+        recent = self.turns[-max_turns:]
+        lines = ["=== Previous Chat Session Memory (Use for follow-ups & context) ==="]
+        for i, t in enumerate(recent, 1):
+            lines.append(f"Turn {i}:")
+            lines.append(f"  User: \"{t.user_prompt}\"")
+            if t.sql:
+                lines.append(f"  Assistant (SQL): {t.sql.strip()}")
+            elif t.assistant_response:
+                clean_resp = t.assistant_response.strip().replace("\n", " ")
+                preview = (clean_resp[:120] + "...") if len(clean_resp) > 120 else clean_resp
+                lines.append(f"  Assistant: \"{preview}\"")
+        lines.append("=== End Chat Session Memory ===")
 
         return "\n".join(lines)
 

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/dbagent/agent/generator.py

@@ -169,9 +169,10 @@ class ScriptGenerator:
         self,
         schema: DatabaseSchema,
         user_prompt: str,
+        conversation_history: Optional[str] = None,
         model: Optional[str] = None,
     ) -> str:
-        """Generate a natural conversational response to the user's message."""
+        """Generate a natural conversational response to the user's message with full session memory."""
         table_names = [t.name for t in schema.tables]
         table_summary = ", ".join(table_names[:30])
         if len(table_names) > 30:
@@ -183,8 +184,12 @@ class ScriptGenerator:
             table_summary=table_summary if table_summary else "No tables found",
         )
 
+        full_prompt = user_prompt
+        if conversation_history:
+            full_prompt = f"{conversation_history}\n\nUser: {user_prompt}"
+
         return self.llm.generate(
-            prompt=user_prompt,
+            prompt=full_prompt,
             system_prompt=system_prompt,
             model=model,
         )

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/dbagent/agent/pipeline.py

@@ -184,16 +184,18 @@ class QueryPipeline:
                     schema = self.connector.inspect_schema(table_names=quick_tables, include_samples=False, max_samples=0)
 
                 result.schema = schema
+                conv_history = self.conversation.format_context_for_prompt()
                 result.message = self.generator.generate_chat_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 = "chat"
                 result.execution_time_ms = (time.time() - t_start) * 1000
-                self.conversation.add_turn(user_prompt=user_prompt, sql="", is_chat=True)
+                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"Chat generation failed: {str(e)}"

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/dbagent_cli.egg-info/PKG-INFO

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

{dbagent_cli-0.7.1 → dbagent_cli-0.7.2}/pyproject.toml

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