|
|
|
@ -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:
|
|
|
|
|