AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-16 11:15:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fixed: The choices may be empty. (#876)

Browse Source 
### What problem does this PR solve?


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
dashi6174 2024-05-22 15:29:07 +08:00 committed by GitHub
parent be13429d05
commit 21453ffff0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
rag/llm/chat_model.py
View File

@ -56,6 +56,7 @@ class Base(ABC):
stream=True, stream=True,
**gen_conf) **gen_conf)
for resp in response: for resp in response:
if len(resp.choices) == 0:continue
if not resp.choices[0].delta.content:continue if not resp.choices[0].delta.content:continue
ans += resp.choices[0].delta.content ans += resp.choices[0].delta.content
total_tokens += 1 total_tokens += 1