Update middlewares/utils.py

middlewares/utils.py

@@ -71,16 +71,19 @@ def gen_augmented_prompt_via_websearch(
     chunk_size=512,
 ):
    
-
-    search_results = []
-    reranked_results = []
-    if search_vendor == "Google":
-        search_results = googleSearchClient.search(prompt, n_crawl)
-    elif search_vendor == "Bing":
-        search_results = bingSearchClient.search(prompt, n_crawl)
-
-    if len(search_results) > 0:
-        reranked_results = rerank(prompt, top_k, search_results, chunk_size)
+    try:
+        search_results = []
+        reranked_results = []
+        if search_vendor == "Google":
+            search_results = googleSearchClient.search(prompt, n_crawl)
+        elif search_vendor == "Bing":
+            print('[Bing search enabled]')
+            search_results = bingSearchClient.search(prompt, n_crawl)
+            print('[Bing search completed]') 
+        if len(search_results) > 0:
+            reranked_results = rerank(prompt, top_k, search_results, chunk_size)
+    except Exception as e:
+        print(e)
 
     links = []
     context = ""