fix: do not return raw search exception due to API keys in URLs

This commit is contained in:
Jun Siang Cheah 2024-05-06 16:54:15 +08:00
parent 99e4edd364
commit 83f086ccdd
2 changed files with 12 additions and 1 deletions

View File

@ -556,7 +556,14 @@ def resolve_hostname(hostname):
@app.post("/websearch")
def store_websearch(form_data: SearchForm, user=Depends(get_current_user)):
try:
web_results = search_web(form_data.query)
try:
web_results = search_web(form_data.query)
except Exception as e:
log.exception(e)
raise HTTPException(
status_code=status.HTTP_400_BAD_REQUEST,
detail=ERROR_MESSAGES.WEB_SEARCH_ERROR,
)
urls = [result.link for result in web_results]
loader = get_web_loader(urls)
data = loader.load()

View File

@ -75,3 +75,7 @@ class ERROR_MESSAGES(str, Enum):
INVALID_URL = (
"Oops! The URL you provided is invalid. Please double-check and try again."
)
WEB_SEARCH_ERROR = (
"Oops! Something went wrong while searching the web. Please try again later."
)