Upload embeddings.py

embeddings.py

@@ -66,8 +66,6 @@ class EmbeddingManager:
             self._prefix_p = ""
             self._prefix_q = ""
 
-        # lower !!!!!!!!!!!!!!
-
         # Creation du Sentence transformer model
         logger.info(f"Initialisation du modèle d'embeddings: {model_name}")
         device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -169,12 +167,9 @@ class EmbeddingManager:
 
             if theme not in seen_themes:
                 seen_themes.add(theme)
-                normalized_document = self._normalize_text(document)
-                if self._prefix_p and normalized_document.startswith(self._prefix_p):
-                    normalized_document = normalized_document[len(self._prefix_p):]
                 candidates.append(
                     SemanticSearchCandidate(
-                        formulation=normalized_document,
+                        formulation=document.removeprefix(self._prefix_p),
                         theme=theme,
                         response=metadata["response"],
                         similarity_score=similarity,