test: chat and text completion stream tests

Author: srcnalt

Tests/ChatApiTests.cs

@@ -1,6 +1,9 @@
-using System.Collections.Generic;
 using NUnit.Framework;
+using System.Threading;
 using System.Threading.Tasks;
+using System.Collections.Generic;
+using System.Linq;
+using UnityEngine;
 
 namespace OpenAI.Tests
 {
@@ -22,5 +25,37 @@ public async Task Create_Chat_Completion()
             var res = await openai.CreateChatCompletion(req);
             Assert.NotNull(res);
         }
+        
+        [Test]
+        public async Task Create_Chat_Completion_Stream()
+        {
+            bool responseReceived = false;
+            float timeout = 10;
+            float time = 0;
+            
+            var req = new CreateChatCompletionRequest
+            {
+                Model = "gpt-3.5-turbo",
+                Messages = new List<ChatMessage>()
+                {
+                    new ChatMessage() { Role = "user", Content = "Hello!" }
+                },
+                Temperature = 0,
+                Stream = true
+            };
+            
+            openai.CreateChatCompletionAsync(req, null, () =>
+            {
+                responseReceived = true;
+            }, new CancellationTokenSource());
+            
+            while (!responseReceived && time < timeout)
+            {
+                await Task.Delay(100);
+                time += 0.1f;
+            }
+            
+            Assert.IsTrue(responseReceived);
+        }
     }
 }

Tests/CompletionsApiTests.cs

@@ -1,4 +1,5 @@
 using NUnit.Framework;
+using System.Threading;
 using System.Threading.Tasks;
 
 namespace OpenAI.Tests
@@ -8,7 +9,7 @@ public class CompletionsApiTests
         private OpenAIApi openai = new OpenAIApi();
 
         [Test]
-        public async Task Create_Completion()
+        public async Task Create_Text_Completion()
         {
             var req = new CreateCompletionRequest
             {
@@ -20,5 +21,35 @@ public async Task Create_Completion()
             var res = await openai.CreateCompletion(req);
             Assert.NotNull(res);
         }
+        
+        [Test]
+        public async Task Create_Text_Completion_Stream()
+        {
+            bool responseReceived = false;
+            float timeout = 5;
+            float time = 0;
+            
+            var req = new CreateCompletionRequest
+            {
+                Model = "text-davinci-003",
+                Prompt = "Say this is a test",
+                MaxTokens = 7,
+                Temperature = 0,
+                Stream = true
+            };
+            
+            openai.CreateCompletionAsync(req, null, () => 
+            {
+                responseReceived = true;
+            }, new CancellationTokenSource());
+            
+            while (!responseReceived && time < timeout)
+            {
+                await Task.Delay(100);
+                time += 0.1f;
+            }
+            
+            Assert.IsTrue(responseReceived);
+        }
     }
 }