Fix error handling when API returns empty

yym68686 · Aug 1, 2024 · e7d701b · e7d701b
1 parent 6916ae4
commit e7d701b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/utils.py b/utils.py
@@ -105,8 +105,7 @@ async def new_generator():
 
         return new_generator()
     except StopAsyncIteration:
-        # 处理生成器为空的情况
-        return async_generator(["data: {'error': 'No data returned'}\n\n"])
+        raise HTTPException(status_code=status_code, detail="data: {'error': 'No data returned'}")
 
 def post_all_models(token, config, api_list):
     all_models = []