UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-16 13:40:55 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend/apps/rag/main.py
View File

@ -556,7 +556,14 @@ def resolve_hostname(hostname):
@app.post("/websearch") @app.post("/websearch")
def store_websearch(form_data: SearchForm, user=Depends(get_current_user)): def store_websearch(form_data: SearchForm, user=Depends(get_current_user)):
try: 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] urls = [result.link for result in web_results]
loader = get_web_loader(urls) loader = get_web_loader(urls)
data = loader.load() data = loader.load()

4
backend/constants.py
View File

@ -75,3 +75,7 @@ class ERROR_MESSAGES(str, Enum):
INVALID_URL = ( INVALID_URL = (
"Oops! The URL you provided is invalid. Please double-check and try again." "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."
)