From 6c8c86c6c81355ba9e29d4df53d813c8ebc0d59e Mon Sep 17 00:00:00 2001 From: Kye Date: Sat, 22 Jul 2023 11:49:35 -0400 Subject: [PATCH] clean up Former-commit-id: e7fdc1ec2c2585dbb6a17b7d32736c327a9786ca --- swarms/__init__.py | 3 ++- swarms/agents/tools/main.py | 24 ++++++++++++------------ 2 files changed, 14 insertions(+), 13 deletions(-) diff --git a/swarms/__init__.py b/swarms/__init__.py index 43030c2c..629a11f2 100644 --- a/swarms/__init__.py +++ b/swarms/__init__.py @@ -2,4 +2,5 @@ from swarms.swarms import Swarms, swarm from swarms.agents import worker_node from swarms.agents.workers.worker_ultra_node import WorkerUltraNode, WorkerUltra -from swarms.agents.workers.worker_agent_ultra import worker_ultra_node \ No newline at end of file +from swarms.agents.workers.worker_agent_ultra import worker_ultra_node +from swarms.agents.workers.WorkerNode import WorkerNode, worker_node \ No newline at end of file diff --git a/swarms/agents/tools/main.py b/swarms/agents/tools/main.py index 667fde7e..f50dc286 100644 --- a/swarms/agents/tools/main.py +++ b/swarms/agents/tools/main.py @@ -2304,18 +2304,18 @@ list_tool.run({}) ###########=========================> #======> 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 -# from langchain.tools.human.tool import HumanInputRun -# from langchain.tools import BaseTool, DuckDuckGoSearchRun +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 +from langchain.tools.human.tool import HumanInputRun +from langchain.tools import BaseTool, DuckDuckGoSearchRun