55 lines
2.0 KiB
Python
55 lines
2.0 KiB
Python
from __future__ import annotations
|
|
|
|
import sys
|
|
import unittest
|
|
from pathlib import Path
|
|
|
|
|
|
sys.path.insert(0, str(Path(__file__).resolve().parent))
|
|
|
|
import domain_case_loop as dcl
|
|
|
|
|
|
class DomainCaseLoopStepStateTests(unittest.TestCase):
|
|
def test_preserves_mcp_catalog_alignment_debug_fields(self) -> None:
|
|
step_state = dcl.build_scenario_step_state(
|
|
scenario_id="planner_alignment_demo",
|
|
domain="planner_autonomy",
|
|
step={
|
|
"step_id": "step_01",
|
|
"title": "Alignment visibility",
|
|
"depends_on": [],
|
|
"question_template": "show planner alignment",
|
|
},
|
|
step_index=1,
|
|
question_resolved="show planner alignment",
|
|
analysis_context={},
|
|
turn_artifact={
|
|
"assistant_message": {
|
|
"reply_type": "factual",
|
|
"text": "Confirmed answer",
|
|
"message_id": "msg-1",
|
|
"trace_id": "trace-1",
|
|
},
|
|
"technical_debug_payload": {
|
|
"detected_mode": "address_query",
|
|
"detected_intent": "counterparty_turnover",
|
|
"selected_recipe": "counterparty_turnover_by_period",
|
|
"capability_id": "confirmed_counterparty_turnover",
|
|
"mcp_discovery_catalog_chain_alignment_status": "selected_matches_top",
|
|
"mcp_discovery_catalog_chain_top_match": "value_flow",
|
|
"mcp_discovery_catalog_chain_selected_matches_top": True,
|
|
},
|
|
"session_summary": {},
|
|
},
|
|
entries=[],
|
|
)
|
|
|
|
self.assertEqual(step_state["mcp_discovery_catalog_chain_alignment_status"], "selected_matches_top")
|
|
self.assertEqual(step_state["mcp_discovery_catalog_chain_top_match"], "value_flow")
|
|
self.assertTrue(step_state["mcp_discovery_catalog_chain_selected_matches_top"])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
unittest.main()
|