fix test

2026-07-15 12:53:50 +08:00
parent a5aeb8889c
commit 24992fc9fb
2 changed files with 12 additions and 6 deletions
@@ -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": [],
                }
            ],
        )