admin/hermione
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Bug Rag Ranker (sometimes the Rank for a document is not only a number)

Browse Source
This commit is contained in:
Emanuele Ferrelli
2025-04-09 16:15:43 +02:00
parent 1e698db368
commit 7501a2d197
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/main/java/com/olympus/hermione/utility/RAGDocumentRanker.java
View File

@@ -31,7 +31,7 @@ public class RAGDocumentRanker {
rankingPrompt +="Answer with a number between 1 and 10 and nothing else."; rankingPrompt +="Answer with a number between 1 and 10 and nothing else.";
OlympusChatClientResponse resp = chatClient.getChatCompletion(rankingPrompt); OlympusChatClientResponse resp = chatClient.getChatCompletion(rankingPrompt);
String rank = resp.getContent(); String rank = resp.getContent().substring(0, 1);
int rankInt = Integer.parseInt(rank); int rankInt = Integer.parseInt(rank);
logger.info("Rank: " + rankInt); logger.info("Rank: " + rankInt);