Роутер работает нормально в process v2

This commit is contained in:
2026-04-07 14:09:51 +03:00
parent 5d77ab1a88
commit 6b74d410cd
1748 changed files with 216679 additions and 14208 deletions
@@ -0,0 +1,26 @@
"""Smoke-тест стандартного retrieval API: один embed и вызов repository."""
from __future__ import annotations
import asyncio
from unittest.mock import MagicMock
from app.core.rag.embedding.gigachat_embedder import GigaChatEmbedder
from app.core.rag.retrieval.session_retriever import RagSessionRetriever, RetrievalPlan
def test_rag_session_retriever_calls_repository() -> None:
embedder = MagicMock(spec=GigaChatEmbedder)
embedder.embed = MagicMock(return_value=[[0.1, 0.2]])
repo = MagicMock()
repo.retrieve = MagicMock(return_value=[{"path": "a.md", "layer": "D0_DOC_CHUNKS"}])
retriever = RagSessionRetriever(repository=repo, embedder=embedder)
plan = RetrievalPlan(profile="test", layers=["D0_DOC_CHUNKS", "D1_DOCUMENT_CATALOG"], limit=5)
rows = asyncio.run(retriever.retrieve("sid-1", "hello", plan))
assert len(rows) == 1
assert embedder.embed.called
assert repo.retrieve.called
call_kw = repo.retrieve.call_args
assert call_kw[0][0] == "sid-1"
assert call_kw[1]["layers"] == plan.layers
assert call_kw[1]["limit"] == 5