handle none type in llm.py

2024-05-25 00:32:13 +05:30 · 2024-05-25 00:32:13 +05:30 · c5695975c7
commit c5695975c7
parent 23e631765e
1 changed files with 4 additions and 3 deletions
--- a/backend/llm.py
+++ b/backend/llm.py
@ -59,6 +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:
            content = chunk.choices[0].delta.content or ""
            full_response += content
            await callback(content)