From db8b4a7b56bc8620c45d62d4d6f5b5891e8418dd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Johan=20Sebastian=20Blandon=20Ordo=C3=B1ez?=
 <45244286+SebastianBlandon@users.noreply.github.com>
Date: Fri, 27 Oct 2023 12:10:08 -0500
Subject: [PATCH] fix: Update OpenAIApi.cs for chunks incomplete bug

---
 Runtime/OpenAIApi.cs | 11 +++++------
 1 file changed, 5 insertions(+), 6 deletions(-)

diff --git a/Runtime/OpenAIApi.cs b/Runtime/OpenAIApi.cs
index 00e5572..2378eab 100644
--- a/Runtime/OpenAIApi.cs
+++ b/Runtime/OpenAIApi.cs
@@ -112,8 +112,8 @@ private async void DispatchRequest<T>(string path, string method, Action<List<T>
                 request.method = method;
                 request.SetHeaders(Configuration, ContentType.ApplicationJson);
                 
-                var asyncOperation = request.SendWebRequest();
-
+                request.SendWebRequest();
+                bool isDone = false;
                 do
                 {
                     List<T> dataList = new List<T>();
@@ -122,10 +122,9 @@ private async void DispatchRequest<T>(string path, string method, Action<List<T>
                     foreach (string line in lines)
                     {
                         var value = line.Replace("data: ", "");
-                        
-                        if (value.Contains("[DONE]")) 
+                        if (value.Contains("stop")) 
                         {
-                            onComplete?.Invoke();
+                            isDone = true;
                             break;
                         }
                         
@@ -145,7 +144,7 @@ private async void DispatchRequest<T>(string path, string method, Action<List<T>
                     
                     await Task.Yield();
                 }
-                while (!asyncOperation.isDone && !token.IsCancellationRequested);
+                while (!isDone);
                 
                 onComplete?.Invoke();
             }