diff options
Diffstat (limited to 'memory.py')
| -rw-r--r-- | memory.py | 21 |
1 files changed, 21 insertions, 0 deletions
@@ -1,5 +1,26 @@ from dataclasses import dataclass, field +from datetime import UTC, datetime from enum import Enum +from uuid import UUID, uuid7 + + +class MemoryCategory(Enum): + SEMANTIC = "semantic" + EPISODIC = "episodic" + + +@dataclass +class LongTermMemory: + id: UUID + content: str + category: MemoryCategory + created_at: datetime + + @classmethod + def create(cls, content: str, category: MemoryCategory) -> LongTermMemory: + return cls( + id=uuid7(), content=content, category=category, created_at=datetime.now(UTC) + ) class TaskStatus(Enum): |
