mcp-sqlite-memory-bank 1.5.1__py3-none-any.whl → 1.6.0__py3-none-any.whl

mcp_sqlite_memory_bank/prompts.py

@@ -19,26 +19,26 @@ import json
 
 class MemoryBankPrompts:
     """Manages MCP Prompts for the SQLite Memory Bank."""
-    
+
     def __init__(self, mcp_app: FastMCP, db_path: str):
         self.mcp = mcp_app
         self.db_path = db_path
         self._register_prompts()
-    
+
     def _register_prompts(self):
         """Register MCP prompts with the FastMCP app."""
-        
+
         @self.mcp.prompt("analyze-memory-content")
         async def analyze_memory_content(table_name: Optional[str] = None) -> str:
             """Analyze memory bank content and provide insights."""
             db = get_database(self.db_path)
-            
+
             if table_name:
                 # Analyze specific table
                 result = cast(Dict[str, Any], db.read_rows(table_name, {}))
                 if not result.get("success"):
                     return f"Error: Could not access table '{table_name}'. Please check if it exists."
-                
+
                 rows = result.get("rows", [])
                 prompt = f"""Please analyze the content in the '{table_name}' table from the memory bank.
 
@@ -58,21 +58,23 @@ Focus on actionable insights that could help improve how this information is sto
                 tables_result = cast(Dict[str, Any], db.list_tables())
                 if not tables_result.get("success"):
                     return "Error: Could not access memory bank tables."
-                
+
                 tables = tables_result.get("tables", [])
                 overview = {"tables": len(tables), "total_content": []}
-                
+
                 for table in tables[:5]:  # Limit to first 5 tables
                     rows_result = cast(Dict[str, Any], db.read_rows(table, {}))
                     if rows_result.get("success"):
                         rows = rows_result.get("rows", [])
                         total_content = cast(List[Any], overview["total_content"])
-                        total_content.append({
-                            "table": table,
-                            "rows": len(rows),
-                            "sample": rows[:2] if rows else []
-                        })
-                
+                        total_content.append(
+                            {
+                                "table": table,
+                                "rows": len(rows),
+                                "sample": rows[:2] if rows else [],
+                            }
+                        )
+
                 prompt = f"""Please analyze the overall content in this memory bank.
 
 Memory Bank Overview:
@@ -85,31 +87,41 @@ Please provide:
 4. Suggestions for leveraging this content more effectively
 
 Focus on high-level strategic insights about the memory bank's utility and organization."""
-            
+
             return prompt
-        
+
         @self.mcp.prompt("search-and-summarize")
-        async def search_and_summarize(query: str, max_results: Optional[int] = 10) -> str:
+        async def search_and_summarize(
+            query: str, max_results: Optional[int] = 10
+        ) -> str:
             """Search memory content and create a summary prompt."""
             db = get_database(self.db_path)
-            
+
             # Perform search
-            result = cast(Dict[str, Any], db.search_content(query, None, max_results or 10))
+            result = cast(
+                Dict[str, Any], db.search_content(query, None, max_results or 10)
+            )
             if not result.get("success"):
                 return f"Error: Could not search for '{query}'. {result.get('error', 'Unknown error')}"
-            
+
             search_results = result.get("results", [])
             if not search_results:
                 return f"No results found for query: '{query}'. Please try different search terms or check if relevant content exists in the memory bank."
-            
+
             # Format results for prompt
             formatted_results = []
-            for i, result in enumerate(search_results[:max_results or 10], 1):
-                formatted_results.append(f"{i}. Table: {result.get('table', 'unknown')}")
-                formatted_results.append(f"   Content: {result.get('content', 'No content')[:200]}...")
-                formatted_results.append(f"   Relevance: {result.get('relevance', 'N/A')}")
+            for i, result in enumerate(search_results[: max_results or 10], 1):
+                formatted_results.append(
+                    f"{i}. Table: {result.get('table', 'unknown')}"
+                )
+                formatted_results.append(
+                    f"   Content: {result.get('content', 'No content')[:200]}..."
+                )
+                formatted_results.append(
+                    f"   Relevance: {result.get('relevance', 'N/A')}"
+                )
                 formatted_results.append("")
-            
+
             prompt = f"""Based on the search query "{query}", here are the most relevant results from the memory bank:
 
 Search Results:
@@ -122,19 +134,21 @@ Please provide:
 4. Actionable insights based on this content
 
 Use this information to provide a thorough, well-organized response that synthesizes the search results."""
-            
+
             return prompt
-        
+
         @self.mcp.prompt("technical-decision-analysis")
-        async def technical_decision_analysis(decision_topic: Optional[str] = None) -> str:
+        async def technical_decision_analysis(
+            decision_topic: Optional[str] = None,
+        ) -> str:
             """Analyze technical decisions from the memory bank."""
             db = get_database(self.db_path)
-            
+
             # Try to find technical_decisions table
             tables_result = cast(Dict[str, Any], db.list_tables())
             if not tables_result.get("success"):
                 return "Error: Could not access memory bank."
-            
+
             tables = tables_result.get("tables", [])
             if "technical_decisions" not in tables:
                 return """No technical decisions table found in the memory bank. 
@@ -145,34 +159,44 @@ To use this prompt effectively, please:
 3. Try this prompt again
 
 The table should include fields like: decision_name, chosen_approach, rationale, alternatives, timestamp."""
-            
+
             # Get technical decisions
             where_clause = {}
             if decision_topic:
                 # This is a simplified search - in practice you'd want semantic search
                 where_clause = {"decision_name": decision_topic}
-            
+
             result = db.read_rows("technical_decisions", where_clause)
             if not result.get("success"):
                 return "Error: Could not read technical decisions."
-            
+
             decisions = result.get("rows", [])
             if not decisions:
                 topic_msg = f" related to '{decision_topic}'" if decision_topic else ""
                 return f"No technical decisions found{topic_msg}. Consider adding some decisions to the memory bank first."
-            
+
             # Format decisions for analysis
             formatted_decisions = []
             decisions_list = cast(List[Dict[str, Any]], decisions)
             for i, decision in enumerate(decisions_list, 1):
-                formatted_decisions.append(f"{i}. Decision: {decision.get('decision_name', 'Unknown')}")
-                formatted_decisions.append(f"   Approach: {decision.get('chosen_approach', 'Not specified')}")
-                formatted_decisions.append(f"   Rationale: {decision.get('rationale', 'Not provided')}")
-                if decision.get('alternatives'):
-                    formatted_decisions.append(f"   Alternatives: {decision.get('alternatives')}")
-                formatted_decisions.append(f"   Date: {decision.get('timestamp', 'Unknown')}")
+                formatted_decisions.append(
+                    f"{i}. Decision: {decision.get('decision_name', 'Unknown')}"
+                )
+                formatted_decisions.append(
+                    f"   Approach: {decision.get('chosen_approach', 'Not specified')}"
+                )
+                formatted_decisions.append(
+                    f"   Rationale: {decision.get('rationale', 'Not provided')}"
+                )
+                if decision.get("alternatives"):
+                    formatted_decisions.append(
+                        f"   Alternatives: {decision.get('alternatives')}"
+                    )
+                formatted_decisions.append(
+                    f"   Date: {decision.get('timestamp', 'Unknown')}"
+                )
                 formatted_decisions.append("")
-            
+
             prompt = f"""Please analyze these technical decisions from the memory bank:
 
 Technical Decisions{f" (filtered by: {decision_topic})" if decision_topic else ""}:
@@ -186,19 +210,19 @@ Please provide:
 5. Suggestions for improving decision documentation
 
 Focus on actionable insights that can improve technical decision-making processes."""
-            
+
             return prompt
-        
+
         @self.mcp.prompt("memory-bank-context")
         async def memory_bank_context(context_type: str = "full") -> str:
             """Provide memory bank context for AI conversations."""
             db = get_database(self.db_path)
-            
+
             # Get overview
             tables_result = db.list_tables()
             if not tables_result.get("success"):
                 return "Error: Could not access memory bank for context."
-            
+
             tables = tables_result.get("tables", [])
             context_info = {
                 "available_tables": tables,
@@ -206,16 +230,16 @@ Focus on actionable insights that can improve technical decision-making processe
                     "Full-text search across all content",
                     "Semantic search (if embeddings are available)",
                     "Structured data queries",
-                    "Content analytics and insights"
+                    "Content analytics and insights",
                 ],
                 "usage_suggestions": [
                     "Use search_content() for finding specific information",
                     "Use semantic_search() for conceptual queries",
                     "Use read_rows() for structured data access",
-                    "Use explore_tables() to discover available content"
-                ]
+                    "Use explore_tables() to discover available content",
+                ],
             }
-            
+
             if context_type == "brief":
                 tables_list = cast(List[str], tables)
                 prompt = f"""Memory Bank Context (Brief):
@@ -234,11 +258,11 @@ This memory bank contains structured information that can be searched and analyz
                             rows = cast(List[Any], result.get("rows", []))
                             sample_content[table] = {
                                 "row_count": len(rows),
-                                "sample_row": rows[0] if rows else None
+                                "sample_row": rows[0] if rows else None,
                             }
                     except Exception:
                         continue
-                
+
                 prompt = f"""Memory Bank Context (Full):
 
 {json.dumps(context_info, indent=2)}
@@ -247,7 +271,7 @@ Sample Content:
 {json.dumps(sample_content, indent=2)}
 
 This memory bank contains structured information that can be searched, analyzed, and leveraged for various tasks. The content is organized in tables with different types of information. Use the available search and query tools to access specific content as needed for your current task."""
-            
+
             return prompt