diff --git a/src/main/java/com/olympus/hermione/stepSolvers/AdvancedQueryRagSolver.java b/src/main/java/com/olympus/hermione/stepSolvers/AdvancedQueryRagSolver.java index f292190..c09452f 100644 --- a/src/main/java/com/olympus/hermione/stepSolvers/AdvancedQueryRagSolver.java +++ b/src/main/java/com/olympus/hermione/stepSolvers/AdvancedQueryRagSolver.java @@ -128,10 +128,12 @@ public class AdvancedQueryRagSolver extends StepSolver { } List docs = new ArrayList(); + List queries = new ArrayList(); for (String query : elaboratedQuery.split("\n")) { logger.info("Elaborated query: " + query); + queries.add(query); Builder request_builder = SearchRequest.builder() .query(elaboratedQuery) .topK(this.topk) @@ -166,6 +168,7 @@ public class AdvancedQueryRagSolver extends StepSolver { String resultString = String.join("\n", result); 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.setCurrentStepId(this.step.getStepId()); this.scenarioExecution.setNextStepId(this.step.getNextStepId());