check for agent

Former-commit-id: de103eb3e3
workerULTRANODE
Kye 2 years ago
parent a789c276ef
commit 2d03356eb0

@ -41,6 +41,10 @@ class AgentManager:
builder = AgentBuilder(self.toolsets)
builder.build_parser()
agent = builder.get_agent()
if not agent:
raise ValueError("Agent not created")
callbacks = []
eval_callback = EVALCallbackHandler()
eval_callback.set_parser(builder.get_parser())
@ -54,7 +58,7 @@ class AgentManager:
callback_manager = CallbackManager(callbacks)
builder.build_llm(callback_manager, openai_api_key)
builder.build_global_tools()
memory: BaseChatMemory = self.get_or_create_memory(session)
@ -70,7 +74,7 @@ class AgentManager:
tool.callback_manager = callback_manager
executor = AgentExecutor.from_agent_and_tools(
agent=builder.get_agent(),
agent=agent,
tools=tools,
memory=memory,
callback_manager=callback_manager,

Loading…
Cancel
Save