2366104c4f
Matches #5910's text-only footprint. Removes image-vector embedding (encode_images, _stream_image_chunks, the _BGEVLAdapter CLIP shim), the multimodal `mode`/KBMode concept + VL embedders (single text embedder now), the mode selector UI across the KB dialogs + thread settings, the MM badges, the /images serving route, and the dead image rendering in the search tool card. Captioning (figure text spliced into markdown) stays — #5910 keeps it too. DB mode/image columns left dormant (no migration). RagDefaultsSection dropped (no controls left).
46 lines
1.7 KiB
Python
46 lines
1.7 KiB
Python
"""Reingest endpoint tests (Backfill UX).
|
|
|
|
Full end-to-end reingest needs a running studio + a real embedder; that's
|
|
covered manually via the curl smoke flow in the plan. Here we cover the
|
|
parts that are testable without external models: payload validation.
|
|
"""
|
|
|
|
import importlib.util
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
REPO_ROOT = Path(__file__).resolve().parents[2]
|
|
STUDIO_BACKEND = REPO_ROOT / "studio" / "backend"
|
|
if str(STUDIO_BACKEND) not in sys.path:
|
|
sys.path.insert(0, str(STUDIO_BACKEND))
|
|
|
|
|
|
def _rag_route():
|
|
"""Load ``routes/rag.py`` directly, bypassing the ``routes`` package.
|
|
|
|
``from routes.rag import X`` first runs ``routes/__init__.py``, which eagerly
|
|
imports every router — including the datasets router, whose chain does
|
|
``from datasets import IterableDataset`` at import time. On a GPU-less CI
|
|
runner the unsloth bootstrap can leave ``datasets`` half-initialized, so that
|
|
eager import raises. These tests only need pure helpers from rag.py, so load
|
|
the file on its own (it has no intra-``routes`` imports).
|
|
"""
|
|
mod = sys.modules.get("_rag_route_under_test")
|
|
if mod is None:
|
|
spec = importlib.util.spec_from_file_location(
|
|
"_rag_route_under_test", STUDIO_BACKEND / "routes" / "rag.py"
|
|
)
|
|
mod = importlib.util.module_from_spec(spec)
|
|
spec.loader.exec_module(mod)
|
|
sys.modules["_rag_route_under_test"] = mod
|
|
return mod
|
|
|
|
|
|
def test_reingest_request_accepts_all_optional_fields():
|
|
ReingestKBRequest = _rag_route().ReingestKBRequest
|
|
|
|
empty = ReingestKBRequest()
|
|
assert empty.embedding_model is None
|
|
|
|
partial = ReingestKBRequest(embedding_model = "BAAI/bge-small-en-v1.5")
|
|
assert partial.embedding_model == "BAAI/bge-small-en-v1.5"
|