From 612d51a591abf0745d8186516f322a8944449ba7 Mon Sep 17 00:00:00 2001
From: zhangqian <zhangqian@123.com>
Date: 星期六, 12 十月 2024 20:46:56 +0800
Subject: [PATCH] 处理异常
---
app/service/ragflow.py | 9 ++++++---
1 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/app/service/ragflow.py b/app/service/ragflow.py
index 5fbe175..df131f1 100644
--- a/app/service/ragflow.py
+++ b/app/service/ragflow.py
@@ -51,8 +51,11 @@
# 妫�鏌ュ搷搴旂姸鎬佺爜
if response.status_code == 200:
# 娴佸紡璇诲彇鍝嶅簲
- async for answer in response.aiter_text():
- yield answer
+ try:
+ async for answer in response.aiter_text():
+ yield answer
+ except GeneratorExit as e:
+ print(e)
+ return
else:
yield f"Error: {response.status_code}"
-
--
Gitblit v1.8.0