Updated AdvancedAIPromptSolver.java

2024-11-15 12:16:08 +00:00
parent be8b51d404
commit 30bf8e727e
1 changed files with 2 additions and 2 deletions
--- 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");