feat(maria): fir de discutie — continuarea nu mai e tratata ca eroare noua

Fiecare mesaj pornea de la zero. Masurat pe indexul viu, dupa o escaladare pentru
ORA-06550: "da, ma blocheaza complet" primea raspuns despre ordinul de plata la
Trezorerie (cosinus 0,642, "acoperit"), "eram la salvarea unei facturi" despre
corectia unei eFacturi (0,742), iar "am incercat si tot nu merge" cerea din nou
detaliile tocmai date. Patru cuvinte fara context seamana cu ceva din documente,
iar cautarea nu avea de unde sti ca sunt raspunsul la intrebarea Mariei.

rag/fir.py tine ancora (textul erorii), codurile ei, referinta escaladarii si
ultimele 6 schimburi, in ~/.maria-bridge/conversations/ (expira, ca si capturile).
La o continuare: cautarea e pe ancora + mesajul nou, modelul primeste istoricul,
mesajul se adauga la escaladarea deschisa de cate ori e nevoie, iar "mesaj prea
vag" nu se mai aplica. Escaladarea duce firul intreg la suport, nu un mesaj rupt
din context.

Firul se rupe doar la o captura noua sau un cod de eroare diferit — schimbarea
subiectului in cuvinte e prea usor de confundat cu o continuare.

In plus:
- Maria tace 60 min cand preia un om, dar automat doar in grupuri cu >= 2
  participanti: in self-chat totul e `fromMe`. Comenzi explicite oriunde
  ("Maria, stop" / "Maria, continua").
- ALLOWED_GROUP_JIDS: testarea se muta in grupul "Maria Test"
  (120363409761730101@g.us), ca sa nu mai poluam chatul "Eu". Filtru si in punte,
  si in consumer. NU echo-test: puntea lui Echo nu filtreaza fromMe in grupuri,
  deci cei doi boti ar intra in bucla.

Teste: 94 pass. Calibrare 23/23, cu doua cazuri de continuare.

Co-Authored-By: Claude Opus 5 (1M context) <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Q4uzvgm7AyJch5WH8QHRhY
This commit is contained in:
Claude Agent
2026-09-01 06:16:21 +00:00
parent 5342875aa2
commit 4d6ff1940f
9 changed files with 475 additions and 82 deletions

View File

@@ -41,6 +41,15 @@ DEFAULTS: dict[str, str] = {
"MAX_TOKENS": "250",
"POLL_INTERVAL_S": "2",
"TEST_MODE_SELF_CHAT_ONLY": "true",
# Grupuri in care Maria are voie sa raspunda (JID-uri separate prin virgula).
# Restul grupurilor sunt ignorate, si de punte, si de consumer.
"ALLOWED_GROUP_JIDS": "",
# Firul de discutie (vezi rag/fir.py): cat tine minte, cat tace dupa ce a
# preluat un om, si de cati participanti e nevoie ca preluarea sa se aplice
# automat (in self-chat si in grupul de test totul e `fromMe`).
"FIR_TTL_MIN": "120",
"FIR_TACERE_MIN": "60",
"FIR_PRELUARE_MIN_PARTICIPANTI": "2",
# OCR pentru capturile de ecran primite pe WhatsApp (vezi rag/ocr.py).
"OCR_LANGS": "ron+eng",
"OCR_MAX_CHARS": "1500",

View File

@@ -25,6 +25,7 @@ import time
import requests
import config
import fir as fir_mod
import ocr
import rank
import triaj
@@ -75,6 +76,10 @@ ESCALATED_RECORDED = (
)
TACERE_OK = (
'Am inteles, ma opresc. Scrie-mi "Maria, continua" cand vrei sa reiau.'
)
REVENIRE_OK = "Sunt aici. Cu ce te ajut?"
FOLLOWUP_OK = (
"Multumesc, am adaugat si asta la {ref} — echipa vede detaliul cand preia "
"problema."
@@ -154,17 +159,25 @@ def search(index: Index, query: str, top_k: int) -> dict:
}
def ask_llm(chunks: list[str], question: str) -> str:
def ask_llm(chunks: list[str], question: str, istoric: list[dict] | None = None) -> str:
"""Raspunsul modelului. `istoric` = schimburile de pana acum, la o continuare.
Fara istoric, „si acum ce fac?" ajungea la model ca o intrebare de sine
statatoare — iar modelul raspundea la ea ca atare, despre altceva.
"""
context = "\n\n---\n\n".join(chunks) if chunks else "(fara documente indexate)"
user_message = f"CONTEXT:\n{context}\n\nINTREBARE:\n{question}"
mesaje = [{"role": "system", "content": SYSTEM_PROMPT}]
# Istoricul precede contextul: ultimul mesaj trebuie sa fie intrebarea curenta,
# altfel modelul raspunde la penultima.
mesaje += (istoric or [])[:-1]
mesaje.append({"role": "user", "content": user_message})
resp = requests.post(
f"{config.get('LLM_URL')}/v1/chat/completions",
json={
"messages": [
{"role": "system", "content": SYSTEM_PROMPT},
{"role": "user", "content": user_message},
],
"messages": mesaje,
"max_tokens": config.get_int("MAX_TOKENS", 250),
},
timeout=60,
@@ -270,7 +283,7 @@ def reference(when: float, message_id: str | None) -> str:
return f"M-{time.strftime('%y%m%d', time.localtime(when))}-{sufix}"
def escalate(msg: dict, question: str, search_result: dict) -> dict:
def escalate(msg: dict, question: str, search_result: dict, fir: dict | None = None) -> dict:
"""Trimite intrebarea la suport si o inregistreaza in jurnal. Intoarce inregistrarea.
Jurnalul se scrie INTOTDEAUNA, si cand notificarea esueaza sau `SUPPORT_JID`
@@ -295,15 +308,19 @@ def escalate(msg: dict, question: str, search_result: dict) -> dict:
"had_image": bool(media.get("path")),
"notified": False,
}
if fir and fir.get("schimburi"):
record["fir"] = fir["schimburi"]
jid = config.get("SUPPORT_JID") or ""
if jid:
cine = msg.get("pushName") or msg.get("from") or "necunoscut"
discutie = fir_mod.rezumat(fir) if fir else ""
rezumat = (
f"[Maria] Intrebare fara raspuns in documente — {record['ref']}\n"
f"De la: {cine}\n"
f"Motiv: {record['reason']}\n\n"
f"{(msg.get('ocr_text') or msg.get('text') or '').strip()[:1200]}"
+ (f"\n\n--- discutia de pana acum ---\n{discutie}" if discutie else "")
)
try:
if media.get("path") and os.path.exists(media["path"]):
@@ -329,49 +346,6 @@ def escalate(msg: dict, question: str, search_result: dict) -> dict:
return record
PENDING_FILE_NAME = "pending.json"
PENDING_TTL_S = 30 * 60 # cat timp un raspuns scurt mai e „completare", nu intrebare noua
def _pending_path():
return config.STATE_DIR / "escalations" / PENDING_FILE_NAME
def _pending_all() -> dict:
try:
return json.loads(_pending_path().read_text(encoding="utf-8"))
except (OSError, ValueError):
return {}
def _pending_write(data: dict) -> None:
try:
_pending_path().parent.mkdir(parents=True, exist_ok=True)
_pending_path().write_text(json.dumps(data, ensure_ascii=False), encoding="utf-8")
except OSError as exc:
print(f"[consumer] nu pot scrie {PENDING_FILE_NAME}: {exc}", file=sys.stderr)
def pending_set(sender: str, ref: str) -> None:
"""Retine ca l-am intrebat pe om cat e de urgent, ca sa stiu unde duce raspunsul."""
data = _pending_all()
data[sender] = {"ref": ref, "at": time.time()}
_pending_write(data)
def pending_get(sender: str) -> dict | None:
intrare = _pending_all().get(sender)
if not intrare or time.time() - intrare.get("at", 0) > PENDING_TTL_S:
return None
return intrare
def pending_clear(sender: str) -> None:
data = _pending_all()
if data.pop(sender, None) is not None:
_pending_write(data)
def append_followup(ref: str, msg: dict, text: str) -> bool:
"""Adauga raspunsul omului la escaladarea deschisa si il trimite la suport.
@@ -408,6 +382,44 @@ def append_followup(ref: str, msg: dict, text: str) -> bool:
return True
_GRUPURI_CACHE: dict = {"at": 0.0, "date": {}}
GRUPURI_TTL_S = 600
def grup_info(jid: str) -> dict:
"""Cate persoane sunt in grup. Raspunsul se tine 10 minute in memorie."""
if time.time() - _GRUPURI_CACHE["at"] > GRUPURI_TTL_S:
try:
resp = requests.get(f"{bridge_url()}/groups", timeout=15)
resp.raise_for_status()
_GRUPURI_CACHE["date"] = {g["jid"]: g for g in resp.json().get("groups", [])}
_GRUPURI_CACHE["at"] = time.time()
except Exception as exc: # noqa: BLE001
print(f"[consumer] nu pot citi grupurile: {exc}", file=sys.stderr)
return _GRUPURI_CACHE["date"].get(jid, {})
def preluare_de_om(msg: dict) -> bool:
"""Un om din echipa a scris in discutie, deci Maria se retrage.
Doar in grupuri cu mai multi oameni. In self-chat si in grupul de test (un
singur participant) TOT ce se scrie e `fromMe`: acolo regula ar face Maria sa
amuteasca la primul mesaj, deci ramane doar comanda explicita.
"""
if not msg.get("fromMe") or not msg.get("isGroup"):
return False
if (msg.get("text") or "").startswith(REPLY_PREFIX):
return False # propriul raspuns, nu un om
minim = config.get_int("FIR_PRELUARE_MIN_PARTICIPANTI", 2)
return grup_info(msg.get("from") or "").get("participants", 0) >= minim
def grupuri_permise() -> set[str]:
"""JID-urile de grup in care Maria are voie sa raspunda (ALLOWED_GROUP_JIDS)."""
brut = config.get("ALLOWED_GROUP_JIDS", "") or ""
return {j.strip() for j in brut.replace(";", ",").split(",") if j.strip()}
def cleanup_media(msg: dict) -> None:
"""Capturile pot contine date de client — nu raman pe disc dupa ce s-a tratat mesajul."""
path = (msg.get("media") or {}).get("path")
@@ -424,22 +436,50 @@ def handle(index: Index, msg: dict) -> None:
has_image = bool(msg.get("media"))
text = msg.get("text", "") or ""
print(f"[consumer] {sender}: {'[imagine] ' if has_image else ''}{text[:80]}", file=sys.stderr)
react_seen(sender, msg.get("id"), msg.get("fromMe", False))
# Raspunsul la „te blocheaza sau poti continua?" merge la escaladarea deschisa.
# Inainte de orice altceva: nu e o intrebare noua, deci nu se cauta si nu se
# confirma cu „caut informatia".
asteptat = pending_get(sender) if not has_image else None
if asteptat and text.strip() and not rank.codes(text):
if append_followup(asteptat["ref"], msg, text.strip()):
pending_clear(sender)
send_reply(sender, FOLLOWUP_OK.format(ref=asteptat["ref"]))
return
pending_clear(sender)
fir = fir_mod.incarca(sender)
# Comenzi explicite. Merg oriunde, si acolo unde regula automata de mai jos nu
# se aplica (self-chat, grup de test cu un singur om).
cmd = fir_mod.comanda(text)
if cmd and not has_image:
fir = fir or fir_mod.deschide(sender, "")
if cmd == "stop":
fir_mod.taci(fir)
fir_mod.salveaza(fir)
print(f"[consumer] {sender}: tac la cerere", file=sys.stderr)
send_reply(sender, TACERE_OK)
else:
fir_mod.vorbeste(fir)
fir_mod.salveaza(fir)
send_reply(sender, REVENIRE_OK)
return
# Un om din echipa a intrat in discutie: Maria nu vorbeste peste el.
if preluare_de_om(msg):
fir = fir or fir_mod.deschide(sender, "")
fir_mod.adauga(fir, "om", text)
fir_mod.taci(fir)
fir_mod.salveaza(fir)
print(f"[consumer] {sender}: a preluat un om, tac {fir_mod.tacere_s() // 60} min",
file=sys.stderr)
return
if fir_mod.tace(fir):
print(f"[consumer] {sender}: fir preluat de om, nu raspund", file=sys.stderr)
return
react_seen(sender, msg.get("id"), msg.get("fromMe", False))
continuare = fir_mod.este_continuare(fir, text, has_image)
# „Am o eroare", fara sa spuna care: nu ghicim si nu deranjam suportul — intrebam.
if triaj.prea_vag(text, has_image):
# Doar la deschiderea firului: la o continuare, detaliile au fost deja cerute.
if not continuare and triaj.prea_vag(text, has_image):
print("[consumer] mesaj prea vag -> cer detalii", file=sys.stderr)
fir = fir_mod.deschide(sender, text)
fir_mod.adauga(fir, "client", text)
fir_mod.adauga(fir, "maria", triaj.CERE_DETALII)
fir_mod.salveaza(fir)
send_reply(sender, triaj.CERE_DETALII)
return
@@ -452,6 +492,15 @@ def handle(index: Index, msg: dict) -> None:
if not question:
return
if continuare:
fir_mod.adauga(fir, "client", text)
search_query = fir_mod.interogare(fir, search_query)
print(f"[consumer] continuare pe firul deschis; caut dupa {search_query[:90]!r}",
file=sys.stderr)
else:
fir = fir_mod.deschide(sender, msg.get("ocr_text") or text)
fir_mod.adauga(fir, "client", text or "[captura de ecran]")
try:
result = search(index, search_query, config.get_int("TOP_K", 3))
except Exception as exc: # noqa: BLE001
@@ -462,11 +511,24 @@ def handle(index: Index, msg: dict) -> None:
surse = ", ".join(f"{t['source']}({t['cosine']})" for t in result["top"]) or "-"
print(f"[consumer] rank: {surse} | {result['reason']}", file=sys.stderr)
# Firul are deja o escaladare deschisa: ce spune omul acum e completare la ea,
# nu o problema noua. De cate ori e nevoie, nu o singura data.
if continuare and fir.get("ref") and append_followup(fir["ref"], msg, text.strip() or "(captura)"):
raspuns = FOLLOWUP_OK.format(ref=fir["ref"])
fir_mod.adauga(fir, "maria", raspuns)
fir_mod.salveaza(fir)
send_reply(sender, raspuns)
return
if not result["covered"]:
print(f"[consumer] fara acoperire in documente -> suport ({result['reason']})", file=sys.stderr)
record = escalate(msg, question, result)
record = escalate(msg, question, result, fir)
sablon = ESCALATED_SENT if record["notified"] else ESCALATED_RECORDED
send_reply(sender, sablon.format(ref=record["ref"]))
raspuns = sablon.format(ref=record["ref"])
fir["ref"] = record["ref"]
fir_mod.adauga(fir, "maria", raspuns)
fir_mod.salveaza(fir)
send_reply(sender, raspuns)
return
# Eroare care nu se rezolva din aplicatie: raspunsul se compune din dictionar
@@ -474,18 +536,22 @@ def handle(index: Index, msg: dict) -> None:
nivel = triaj.nivel_suport(result["chunks"][0]) if result["chunks"] else None
if nivel:
print(f"[consumer] eroare de nivel '{nivel}' -> raspuns fix + suport", file=sys.stderr)
record = escalate(msg, question, result)
send_reply(sender, triaj.raspuns_escaladat(
result["chunks"][0], nivel, record["ref"], record["notified"],
cu_imagine=has_image))
pending_set(sender, record["ref"])
record = escalate(msg, question, result, fir)
raspuns = triaj.raspuns_escaladat(
result["chunks"][0], nivel, record["ref"], record["notified"], cu_imagine=has_image)
fir["ref"] = record["ref"]
fir_mod.adauga(fir, "maria", raspuns)
fir_mod.salveaza(fir)
send_reply(sender, raspuns)
return
try:
reply = ask_llm(result["chunks"], question)
reply = ask_llm(result["chunks"], question, fir_mod.istoric(fir) if continuare else None)
except Exception as exc: # noqa: BLE001
print(f"[consumer] LLM error: {exc}", file=sys.stderr)
reply = "Scuze, am o problema tehnica momentan. Cineva din echipa te va contacta."
fir_mod.adauga(fir, "maria", reply)
fir_mod.salveaza(fir)
send_reply(sender, reply)
print(f"[consumer] -> raspuns catre {sender}", file=sys.stderr)
@@ -509,7 +575,7 @@ def main() -> None:
resp = requests.get(f"{bridge_url()}/messages", timeout=10)
resp.raise_for_status()
for msg in resp.json().get("messages", []):
if msg.get("isGroup"):
if msg.get("isGroup") and msg.get("from") not in grupuri_permise():
continue
if (msg.get("text") or "").startswith(REPLY_PREFIX):
continue # ecoul propriului raspuns in self-chat, ignorat

View File

@@ -0,0 +1,160 @@
#!/usr/bin/env python3
"""Firul de discutie: al doilea mesaj al omului e continuarea primului.
Fara asta, fiecare mesaj pornea de la zero. Masurat pe indexul viu, dupa o
escaladare pentru ORA-06550:
„da, ma blocheaza complet" -> raspuns despre ordinul de plata la Trezorerie
„eram la salvarea unei facturi" -> raspuns despre corectia unei eFacturi
„si acum ce fac?" -> raspuns despre eFacturi primite gresit
Toate trei „acoperite", cu cosinus peste prag: patru cuvinte fara context seamana
cu ceva din documente, iar cautarea nu are de unde sti ca sunt raspunsul la
intrebarea Mariei, nu o intrebare noua.
Ce tine firul: ancora (textul erorii care l-a deschis), codurile ei, referinta
escaladarii deschise si ultimele schimburi. Ancora intra in interogarea de cautare
la fiecare continuare — asa „da, ma blocheaza" se cauta impreuna cu eroarea, nu
singur.
Ce rupe firul, necontestat: o captura noua sau un cod de eroare diferit. Sunt
singurele semne sigure ca omul a trecut la altceva; restul (schimbarea subiectului
in cuvinte) e prea usor de confundat cu o continuare, iar greseala aia produce
exact raspunsurile de mai sus.
Firele contin text de client (uneori si OCR de pe capturi), deci expira si se sterg
singure, ca si capturile.
"""
from __future__ import annotations
import json
import re
import time
import config
import rank
MAX_SCHIMBURI = 6 # cate replici se tin minte (ale omului si ale Mariei, la un loc)
def ttl_s() -> int:
"""Cat timp de tacere inseamna ca discutia s-a incheiat (FIR_TTL_MIN)."""
return config.get_int("FIR_TTL_MIN", 120) * 60
def tacere_s() -> int:
"""Cat sta Maria deoparte dupa ce a preluat un om (FIR_TACERE_MIN)."""
return config.get_int("FIR_TACERE_MIN", 60) * 60
def _dir():
return config.STATE_DIR / "conversations"
def _cale(jid: str):
return _dir() / (re.sub(r"[^0-9a-zA-Z@._-]", "_", jid) + ".json")
def incarca(jid: str) -> dict | None:
"""Firul deschis cu cineva, sau `None` daca nu exista ori a expirat."""
cale = _cale(jid)
try:
fir = json.loads(cale.read_text(encoding="utf-8"))
except (OSError, ValueError):
return None
if time.time() - fir.get("at", 0) > ttl_s():
cale.unlink(missing_ok=True)
return None
return fir
def salveaza(fir: dict) -> None:
fir["at"] = time.time()
try:
_dir().mkdir(parents=True, exist_ok=True)
_cale(fir["jid"]).write_text(json.dumps(fir, ensure_ascii=False), encoding="utf-8")
except OSError:
pass # firul e o comoditate, nu o obligatie: fara el se comporta ca inainte
def sterge(jid: str) -> None:
_cale(jid).unlink(missing_ok=True)
def deschide(jid: str, ancora: str) -> dict:
return {"jid": jid, "at": time.time(), "ancora": ancora.strip(),
"coduri": rank.codes(ancora), "ref": None, "tacere_pana": 0, "schimburi": []}
def adauga(fir: dict, rol: str, text: str) -> None:
fir.setdefault("schimburi", []).append({"rol": rol, "text": text.strip()[:1500],
"at": time.time()})
del fir["schimburi"][:-MAX_SCHIMBURI]
def este_continuare(fir: dict | None, text: str, are_imagine: bool) -> bool:
"""Mesajul duce mai departe firul deschis, sau incepe altceva?
O captura noua sau un cod pe care firul nu-l are inseamna subiect nou. In rest
— inclusiv la un mesaj lung — e continuare: omul raspunde la ce l-a intrebat
Maria, iar ancora nu strica o cautare care oricum are destul text al ei.
"""
if not fir:
return False
if are_imagine:
return False
noi = rank.codes(text)
if noi and any(c not in (fir.get("coduri") or []) for c in noi):
return False
return True
def interogare(fir: dict, text: str) -> str:
"""Ce se cauta in index la o continuare: ancora firului plus mesajul nou."""
ancora = (fir.get("ancora") or "").strip()
if not ancora:
return text
return f"{ancora}\n{text.strip()}".strip()
def istoric(fir: dict) -> list[dict]:
"""Schimburile de pana acum, in formatul de chat al modelului."""
return [{"role": "assistant" if s["rol"] == "maria" else "user", "content": s["text"]}
for s in fir.get("schimburi", [])]
def rezumat(fir: dict, maxim: int = 900) -> str:
"""Firul, ca text, pentru mesajul catre suport."""
etichete = {"maria": "Maria", "om": "Suport"}
linii = [f"{etichete.get(s['rol'], 'Client')}: {s['text']}"
for s in fir.get("schimburi", [])]
return "\n".join(linii)[-maxim:]
# --- cand preia un om -------------------------------------------------------
_STOP = re.compile(r"^\s*(maria[, ]+)?(stop|pauza|taci|preiau|ma ocup eu)\b")
_START = re.compile(r"^\s*maria[, ]+(start|continua|revino|poti)\b")
def comanda(text: str) -> str | None:
"""`"stop"` / `"start"` / `None` — comenzi explicite de oprire si repornire."""
t = rank.fold(text or "")
if _START.match(t):
return "start"
if _STOP.match(t):
return "stop"
return None
def taci(fir: dict, secunde: int | None = None) -> None:
fir["tacere_pana"] = time.time() + (tacere_s() if secunde is None else secunde)
def vorbeste(fir: dict) -> None:
fir["tacere_pana"] = 0
def tace(fir: dict | None) -> bool:
return bool(fir) and time.time() < (fir.get("tacere_pana") or 0)