diff --git a/swarms/agents/utils/manager.py b/swarms/agents/utils/manager.py index 8eb383a7..0c73ee9e 100644 --- a/swarms/agents/utils/manager.py +++ b/swarms/agents/utils/manager.py @@ -3,7 +3,7 @@ from typing import Dict, Optional from langchain.agents.agent import AgentExecutor from langchain.callbacks.manager import CallbackManager -from langchain.callbacks.base import set_handler +# from langchain.callbacks.base import set_handler from langchain.chains.conversation.memory import ConversationBufferMemory from langchain.memory.chat_memory import BaseChatMemory @@ -13,7 +13,7 @@ from .builder import AgentBuilder from .callback import EVALCallbackHandler, ExecutionTracingCallbackHandler -set_handler(EVALCallbackHandler()) +CallbackManager.set_handler(EVALCallbackHandler()) class AgentManager: