Updated AdvancedAIPromptSolver.java

This commit is contained in:
Andrea Terzani
2024-11-15 12:16:08 +00:00
parent be8b51d404
commit 30bf8e727e

View File

@@ -73,14 +73,14 @@ public class AdvancedAIPromptSolver extends StepSolver {
.call(); .call();
Usage usage = resp.chatResponse().getMetadata().getUsage(); /*Usage usage = resp.chatResponse().getMetadata().getUsage();
if (usage != null) { if (usage != null) {
Long usedTokens = usage.getTotalTokens(); Long usedTokens = usage.getTotalTokens();
this.scenarioExecution.setUsedTokens(usedTokens); this.scenarioExecution.setUsedTokens(usedTokens);
} else { } else {
logger.info("Token usage information is not available."); logger.info("Token usage information is not available.");
} }*/
if(qai_output_entityType!=null && qai_output_entityType.equals("CiaOutputEntity")){ if(qai_output_entityType!=null && qai_output_entityType.equals("CiaOutputEntity")){
logger.info("Output is of type CiaOutputEntity"); logger.info("Output is of type CiaOutputEntity");