1
0
Эх сурвалжийг харах

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

Jun Siang Cheah 1 жил өмнө
parent
commit
83f086ccdd

+ 8 - 1
backend/apps/rag/main.py

@@ -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()

+ 4 - 0
backend/constants.py

@@ -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."
+    )