pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-18 19:46:44 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: in case response.choices[0].message.content is None. (#6190)

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

#6164

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu
2025-03-18 10:00:27 +08:00
committed by GitHub
parent b0b4b7ba33
commit 7e4d693054
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rag/llm/chat_model.py
View File

@ -49,7 +49,7 @@ class Base(ABC):
model=self.model_name, model=self.model_name,
messages=history, messages=history,
**gen_conf) **gen_conf)
if not response.choices: if any([not response.choices, not response.choices[0].message, not response.choices[0].message.content]):
return "", 0 return "", 0
ans = response.choices[0].message.content.strip() ans = response.choices[0].message.content.strip()
if response.choices[0].finish_reason == "length": if response.choices[0].finish_reason == "length":
@ -184,7 +184,6 @@ class BaiChuanChat(Base):
def _format_params(params): def _format_params(params):
return { return {
"temperature": params.get("temperature", 0.3), "temperature": params.get("temperature", 0.3),
"max_tokens": params.get("max_tokens", 2048),
"top_p": params.get("top_p", 0.85), "top_p": params.get("top_p", 0.85),
} }