CodeCow/Scrapegraph-ai
1
0
Fork 0
mirror of https://github.com/VinciGit00/Scrapegraph-ai.git synced 2026-07-01 21:00:48 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

chore: correct search engine name
Some checks failed
/ build (push) Has been cancelled
Details

Browse Source
This commit is contained in:
Federico Aguzzi 2024-07-14 22:39:34 +02:00
parent 5c92186081
commit 7ba2f6ae0b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scrapegraphai/utils/research_web.py
View File

@ -11,9 +11,9 @@ def search_on_web(query: str, search_engine: str = "Google", max_results: int =
Args: Args:
query (str): The search query to find on the internet. query (str): The search query to find on the internet.
search_engine (str, optional): Specifies the search engine to use, options include 'Google', 'DuckDuckGo', 'Bing', or 'SearcNGX'. Default is 'Google'. search_engine (str, optional): Specifies the search engine to use, options include 'Google', 'DuckDuckGo', 'Bing', or 'SearXNG'. Default is 'Google'.
max_results (int, optional): The maximum number of search results to return. max_results (int, optional): The maximum number of search results to return.
port (int, optional): The port number to use when searching with 'SearcNGX'. Default is 8080. port (int, optional): The port number to use when searching with 'SearXNG'. Default is 8080.
Returns: Returns:
List[str]: A list of URLs as strings that are the search results. List[str]: A list of URLs as strings that are the search results.
@ -53,7 +53,7 @@ def search_on_web(query: str, search_engine: str = "Google", max_results: int =
search_results.append(link) search_results.append(link)
return search_results return search_results
elif search_engine.lower() == "searcngx": elif search_engine.lower() == "searxng":
url = f"http://localhost:{port}" url = f"http://localhost:{port}"
params = {"q": query, "format": "json"} params = {"q": query, "format": "json"}
@ -66,4 +66,4 @@ def search_on_web(query: str, search_engine: str = "Google", max_results: int =
return limited_results return limited_results
else: else:
raise ValueError("The only search engines available are DuckDuckGo, Google, Bing, or SearcNGX") raise ValueError("The only search engines available are DuckDuckGo, Google, Bing, or SearXNG")