p3pp01 commited on
Commit
f7eb03b
·
1 Parent(s): f101b02
Files changed (1) hide show
  1. tools/web_search.py +3 -3
tools/web_search.py CHANGED
@@ -1,6 +1,6 @@
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
- import ddgs
4
 
5
  class DuckDuckGoSearchTool(Tool):
6
  name = "web_search"
@@ -12,10 +12,10 @@ class DuckDuckGoSearchTool(Tool):
12
  super().__init__()
13
  self.max_results = max_results
14
  try:
15
- from ddgs import DDGS
16
  except ImportError as e:
17
  raise ImportError(
18
- "You must install package `ddgs` to run this tool: for instance run `pip install duckduckgo-search`."
19
  ) from e
20
  self.ddgs = DDGS(**kwargs)
21
 
 
1
  from typing import Any, Optional
2
  from smolagents.tools import Tool
3
+ from duckduckgo_search import DDGS
4
 
5
  class DuckDuckGoSearchTool(Tool):
6
  name = "web_search"
 
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`."
19
  ) from e
20
  self.ddgs = DDGS(**kwargs)
21