diff --git a/setup.py b/setup.py index dba95e89d..fa2767c77 100644 --- a/setup.py +++ b/setup.py @@ -4,7 +4,7 @@ setup( name = 'swarms', packages = find_packages(exclude=[]), - version = '1.0.6', + version = '1.0.7', license='MIT', description = 'Swarms - Pytorch', author = 'Kye Gomez', diff --git a/swarms/swarms.py b/swarms/swarms.py index 3cf4cdbdc..d468dabbd 100644 --- a/swarms/swarms.py +++ b/swarms/swarms.py @@ -15,6 +15,8 @@ from swarms.utils.task import Task +# TODO: Pass in abstract LLM class that can utilize Hf or Anthropic models +# TODO: Move away from OPENAI class Swarms: def __init__(self, openai_api_key="", use_vectorstore=True, use_async=True, human_in_the_loop=True): #openai_api_key: the openai key. Default is empty @@ -137,16 +139,11 @@ def initialize_boss_node(self, vectorstore, worker_node, llm_class=OpenAI, max_i #math tool 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 = [ Tool(name="TODO", func=todo_chain.run, description="useful for when you need to come up with todo lists. Input: an objective to create a todo list for your objective. Note create a todo list then assign a ranking from 0.0 to 1.0 to each task, then sort the tasks based on the tasks most likely to achieve the objective. The Output: a todo list for that objective with rankings for each step from 0.1 Please be very clear what the objective is!"), worker_node, - math_tool + Tool(name="Calculator", func=llm_math_chain.run, description="useful for when you need to answer questions about math") ] suffix = """Question: {task}\n{agent_scratchpad}"""