Skip to content

Commit

Permalink
clean up setup
Browse files Browse the repository at this point in the history
Former-commit-id: c2a3976
  • Loading branch information
Kye committed Jul 22, 2023
1 parent c91966b commit feced1a
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 7 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.6',
version = '1.0.7',
license='MIT',
description = 'Swarms - Pytorch',
author = 'Kye Gomez',
Expand Down
9 changes: 3 additions & 6 deletions swarms/swarms.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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}"""
Expand Down

0 comments on commit feced1a

Please sign in to comment.