replaced websearch duckduckgo with new driver

This commit is contained in:
Jason Schleifer
2024-07-10 11:19:18 +12:00
parent 9db1f9d805
commit 0a0ffc3abc
+4 -3
View File
@@ -1,5 +1,5 @@
import requests
from griptape.drivers import OpenAiAudioTranscriptionDriver
from griptape.drivers import DuckDuckGoWebSearchDriver, OpenAiAudioTranscriptionDriver
from griptape.engines import AudioTranscriptionEngine
from griptape.tools import (
Calculator,
@@ -7,12 +7,12 @@ from griptape.tools import (
FileManager,
GriptapeCloudKnowledgeBaseClient,
WebScraper,
WebSearch,
)
from griptape.tools.audio_transcription_client.tool import AudioTranscriptionClient
from ..py.griptape_config import get_config
from .base_tool import gtUIBaseTool
from .duckduckgo_client import DuckDuckGoTool
class gtUIAudioTranscriptionClient(gtUIBaseTool):
@@ -85,7 +85,8 @@ class gtUIWebSearch(gtUIBaseTool):
DESCRIPTION = "Search the web using DuckDuckGo."
def create(self, off_prompt):
tool = DuckDuckGoTool(
tool = WebSearch(
web_search_driver=DuckDuckGoWebSearchDriver(),
off_prompt=off_prompt,
)
return ([tool],)