diff options
author | Abhijit Balaji <balaabhijit5@gmail.com> | 2023-10-05 22:46:51 -0600 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-10-05 21:46:51 -0700 |
commit | 134ee46f5bac32f33c86ef2b21ab8049db025da5 (patch) | |
tree | 3ed8daab7b15725bc9de6648ca71d53f509f4877 | |
parent | e7c9b0fa42a8896130da1f18a5df58a1937997e0 (diff) | |
download | sncontinue-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.py | 2 |
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 |