Nymbo commited on
Commit
29e3c54
·
verified ·
1 Parent(s): 2add32c

disabling safesearch by default

Browse files
Files changed (1) hide show
  1. Modules/Web_Search.py +8 -5
Modules/Web_Search.py CHANGED
@@ -17,6 +17,9 @@ TOOL_SUMMARY = (
17
  )
18
 
19
 
 
 
 
20
  def _extract_date_from_snippet(snippet: str) -> str:
21
  if not snippet:
22
  return ""
@@ -142,15 +145,15 @@ def Web_Search(
142
  _search_rate_limiter.acquire()
143
  with DDGS() as ddgs:
144
  if stype == "text":
145
- raw_gen = ddgs.text(query, max_results=total_needed + 10)
146
  elif stype == "news":
147
- raw_gen = ddgs.news(query, max_results=total_needed + 10)
148
  elif stype == "images":
149
- raw_gen = ddgs.images(query, max_results=total_needed + 10)
150
  elif stype == "videos":
151
- raw_gen = ddgs.videos(query, max_results=total_needed + 10)
152
  else:
153
- raw_gen = ddgs.books(query, max_results=total_needed + 10)
154
  try:
155
  return list(raw_gen)
156
  except Exception as inner_exc:
 
17
  )
18
 
19
 
20
+ _SAFESEARCH_LEVEL = "off"
21
+
22
+
23
  def _extract_date_from_snippet(snippet: str) -> str:
24
  if not snippet:
25
  return ""
 
145
  _search_rate_limiter.acquire()
146
  with DDGS() as ddgs:
147
  if stype == "text":
148
+ raw_gen = ddgs.text(query, max_results=total_needed + 10, safesearch=_SAFESEARCH_LEVEL)
149
  elif stype == "news":
150
+ raw_gen = ddgs.news(query, max_results=total_needed + 10, safesearch=_SAFESEARCH_LEVEL)
151
  elif stype == "images":
152
+ raw_gen = ddgs.images(query, max_results=total_needed + 10, safesearch=_SAFESEARCH_LEVEL)
153
  elif stype == "videos":
154
+ raw_gen = ddgs.videos(query, max_results=total_needed + 10, safesearch=_SAFESEARCH_LEVEL)
155
  else:
156
+ raw_gen = ddgs.books(query, max_results=total_needed + 10, safesearch=_SAFESEARCH_LEVEL)
157
  try:
158
  return list(raw_gen)
159
  except Exception as inner_exc: