Update helper.py

helper.py

@@ -39,9 +39,8 @@ def get_fn(model_name: str, **model_kwargs) -> Callable:
             response_text = ""
             # Iterate over the streaming response
             for chunk in response:
-                if 'choices' in chunk and len(chunk['choices']) > 0:
-                    delta = chunk['choices'][0].get('delta', {})
-                    content = delta.get('content', '')
+                if len(chunk.choices[0]) > 0:
+                    content = chunk.choices[0].delta.content
                     if content:
                         response_text += content
                         yield response_text.strip()