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

added logic for retrieving aiModel

Browse Source
This commit is contained in:
Emanuele Ferrelli
2024-11-04 17:20:48 +01:00
parent ce496fa9f7
commit a937440832
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/main/java/com/olympus/hermione/services/ScenarioExecutionService.java
View File

@@ -132,18 +132,19 @@ public class ScenarioExecutionService {
HashMap<String, Object> execSharedMap = new HashMap<String, Object>(); HashMap<String, Object> execSharedMap = new HashMap<String, Object>();
execSharedMap.put("user_input", inputs); execSharedMap.put("user_input", inputs);
scenarioExecution.setExecSharedMap(execSharedMap); scenarioExecution.setExecSharedMap(execSharedMap);
scenarioExecutionRepository.save(scenarioExecution);
AiModel aiModel; AiModel aiModel;
if(scenario.getAiModel() != null){ if(scenario.getAiModel() != null){
aiModel = scenario.getAiModel(); aiModel = scenario.getAiModel();
}else { }else {
aiModel = aiModelRepository.findByIsDefault(true); aiModel = aiModelRepository.findByIsDefault(true);
//scenario.setAiModel(aiModel); scenario.setAiModel(aiModel);
//scenarioRepository.save(scenario); scenarioExecution.setScenario(scenario);
} }
ChatModel chatModel = createChatModel(aiModel);
scenarioExecutionRepository.save(scenarioExecution);
ChatModel chatModel = createChatModel(aiModel);
if(scenario.isUseChatMemory()){ if(scenario.isUseChatMemory()){
logger.info("Initializing chatClient with chat-memory advisor"); logger.info("Initializing chatClient with chat-memory advisor");