Merge branch 'staging' of https://github.com/onepointconsulting/data-questionnaire-agent into staging

Author: onepointconsulting

data_questionnaire_agent/model/confidence_schema.py

@@ -14,6 +14,15 @@ class ConfidenceDegree(StrEnum):
     low = "low"
 
 
+CONFIDENCE_DEGREE_DICT = {
+    ConfidenceDegree.outstanding: 5,
+    ConfidenceDegree.high: 4,
+    ConfidenceDegree.medium: 3,
+    ConfidenceDegree.mediocre: 2,
+    ConfidenceDegree.low: 1,
+}
+
+
 class ConfidenceRating(BaseModel):
     """Represents a rating of how confident the model is to give advice to a customer based on a questionnaire"""
 
@@ -30,6 +39,29 @@ class ConfidenceRating(BaseModel):
         description="The confidence rating of the model to give advice to a customer based on a questionnaire",
     )
 
+    def _value(self) -> int:
+        return CONFIDENCE_DEGREE_DICT[self.rating]
+
+    def __lt__(self, other):
+        if isinstance(other, ConfidenceRating):
+            return self._value() < other._value()
+        return NotImplemented
+
+    def __le__(self, other):
+        if isinstance(other, ConfidenceRating):
+            return self._value() <= other._value()
+        return NotImplemented
+
+    def __gt__(self, other):
+        if isinstance(other, ConfidenceRating):
+            return self._value() > other._value()
+        return NotImplemented
+
+    def __ge__(self, other):
+        if isinstance(other, ConfidenceRating):
+            return self._value() >= other._value()
+        return NotImplemented
+
     def to_markdown(self, locale: str = "en") -> str:
         return f"""
 # {t("Confidence Degree", locale=locale)}
@@ -49,3 +81,5 @@ def to_html(self, language: str = "en") -> str:
 
 <p>{self.reasoning}</p>
 """
+
+    

data_questionnaire_agent/server/questionnaire_server.py

@@ -412,10 +412,19 @@ async def save_confidence_rating(
     questionnaire: Questionnaire,
 ):
     if confidence_rating is not None:
+        step = len(questionnaire.questions)
+
+        # Prevent the confidence rating from decreasing
+        previous_confidence_rating = None
+        if step is not None and step > 2:
+            previous_confidence_rating = await select_confidence(session_id, step - 1)
+            if previous_confidence_rating is not None and previous_confidence_rating is not None:
+                if previous_confidence_rating > confidence_rating:
+                    confidence_rating = previous_confidence_rating
+        
         # Save the confidence
         if conditional_advice:
             conditional_advice.confidence = confidence_rating
-        step = len(questionnaire.questions)
         await save_confidence(session_id, step, confidence_rating)