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),