Skip to content

Commit

Permalink
claen up swarms
Browse files Browse the repository at this point in the history
Former-commit-id: 2f361d6
  • Loading branch information
kyegomez committed Jul 22, 2023
1 parent feced1a commit e80dd5c
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
setup(
name = 'swarms',
packages = find_packages(exclude=[]),
version = '1.0.7',
version = '1.0.8',
license='MIT',
description = 'Swarms - Pytorch',
author = 'Kye Gomez',
Expand Down
4 changes: 2 additions & 2 deletions swarms/swarms.py
Original file line number Diff line number Diff line change
Expand Up @@ -138,12 +138,12 @@ def initialize_boss_node(self, vectorstore, worker_node, llm_class=OpenAI, max_i
todo_chain = LLMChain(llm=llm, prompt=todo_prompt)

#math tool
llm_math_chain = LLMMathChain.from_llm(llm=llm, verbose=True)
# llm_math_chain = LLMMathChain.from_llm(llm=llm, verbose=True)

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,
Tool(name="Calculator", func=llm_math_chain.run, description="useful for when you need to answer questions about math")
# 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}"""
Expand Down

0 comments on commit e80dd5c

Please sign in to comment.