[ Add ] added support for time outs to include stream methods

anasfik · Nov 22, 2023 · 4539d6b · 4539d6b
1 parent f63803d
commit 4539d6b
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/src/core/networking/client.dart b/lib/src/core/networking/client.dart
@@ -159,7 +159,10 @@ abstract class OpenAINetworkingClient {
       ]);
     }
 
-    clientForUse.send(request).then((streamedResponse) {
+    clientForUse
+        .send(request)
+        .timeout(OpenAIConfig.requestsTimeOut)
+        .then((streamedResponse) {
       streamedResponse.stream.listen(
         (value) {
           final data = utf8.decode(value);
@@ -362,7 +365,8 @@ abstract class OpenAINetworkingClient {
       }
 
       OpenAILogger.logStartRequest(to);
-      clientForUse.send(request).then(
+
+      clientForUse.send(request).timeout(OpenAIConfig.requestsTimeOut).then(
         (respond) {
           OpenAILogger.startReadStreamResponse();