From 0349266585e9d45ac2108bdd478106ed72e35557 Mon Sep 17 00:00:00 2001 From: Brandon Hancock Date: Fri, 10 Jan 2025 14:38:07 -0500 Subject: [PATCH] Drop token calc handler changes --- src/crewai/utilities/token_counter_callback.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/src/crewai/utilities/token_counter_callback.py b/src/crewai/utilities/token_counter_callback.py index e6f29389a..ee26d41ab 100644 --- a/src/crewai/utilities/token_counter_callback.py +++ b/src/crewai/utilities/token_counter_callback.py @@ -1,4 +1,3 @@ -import threading import warnings from typing import Any, Dict, Optional @@ -11,7 +10,6 @@ from crewai.agents.agent_builder.utilities.base_token_process import TokenProces class TokenCalcHandler(CustomLogger): def __init__(self, token_cost_process: Optional[TokenProcess]): self.token_cost_process = token_cost_process - self.event = threading.Event() def log_success_event( self, @@ -33,8 +31,3 @@ class TokenCalcHandler(CustomLogger): self.token_cost_process.sum_cached_prompt_tokens( usage.prompt_tokens_details.cached_tokens ) - - def on_llm_end(self, response): - # Process the response - self.token_cost_process.update(response) - self.event.set()