From e64f1c68bc088a4a2711cd4ee9679abd0ef03d3e Mon Sep 17 00:00:00 2001 From: Kye Date: Tue, 11 Jul 2023 14:38:09 -0400 Subject: [PATCH] clean up Former-commit-id: e917a3dbd9e04ae9f8698a02713e4d672ba6fe9c --- swarms/agents/utils/manager.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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: