memory_layer/models.py at master · staru09/memory_layer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
from dataclasses import dataclass, field
from datetime import datetime
from typing import Optional


@dataclass
class Foresight:
    id: Optional[int] = None
    description: str = ""
    valid_from: Optional[datetime] = None
    valid_until: Optional[datetime] = None
    evidence: str = ""
    duration_days: Optional[int] = None
    is_active: bool = True
    created_at: Optional[datetime] = None


@dataclass
class ConflictLog:
    id: Optional[int] = None
    category: str = ""
    old_value: str = ""
    new_value: str = ""
    resolution: str = "recency_wins"
    detected_at: Optional[datetime] = None


@dataclass
class UserProfile:
    id: Optional[int] = None
    profile_text: str = ""
    updated_at: Optional[datetime] = None


@dataclass
class ChatThread:
    id: str = ""
    title: Optional[str] = None
    created_at: Optional[datetime] = None
    updated_at: Optional[datetime] = None


@dataclass
class ChatMessage:
    id: Optional[int] = None
    thread_id: str = ""
    role: str = ""  # 'user' or 'assistant'
    content: str = ""
    created_at: Optional[datetime] = None
    ingested: bool = False


@dataclass
class QueryLog:
    id: Optional[int] = None
    thread_id: Optional[str] = None
    query_text: str = ""
    response_text: str = ""
    memory_context: str = ""
    retrieval_metadata: Optional[dict] = field(default_factory=dict)
    created_at: Optional[datetime] = None
    query_time: Optional[datetime] = None