diff options
| author | Nate Sesti <sestinj@gmail.com> | 2023-09-28 10:30:35 -0700 |
|---|---|---|
| committer | Nate Sesti <sestinj@gmail.com> | 2023-09-28 10:30:35 -0700 |
| commit | a8e69a02e6897689a1727fb7542ed5684b1348e2 (patch) | |
| tree | b628a2d683a698fa492a2488c73e2b5aeac47932 | |
| parent | a95d465c2f081c1318f70343230d984f6478ea85 (diff) | |
| download | sncontinue-a8e69a02e6897689a1727fb7542ed5684b1348e2.tar.gz sncontinue-a8e69a02e6897689a1727fb7542ed5684b1348e2.tar.bz2 sncontinue-a8e69a02e6897689a1727fb7542ed5684b1348e2.zip | |
fix: :bug: fix azure openai bug for 07 version
| -rw-r--r-- | continuedev/src/continuedev/libs/llm/openai.py | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/continuedev/src/continuedev/libs/llm/openai.py b/continuedev/src/continuedev/libs/llm/openai.py index 6afa4e77..d21bbde1 100644 --- a/continuedev/src/continuedev/libs/llm/openai.py +++ b/continuedev/src/continuedev/libs/llm/openai.py @@ -118,11 +118,12 @@ class OpenAI(LLM): messages=[{"role": "user", "content": prompt}], **args, ): - if "content" in chunk.choices[0].delta: + if len(chunk.choices) > 0 and "content" in chunk.choices[0].delta: yield chunk.choices[0].delta.content else: async for chunk in await openai.Completion.acreate(prompt=prompt, **args): - yield chunk.choices[0].text + if len(chunk.choices) > 0: + yield chunk.choices[0].text async def _stream_chat(self, messages: List[ChatMessage], options): args = self.collect_args(options) |