13 lines
564 B
Python
13 lines
564 B
Python
from app.modules.rag.contracts.enums import RetrievalMode
|
|
from app.modules.rag.retrieval.query_router import RagQueryRouter
|
|
|
|
|
|
def test_query_router_uses_docs_by_default() -> None:
|
|
router = RagQueryRouter()
|
|
assert router.resolve_mode("Какие есть требования по биллингу?") == RetrievalMode.DOCS
|
|
|
|
|
|
def test_query_router_switches_to_code_on_explicit_code_requests() -> None:
|
|
router = RagQueryRouter()
|
|
assert router.resolve_mode("Объясни как работает код endpoint create invoice") == RetrievalMode.CODE
|