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