Spaces:
Runtime error
Runtime error
from typing import Any, Dict, List | |
from pydantic import BaseModel | |
from langchain.schema import BaseMemory | |
class SimpleMemory(BaseMemory, BaseModel): | |
"""Simple memory for storing context or other bits of information that shouldn't | |
ever change between prompts. | |
""" | |
memories: Dict[str, Any] = dict() | |
def memory_variables(self) -> List[str]: | |
return list(self.memories.keys()) | |
def load_memory_variables(self, inputs: Dict[str, Any]) -> Dict[str, str]: | |
return self.memories | |
def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None: | |
"""Nothing should be saved or changed, my memory is set in stone.""" | |
pass | |
def clear(self) -> None: | |
"""Nothing to clear, got a memory like a vault.""" | |
pass | |