Fix the bug with incomplete error handling.

yym68686 · Aug 8, 2024 · d738fa5 · d738fa5
1 parent 33bccc2
commit d738fa5
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/utils.py b/utils.py
@@ -109,6 +109,11 @@ async def new_generator():
         return new_generator()
     except StopAsyncIteration:
         raise HTTPException(status_code=status_code, detail="data: {'error': 'No data returned'}")
+    except Exception as e:
+        logger.error(f"error_handling_wrapper Exception: {str(e)}")
+        import traceback
+        logger.error(f"error_handling_wrapper Exception: {traceback.print_exc()}")
+        raise HTTPException(status_code=status_code, detail="data: {'error': 'No data returned'}")
 
 def post_all_models(token, config, api_list):
     all_models = []