Fix bug in playground for Azure OpenAI (#493)

* Fix bug in playground for Azure OpenAI * Update openai.py Use hasattr instead of checking for choices in response
Chainlit · Oct 22, 2023 · f8e649a · f8e649a
1 parent 2b9044b
commit f8e649a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/backend/chainlit/playground/providers/openai.py b/backend/chainlit/playground/providers/openai.py
@@ -145,8 +145,11 @@ async def create_completion(self, request):
 
         async def create_event_stream():
             async for stream_resp in response:
-                token = stream_resp.choices[0]["delta"].get("content", "")
-                yield token
+                if hasattr(stream_resp, 'choices') and len(stream_resp.choices) > 0:
+                    token = stream_resp.choices[0]["delta"].get("content", "")
+                    yield token
+                else:
+                    continue
 
         return StreamingResponse(create_event_stream())