From a7ef1c5f0bb2c6b4283e8a3f467289a2179bb47b Mon Sep 17 00:00:00 2001 From: Nick Galluzzo Date: Tue, 5 Aug 2025 17:39:10 +0700 Subject: [PATCH] fix(evaluation): Adjust quality rating thresholds and add missing color mapping The change modifies the quality rating color logic to include a blue color for "GOOD" scores and adds proper handling for "AVERAGE" scores, ensuring consistent visual feedback across all rating levels. --- src/diffmage/evaluation/models.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/diffmage/evaluation/models.py b/src/diffmage/evaluation/models.py index 418f3cc..eba2f4a 100644 --- a/src/diffmage/evaluation/models.py +++ b/src/diffmage/evaluation/models.py @@ -43,9 +43,11 @@ def get_quality_level(score: float) -> str: @staticmethod def get_rating_color(score: float) -> str: """Color for the quality rating""" - if score > ScoreThresholds.EXCELLENT: + if score >= ScoreThresholds.EXCELLENT: return "green" elif score >= ScoreThresholds.GOOD: + return "blue" + elif score >= ScoreThresholds.AVERAGE: return "yellow" else: return "red"