From 134ee46f5bac32f33c86ef2b21ab8049db025da5 Mon Sep 17 00:00:00 2001 From: Abhijit Balaji Date: Thu, 5 Oct 2023 22:46:51 -0600 Subject: check if attribute choices is present (#535) --- continuedev/src/continuedev/libs/llm/openai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/continuedev/src/continuedev/libs/llm/openai.py b/continuedev/src/continuedev/libs/llm/openai.py index db3c9852..7541d92e 100644 --- a/continuedev/src/continuedev/libs/llm/openai.py +++ b/continuedev/src/continuedev/libs/llm/openai.py @@ -134,7 +134,7 @@ class OpenAI(LLM): stream=True, **args, ): - if len(chunk.choices) == 0: + if not hasattr(chunk, "choices") or len(chunk.choices) == 0: continue yield chunk.choices[0].delta -- cgit v1.2.3-70-g09d2