Spaces:
Runtime error
Runtime error
File size: 1,309 Bytes
fb9c306 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
from dataclasses import dataclass
import requests
from fastapi import HTTPException
from graphgen.utils import logger
BING_SEARCH_V7_ENDPOINT = "https://api.bing.microsoft.com/v7.0/search"
BING_MKT = "en-US"
@dataclass
class BingSearch:
"""
Bing Search client to search with Bing.
"""
subscription_key: str
def search(self, query: str, num_results: int = 1):
"""
Search with Bing and return the contexts.
:param query: The search query.
:param num_results: The number of results to return.
:return: A list of search results.
"""
params = {"q": query, "mkt": BING_MKT, "count": num_results}
response = requests.get(
BING_SEARCH_V7_ENDPOINT,
headers={"Ocp-Apim-Subscription-Key": self.subscription_key},
params=params,
timeout=10,
)
if not response.ok:
logger.error("Search engine error: %s", response.text)
raise HTTPException(response.status_code, "Search engine error.")
json_content = response.json()
try:
contexts = json_content["webPages"]["value"][:num_results]
except KeyError:
logger.error("Error encountered: %s", json_content)
return []
return contexts
|