diff --git a/verifiers/rubrics/judge_rubric.py b/verifiers/rubrics/judge_rubric.py index 9a8b737062..5fbdc12a25 100644 --- a/verifiers/rubrics/judge_rubric.py +++ b/verifiers/rubrics/judge_rubric.py @@ -60,11 +60,10 @@ async def judge( state: State | None = None, ) -> str: if isinstance(prompt, list): + # Rollouts normalize prompts to Pydantic Messages (not raw dicts). last_msg = prompt[-1] - if isinstance(last_msg, dict) and "content" in last_msg: - question = str(last_msg["content"]) - else: - question = "" + content = self.parser._message_field(last_msg, "content", "") + question = self.parser._content_to_text(content) else: question = str(prompt) response = self.parser.parse_answer(completion)