File size: 2,088 Bytes
732e77c | 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 63 64 65 66 67 | # /memory/profile.py
from __future__ import annotations
from dataclasses import dataclass, asdict
from typing import Dict, List, Optional
from pathlib import Path
import json, time
PROFILE_DIR = Path("memory/.profiles")
PROFILE_DIR.mkdir(parents=True, exist_ok=True)
@dataclass
class Note:
key: str
value: str
created_at: float
updated_at: float
tags: List[str]
@dataclass
class Profile:
user_id: str
display_name: Optional[str] = None
notes: Dict[str, Note] = None
@classmethod
def load(cls, user_id: str) -> "Profile":
p = PROFILE_DIR / f"{user_id}.json"
if not p.exists():
return Profile(user_id=user_id, notes={})
data = json.loads(p.read_text(encoding="utf-8"))
notes = {k: Note(**v) for k, v in data.get("notes", {}).items()}
return Profile(user_id=data["user_id"], display_name=data.get("display_name"), notes=notes)
def save(self) -> None:
p = PROFILE_DIR / f"{self.user_id}.json"
data = {
"user_id": self.user_id,
"display_name": self.display_name,
"notes": {k: asdict(v) for k, v in (self.notes or {}).items()},
}
p.write_text(json.dumps(data, ensure_ascii=False, indent=2), encoding="utf-8")
# --- memory operations (explicit user consent) ---
def remember(self, key: str, value: str, tags: Optional[List[str]] = None) -> None:
now = time.time()
note = self.notes.get(key)
if note:
note.value, note.updated_at = value, now
if tags: note.tags = tags
else:
self.notes[key] = Note(key=key, value=value, tags=tags or [], created_at=now, updated_at=now)
self.save()
def forget(self, key: str) -> bool:
ok = key in self.notes
if ok:
self.notes.pop(key)
self.save()
return ok
def recall(self, key: str) -> Optional[str]:
n = self.notes.get(key)
return n.value if n else None
def list_notes(self) -> List[str]:
return sorted(self.notes.keys())
|