phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-02 00:34:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: message.content can be None

Browse source
This commit is contained in:
seyeong-han 2024-11-24 01:20:04 -06:00
parent a7df3e539c
commit 061a1cc790
1 changed files with 4 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/utils/inference/openai_compat.py
View file

@ -47,7 +47,10 @@ def text_from_choice(choice) -> str:
return choice.delta.content return choice.delta.content
if hasattr(choice, "message"): if hasattr(choice, "message"):
return choice.message.content try:
return choice.message.content
except:
return choice.text
return choice.text return choice.text