refactor(ai): OCR sidecar canonical naming cleanup — typhoon→np-dms, remove hardcoded keys, asyncio.to_thread, ADR-040/041

2026-06-20 16:37:04 +07:00
parent d418d791a4
commit a80ebef285
70 changed files with 5762 additions and 452 deletions
@@ -0,0 +1,49 @@
+# File: tests/integration/ocr-sidecar/test_cpu_fallback.py
+# Change Log:
+# - 2026-06-20: Added ADR-040 CPU fallback integration coverage for retrieval endpoints.
+
+from pathlib import Path
+from unittest.mock import MagicMock, patch
+
+from fastapi.testclient import TestClient
+
+import sys
+
+UNIT_DIR = Path(__file__).resolve().parents[2] / "unit" / "ocr-sidecar"
+if str(UNIT_DIR) not in sys.path:
+    sys.path.insert(0, str(UNIT_DIR))
+
+from test_path_traversal import load_app
+
+
+def test_embed_uses_cpu_when_vram_headroom_is_low(tmp_path: Path) -> None:
+    app_module = load_app(tmp_path)
+    client = TestClient(app_module.app)
+    bge_model = MagicMock()
+    bge_model.encode.return_value = {
+        "dense_vecs": [[0.1, 0.2]],
+        "lexical_weights": [{"101": 0.5}],
+    }
+    headroom = MagicMock(total_mb=16384.0, used_mb=15000.0, available_mb=1000.0, query_success=True)
+    with patch.object(app_module, "bge_model", bge_model), patch.object(app_module, "get_vram_headroom", return_value=headroom):
+        response = client.post("/embed", json={"text": "hello"}, headers={"X-API-Key": "test-key"})
+    assert response.status_code == 200
+    assert response.json()["device"] == "cpu"
+    bge_model.model.to.assert_called_with("cpu")
+
+
+def test_rerank_uses_cpu_when_vram_headroom_is_low(tmp_path: Path) -> None:
+    app_module = load_app(tmp_path)
+    client = TestClient(app_module.app)
+    reranker = MagicMock()
+    reranker.compute_score.return_value = [0.9]
+    headroom = MagicMock(total_mb=16384.0, used_mb=15000.0, available_mb=1000.0, query_success=True)
+    with patch.object(app_module, "reranker", reranker), patch.object(app_module, "get_vram_headroom", return_value=headroom):
+        response = client.post(
+            "/rerank",
+            json={"query": "q", "chunks": ["chunk"]},
+            headers={"X-API-Key": "test-key"},
+        )
+    assert response.status_code == 200
+    assert response.json()["device"] == "cpu"
+    reranker.model.to.assert_called_with("cpu")