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

### 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
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

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