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

added memory_id

Browse Source
This commit is contained in:
Emanuele Ferrelli
2024-10-09 18:09:46 +02:00
parent d2d9c33ec7
commit dd58495fbf
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
src/main/java/com/olympus/hermione/stepSolvers/BasicAIPromptSolver.java
View File

@@ -2,6 +2,7 @@ package com.olympus.hermione.stepSolvers;
import java.util.List; import java.util.List;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import org.springframework.ai.chat.client.ChatClient.CallResponseSpec;
import org.springframework.ai.chat.messages.Message; import org.springframework.ai.chat.messages.Message;
import org.springframework.ai.chat.messages.SystemMessage; import org.springframework.ai.chat.messages.SystemMessage;
import org.springframework.ai.chat.messages.UserMessage; import org.springframework.ai.chat.messages.UserMessage;
@@ -56,12 +57,11 @@ public class BasicAIPromptSolver extends StepSolver {
Message userMessage = new UserMessage(this.qai_user_input); Message userMessage = new UserMessage(this.qai_user_input);
Message systemMessage = new SystemMessage(this.qai_system_prompt_template); Message systemMessage = new SystemMessage(this.qai_system_prompt_template);
Prompt prompt = new Prompt(List.of(userMessage, systemMessage)); CallResponseSpec resp = chatClient.prompt()
.messages(userMessage,systemMessage)
List<Generation> response = chatModel.call(prompt).getResults(); .call();
String output = response.get(0).getOutput().getContent();
String output = resp.content();
this.scenarioExecution.getExecSharedMap().put(this.qai_output_variable, output); this.scenarioExecution.getExecSharedMap().put(this.qai_output_variable, output);
this.scenarioExecution.setNextStepId(this.step.getNextStepId()); this.scenarioExecution.setNextStepId(this.step.getNextStepId());