Allow http client options

sergix44 · Mar 23, 2024 · 00c6b99 · 00c6b99
1 parent 64017a0
commit 00c6b99
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/Client.php b/src/Client.php
@@ -37,9 +37,9 @@ class Client extends RemoteClient
 
     private ?string $hfToken;
 
-    public function __construct(string $src, ?string $hfToken = null, ?Config $config = null)
+    public function __construct(string $src, ?string $hfToken = null, ?Config $config = null, array $httpClientOptions = [])
     {
-        parent::__construct($src);
+        parent::__construct($src, $httpClientOptions);
         $this->config = $config ?? $this->http('get', self::HTTP_CONFIG, dto: Config::class);
         $this->loadEndpoints($this->config->dependencies);
         $this->sessionHash = substr(md5(microtime()), 0, 11);

diff --git a/src/Client/RemoteClient.php b/src/Client/RemoteClient.php
@@ -17,7 +17,7 @@ abstract class RemoteClient extends RegisterEvents
 
     protected HydratorInterface $hydrator;
 
-    public function __construct(string $src)
+    public function __construct(string $src, array $httpClientOptions = [])
     {
         if (
             ! str_starts_with($src, 'http://') &&
@@ -32,13 +32,13 @@ public function __construct(string $src)
 
         $this->hydrator = new Hydrator();
 
-        $this->httpClient = new Guzzle([
+        $this->httpClient = new Guzzle(array_merge([
             'base_uri' => str_replace('ws', 'http', $this->src),
             'headers' => [
                 'User-Agent' => 'gradio_client_php/1.0',
                 'Accept' => 'application/json',
             ],
-        ]);
+        ], $httpClientOptions));
     }
 
     protected function http(string $method, string $uri, array $params = [], array $opt = [], ?string $dto = null)