Handling finish_reason when using SSE

polterguy · Sep 24, 2023 · 813825a · 813825a
1 parent 85b12b6
commit 813825a
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/backend/files/system/openai/front.files/chat/default.js b/backend/files/system/openai/front.files/chat/default.js
@@ -273,6 +273,14 @@ function aista_submit_form(speech) {
         return; // Returning early ...
       }
 
+      if (obj.finish_reason) {
+
+        // We've got a finish reason from OpenAI.
+        const answer = window.document.getElementsByClassName('ainiro-has-more')[0];
+        answer.className = answer.className + ' ' + obj.finish_reason;
+        return;
+      }
+
       // Appending response to temporary result.
       ainiroTempContent += obj.message;
 

diff --git a/backend/files/system/openai/magic.startup/magic.ai.chat.hl b/backend/files/system/openai/magic.startup/magic.ai.chat.hl
@@ -283,6 +283,12 @@ select prompt, max(completion) as completion
                                        set-value:x:@.finish_reason
                                           get-value:x:@json2lambda/*/choices/0/*/finish_reason
 
+                                       // Signaling client
+                                       unwrap:x:+/*/*
+                                       sockets.signal:x:@.session
+                                          args
+                                             finish_reason:x:@.finish_reason
+
                                  // Making sure we have a message.
                                  if
                                     and