Update openai.py

Use hasattr instead of checking for choices in response
Chainlit · Oct 20, 2023 · 0628cd6 · 0628cd6
1 parent d52dcff
commit 0628cd6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/chainlit/playground/providers/openai.py b/backend/chainlit/playground/providers/openai.py
@@ -145,7 +145,7 @@ async def create_completion(self, request):
 
         async def create_event_stream():
             async for stream_resp in response:
-                if 'choices' in stream_resp and len(stream_resp.choices) > 0:
+                if hasattr(stream_resp, 'choices') and len(stream_resp.choices) > 0:
                     token = stream_resp.choices[0]["delta"].get("content", "")
                     yield token
                 else: