diff --git a/swarms/agents/tools/__init__.py b/swarms/agents/tools/__init__.py index b59c3951..6f4d2f9b 100644 --- a/swarms/agents/tools/__init__.py +++ b/swarms/agents/tools/__init__.py @@ -1,5 +1,5 @@ from swarms.agents.tools.base import BaseTool, Tool, StructuredTool, ToolWrapper, BaseToolSet, ToolCreator, GlobalToolsCreator, SessionToolsCreator, ToolsFactory -from swarms.agents.tools.autogpt import pushd, process_csv, async_load_playwright, run_async, browse_web_page, WebpageQATool, web_search +from swarms.agents.tools.autogpt import pushd, process_csv, async_load_playwright, run_async, browse_web_page, WebpageQATool, web_search, query_website_tool from swarms.agents.tools.exit_conversation import ExitConversation from swarms.agents.tools.models import MaskFormer, ImageEditing, InstructPix2Pix, Text2Image, VisualQuestionAnswering, ImageCaptioning diff --git a/swarms/agents/tools/autogpt.py b/swarms/agents/tools/autogpt.py index 6df6e686..b12ba7b8 100644 --- a/swarms/agents/tools/autogpt.py +++ b/swarms/agents/tools/autogpt.py @@ -130,7 +130,7 @@ class WebpageQATool(BaseTool): raise NotImplementedError -# 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()