doggdad commited on
Commit
c18f4df
·
verified ·
1 Parent(s): 3189339

updated websearch

Browse files
Files changed (1) hide show
  1. tools/web_search.py +3 -2
tools/web_search.py CHANGED
@@ -1,6 +1,7 @@
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
- import duckduckgo_search
 
4
 
5
  class DuckDuckGoSearchTool(Tool):
6
  name = "web_search"
@@ -12,7 +13,7 @@ class DuckDuckGoSearchTool(Tool):
12
  super().__init__()
13
  self.max_results = max_results
14
  try:
15
- from duckduckgo_search import DDGS
16
  except ImportError as e:
17
  raise ImportError(
18
  "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."
 
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
+ #import duckduckgo_search
4
+ import ddgs
5
 
6
  class DuckDuckGoSearchTool(Tool):
7
  name = "web_search"
 
13
  super().__init__()
14
  self.max_results = max_results
15
  try:
16
+ from ddgs import DDGS
17
  except ImportError as e:
18
  raise ImportError(
19
  "You must install package `duckduckgo_search` to run this tool: for instance run `pip install duckduckgo-search`."