1"""
2context_builder_llm.py — LLM-саммари чатов и лично известных людей.
3 
4Логика:
5- Чаты: саммари чата + краткое досье на ВСЕХ участников сразу (один блок)
6- Люди: подробный профиль только если есть личка (chat_type=private)
7"""
8 
9import json
10import subprocess
11import time
12from pathlib import Path
13from collections import defaultdict
14from datetime import datetime, timezone
15 
16CONTEXT_DIR = Path(__file__).parent / "context"
17MESSAGES_FILE = CONTEXT_DIR / "messages.jsonl"
18CHATS_DIR = CONTEXT_DIR / "chats"
19PEOPLE_DIR = CONTEXT_DIR / "people"
20 
21CHATS_DIR.mkdir(parents=True, exist_ok=True)
22PEOPLE_DIR.mkdir(parents=True, exist_ok=True)
23 
24OWNER_ID = int(os.getenv("TELEGRAM_OWNER_ID", "0"))
25MAX_MSGS = 80
26PAUSE = 8
27 
28 
29def ask_claude(prompt: str) -> str:
30    time.sleep(PAUSE)
31    result = subprocess.run(
32        ["openclaw", "agent", "--agent", "main", "--message", prompt, "--json"],
33        capture_output=True, text=True, timeout=300
34    )
35    try:
36        data = json.loads(result.stdout)
37        return data["result"]["payloads"][0]["text"]
38    except Exception:
39        return result.stdout.strip() or "ошибка"
40 
41 
42def fmt_msgs(msgs: list) -> str:
43    lines = []
44    for m in msgs[-MAX_MSGS:]:
45        d = "→" if m.get("direction") == "out" else "←"
46        name = m.get("sender_name", "?")
47        text = (m.get("text") or "").strip()
48        if not text:
49            continue
50        ts = m.get("ts", "")[:10]
51        reply = f" [↩ {m['reply_to_text'][:40]}]" if m.get("reply_to_text") else ""
52        lines.append(f"{d} {name} [{ts}]: {text}{reply}")
53    return "\n".join(lines)
54 
55 
56def load_messages():
57    if not MESSAGES_FILE.exists():
58        return []
59    msgs = []
60    with open(MESSAGES_FILE, encoding="utf-8") as f:
61        for line in f:
62            try:
63                msgs.append(json.loads(line))
64            except Exception:
65                pass
66    return msgs
67 
68 
69def main():
70    print("[llm_builder] Загружаю...")
71    msgs = load_messages()
72    if not msgs:
73        print("[llm_builder] Нет сообщений.")
74        return
75    print(f"[llm_builder] {len(msgs)} записей")
76 
77    # Группируем по чатам
78    chats = defaultdict(lambda: {"title": "", "type": "", "msgs": []})
79    for m in msgs:
80        cid = m.get("chat_id")
81        chats[cid]["title"] = m.get("chat_title", str(cid))
82        chats[cid]["type"] = m.get("chat_type", "")
83        chats[cid]["msgs"].append(m)
84 
85    # Находим личные чаты (не с собой)
86    private_chat_ids = set()
87    for cid, info in chats.items():
88        if info["type"] == "private" and cid != OWNER_ID:
89            private_chat_ids.add(cid)
90 
91    updated = datetime.now(timezone.utc).strftime("%Y-%m-%d %H:%M UTC")
92 
93    # Саммари групповых чатов
94    for cid, info in chats.items():
95        if info["type"] != "group":
96            continue
97        if len(info["msgs"]) < 5:
98            continue
99 
100        print(f"[llm_builder] Чат: {info['title']}")
101        text = fmt_msgs(info["msgs"])
102 
103        # Собираем участников (не owner)
104        participants = defaultdict(list)
105        for m in info["msgs"]:
106            sid = m.get("sender_id")
107            if sid and sid != OWNER_ID:
108                name = m.get("sender_name", str(sid))
109                t = (m.get("text") or "").strip()
110                if t:
111                    participants[name].append(t[:100])
112 
113        # Краткие характеристики участников
114        people_summary = ""
115        if participants:
116            people_lines = []
117            for name, texts in participants.items():
118                sample = " / ".join(texts[:3])
119                people_lines.append(f"- {name}: «{sample}»")
120            people_summary = "\n".join(people_lines)
121 
122        prompt = f"""Чат "{info['title']}" (тип: {info['type']}). Последние сообщения (→ outgoing (owner), ← incoming):
123 
124{text}
125 
126Участники и примеры их сообщений:
127{people_summary}
128 
129Дай структурированное резюме:
1301. О чём этот чат и какова его атмосфера (2-3 предложения)
1312. Краткое досье на каждого участника (1 строка: имя — кто такой, роль в чате)
1323. Актуальные темы на сейчас
133 
134Максимум 200 слов, без лишней воды."""
135 
136        summary = ask_claude(prompt)
137 
138        chat_file = CHATS_DIR / f"{cid}.md"
139        existing = chat_file.read_text(encoding="utf-8") if chat_file.exists() else ""
140        if "## LLM Анализ" in existing:
141            existing = existing[:existing.index("## LLM Анализ")]
142        chat_file.write_text(
143            existing.rstrip() + f"\n\n## LLM Анализ\n*{updated}*\n\n{summary}\n",
144            encoding="utf-8"
145        )
146 
147    # Профили людей только из личек
148    for cid, info in chats.items():
149        if cid not in private_chat_ids:
150            continue
151        if len(info["msgs"]) < 3:
152            continue
153 
154        # Находим собеседника
155        other_name = ""
156        other_id = cid  # для личек chat_id = user_id собеседника
157        for m in info["msgs"]:
158            sid = m.get("sender_id")
159            if sid and sid != OWNER_ID:
160                other_name = m.get("sender_name", str(cid))
161                other_id = sid
162                break
163 
164        if not other_name:
165            continue
166 
167        print(f"[llm_builder] Личка: {other_name}")
168        text = fmt_msgs(info["msgs"])
169 
170        prompt = f"""Личная переписка between owner (→) и {other_name} (←):
171 
172{text}
173 
174Составь подробный профиль {other_name}:
1751. Кто этот человек — возраст, характер, роль in owner's life
1762. Как они общаются, какие темы
1773. Что сейчас происходит у этого человека
1784. Relationship with owner — близость, история, эмоциональный тон
179 
180Максимум 150 слов."""
181 
182        summary = ask_claude(prompt)
183 
184        person_file = PEOPLE_DIR / f"{other_id}.md"
185        existing = person_file.read_text(encoding="utf-8") if person_file.exists() else ""
186        if "## LLM Анализ" in existing:
187            existing = existing[:existing.index("## LLM Анализ")]
188        person_file.write_text(
189            existing.rstrip() + f"\n\n## LLM Анализ (личка)\n*{updated}*\n\n{summary}\n",
190            encoding="utf-8"
191        )
192 
193    print("[llm_builder] Готово!")
194 
195 
196if __name__ == "__main__":
197    main()
198