wehub-resource-sync
e4dcfc49aa
Tests / Import Check (Python 3.13) (push) Has been cancelled
Tests / Import Check (Python 3.14) (push) Has been cancelled
Tests / Python Tests (Python 3.11) (push) Has been cancelled
Tests / Python Tests (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.14) (push) Has been cancelled
Tests / Test Summary (push) Has been cancelled
Tests / Lint and Format (push) Has been cancelled
Tests / Web Node Tests (push) Has been cancelled
Tests / Import Check (Python 3.11) (push) Has been cancelled
Tests / Import Check (Python 3.12) (push) Has been cancelled
Tests / Python Tests (Python 3.13) (push) Has been cancelled
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
"""LlamaIndex engine knobs stored in RuntimeSettingsService."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
from deeptutor.services.config.runtime_settings import RuntimeSettingsService
|
|
|
|
|
|
def test_llamaindex_defaults_when_absent(tmp_path: Path) -> None:
|
|
svc = RuntimeSettingsService(tmp_path, process_env={})
|
|
loaded = svc.load_llamaindex(include_process_overrides=False)
|
|
assert loaded["retrieval_profile"] == "hybrid"
|
|
assert loaded["top_k"] == 5
|
|
assert loaded["vector_top_k_multiplier"] == 2
|
|
assert loaded["bm25_top_k_multiplier"] == 2
|
|
assert loaded["chunk_size"] == 512
|
|
assert loaded["chunk_overlap"] == 50
|
|
|
|
|
|
def test_llamaindex_roundtrip(tmp_path: Path) -> None:
|
|
svc = RuntimeSettingsService(tmp_path, process_env={})
|
|
svc.save_llamaindex(
|
|
{"retrieval_profile": "vector", "top_k": 8, "chunk_size": 1024, "chunk_overlap": 0}
|
|
)
|
|
|
|
loaded = svc.load_llamaindex(include_process_overrides=False)
|
|
assert loaded["retrieval_profile"] == "vector"
|
|
assert loaded["top_k"] == 8
|
|
assert loaded["chunk_size"] == 1024
|
|
assert loaded["chunk_overlap"] == 0
|
|
# Its own file beside the other per-feature settings.
|
|
assert (tmp_path / "llamaindex.json").exists()
|
|
|
|
|
|
def test_llamaindex_clamps_out_of_range(tmp_path: Path) -> None:
|
|
svc = RuntimeSettingsService(tmp_path, process_env={})
|
|
svc.save_llamaindex(
|
|
{
|
|
"retrieval_profile": "nonsense",
|
|
"top_k": 999,
|
|
"bm25_top_k_multiplier": 0,
|
|
"chunk_size": 8,
|
|
"chunk_overlap": 99999,
|
|
}
|
|
)
|
|
loaded = svc.load_llamaindex(include_process_overrides=False)
|
|
# Unknown profile falls back to the safe default.
|
|
assert loaded["retrieval_profile"] == "hybrid"
|
|
assert loaded["top_k"] == 50
|
|
assert loaded["bm25_top_k_multiplier"] == 1
|
|
assert loaded["chunk_size"] == 64
|
|
# Overlap is clamped below the chunk size so chunking never degenerates.
|
|
assert loaded["chunk_overlap"] == 63
|
|
|
|
|
|
def test_llamaindex_profile_env_override(tmp_path: Path) -> None:
|
|
svc = RuntimeSettingsService(tmp_path, process_env={})
|
|
svc.save_llamaindex({"retrieval_profile": "vector"})
|
|
|
|
overridden = RuntimeSettingsService(tmp_path, process_env={"RAG_RETRIEVAL_PROFILE": "hybrid"})
|
|
loaded = overridden.load_llamaindex(include_process_overrides=True)
|
|
assert loaded["retrieval_profile"] == "hybrid"
|
|
|
|
|
|
def test_chunk_geometry_preserves_zero_overlap(monkeypatch) -> None:
|
|
from deeptutor.services.rag.pipelines.llamaindex import config
|
|
|
|
monkeypatch.setattr(
|
|
config,
|
|
"_load_runtime_settings",
|
|
lambda: {"chunk_size": 512, "chunk_overlap": 0},
|
|
)
|
|
|
|
assert config.chunk_geometry() == (512, 0)
|