From 30bf8e727e449dcdf80e82c616cdcb8c46c500a7 Mon Sep 17 00:00:00 2001 From: Andrea Terzani Date: Fri, 15 Nov 2024 12:16:08 +0000 Subject: [PATCH] Updated AdvancedAIPromptSolver.java --- .../olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java b/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java index b38858f..7e9dcf5 100644 --- a/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java +++ b/src/main/java/com/olympus/hermione/stepSolvers/AdvancedAIPromptSolver.java @@ -73,14 +73,14 @@ public class AdvancedAIPromptSolver extends StepSolver { .call(); - Usage usage = resp.chatResponse().getMetadata().getUsage(); + /*Usage usage = resp.chatResponse().getMetadata().getUsage(); if (usage != null) { Long usedTokens = usage.getTotalTokens(); this.scenarioExecution.setUsedTokens(usedTokens); } else { logger.info("Token usage information is not available."); - } + }*/ if(qai_output_entityType!=null && qai_output_entityType.equals("CiaOutputEntity")){ logger.info("Output is of type CiaOutputEntity");