Files
ThothII/harness/tests/l2/test_memory_save_one_real.py
T
marcopanandClaude Fable 5 c4951e2aa5 chore: hygiene pass — ruff clean, docs storage-model truth, replay /me, failSession log
Audit findings 6.1-6.4 + the audit's remediation plan itself
(docs/superpowers/plans/2026-07-20-full-audit-remediation-plan.md).

- ruff: 34 → 0 (unused imports/f-strings auto-fixed; E702 semicolon lines
  split in test files; one unused local dropped). Suite still 819 green.
- CLAUDE.md + PROJECT_STATE.md no longer claim "no database / settings in
  settings.json": the harness selects filesystem OR PostgreSQL session
  storage (repository.py, server mode), and settings flow through harness
  preferences with the JSON file as fallback only.
- tools/replay: stub /me (SPA boot was parsing the SPA's own HTML as JSON)
  and /runtime/prewarm.
- failSession best-effort persistence now logs its failure server-side
  instead of vanishing.

Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
2026-07-20 01:55:41 +02:00

53 lines
2.3 KiB
Python

"""L2: tht memory save-one against real pgvector (spec D11, L2).
Validates D11 end-to-end: a single promoted decision is upserted to the real
pgvector via the WRITER key (not a full resync), and a subsequent search_similar
finds the memory. L1 tested the pure save_one_memory core; here the REST writer +
real pgvector + real embeddings are in the loop.
Run: pytest -m l2 tests/l2/test_memory_save_one_real.py -s (needs .env + VPN + Ollama)
"""
from datetime import datetime
from pathlib import Path
import pytest
from tht.memory import MemoryRecord, save_one_memory
from tht.workspace import load_workspace
pytestmark = [pytest.mark.l2]
WORKSPACE = Path(__file__).resolve().parents[2] / "workspaces" / "tht-test.yaml"
def test_save_one_upserts_to_real_pgvector(l2_env):
"""save_one_memory pushes one row to the real pgvector via the writer key, and
a subsequent search_similar retrieves it. Idempotent (re-running upserts >= 0)."""
from tht.vectorstore.embeddings import OllamaEmbeddings
from tht.vectorstore.rest_client import VectorRestClient
ws = load_workspace(WORKSPACE)
if not ws.vector_write_rest or not ws.vector_write_rest.api_key.strip():
pytest.skip("vector_write_rest not configured (no writer key)")
writer = VectorRestClient(ws.vector_write_rest)
embedder = OllamaEmbeddings(ws.embeddings)
record = MemoryRecord(
id="mem-l2test", ts=datetime.now(), session_id="l2-self-test",
decision_seq=999, type="table_promoted", subject="fct_ricoveri",
detail="ablazione", rationale="L2 self-test (idempotent)",
question_context="ablazione 2025", tables=["fct_ricoveri"], concepts=[],
)
from tht.adapters.vector import ThothHttpVectorStore
store = ThothHttpVectorStore(reader=writer, writer=writer)
upserted = save_one_memory([record], decision_seq=999, store=store, embedder=embedder)
assert upserted >= 0 # idempotent: 0 on unchanged, >=1 on new/updated
# read it back via the READER key (vector_rest, path /vector/v1/)
reader = VectorRestClient(ws.vector_rest)
qvec = embedder.embed_query("ablazione")
hits = reader.search_similar("memory", qvec, 10)
ids = {h.get("metadata", {}).get("record_key", "") for h in hits}
assert "memory:mem-l2test" in ids, "upserted memory not retrievable via search_similar"