Makes use of streams on API responses

cloudstudio · Jan 11, 2024 · 7d08354 · 7d08354
1 parent 656e38e
commit 7d08354
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/src/Traits/MakesHttpRequests.php b/src/Traits/MakesHttpRequests.php
@@ -3,6 +3,7 @@
 namespace Cloudstudio\Ollama\Traits;
 
 use Illuminate\Support\Facades\Http;
+use GuzzleHttp\Client;
 
 trait MakesHttpRequests
 {
@@ -17,8 +18,19 @@ trait MakesHttpRequests
     protected function sendRequest(string $urlSuffix, array $data, string $method = 'post')
     {
         $url = config('ollama-laravel.url') . $urlSuffix;
-        $response = Http::timeout(config('ollama-laravel.connection.timeout'))->$method($url, $data);
 
-        return $response->json();
+        if (!empty($data['stream']) && $data['stream'] === true) {
+            $client = new Client();
+            $response = $client->request($method, $url, [
+                'json' => $data,
+                'stream' => true,
+                'timeout' => config('ollama-laravel.connection.timeout'),
+            ]);
+
+            return $response;
+        } else {
+            $response = Http::timeout(config('ollama-laravel.connection.timeout'))->$method($url, $data);
+            return $response->json();
+        }
     }
 }