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);