|
| 1 | +from memory import LongMemoryStore, ShortMemory |
| 2 | +from tools import get_orders_count, get_sales_total, render_report |
| 3 | + |
| 4 | +DEFAULT_PREFS = { |
| 5 | + "report_format": "default", |
| 6 | + "currency": "USD", |
| 7 | +} |
| 8 | + |
| 9 | + |
| 10 | +def save_user_preferences( |
| 11 | + *, |
| 12 | + user_key: str, |
| 13 | + prefs: dict[str, str], |
| 14 | + short_memory: ShortMemory, |
| 15 | + long_memory: LongMemoryStore, |
| 16 | +) -> None: |
| 17 | + short_memory.add("user", f"Save prefs: {prefs}") |
| 18 | + long_memory.save_prefs(user_key, prefs) |
| 19 | + short_memory.add("assistant", "Preferences saved to long-term memory") |
| 20 | + |
| 21 | + |
| 22 | +def parse_prefs_from_short_memory(short_memory: ShortMemory) -> dict[str, str]: |
| 23 | + # Simplified parser for learning: looks for lines like "pref:key=value". |
| 24 | + parsed: dict[str, str] = {} |
| 25 | + for item in short_memory.snapshot(): |
| 26 | + content = item["content"] |
| 27 | + if "pref:" not in content: |
| 28 | + continue |
| 29 | + payload = content.split("pref:", 1)[1] |
| 30 | + if "=" not in payload: |
| 31 | + continue |
| 32 | + key, value = payload.split("=", 1) |
| 33 | + parsed[key.strip()] = value.strip() |
| 34 | + return parsed |
| 35 | + |
| 36 | + |
| 37 | +def build_weekly_report( |
| 38 | + *, |
| 39 | + user_id: int, |
| 40 | + user_key: str, |
| 41 | + request: str, |
| 42 | + short_memory: ShortMemory, |
| 43 | + long_memory: LongMemoryStore, |
| 44 | + use_long_memory: bool, |
| 45 | +) -> dict: |
| 46 | + trace: list[str] = [] |
| 47 | + |
| 48 | + short_memory.add("user", request) |
| 49 | + trace.append(f"request={request}") |
| 50 | + |
| 51 | + short_prefs = parse_prefs_from_short_memory(short_memory) |
| 52 | + trace.append(f"short_prefs={short_prefs}") |
| 53 | + |
| 54 | + long_prefs = long_memory.load_prefs(user_key) if use_long_memory else {} |
| 55 | + trace.append(f"long_prefs={long_prefs}") |
| 56 | + |
| 57 | + prefs = {**DEFAULT_PREFS, **short_prefs, **long_prefs} |
| 58 | + trace.append(f"resolved_prefs={prefs}") |
| 59 | + |
| 60 | + total = get_sales_total(user_id) |
| 61 | + orders = get_orders_count(user_id) |
| 62 | + |
| 63 | + report = render_report( |
| 64 | + total=total, |
| 65 | + orders=orders, |
| 66 | + currency=prefs["currency"], |
| 67 | + report_format=prefs["report_format"], |
| 68 | + ) |
| 69 | + |
| 70 | + short_memory.add("assistant", f"Report generated with prefs={prefs}") |
| 71 | + |
| 72 | + return { |
| 73 | + "prefs": prefs, |
| 74 | + "report": report, |
| 75 | + "trace": trace, |
| 76 | + "short_memory_snapshot": short_memory.snapshot(), |
| 77 | + } |
0 commit comments