pkraman06 commited on
Commit
eb4c6e5
·
verified ·
1 Parent(s): 81b2f25

Update tools/web_search_tool.py

Browse files
Files changed (1) hide show
  1. tools/web_search_tool.py +18 -6
tools/web_search_tool.py CHANGED
@@ -1,18 +1,30 @@
1
- from crewai.tools import BaseTool
 
2
  import requests
3
  from bs4 import BeautifulSoup
4
 
5
 
6
  class WebSearchTool(BaseTool):
7
- name: str = "Web Search Tool"
8
- description: str = "Fetches web content for debugging help"
9
 
10
  def _run(self, query: str):
 
11
  url = f"https://duckduckgo.com/html/?q={query}"
12
 
13
  response = requests.get(url)
14
- soup = BeautifulSoup(response.text, "html.parser")
15
 
16
- results = soup.find_all("a", class_="result__a", limit=5)
 
 
 
 
 
 
 
 
 
17
 
18
- return "\n".join([result.get_text() for result in results])
 
 
 
1
+ from langchain.tools import BaseTool
2
+
3
  import requests
4
  from bs4 import BeautifulSoup
5
 
6
 
7
  class WebSearchTool(BaseTool):
8
+ name = "Web Search Tool"
9
+ description = "Searches web content"
10
 
11
  def _run(self, query: str):
12
+
13
  url = f"https://duckduckgo.com/html/?q={query}"
14
 
15
  response = requests.get(url)
 
16
 
17
+ soup = BeautifulSoup(
18
+ response.text,
19
+ "html.parser"
20
+ )
21
+
22
+ results = soup.find_all(
23
+ "a",
24
+ class_="result__a",
25
+ limit=5
26
+ )
27
 
28
+ return "\n".join(
29
+ [result.get_text() for result in results]
30
+ )