Skip to content

Commit f1f1662

Browse files
authored
Added a default value for the "content" field in ChatCompletionResponse if it is empty or missing. (#25)
1 parent d4d5e16 commit f1f1662

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

src/inference_endpoint/openai/openai_adapter.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,4 +158,9 @@ def decode_endpoint_response(
158158
# Set default values for optional fields if missing
159159
response_dict["choices"][0]["message"]["refusal"] = "None"
160160
response_dict["choices"][0]["logprobs"] = {"content": [], "refusal": []}
161+
if (
162+
"content" not in response_dict["choices"][0]["message"]
163+
or response_dict["choices"][0]["message"]["content"] is None
164+
):
165+
response_dict["choices"][0]["message"]["content"] = "None"
161166
return CreateChatCompletionResponse(**response_dict, ignore_extra=True)

0 commit comments

Comments
 (0)