pull/58/head
Kye 2 years ago
parent be16da4952
commit 225eef297c

@ -126,15 +126,9 @@ class WebpageQATool(BaseTool):
async def _arun(self, url: str, question: str) -> str: async def _arun(self, url: str, question: str) -> str:
raise NotImplementedError raise NotImplementedError
llm = ChatOpenAI(model_name="gpt-4", temperature=1.0) # llm = ChatOpenAI(model_name="gpt-4", temperature=1.0)
query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(llm)) # 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 # # @tool
# code_intepret = CodeInterpreter() # code_intepret = CodeInterpreter()

@ -1,6 +1,5 @@
import faiss import faiss
from langchain.chat_models import ChatOpenAI
from langchain.docstore import InMemoryDocstore from langchain.docstore import InMemoryDocstore
from langchain.embeddings import OpenAIEmbeddings from langchain.embeddings import OpenAIEmbeddings
from langchain.tools.human.tool import HumanInputRun 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.agents.message import Message
from swarms.tools.autogpt import ( from swarms.tools.autogpt import (
ReadFileTool, ReadFileTool,
VQAinference,
WriteFileTool, WriteFileTool,
compile, compile,
process_csv, process_csv,
query_website_tool, load_qa_with_sources_chain,
WebpageQATool
) )
from swarms.utils.decorators import error_decorator, log_decorator, timing_decorator from swarms.utils.decorators import error_decorator, log_decorator, timing_decorator
@ -129,6 +128,8 @@ class Worker:
temperature=0.5) temperature=0.5)
``` ```
""" """
query_website_tool = WebpageQATool(qa_chain=load_qa_with_sources_chain(self.llm))
self.tools = [ self.tools = [
WriteFileTool(root_dir=ROOT_DIR), WriteFileTool(root_dir=ROOT_DIR),
ReadFileTool(root_dir=ROOT_DIR), ReadFileTool(root_dir=ROOT_DIR),

Loading…
Cancel
Save