AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-14 05:55:59 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix OpenAI llm return bug (#1728)

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

fix OpenAI llm return bug

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
黄腾 2024-07-29 09:21:31 +08:00 committed by GitHub
parent 549d67e281
commit 61096596bc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/chat_model.py
View File

@ -62,6 +62,8 @@ class Base(ABC):
**gen_conf) **gen_conf)
for resp in response: for resp in response:
if not resp.choices:continue if not resp.choices:continue
if not resp.choices[0].delta.content:
resp.choices[0].delta.content = ""
ans += resp.choices[0].delta.content ans += resp.choices[0].delta.content
total_tokens = ( total_tokens = (
( (