Upload 2 files

plugins/memory/conversation_memory.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+"""Conversation Memory Plugin"""
+from typing import List, Dict, Any
+from collections import deque
+import pandas as pd
+
+class ConversationMemory:
+    """Track conversation history and context."""
+    def __init__(self, max_history: int = 50):
+        self.history = deque(maxlen=max_history)
+        self.context = {}
+    
+    def add_message(self, role: str, content: str, metadata: Dict[str, Any] = None):
+        message = {"role": role, "content": content, "timestamp": pd.Timestamp.now().isoformat()}
+        if metadata:
+            message["metadata"] = metadata
+        self.history.append(message)
+    
+    def get_history(self, last_n: int = None) -> List[Dict[str, Any]]:
+        return list(self.history)[-last_n:] if last_n else list(self.history)
+    
+    def clear(self):
+        self.history.clear()
+        self.context.clear()

plugins/memory/document_memory.py

@@ -0,0 +1,27 @@
+#!/usr/bin/env python3
+"""Document Memory Plugin"""
+from typing import Dict, Any, Optional
+
+class DocumentMemory:
+    """Cache and manage uploaded documents."""
+    def __init__(self):
+        self.documents = {}
+        self.metadata = {}
+    
+    def store_document(self, doc_id: str, content: Any, metadata: Dict[str, Any]):
+        """Store document with metadata."""
+        self.documents[doc_id] = content
+        self.metadata[doc_id] = metadata
+    
+    def get_document(self, doc_id: str) -> Optional[Any]:
+        """Retrieve document by ID."""
+        return self.documents.get(doc_id)
+    
+    def list_documents(self) -> Dict[str, Dict[str, Any]]:
+        """List all stored documents with metadata."""
+        return {doc_id: self.metadata[doc_id] for doc_id in self.documents.keys()}
+    
+    def clear(self):
+        """Clear all documents."""
+        self.documents.clear()
+        self.metadata.clear()