| |
| """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() |
|
|
