From 7501a2d1972127030d871a751370e5eef3f1c73e Mon Sep 17 00:00:00 2001
From: Emanuele Ferrelli <emanuele.ferrelli@accenture.com>
Date: Wed, 9 Apr 2025 16:15:43 +0200
Subject: [PATCH] Bug Rag Ranker (sometimes the Rank for a document is not only
 a number)

---
 .../java/com/olympus/hermione/utility/RAGDocumentRanker.java    | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/main/java/com/olympus/hermione/utility/RAGDocumentRanker.java b/src/main/java/com/olympus/hermione/utility/RAGDocumentRanker.java
index 01c5230..99168e6 100644
--- a/src/main/java/com/olympus/hermione/utility/RAGDocumentRanker.java
+++ b/src/main/java/com/olympus/hermione/utility/RAGDocumentRanker.java
@@ -31,7 +31,7 @@ public class RAGDocumentRanker {
             rankingPrompt +="Answer with a number between 1 and 10 and nothing else.";
         
             OlympusChatClientResponse resp = chatClient.getChatCompletion(rankingPrompt);
-            String rank = resp.getContent();   
+            String rank = resp.getContent().substring(0, 1);  
 
             int rankInt = Integer.parseInt(rank);
             logger.info("Rank: " + rankInt);