From a0ba9a81b2f17a473e8af5eab0d5a7b00b1ebc16 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Fri, 5 Apr 2024 08:24:20 -0300 Subject: [PATCH] not overriding llm callbacks --- src/crewai/agent.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/crewai/agent.py b/src/crewai/agent.py index 7a4a87c9d..27b5a9076 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -156,9 +156,12 @@ class Agent(BaseModel): def set_agent_executor(self) -> "Agent": """set agent executor is set.""" if hasattr(self.llm, "model_name"): - self.llm.callbacks = [ - TokenCalcHandler(self.llm.model_name, self._token_process) - ] + token_handler = TokenCalcHandler(self.llm.model_name, self._token_process) + if isinstance(self.llm.callbacks, list): + self.llm.callbacks.append(token_handler) + else: + self.llm.callbacks = [token_handler] + if not self.agent_executor: if not self.cache_handler: self.cache_handler = CacheHandler()