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

Aggiungi gestione delle query elaborate in AdvancedQueryRagSolver

Browse Source
This commit is contained in:
andrea.terzani
2025-03-25 10:13:38 +01:00
parent 557f9d05c1
commit 1525f17580
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/main/java/com/olympus/hermione/stepSolvers/AdvancedQueryRagSolver.java
View File

@@ -128,10 +128,12 @@ public class AdvancedQueryRagSolver extends StepSolver {
} }
List<Document> docs = new ArrayList<Document>(); List<Document> docs = new ArrayList<Document>();
List<String> queries = new ArrayList<String>();
for (String query : elaboratedQuery.split("\n")) { for (String query : elaboratedQuery.split("\n")) {
logger.info("Elaborated query: " + query); logger.info("Elaborated query: " + query);
queries.add(query);
Builder request_builder = SearchRequest.builder() Builder request_builder = SearchRequest.builder()
.query(elaboratedQuery) .query(elaboratedQuery)
.topK(this.topk) .topK(this.topk)
@@ -166,6 +168,7 @@ public class AdvancedQueryRagSolver extends StepSolver {
String resultString = String.join("\n", result); String resultString = String.join("\n", result);
this.scenarioExecution.getExecSharedMap().put("tech_rag_source_documents", source_doc); this.scenarioExecution.getExecSharedMap().put("tech_rag_source_documents", source_doc);
this.scenarioExecution.getExecSharedMap().put("tech_rag_query", queries);
this.scenarioExecution.getExecSharedMap().put(this.outputField, resultString); this.scenarioExecution.getExecSharedMap().put(this.outputField, resultString);
this.scenarioExecution.setCurrentStepId(this.step.getStepId()); this.scenarioExecution.setCurrentStepId(this.step.getStepId());
this.scenarioExecution.setNextStepId(this.step.getNextStepId()); this.scenarioExecution.setNextStepId(this.step.getNextStepId());