Skip to content

Commit a52da0b

Browse files
vertex-sdk-botcopybara-github
authored andcommitted
fix: GenAI Client(evals) - fix hallucination visualization
PiperOrigin-RevId: 824638559
1 parent 3da6bc2 commit a52da0b

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

vertexai/_genai/_evals_visualization.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -460,9 +460,13 @@ def _get_evaluation_html(eval_result_json: str) -> str:
460460
if (Array.isArray(explanationData) && explanationData.length > 0 && explanationData[0].sentence) {{
461461
bubbles += '<div class="rubric-bubble-container" style="margin-top: 8px;">';
462462
explanationData.forEach(item => {{
463-
const sentence = item.sentence || 'N/A';
463+
let sentence = item.sentence || 'N/A';
464464
const label = item.label ? item.label.toLowerCase() : '';
465-
const verdictText = label === 'no_rad' ? '<span class="pass">Pass</span>' : '<span class="fail">Fail</span>';
465+
const isPass = label === 'no_rad' || label === 'supported';
466+
const verdictText = isPass ? '<span class="pass">Pass</span>' : '<span class="fail">Fail</span>';
467+
if (isPass) {{
468+
sentence = `"${{sentence}}" is grounded`;
469+
}}
466470
const rationale = item.rationale || 'N/A';
467471
const itemJson = JSON.stringify(item, null, 2);
468472
bubbles += `

0 commit comments

Comments
 (0)