14 lines
343 B
Python
14 lines
343 B
Python
# agent/history.py
|
|
class ConversationHistory:
|
|
def __init__(self):
|
|
self.messages: list[dict] = []
|
|
|
|
def add_message(self, role: str, content: str):
|
|
self.messages.append({"role": role, "content": content})
|
|
|
|
def get_all(self) -> list[dict]:
|
|
return self.messages
|
|
|
|
def clear(self):
|
|
self.messages = []
|