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

Correggi il logger per AdvancedAIPromptSolver e modifica il metodo di invio dei messaggi

Browse Source
This commit is contained in:
andrea.terzani
2025-02-21 14:39:46 +01:00
parent 3402b4ada2
commit b4fa5025ca
2 changed files with 9 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,7 +23,7 @@ public class AdvancedAIPromptSolver extends StepSolver {
private String qai_custom_memory_id;
Logger logger = (Logger) LoggerFactory.getLogger(BasicQueryRagSolver.class);
Logger logger = (Logger) LoggerFactory.getLogger(AdvancedAIPromptSolver.class);
private void loadParameters(){
logger.info("Loading parameters");
@@ -59,14 +59,16 @@ public class AdvancedAIPromptSolver extends StepSolver {
this.scenarioExecution.setCurrentStepId(this.step.getStepId());
loadParameters();
/*
String userText = this.qai_user_input;
Message userMessage = new UserMessage(userText);
Message systemMessage = new SystemMessage(this.qai_system_prompt_template);
*/
CallResponseSpec resp = chatClient.prompt()
.messages(userMessage,systemMessage)
.user(this.qai_user_input)
.system(this.qai_system_prompt_template)
//.messages(userMessage,systemMessage)
.advisors(advisor -> advisor
.param("chat_memory_conversation_id", this.scenarioExecution.getId()+this.qai_custom_memory_id)
.param("chat_memory_response_size", 100))

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

@@ -57,8 +57,9 @@ public class BasicAIPromptSolver extends StepSolver {
Message userMessage = new UserMessage(this.qai_user_input);
Message systemMessage = new SystemMessage(this.qai_system_prompt_template);
CallResponseSpec resp = chatClient.prompt()
.messages(userMessage,systemMessage)
CallResponseSpec resp = chatClient.prompt().user(this.qai_user_input)
.system(this.qai_system_prompt_template)
//.messages(userMessage,systemMessage)
.call();
String output = resp.content();