Update helper.py

helper.py

@@ -46,7 +46,7 @@ def get_fn(model_name: str, **model_kwargs):
                 chunk_message = chunk.choices[0].delta.content
                 if chunk_message:
                     response_text += chunk_message
-                    yield assistant_message.strip()
+                    yield response_text.strip()
         except Exception as e:
             print(f"Error during generation: {str(e)}")
             yield f"An error occurred: {str(e)}"