Spaces:
Runtime error
Runtime error
| from __future__ import annotations | |
| from pathlib import Path | |
| import sys | |
| import pytest | |
| ROOT = Path(__file__).resolve().parents[1] | |
| SRC = ROOT / "src" | |
| if str(SRC) not in sys.path: | |
| sys.path.insert(0, str(SRC)) | |
| from apps.p5_memory_quilt import llm_backend | |
| from apps.p5_memory_quilt.prompt_rewriter import rewrite_memory | |
| def test_rewrite_memory_uses_llm_backend_and_records_inference_meta(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None: | |
| checkpoint = tmp_path / "checkpoint" | |
| checkpoint.mkdir() | |
| monkeypatch.setenv("P5_MEMORY_QUILT_PRIMARY_MODEL_PATH", str(checkpoint)) | |
| monkeypatch.delenv("P5_MEMORY_QUILT_FALLBACK_MODEL_PATH", raising=False) | |
| captured: dict[str, object] = {} | |
| def fake_try_rewrite(memory_text: str, location_tag: str, style: str, cfg: llm_backend.LLMConfig): | |
| captured["memory_text"] = memory_text | |
| captured["location_tag"] = location_tag | |
| captured["style"] = style | |
| captured["cfg"] = cfg | |
| return ( | |
| { | |
| "caption": "Corner Store", | |
| "story": "A stitched memory of the corner store.", | |
| "flux_prompt": "fabric quilt scene of a corner store at soft evening light", | |
| "style_hint": "stitched fabric texture", | |
| }, | |
| { | |
| "adapter_name": "local-transformers", | |
| "backend": "local-transformers", | |
| "model_id": cfg.model_id, | |
| "checkpoint_path": cfg.checkpoint_path, | |
| "generation_stats": { | |
| "prompt_tokens": 11, | |
| "generated_tokens": 22, | |
| "elapsed_ms": 12.5, | |
| }, | |
| }, | |
| ) | |
| monkeypatch.setattr(llm_backend, "try_rewrite", fake_try_rewrite) | |
| result = rewrite_memory("Every Friday the tamale cart parked outside the blue house.", "corner store", "Fabric Quilt") | |
| assert captured["memory_text"] == "Every Friday the tamale cart parked outside the blue house." | |
| assert captured["location_tag"] == "corner store" | |
| assert captured["style"] == "Fabric Quilt" | |
| assert isinstance(captured["cfg"], llm_backend.LLMConfig) | |
| assert result.selected_model_id == "jetbrains/Mellm-2-12B-Instruct" | |
| assert result.checkpoint_path == str(checkpoint) | |
| assert result.prompt_source == "local-transformers" | |
| assert result.caption == "Corner Store" | |
| assert result.story == "A stitched memory of the corner store." | |
| assert result.flux_prompt.startswith("fabric quilt scene") | |
| assert result.inference_meta["model_id"] == "jetbrains/Mellm-2-12B-Instruct" | |
| assert result.inference_meta["checkpoint_source"] == "primary-checkpoint" | |
| assert result.inference_meta["generation_stats"]["generated_tokens"] == 22 | |
| def test_rewrite_memory_rejects_empty_memory_text() -> None: | |
| with pytest.raises(ValueError, match="memory_text"): | |
| rewrite_memory("", "corner store", "Fabric Quilt") | |
| def test_rewrite_memory_rejects_unknown_style(tmp_path: Path, monkeypatch: pytest.MonkeyPatch) -> None: | |
| checkpoint = tmp_path / "checkpoint" | |
| checkpoint.mkdir() | |
| monkeypatch.setenv("P5_MEMORY_QUILT_PRIMARY_MODEL_PATH", str(checkpoint)) | |
| monkeypatch.delenv("P5_MEMORY_QUILT_FALLBACK_MODEL_PATH", raising=False) | |
| with pytest.raises(ValueError, match="Unknown style"): | |
| rewrite_memory("Every Friday the tamale cart parked outside the blue house.", "corner store", "Not A Style") | |