фиксирую состояние

tests/pipeline_setup_v4/core/models.py

@@ -5,7 +5,13 @@ from pathlib import Path
 from typing import Literal
 
 
-ComponentKind = Literal["process_v2_intent_router"]
+ComponentKind = Literal[
+    "process_v2_intent_router",
+    "process_v2_retrieval_policy_resolver",
+    "process_v2_router_plus_retrieval_policy",
+    "process_v2_router_plus_retrieval_policy_rag",
+    "process_v2_full_chain",
+]
 
 
 @dataclass(slots=True, frozen=True)
@@ -15,21 +21,41 @@ class RouterExpectation:
     sub_intent: str | None = None
 
 
+@dataclass(slots=True, frozen=True)
+class RetrievalPlanExpectation:
+    profile: str | None = None
+    layers: tuple[str, ...] = ()
+    limit: int | None = None
+    filters: dict[str, object] = field(default_factory=dict)
+
+
 @dataclass(slots=True, frozen=True)
 class CaseExpectations:
     router: RouterExpectation = RouterExpectation()
+    retrieval_plan: RetrievalPlanExpectation = field(default_factory=RetrievalPlanExpectation)
+    route_assertions: dict[str, object] = field(default_factory=dict)
+    retrieval_plan_assertions: dict[str, object] = field(default_factory=dict)
+    rag_assertions: dict[str, object] = field(default_factory=dict)
+    pipeline_assertions: dict[str, object] = field(default_factory=dict)
+    llm_assertions: dict[str, object] = field(default_factory=dict)
 
 
 @dataclass(slots=True, frozen=True)
 class V4Case:
     case_id: str
     component: ComponentKind
-    query: str
     source_file: Path
-    expectations: CaseExpectations = CaseExpectations()
+    query: str = ""
+    rag_session_id: str | None = None
+    route: dict[str, object] = field(default_factory=dict)
+    expectations: CaseExpectations = field(default_factory=CaseExpectations)
     notes: str = ""
     tags: tuple[str, ...] = ()
 
+    @property
+    def display_input(self) -> str:
+        return self.query or self.route.get("user_query") or self.case_id
+
 
 @dataclass(slots=True, frozen=True)
 class ExecutionPayload: