From 140a3ef943aeb47cf3f058c93b12f705d256e10a Mon Sep 17 00:00:00 2001 From: Debanjum Date: Wed, 17 Sep 2025 17:23:52 +0530 Subject: [PATCH] Avoid unbound chunk variable error in ai api call from completion func --- src/khoj/processor/conversation/openai/utils.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/khoj/processor/conversation/openai/utils.py b/src/khoj/processor/conversation/openai/utils.py index df4f181a..46e275ff 100644 --- a/src/khoj/processor/conversation/openai/utils.py +++ b/src/khoj/processor/conversation/openai/utils.py @@ -159,6 +159,7 @@ def completion_with_backoff( tool_calls: list[ToolCall] = [] thoughts = "" aggregated_response = "" + chunk = None if stream: with client.beta.chat.completions.stream( messages=formatted_messages, # type: ignore @@ -966,6 +967,8 @@ async def astream_thought_processor( chunk_data = chunk.model_dump() # Skip chunks that don't have the required object field or have invalid values + if "object" in chunk_data and chunk_data.get("object") == "": + chunk_data["object"] = "chat.completion.chunk" if not chunk_data.get("object") or chunk_data.get("object") != "chat.completion.chunk": logger.warning(f"Skipping invalid chunk with object field: {chunk_data.get('object', 'missing')}") continue