Skip to content

Commit

Permalink
setup 7.8
Browse files Browse the repository at this point in the history
  • Loading branch information
Kye committed Jul 6, 2023
1 parent 4f559bd commit 54eea60
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 21 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 = '0.7.7',
version = '0.7.8',
license='MIT',
description = 'Swarms - Pytorch',
author = 'Kye Gomez',
Expand Down
40 changes: 20 additions & 20 deletions swarms/tools/agent_tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,24 +36,24 @@
ROOT_DIR = "./data/"

# ---------- Tools ----------
openai_api_key = os.environ["OPENAI_API_KEY"]
llm = ChatOpenAI(model_name="gpt-4", temperature=1.0, openai_api_key=openai_api_key)

worker_tools = [
DuckDuckGoSearchRun(),
WriteFileTool(root_dir=ROOT_DIR),
ReadFileTool(root_dir=ROOT_DIR),
process_csv,
# openai_api_key = os.environ["OPENAI_API_KEY"]
# llm = ChatOpenAI(model_name="gpt-4", temperature=1.0, openai_api_key=openai_api_key)

# worker_tools = [
# DuckDuckGoSearchRun(),
# WriteFileTool(root_dir=ROOT_DIR),
# ReadFileTool(root_dir=ROOT_DIR),
# process_csv,

WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)),

# Tool(name='terminal', func=Terminal.execute, description='Operates a terminal'),
# Tool(name='code_writer', func=CodeWriter(), description='Writes code'),
# Tool(name='code_editor', func=CodeEditor(), description='Edits code'),#
]

# ---------- Vector Store ----------
embeddings_model = OpenAIEmbeddings()
embedding_size = 1536
index = faiss.IndexFlatL2(embedding_size)
vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})
# WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)),

# # Tool(name='terminal', func=Terminal.execute, description='Operates a terminal'),
# # Tool(name='code_writer', func=CodeWriter(), description='Writes code'),
# # Tool(name='code_editor', func=CodeEditor(), description='Edits code'),#
# ]

# # ---------- Vector Store ----------
# embeddings_model = OpenAIEmbeddings()
# embedding_size = 1536
# index = faiss.IndexFlatL2(embedding_size)
# vectorstore = FAISS(embeddings_model.embed_query, index, InMemoryDocstore({}), {})

0 comments on commit 54eea60

Please sign in to comment.