Update runtime_intentbench.py

runtime_intentbench.py

@@ -345,6 +345,8 @@ class RemoteAPIBackend(InferenceBackend):
                 "message": "Collector finished via remote API.",
                 "backend_mode": self.mode,
                 "collector_summary": payload.get("collector_summary", ""),
+                "collector_raw": payload.get("collector_raw", payload.get("collector_summary", "")),
+                "raw_prompt": payload.get("collector_prompt", ""),
             }
         if payload.get("planner_decision"):
             yield {
@@ -352,7 +354,9 @@ class RemoteAPIBackend(InferenceBackend):
                 "message": "Planner finished via remote API.",
                 "backend_mode": self.mode,
                 "planner_decision": payload.get("planner_decision", ""),
+                "planner_raw": payload.get("planner_raw", payload.get("planner_decision", "")),
                 "grounder_span_text": payload.get("grounder_span", ""),
+                "raw_prompt": payload.get("planner_prompt", ""),
             }
         if payload.get("grounder_span"):
             yield {
@@ -360,20 +364,25 @@ class RemoteAPIBackend(InferenceBackend):
                 "message": "Grounder finished via remote API.",
                 "backend_mode": self.mode,
                 "grounder_span_text": payload.get("grounder_span", ""),
+                "grounder_raw": payload.get("grounder_raw", payload.get("grounder_span", "")),
                 "grounded_video": payload.get("highlight_clip_path"),
+                "raw_prompt": payload.get("grounder_prompt", ""),
             }
         yield {
             "stage": "answer",
             "message": "Answer received from remote API.",
             "backend_mode": self.mode,
             "final_answer": payload.get("final_answer", ""),
-            "answer_raw": payload.get("answer_raw", ""),
+            "answer_raw": payload.get("answer_raw", payload.get("final_answer", "")),
+            "raw_prompt": payload.get("answer_prompt", ""),
         }
         yield {
             "stage": "review",
             "message": "Review received from remote API.",
             "backend_mode": self.mode,
             "review_summary": payload.get("review_summary", ""),
+            "review_raw": payload.get("review_raw", payload.get("review_summary", "")),
+            "raw_prompt": payload.get("review_prompt", ""),
         }
         yield {
             "stage": "done",
@@ -652,6 +661,7 @@ class CPUFallbackBackend(LocalQwenVLBackend):
                     "message": "Collector finished.",
                     "backend_mode": self.mode,
                     "collector_summary": collector_summary,
+                    "collector_raw": collector_summary,
                     "raw_prompt": format_messages_markdown("Collector Input", collector_messages),
                 }
 
@@ -672,14 +682,14 @@ class CPUFallbackBackend(LocalQwenVLBackend):
                 planner_text = (
                     f"Use Grounder: {use_grounder}\n"
                     f"Grounding Query: {grounding_query or '-'}\n"
-                    f"Reason: {planner_json.get('reason', '-')}\n\n"
-                    f"Raw:\n{planner_raw}"
+                    f"Reason: {planner_json.get('reason', '-')}"
                 )
                 yield {
                     "stage": "planner",
                     "message": "Planner finished.",
                     "backend_mode": self.mode,
                     "planner_decision": planner_text,
+                    "planner_raw": planner_raw,
                     "raw_prompt": format_messages_markdown("Planner Input", planner_messages),
                 }
 
@@ -712,6 +722,7 @@ class CPUFallbackBackend(LocalQwenVLBackend):
                         "message": "Grounder finished.",
                         "backend_mode": self.mode,
                         "grounder_span_text": grounded_span_text,
+                        "grounder_raw": grounder_raw,
                         "grounded_video": grounded_video,
                         "raw_prompt": format_messages_markdown("Grounder Input", grounder_messages),
                     }
@@ -721,6 +732,7 @@ class CPUFallbackBackend(LocalQwenVLBackend):
                         "message": "Grounder skipped.",
                         "backend_mode": self.mode,
                         "grounder_span_text": grounded_span_text,
+                        "grounder_raw": grounded_span_text,
                         "grounded_video": None,
                     }
 
@@ -760,6 +772,7 @@ class CPUFallbackBackend(LocalQwenVLBackend):
                     "message": "Review finished.",
                     "backend_mode": self.mode,
                     "review_summary": review_summary,
+                    "review_raw": review_raw,
                     "raw_prompt": format_messages_markdown("Review Input", review_messages),
                 }
                 yield {