Handle end of chunk logic in openai stream processor

2026-03-04 21:29:12 +00:00 · 2024-08-20 10:50:09 -07:00
parent 029775420c
commit f6ce2fd432
1 changed files with 2 additions and 0 deletions
--- a/src/khoj/processor/conversation/openai/utils.py
+++ b/src/khoj/processor/conversation/openai/utils.py
@@ -56,6 +56,8 @@ def completion_with_backoff(
    )
    aggregated_response = ""
    for chunk in chat:
+        if len(chunk.choices) == 0:
+            continue
        delta_chunk = chunk.choices[0].delta  # type: ignore
        if isinstance(delta_chunk, str):
            aggregated_response += delta_chunk