null check for chunk.choices

This commit is contained in:
Naman 2024-05-25 07:02:02 +05:30
parent c5695975c7
commit f0ee686660

View File

@ -59,7 +59,7 @@ async def stream_openai_response(
full_response = ""
async for chunk in stream: # type: ignore
assert isinstance(chunk, ChatCompletionChunk)
if chunk.choices and chunk.choices[0].delta.content:
if chunk.choices and len(chunk.choices) > 0 and chunk.choices[0].delta and chunk.choices[0].delta.content:
content = chunk.choices[0].delta.content or ""
full_response += content
await callback(content)