fix: fix infinite retry loop in async_chat_streamly when success

The original code continues the retry loop even after a successful response.
This commit is contained in:
N0bodycan 2025-12-09 14:01:30 +08:00 committed by GitHub
parent c51e6b2a58
commit a393a5aa35
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -187,6 +187,9 @@ class Base(ABC):
ans = delta_ans
total_tokens += tol
yield ans
yield total_tokens
return
except Exception as e:
e = await self._exceptions_async(e, attempt)
if e:
@ -194,8 +197,6 @@ class Base(ABC):
yield total_tokens
return
yield total_tokens
def _length_stop(self, ans):
if is_chinese([ans]):
return ans + LENGTH_NOTIFICATION_CN