stream completion log probs

This commit is contained in:
Xi Yan 2025-01-15 16:50:58 -08:00
parent 9c13a7b76b
commit 44069404fd

View file

@ -178,6 +178,7 @@ async def process_completion_stream_response(
yield CompletionResponseStreamChunk(
delta=text,
stop_reason=stop_reason,
logprobs=convert_openai_completion_logprobs(choice.logprobs),
)
if finish_reason:
if finish_reason in ["stop", "eos", "eos_token"]: