diff --git a/src/App.vue b/src/App.vue
index 9c5ceb197fbc25eca192e1b02dc18fd82640b6fc..a9bc0bd390c85f9f1c2ba072a1682e6f7ef9c029 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -83,7 +83,8 @@ export default {
         },
         onmessage: (msg) => {
           const data = JSON.parse(msg.data);
-          const finish = data.choices[0].finish_reason === 'stop';
+          const finish_reason = data.choices[0].finish_reason;
+          const finish = finish_reason === 'stop' || finish_reason === 'length';
           const content = data.choices[0].delta.content;
 
           if (finish) {