Hitonabi
46f108b6f3
Schliesst den offenen "Update"-Punkt aus dem Upgrade-Playbook (Backup war Schicht 1): - mem0_service/requirements.txt jetzt GEPINNT (mem0ai==2.0.8, chromadb==1.5.9, fastapi==0.138.1, uvicorn==0.49.0) -> Upgrades nur absichtlich. Grund: der Sidecar nutzt mem0-Interna (NoThink-Swap, Roh-Chroma-Zugriff), die ein blindes Upgrade still brechen koennte. - mem0_service/smoke_test.py: prueft add/Suche/Auto-Lernen/Deutsch/Auto-Einordnung/Graph gegen eine WEGWERF-Collection (fasst /srv/models/mem0 nicht an). Vor/nach mem0-Upgrades laufen lassen. Live: alle Tests gruen. - docs/UPGRADE.md: sichere Upgrade-Prozedur (Backup -> Pin -> install -> smoke_test -> gruen=restart / rot=anpassen oder restore) + reagraph/React- und Hermes-Plugin-Hinweise. Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
72 lines
2.8 KiB
Python
72 lines
2.8 KiB
Python
#!/usr/bin/env python3
|
|
"""
|
|
Smoke-Test für den Mem0-Sidecar — VOR und NACH jedem mem0/chromadb-Upgrade laufen lassen.
|
|
Prüft genau die Pfade, die mem0-INTERNA nutzen (LLM-Extraktion/NoThink, Auto-Einordnung,
|
|
Embeddings/Suche, Roh-Chroma-Graph). Nutzt eine WEGWERF-Collection unter /tmp — das echte
|
|
Gedächtnis (/srv/models/mem0) wird NICHT angefasst.
|
|
|
|
Aufruf: ~/.mem0/venv/bin/python ~/mission-control-v2/mem0_service/smoke_test.py
|
|
Exit 0 = alles grün, sonst 1.
|
|
"""
|
|
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
|
|
# WICHTIG: Env vor dem Import von app setzen (app liest die Pfade beim Modul-Load).
|
|
_TMP = tempfile.mkdtemp(prefix="mem0_smoke_")
|
|
os.environ["MEM0_CHROMA_PATH"] = os.path.join(_TMP, "chroma")
|
|
os.environ["MEM0_HISTORY_DB"] = os.path.join(_TMP, "history.db")
|
|
os.environ["MEM0_COLLECTION"] = f"smoke_{int(time.time())}"
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
|
|
import app # noqa: E402
|
|
|
|
_fails: list[str] = []
|
|
|
|
|
|
def check(name: str, cond: bool, detail: str = "") -> None:
|
|
print(("PASS" if cond else "FAIL"), "—", name, ("" if cond else f":: {detail}"))
|
|
if not cond:
|
|
_fails.append(name)
|
|
|
|
|
|
def main() -> int:
|
|
app.mem() # Memory bauen (öffnet Chroma, pingt Modelle)
|
|
|
|
# 1) Verbatim add + Rücklesen
|
|
r = app.add_memory(app.MemIn(content="Smoke: der Himmel ist blau.", category="knowledge", source="smoke"))
|
|
check("verbatim add + get", bool(r.get("id")) and "blau" in r.get("content", ""), str(r)[:160])
|
|
|
|
# 2) Embeddings / semantische Suche (Bedeutung, nicht Wort)
|
|
res = app.list_memory(q="Welche Farbe hat der Himmel?")
|
|
check("semantische Suche", any("blau" in i.get("content", "") for i in res), str(res)[:160])
|
|
|
|
# 3) Auto-Lernen: LLM-Extraktion (NoThink) + Deutsch erzwungen
|
|
lr = app.learn(app.LearnIn(text="Ich heiße Max und programmiere am liebsten in Go.", source="smoke"))
|
|
items = lr.get("results", [])
|
|
check("Auto-Lernen extrahiert Fakten", len(items) > 0, str(lr)[:160])
|
|
joined = " ".join((i.get("content") or "") for i in items).lower()
|
|
check("Extraktion auf Deutsch", any(w in joined for w in ("heißt", "programmiert", "mag", "bevorzugt", "name", "nutzt")), joined[:160])
|
|
|
|
# 4) Auto-Einordnung in die 4 gültigen Kategorien
|
|
cats = {i.get("category") for i in items if i.get("category")}
|
|
check("Auto-Einordnung gültige Kategorie", bool(cats) and cats <= set(app.CATEGORIES), str(cats))
|
|
|
|
# 5) Graph aus den Roh-Embeddings (Chroma-Interna)
|
|
g = app.graph()
|
|
check("Graph liefert Knoten", len(g.get("nodes", [])) >= 1, str({k: len(v) for k, v in g.items()}))
|
|
|
|
print("\n" + ("ALLE TESTS GRÜN ✓" if not _fails else f"FEHLGESCHLAGEN: {_fails}"))
|
|
return 0 if not _fails else 1
|
|
|
|
|
|
if __name__ == "__main__":
|
|
try:
|
|
code = main()
|
|
finally:
|
|
shutil.rmtree(_TMP, ignore_errors=True)
|
|
sys.exit(code)
|