Update src/Reranker.py

src/Reranker.py

@@ -49,14 +49,8 @@ Példa a helyes formátumra:
         # Formatage du prompt
         formatted_prompt = prompt.format_prompt(query=query, context=context)
 
-# Appeler le modèle avec les messages formatés
-        response_message = self.llm.invoke(formatted_prompt.to_messages())
-        
-        # Récupérer la chaîne JSON dans la réponse
-        json_str = response_message.content
-        
-        # Parser la réponse en objet structuré
-        parsed_result = self.parser.parse(json_str)
-        
-        return parsed_result
+    # Appel du modèle avec parsing automatique
+        result = self.parser.invoke(formatted_prompt.to_messages())
+    
+        return result