Hitonabi
47f7a85510
Die Ampel-Nachruestung deckte 317/337 vorbestehende ruff-Verstoesse im ganzen Repo auf. Aufgeraeumt: - ruff.toml: intentionale Muster als Projekt-Politik ausgenommen (BLE001 blind-except, S110/S112 try-except-pass/continue, PLW1510 subprocess-best-effort, B008 FastAPI- Depends/File-Idiom, EXE001 Shebang, + wenige Stil-Regeln). __init__.py-Re-Exports geschuetzt (F401). - ruff --fix: 128 mechanische (Import-Sortierung, PEP585/604-Annotationen, tote Imports, ueberfluessige noqa) auto-behoben. - 12 echte Reste von Hand: PERF402/102, PLC3002 (Lambda->walrus), ISC004 (String-Concat geklammert), F841/RUF059 (ungenutzte Vars), PIE810 (startswith-Tuple), UP031 (f-string), UP035 (veraltete typing-Imports). Ergebnis: 'ruff check .' = 0, 'compileall' grün. Kein Verhaltenswechsel (nur Stil/Modernisierung). 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
|
|
|
|
_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)
|