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

basic steps error handling

Browse Source
This commit is contained in:
andrea.terzani
2024-10-29 08:43:28 +01:00
parent ba2f67202c
commit 78a36e8fcc
3 changed files with 34 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/main/java/com/olympus/hermione/controllers/ScenarioController.java
View File

@@ -58,6 +58,7 @@ public class ScenarioController {
@PostMapping("scenarios/execute")
public ScenarioOutput executeScenario(@RequestBody ScenarioExecutionInput scenarioExecutionInput) {
ScenarioOutput preOutput = scenarioExecutionService.prepareScrenarioExecution(scenarioExecutionInput);
scenarioExecutionService.executeScenarioAsync(preOutput);

3
src/main/java/com/olympus/hermione/models/ScenarioExecution.java
View File

@@ -29,7 +29,8 @@ public class ScenarioExecution {
private String currentStepDescription;
private String executedByUserId;
private String latestStepStatus;
private String latestStepOutput;
private ScenarioExecutionInput scenarioExecutionInput;
}

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

@@ -97,6 +97,12 @@ public class ScenarioExecutionService {
scenarioOutput.setStatus("IN_PROGRESS");
}
if(scenarioExecution.getLatestStepStatus().equals("ERROR")){
scenarioOutput.setScenarioExecution_id(scenarioExecution.getId());
scenarioOutput.setStatus("ERROR");
scenarioOutput.setMessage(scenarioExecution.getLatestStepOutput());
}
}else{
scenarioOutput.setScenarioExecution_id(null);
scenarioOutput.setStatus("ERROR");
@@ -168,9 +174,12 @@ public class ScenarioExecutionService {
while (scenarioExecution.getNextStepId()!=null) {
ScenarioStep step = steps.stream().filter(s -> s.getStepId().equals(scenarioExecution.getNextStepId())).findFirst().orElse(null);
executeScenarioStep(step, scenarioExecution);
}
if(scenarioExecution.getLatestStepStatus().equals("ERROR")){
logger.error("Error while executing step: " + step.getStepId());
break;
}
}
}
}
@@ -204,18 +213,30 @@ public class ScenarioExecutionService {
logger.info("Initializing step: " + step.getStepId());
solver.init(step, scenarioExecution,
vectorStore,chatModel,chatClient,graphDriver,neo4JUitilityService
,discoveryClient);
logger.info("Solving step: " + step.getStepId());
scenarioExecution.setCurrentStepId(step.getStepId());
scenarioExecution.setCurrentStepDescription(step.getName());
ScenarioExecution scenarioExecutionNew = solver.solveStep();
logger.info("Step solved: " + step.getStepId());
logger.info("Next step: " + scenarioExecutionNew.getNextStepId());
ScenarioExecution scenarioExecutionNew=scenarioExecution;
try{
solver.init(step, scenarioExecution,
vectorStore,chatModel,chatClient,graphDriver,neo4JUitilityService
,discoveryClient);
scenarioExecutionNew = solver.solveStep();
logger.info("Step solved: " + step.getStepId());
logger.info("Next step: " + scenarioExecutionNew.getNextStepId());
}catch (Exception e){
logger.error("Error while solving step: " + step.getStepId() + " => " + e.getMessage());
scenarioExecutionNew.setNextStepId(null);
scenarioExecutionNew.setLatestStepStatus("ERROR");
scenarioExecutionNew.setLatestStepOutput(e.getMessage());
}
scenarioExecutionRepository.save(scenarioExecutionNew);