fix test

2026-07-15 12:53:50 +08:00
parent a5aeb8889c
commit 24992fc9fb
2 changed files with 12 additions and 6 deletions
@@ -216,9 +216,12 @@ def make_task_detail_context(
                "index": 0,
                "timestamp": "2026-01-01T00:01:00Z",
                "prompt": "Tap the Settings icon",
-                "tool_call_text": '{"action": "tap", "x": 100, "y": 200}',
-                "result_text": '{"ok": true}',
-                "screenshot_src": None,
+                "tool_call": {"action": "tap", "x": 100, "y": 200},
+                "result": {"ok": True},
+                "before_screenshot_src": None,
+                "after_screenshot_src": None,
+                "ocr_results": [],
+                "ui_tree_nodes": [],
            },
        ]
    return {
@@ -170,9 +170,12 @@ def _xss_context(template: str, session: SessionState) -> dict[str, Any]:
                    "index": 0,
                    "timestamp": XSS_PROBE,
                    "prompt": XSS_PROBE,
-                    "tool_call_text": XSS_PROBE,
-                    "result_text": XSS_PROBE,
-                    "screenshot_src": None,
+                    "tool_call": {"action": XSS_PROBE, "description": XSS_PROBE},
+                    "result": {"ok": XSS_PROBE},
+                    "before_screenshot_src": None,
+                    "after_screenshot_src": None,
+                    "ocr_results": [],
+                    "ui_tree_nodes": [],
                }
            ],
        )