Browse Source

fix OpenAI llm return bug (#1728)

### What problem does this PR solve?

fix OpenAI llm return bug

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.9.0
黄腾 1 year ago
parent
commit
61096596bc
No account linked to committer's email address
1 changed files with 2 additions and 0 deletions
  1. 2
    0
      rag/llm/chat_model.py

+ 2
- 0
rag/llm/chat_model.py View File

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

Loading…
Cancel
Save