main
Kye 2 years ago
parent 88a50e68a2
commit 94beff25d7

@ -42,13 +42,13 @@ tools = [
DuckDuckGoSearchRun(),
WriteFileTool(root_dir=ROOT_DIR),
ReadFileTool(root_dir=ROOT_DIR),
process_csv, description=('Processes a CSV file'),
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'),
# Tool(name='code_editor', func=CodeEditor(), description='Edits code'),#
]
# ---------- Vector Store ----------
@ -152,6 +152,7 @@ class Swarms:
def initialize_llm(self):
return ChatOpenAI(model_name="gpt-4", temperature=1.0, openai_api_key=self.openai_api_key)
#
def initialize_tools(self, llm):
web_search = DuckDuckGoSearchRun()

Loading…
Cancel
Save