null check for chunk.choices

2024-05-25 07:02:02 +05:30 · 2024-05-25 07:02:02 +05:30 · f0ee686660
commit f0ee686660
parent c5695975c7
1 changed files with 1 additions and 1 deletions
--- a/backend/llm.py
+++ b/backend/llm.py
@ -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)