From 1c7b3e18e5170291ab0ab8b18b62a0700bb92c9b Mon Sep 17 00:00:00 2001 From: Emanuele Ferrelli Date: Wed, 9 Oct 2024 18:10:46 +0200 Subject: [PATCH] added memory_id --- .../olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java b/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java index 98e9828..77789a8 100644 --- a/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java +++ b/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java @@ -21,6 +21,7 @@ public class AdvancedAIPromptSolver extends StepSolver { private String qai_output_variable; private boolean qai_load_graph_schema=false; private String qai_output_entityType; + private String qai_custom_memory_id; Logger logger = (Logger) LoggerFactory.getLogger(BasicQueryRagSolver.class); @@ -46,6 +47,7 @@ public class AdvancedAIPromptSolver extends StepSolver { this.qai_output_variable = (String) this.step.getAttributes().get("qai_output_variable"); this.qai_output_entityType = (String) this.step.getAttributes().get("qai_output_entityType"); + this.qai_custom_memory_id = (String) this.step.getAttributes().get("qai_custom_memory_id"); //TODO: Add memory ID attribute to have the possibility of multiple conversations } @@ -66,7 +68,8 @@ public class AdvancedAIPromptSolver extends StepSolver { CallResponseSpec resp = chatClient.prompt() .messages(userMessage,systemMessage) - .advisors(advisor -> advisor.param("chat_memory_conversation_id", this.scenarioExecution.getId()) + .advisors(advisor -> advisor + .param("chat_memory_conversation_id", this.scenarioExecution.getId()+this.qai_custom_memory_id) .param("chat_memory_response_size", 100)) .call();