refs #732: preserve Select AI results in HMM reports

ai-web-agent-console/tests/test_scenarios.py

@@ -1,8 +1,10 @@
 from __future__ import annotations
 
+import ast
 import json
 from pathlib import Path
 import tempfile
+from typing import Any, Mapping
 import unittest
 from unittest.mock import patch
 
@@ -140,6 +142,38 @@ class DemoScenarioConfigTest(unittest.TestCase):
         self.assertIn("render_hmm_carrier_report", server["tool_allowlist"])
         self.assertNotIn('tool.name == "render_hmm_carrier_report"', source)
 
+    def test_hmm_report_rows_accept_nested_select_ai_json_array(self) -> None:
+        source = (Path(__file__).parents[1] / "app.py").read_text(encoding="utf-8")
+        tree = ast.parse(source)
+        helper = next(
+            node
+            for node in tree.body
+            if isinstance(node, ast.FunctionDef)
+            and node.name == "_mcp_structured_rows"
+        )
+        namespace: dict[str, Any] = {
+            "Any": Any,
+            "Mapping": Mapping,
+            "json": json,
+        }
+        exec(compile(ast.Module(body=[helper], type_ignores=[]), "app.py", "exec"), namespace)
+
+        rows = namespace["_mcp_structured_rows"](
+            {
+                "response": {
+                    "result": json.dumps(
+                        [
+                            {"EMPLOYEE_CODE": "E9001", "CARRIER_CODE": "C901"},
+                            {"EMPLOYEE_CODE": "E9002", "CARRIER_CODE": "C902"},
+                        ]
+                    )
+                }
+            }
+        )
+
+        self.assertEqual(len(rows), 2)
+        self.assertEqual(rows[0]["CARRIER_CODE"], "C901")
+
     def test_hmm_report_template_contains_only_dynamic_payload_slot(self) -> None:
         template = (
             Path(__file__).parents[1]