From 956a6b53f6d3eff29fda2c2b523029d84747bf2d Mon Sep 17 00:00:00 2001 From: Kye Date: Wed, 4 Oct 2023 16:39:30 -0400 Subject: [PATCH] worker fix Former-commit-id: 225eef297c046fc6f48f7ac88a35b1951b82bf50 --- swarms/tools/autogpt.py | 10 ++-------- swarms/workers/worker.py | 7 ++++--- 2 files changed, 6 insertions(+), 11 deletions(-) diff --git a/swarms/tools/autogpt.py b/swarms/tools/autogpt.py index 34c3d428..bd78e936 100644 --- a/swarms/tools/autogpt.py +++ b/swarms/tools/autogpt.py @@ -126,15 +126,9 @@ class WebpageQATool(BaseTool): async def _arun(self, url: str, question: str) -> str: raise NotImplementedError -llm = ChatOpenAI(model_name="gpt-4", temperature=1.0) -query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)) +# llm = ChatOpenAI(model_name="gpt-4", temperature=1.0) +# query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)) -# !pip install duckduckgo_search -# web_search = DuckDuckGoSearchRun() - - - -# from swarms.tools.code_intepretor import CodeInterpreter # # @tool # code_intepret = CodeInterpreter() diff --git a/swarms/workers/worker.py b/swarms/workers/worker.py index a2cacead..0d659859 100644 --- a/swarms/workers/worker.py +++ b/swarms/workers/worker.py @@ -1,6 +1,5 @@ import faiss -from langchain.chat_models import ChatOpenAI from langchain.docstore import InMemoryDocstore from langchain.embeddings import OpenAIEmbeddings from langchain.tools.human.tool import HumanInputRun @@ -10,11 +9,11 @@ from langchain_experimental.autonomous_agents import AutoGPT from swarms.agents.message import Message from swarms.tools.autogpt import ( ReadFileTool, - VQAinference, WriteFileTool, compile, process_csv, - query_website_tool, + load_qa_with_sources_chain, + WebpageQATool ) from swarms.utils.decorators import error_decorator, log_decorator, timing_decorator @@ -129,6 +128,8 @@ class Worker: temperature=0.5) ``` """ + query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(self.llm)) + self.tools = [ WriteFileTool(root_dir=ROOT_DIR), ReadFileTool(root_dir=ROOT_DIR),