Update helper.py

helper.py

@@ -41,8 +41,13 @@ def get_fn(model_name: str, **model_kwargs):
                 stream=True, 
                 response_format={"type": "text"},
             )
-            assistant_message = response.choices[0].message.content
-            yield assistant_message.strip()
+
+            response_text = ""
+            for chunk in response:
+                chunk_message = chunk.choices[0].delta.content
+                if chunk_message:
+                    response_text += chunk_message
+                    yield assistant_message.strip()
         except Exception as e:
             print(f"Error during generation: {str(e)}")
             yield f"An error occurred: {str(e)}"