Spaces:
Runtime error
Runtime error
File size: 645 Bytes
716048e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from typing import List, Dict
class MemoryLayer:
"""
The 'Memory' of the agent.
Responsibility: Store and retrieve conversation history.
"""
def __init__(self):
self.history: List[Dict[str, str]] = []
def add_entry(self, role: str, content: str):
self.history.append({"role": role, "content": content})
def get_history(self) -> List[Dict[str, str]]:
return self.history
def set_history(self, messages: List[Dict[str, str]]):
"""Allows initializing/overwriting memory from external source (e.g. Streamlit session)"""
self.history = messages
|