Sauten commited on
Commit
99df9c9
·
verified ·
1 Parent(s): 7f54e54

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -21
app.py CHANGED
@@ -6,6 +6,8 @@ import pandas as pd
6
  import json
7
  from urllib.parse import quote
8
  from bs4 import BeautifulSoup
 
 
9
 
10
 
11
  from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, load_tool, tool, FinalAnswerTool, OpenAIServerModel, VisitWebpageTool
@@ -45,28 +47,8 @@ def simple_web_search(query: str, num_results: int = 5) -> str:
45
  num_results: Result count.
46
  """
47
  try:
48
- url = f"https://html.duckduckgo.com/html/?q={quote(query)}"
49
- headers = {
50
- 'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; rv:91.0) Gecko/20100101 Firefox/91.0'
51
- }
52
- response = requests.get(url, headers=headers, timeout=10)
53
- response.raise_for_status()
54
-
55
- soup = BeautifulSoup(response.text, 'html.parser')
56
- results = []
57
-
58
- for result in soup.select('.result')[:num_results]:
59
- title = result.select_one('.result__title a')
60
- snippet = result.select_one('.result__snippet')
61
- if title and snippet:
62
- results.append({
63
- 'title': title.get_text(strip=True),
64
- 'link': title['href'],
65
- 'snippet': snippet.get_text(strip=True)
66
- })
67
-
68
  return json.dumps(results if results else {"error": "No results found"})
69
-
70
  except Exception as e:
71
  return json.dumps({"error": str(e)})
72
  #simple_web_search = simple_web_search()
 
6
  import json
7
  from urllib.parse import quote
8
  from bs4 import BeautifulSoup
9
+ from duckduckgo_search import ddg
10
+
11
 
12
 
13
  from smolagents import ToolCallingAgent, DuckDuckGoSearchTool, load_tool, tool, FinalAnswerTool, OpenAIServerModel, VisitWebpageTool
 
47
  num_results: Result count.
48
  """
49
  try:
50
+ results = ddg(query, max_results=num_results)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  return json.dumps(results if results else {"error": "No results found"})
 
52
  except Exception as e:
53
  return json.dumps({"error": str(e)})
54
  #simple_web_search = simple_web_search()