Новый раг

tests/rag/test_query_router.py

@@ -0,0 +1,12 @@
+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