omni agent added to swarm classs

main
Kye 2 years ago
parent df2e4690a4
commit 9a4d748fb5

@ -35,9 +35,7 @@ class Swarms:
ReadFileTool(root_dir=ROOT_DIR), ReadFileTool(root_dir=ROOT_DIR),
process_csv, process_csv,
WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)), WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)),
# self.omni_worker_agent.chat # Add the OmniWorkerAgent's chat method as a tool
] ]
if self.omni_worker_agent: if self.omni_worker_agent:
tools.append(self.omni_worker_agent.chat) #add omniworker agent class tools.append(self.omni_worker_agent.chat) #add omniworker agent class
return tools return tools

Loading…
Cancel
Save