Fix confidence score fallback in RAGService and update test cases for accuracy

Author: thanhtulearncode

app/models.py

@@ -0,0 +1,8 @@
+from typing import Optional
+from sqlmodel import SQLModel, Field
+
+class FAQItem(SQLModel, table=True):
+    __table_args__ = {"extend_existing": True} 
+    id: Optional[int] = Field(default=None, primary_key=True)
+    question: str
+    answer: str

app/services/rag_engine.py

@@ -156,6 +156,6 @@ def search(self, query: str, threshold: float = 0.45) -> Dict:
         # Fallback sous le seuil
         return {
             "answer": None,
-            "confidence": similarity,
+            "confidence": 0.0,
             "matched_question": metadata["original_question"],
         }

tests/test_retrieval.py

@@ -28,32 +28,32 @@ def test_rag_search_nominal(session: Session):
     # Recharger le moteur RAG (synchronisation avec la DB)
     engine = RAGService()
     engine.reload_from_db(session)
-    # MOCKER la collection ChromaDB pour éviter le TypeError.
     engine.collection = MagicMock()
     engine.collection.query.return_value = {
         "ids": [["1"]],
-        "distances": [[0.1]],  # Doit être un float, pas un Mock
+        "distances": [[0.1]], 
         "metadatas": [[{
             "answer": "Allez sur la page inscription.", 
-            "question": "Comment créer un compte ?"
+            "original_question": "Comment créer un compte ?"
         }]],
     }
-    # Tester une recherche pertinente
+    # Test
     result = engine.search("créer compte utilisateur")
-    # Le score devrait être élevé car la question est proche
-    assert result["confidence"] > 0.0
-    assert result["answer"] == "Allez sur la page inscription."
+    assert result["confidence"] > 0.8
+    assert "inscription" in result["answer"]
 
 def test_rag_search_no_match(session: Session):
     """Vérifie le comportement quand rien ne correspond."""
     engine = RAGService()
     engine.reload_from_db(session)
-    # MOCKER la collection pour simuler une distance élevée (pas de match).
     engine.collection = MagicMock()
     engine.collection.query.return_value = {
         "ids": [["1"]],
         "distances": [[1.5]], 
-        "metadatas": [[{"answer": "Réponse non pertinente"}]]
+        "metadatas": [[{
+            "answer": "Réponse non pertinente",
+            "original_question": "Question non pertinente"
+        }]]
     }
     result = engine.search("Une question qui n'a aucun sens ici")
     assert result["answer"] is None