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

fix MiniMax chat bug (#1733)

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

#1717   fix MiniMax chat bug

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
This commit is contained in:
黄腾 2024-07-29 19:35:16 +08:00 committed by GitHub
parent 6012f376ca
commit 29f7f8b81e
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
rag/llm/chat_model.py
View File

@ -564,12 +564,15 @@ class MiniMaxChat(Base):
) )
for resp in response.text.split("\n\n")[:-1]: for resp in response.text.split("\n\n")[:-1]:
resp = json.loads(resp[6:]) resp = json.loads(resp[6:])
if "delta" in resp["choices"][0]: text = ""
if "choices" in resp and "delta" in resp["choices"][0]:
text = resp["choices"][0]["delta"]["content"] text = resp["choices"][0]["delta"]["content"]
else:
continue
ans += text ans += text
total_tokens += num_tokens_from_string(text) total_tokens = (
total_tokens + num_tokens_from_string(text)
if "usage" not in resp
else resp["usage"]["total_tokens"]
)
yield ans yield ans
except Exception as e: except Exception as e: