From 1525f17580f96268eddec10e9c880bf65a64f00d Mon Sep 17 00:00:00 2001
From: "andrea.terzani" <andrea.terzani@accenture.com>
Date: Tue, 25 Mar 2025 10:13:38 +0100
Subject: [PATCH] Aggiungi gestione delle query elaborate in
 AdvancedQueryRagSolver

---
 .../olympus/hermione/stepSolvers/AdvancedQueryRagSolver.java   | 3 +++
 1 file changed, 3 insertions(+)

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<Document> docs = new ArrayList<Document>();
+        List<String> queries = new ArrayList<String>();
 
         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());