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

This commit is contained in:
2026-04-07 14:09:51 +03:00
parent 0a25e42ea1
commit 8b7b72967e
1746 changed files with 216414 additions and 14037 deletions
@@ -0,0 +1,32 @@
import pytest
from tests.unit_tests.rag.intent_router_testkit import run_sequence
pytestmark = pytest.mark.intent_router
@pytest.mark.parametrize(
("query", "plan_id", "primary_doc_types", "secondary_doc_types", "expected_filter_key", "expected_filter_value"),
[
("объясни /health", "docs_api_method_explain_v1", ["api_method"], ["logic_block", "domain_entity", "architecture_overview"], "endpoint_path", "/health"),
("какие методы в notifications", "docs_list_api_methods_v1", ["api_method"], [], "domain_name", "notifications"),
("найди документацию по telegram_delivery", "docs_find_documents_by_domain_v1", ["index_page", "architecture_overview", "api_method", "logic_block", "domain_entity"], [], "domain_name", "telegram_delivery"),
("сгенерируй openapi по /send", "docs_generate_openapi_v1", ["api_method"], ["domain_entity", "logic_block"], "endpoint_path", "/send"),
("как устроен сервис", "docs_general_docs_qa_v1", ["index_page", "architecture_overview"], ["logic_block", "domain_entity", "api_method"], "scope_level", "project"),
],
)
def test_docs_retrieval_plan_contracts(
query: str,
plan_id: str,
primary_doc_types: list[str],
secondary_doc_types: list[str],
expected_filter_key: str,
expected_filter_value: str,
) -> None:
result = run_sequence([query])[0]
assert result.retrieval_plan is not None
assert result.retrieval_plan.plan_id == plan_id
assert result.retrieval_plan.primary_doc_types == primary_doc_types
assert result.retrieval_plan.secondary_doc_types == secondary_doc_types
assert result.retrieval_plan.filters[expected_filter_key] == expected_filter_value