diff --git a/swarms/agents/swarms.py b/swarms/agents/swarms.py index 678db059..7364f128 100644 --- a/swarms/agents/swarms.py +++ b/swarms/agents/swarms.py @@ -93,9 +93,22 @@ query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)) # !pip install duckduckgo_search web_search = DuckDuckGoSearchRun() -# +#======> multimodal_agent_tool = MultiModalVisualAgentTool(MultiModalVisualAgent) + +#======> Calculator +from langchain import LLMMathChain + +llm_math_chain = LLMMathChain.from_llm(llm=llm, verbose=True) +math_tool = Tool( + name="Calculator", + func=llm_math_chain.run, + description="useful for when you need to answer questions about math" + ), + + + tools = [ web_search, @@ -108,7 +121,9 @@ tools = [ Terminal, CodeWriter, - CodeEditor + CodeEditor, + + math_tool # HumanInputRun(), # Activate if you want the permit asking for help from the human ] @@ -133,8 +148,6 @@ class WorkerNode: def create_agent(self, ai_name, ai_role, human_in_the_loop, search_kwargs): - - # Instantiate the agent self.agent = AutoGPT.from_llm_and_tools( ai_name=ai_name, @@ -155,6 +168,19 @@ class WorkerNode: +# #inti worker node with llm +# worker_node = WorkerNode(llm=llm, tools=tools, vectorstore=vectorstore) + +# #create an agent within the worker node +# worker_node.create_agent(ai_name="AI Assistant", ai_role="Assistant", human_in_the_loop=True, search_kwargs={}) + +# #use the agent to perform a task +# worker_node.run_agent("Find 20 potential customers for a Swarms based AI Agent automation infrastructure") + + +#======================================> WorkerNode + + class MetaWorkerNode: def __init__(self, llm, tools, vectorstore): self.llm = llm