Новый раг

tests/agent/orchestrator/test_explain_actions.py

@@ -0,0 +1,131 @@
+from app.modules.agent.engine.orchestrator.actions.explain_actions import ExplainActions
+from app.modules.agent.engine.orchestrator.execution_context import ExecutionContext
+from app.modules.agent.engine.orchestrator.models import (
+    ExecutionPlan,
+    OutputContract,
+    RoutingMeta,
+    Scenario,
+    TaskConstraints,
+    TaskSpec,
+)
+
+
+def _ctx(rag_items: list[dict]) -> ExecutionContext:
+    task = TaskSpec(
+        task_id="task-1",
+        dialog_session_id="dialog-1",
+        rag_session_id="rag-1",
+        user_message="Объясни по коду как работает task_processor",
+        scenario=Scenario.EXPLAIN_PART,
+        routing=RoutingMeta(domain_id="project", process_id="qa", confidence=0.9, reason="test"),
+        constraints=TaskConstraints(),
+        output_contract=OutputContract(result_type="answer"),
+        metadata={
+            "rag_items": rag_items,
+            "rag_context": "",
+            "confluence_context": "",
+            "files_map": {},
+        },
+    )
+    plan = ExecutionPlan(
+        plan_id="plan-1",
+        task_id="task-1",
+        scenario=Scenario.EXPLAIN_PART,
+        template_id="tpl",
+        template_version="1",
+        steps=[],
+    )
+    return ExecutionContext(task=task, plan=plan, graph_resolver=lambda *_: None, graph_invoker=lambda *_: {})
+
+
+def test_explain_actions_switch_to_code_profile_when_code_layers_present() -> None:
+    ctx = _ctx(
+        [
+            {
+                "source": "app/task_processor.py",
+                "layer": "C1_SYMBOL_CATALOG",
+                "title": "task_processor.process_task",
+                "content": "function task_processor.process_task(task)",
+                "metadata": {"qname": "task_processor.process_task", "kind": "function"},
+            },
+            {
+                "source": "app/task_processor.py",
+                "layer": "C2_DEPENDENCY_GRAPH",
+                "title": "task_processor.process_task:calls",
+                "content": "task_processor.process_task calls queue.publish",
+                "metadata": {"edge_type": "calls"},
+            },
+        ]
+    )
+    actions = ExplainActions()
+
+    actions.collect_sources(ctx)
+    actions.extract_logic(ctx)
+    actions.summarize(ctx)
+
+    sources = ctx.artifacts.get_content("sources", {})
+    assert sources["source_profile"] == "code"
+    answer = str(ctx.artifacts.get_content("final_answer", ""))
+    assert "кодовых слоев индекса" not in answer
+    assert "CodeRAG" not in answer
+    assert "app/task_processor.py" in answer
+    assert "requirements/docs context" not in answer
+
+
+def test_explain_actions_add_code_details_block() -> None:
+    ctx = _ctx(
+        [
+            {
+                "source": "src/config_manager/__init__.py",
+                "layer": "C1_SYMBOL_CATALOG",
+                "title": "ConfigManager",
+                "content": "const ConfigManager\nConfigManager = config_manager.v2.ConfigManagerV2",
+                "metadata": {
+                    "qname": "ConfigManager",
+                    "kind": "const",
+                    "lang_payload": {"imported_from": "v2.ConfigManagerV2", "import_alias": True},
+                },
+            },
+            {
+                "source": "src/config_manager/v2/control/base.py",
+                "layer": "C1_SYMBOL_CATALOG",
+                "title": "ControlChannel",
+                "content": "class ControlChannel\nControlChannel(ABC)",
+                "metadata": {"qname": "ControlChannel", "kind": "class"},
+            },
+            {
+                "source": "src/config_manager/v2/core/control_bridge.py",
+                "layer": "C1_SYMBOL_CATALOG",
+                "title": "ControlChannelBridge",
+                "content": "class ControlChannelBridge\nПредоставляет halt и status как обработчики start/stop/status",
+                "metadata": {"qname": "ControlChannelBridge", "kind": "class"},
+            },
+            {
+                "source": "src/config_manager/v2/core/control_bridge.py",
+                "layer": "C2_DEPENDENCY_GRAPH",
+                "title": "ControlChannelBridge.on_start:calls",
+                "content": "ControlChannelBridge.on_start calls self._start_runtime",
+                "metadata": {"src_qname": "ControlChannelBridge.on_start", "dst_ref": "self._start_runtime"},
+            },
+            {
+                "source": "src/config_manager/v2/__init__.py",
+                "layer": "C0_SOURCE_CHUNKS",
+                "title": "src/config_manager/v2/__init__.py:1-6",
+                "content": '"""Контракт: управление через API (config.yaml, секция management)."""',
+                "metadata": {},
+            },
+        ]
+    )
+    actions = ExplainActions()
+
+    actions.collect_sources(ctx)
+    actions.extract_logic(ctx)
+    actions.summarize(ctx)
+
+    answer = str(ctx.artifacts.get_content("final_answer", ""))
+    assert "### Что видно по коду" in answer
+    assert "ConfigManager` в проекте доступен как alias" in answer
+    assert "ControlChannelBridge.on_start" in answer
+    assert "### Где смотреть в проекте" in answer
+    assert "В индексе нет точного символа" not in answer
+    assert "отдельный интерфейс управления" in answer

tests/rag/test_code_indexing_pipeline.py

@@ -0,0 +1,57 @@
+from app.modules.rag.contracts.enums import RagLayer
+from app.modules.rag.indexing.code.pipeline import CodeIndexingPipeline
+
+
+def test_code_pipeline_builds_source_symbols_edges_and_entrypoints() -> None:
+    pipeline = CodeIndexingPipeline()
+    content = """
+from fastapi import APIRouter
+
+router = APIRouter()
+
+class UserService:
+    def get_user(self, user_id):
+        return user_id
+
+@router.get("/users/{user_id}")
+async def get_user(user_id: str):
+    service = UserService()
+    return service.get_user(user_id)
+"""
+    docs = pipeline.index_file(
+        repo_id="acme/proj",
+        commit_sha="abc123",
+        path="app/api/users.py",
+        content=content,
+    )
+
+    layers = {doc.layer for doc in docs}
+    assert RagLayer.CODE_SOURCE_CHUNKS in layers
+    assert RagLayer.CODE_SYMBOL_CATALOG in layers
+    assert RagLayer.CODE_DEPENDENCY_GRAPH in layers
+    assert RagLayer.CODE_ENTRYPOINTS in layers
+
+    symbol_doc = next(doc for doc in docs if doc.layer == RagLayer.CODE_SYMBOL_CATALOG and doc.metadata["kind"] == "function")
+    assert "get_user" in symbol_doc.metadata["qname"]
+
+    edge_doc = next(doc for doc in docs if doc.layer == RagLayer.CODE_DEPENDENCY_GRAPH)
+    assert edge_doc.metadata["edge_type"] in {"calls", "imports", "inherits"}
+
+    entry_doc = next(doc for doc in docs if doc.layer == RagLayer.CODE_ENTRYPOINTS)
+    assert entry_doc.metadata["framework"] == "fastapi"
+
+
+def test_code_pipeline_indexes_import_alias_as_symbol() -> None:
+    pipeline = CodeIndexingPipeline()
+    content = "from .v2 import ConfigManagerV2 as ConfigManager\n"
+
+    docs = pipeline.index_file(
+        repo_id="acme/proj",
+        commit_sha="abc123",
+        path="src/config_manager/__init__.py",
+        content=content,
+    )
+
+    alias_doc = next(doc for doc in docs if doc.layer == RagLayer.CODE_SYMBOL_CATALOG and doc.metadata["qname"] == "ConfigManager")
+    assert alias_doc.metadata["kind"] == "const"
+    assert alias_doc.metadata["lang_payload"]["import_alias"] is True

tests/rag/test_docs_indexing_pipeline.py

@@ -0,0 +1,63 @@
+from app.modules.rag.contracts.enums import RagLayer
+from app.modules.rag.indexing.docs.pipeline import DocsIndexingPipeline
+
+
+def test_docs_pipeline_builds_catalog_facts_sections_and_policy() -> None:
+    pipeline = DocsIndexingPipeline()
+    content = """---
+id: api.billing.create_invoice
+type: policy
+domain: billing
+links:
+  calls_api:
+    - api.billing.validate_invoice
+tags: [billing]
+status: active
+---
+# Create Invoice
+
+## Spec Summary
+
+Creates an invoice in billing.
+
+## Request Contract
+
+| field | type | required | validation |
+| --- | --- | --- | --- |
+| amount | decimal | yes | > 0 |
+
+## Error Matrix
+
+| status | error | client action |
+| --- | --- | --- |
+| 400 | invalid_amount | fix request |
+
+## Rules
+
+- metric: billing.invoice.created
+- rule: amount must be positive
+"""
+    docs = pipeline.index_file(
+        repo_id="acme/proj",
+        commit_sha="abc123",
+        path="docs/billing/create_invoice.md",
+        content=content,
+    )
+
+    layers = {doc.layer for doc in docs}
+    assert RagLayer.DOCS_MODULE_CATALOG in layers
+    assert RagLayer.DOCS_FACT_INDEX in layers
+    assert RagLayer.DOCS_SECTION_INDEX in layers
+    assert RagLayer.DOCS_POLICY_INDEX in layers
+
+    module_doc = next(doc for doc in docs if doc.layer == RagLayer.DOCS_MODULE_CATALOG)
+    assert module_doc.metadata["module_id"] == "api.billing.create_invoice"
+    assert module_doc.metadata["type"] == "policy"
+
+    fact_texts = [doc.text for doc in docs if doc.layer == RagLayer.DOCS_FACT_INDEX]
+    assert any("calls_api" in text for text in fact_texts)
+    assert any("has_field" in text for text in fact_texts)
+    assert any("returns_error" in text for text in fact_texts)
+
+    section_doc = next(doc for doc in docs if doc.layer == RagLayer.DOCS_SECTION_INDEX)
+    assert section_doc.metadata["section_path"]

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

tests/rag/test_query_terms.py

@@ -0,0 +1,9 @@
+from app.modules.rag.retrieval.query_terms import extract_query_terms
+
+
+def test_extract_query_terms_from_code_question() -> None:
+    terms = extract_query_terms("Объясни по коду как можно управлять COnfigmanager?")
+
+    assert "configmanager" in terms
+    assert "config_manager" in terms
+    assert "control" in terms