From c2ffa86605a3f194b7cea96dccb95c9e96396184 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Pedro=20Scarmignani=20Lumi?=
 <jpedro.lumi@gmail.com>
Date: Wed, 3 Apr 2024 14:29:43 -0300
Subject: [PATCH 1/2] FIX: request done before all chunks are received

---
 Runtime/OpenAIApi.cs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/Runtime/OpenAIApi.cs b/Runtime/OpenAIApi.cs
index 5dd54f6..88ec14b 100644
--- a/Runtime/OpenAIApi.cs
+++ b/Runtime/OpenAIApi.cs
@@ -112,6 +112,7 @@ private async void DispatchRequest<T>(string path, string method, Action<List<T>
                 request.SetHeaders(Configuration, ContentType.ApplicationJson);
                 
                 var asyncOperation = request.SendWebRequest();
+                bool isDone = false;
 
                 do
                 {
@@ -124,6 +125,7 @@ private async void DispatchRequest<T>(string path, string method, Action<List<T>
                         
                         if (value.Contains("[DONE]")) 
                         {
+                            isDone = true;
                             onComplete?.Invoke();
                             break;
                         }
@@ -144,7 +146,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();
             }

From 73f89d07a5e362acfc2c72b5a97567bf0aed7951 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jo=C3=A3o=20Pedro=20Scarmignani=20Lumi?=
 <jpedro.lumi@gmail.com>
Date: Wed, 3 Apr 2024 15:27:37 -0300
Subject: [PATCH 2/2] FIX: onComplete action being invoked twice

---
 Runtime/OpenAIApi.cs | 1 -
 1 file changed, 1 deletion(-)

diff --git a/Runtime/OpenAIApi.cs b/Runtime/OpenAIApi.cs
index 88ec14b..6468957 100644
--- a/Runtime/OpenAIApi.cs
+++ b/Runtime/OpenAIApi.cs
@@ -126,7 +126,6 @@ private async void DispatchRequest<T>(string path, string method, Action<List<T>
                         if (value.Contains("[DONE]")) 
                         {
                             isDone = true;
-                            onComplete?.Invoke();
                             break;
                         }