mirror of
https://github.com/JasonYANG170/CodeGeeX4.git
synced 2024-11-23 20:26:29 +00:00
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
|
"""
|
||
|
|
Bing Search
|
||
|
|
"""
|
||
|
|
import os
|
||
|
|
|
||
|
|
import requests
|
||
|
|
from backend.apis.api import API
|
||
|
|
|
||
|
|
BING_API_KEY = os.getenv('BING_API_KEY')
|
||
|
|
|
||
|
|
|
||
|
|
class BingSearchAPI(API):
|
||
|
|
def __init__(self):
|
||
|
|
self.url = "https://api.bing.microsoft.com/v7.0/search"
|
||
|
|
|
||
|
|
def search(self, query, freshness=None):
|
||
|
|
"""
|
||
|
|
Search with bing
|
||
|
|
|
||
|
|
References: https://docs.microsoft.com/en-us/bing/search-apis/bing-web-search/overview
|
||
|
|
"""
|
||
|
|
response = requests.get(
|
||
|
|
url=self.url,
|
||
|
|
headers={"Ocp-Apim-Subscription-Key": BING_API_KEY},
|
||
|
|
params={
|
||
|
|
"q": query,
|
||
|
|
"mkt": 'zh-CN',
|
||
|
|
"freshness": freshness,
|
||
|
|
},
|
||
|
|
timeout=10,
|
||
|
|
)
|
||
|
|
try:
|
||
|
|
json_content = response.json()
|
||
|
|
contexts = json_content['webPages']['value'][:4]
|
||
|
|
search_res = [{
|
||
|
|
"url": item['url'],
|
||
|
|
"title": item['name'],
|
||
|
|
"snippet": item['snippet']
|
||
|
|
} for item in contexts]
|
||
|
|
return search_res
|
||
|
|
except Exception as e:
|
||
|
|
print(f"Searching failed, caused by {e}")
|
||
|
|
return []
|