summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAbhijit Balaji <balaabhijit5@gmail.com>2023-10-05 22:46:51 -0600
committerGitHub <noreply@github.com>2023-10-05 21:46:51 -0700
commit134ee46f5bac32f33c86ef2b21ab8049db025da5 (patch)
tree3ed8daab7b15725bc9de6648ca71d53f509f4877
parente7c9b0fa42a8896130da1f18a5df58a1937997e0 (diff)
downloadsncontinue-134ee46f5bac32f33c86ef2b21ab8049db025da5.tar.gz
sncontinue-134ee46f5bac32f33c86ef2b21ab8049db025da5.tar.bz2
sncontinue-134ee46f5bac32f33c86ef2b21ab8049db025da5.zip
check if attribute choices is present (#535)
-rw-r--r--continuedev/src/continuedev/libs/llm/openai.py2
1 files changed, 1 insertions, 1 deletions
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