Merge pull request #5 from utopia-php/make-timeouts-configurable

chore: update timeouts

Author: eldadfux

src/Agents/Adapter.php

@@ -25,6 +25,13 @@ abstract class Adapter
      */
     protected int $outputTokens = 0;
 
+    /**
+     * Request timeout in seconds
+     *
+     * @var int
+     */
+    protected int $timeout = 90;
+
     /**
      * Get the adapter name
      *
@@ -145,4 +152,27 @@ public function getTotalTokens(): int
     {
         return $this->inputTokens + $this->outputTokens;
     }
+
+    /**
+     * Set timeout in seconds
+     *
+     * @param  int  $timeout
+     * @return self
+     */
+    public function setTimeout(int $timeout): self
+    {
+        $this->timeout = $timeout;
+
+        return $this;
+    }
+
+    /**
+     * Get timeout in seconds
+     *
+     * @return int
+     */
+    public function getTimeout(): int
+    {
+        return $this->timeout;
+    }
 }

src/Agents/Adapters/Anthropic.php

@@ -57,18 +57,21 @@ class Anthropic extends Adapter
      * @param  string  $model
      * @param  int  $maxTokens
      * @param  float  $temperature
+     * @param  int  $timeout
      *
      * @throws \Exception
      */
     public function __construct(
         string $apiKey,
         string $model = self::MODEL_CLAUDE_3_SONNET,
         int $maxTokens = 1024,
-        float $temperature = 1.0
+        float $temperature = 1.0,
+        int $timeout = 90
     ) {
         $this->apiKey = $apiKey;
         $this->maxTokens = $maxTokens;
         $this->temperature = $temperature;
+        $this->timeout = $timeout;
         $this->setModel($model);
     }
 
@@ -89,7 +92,7 @@ public function send(array $messages, ?callable $listener = null): Message
 
         $client = new Client();
         $client
-            ->setTimeout(90)
+            ->setTimeout($this->timeout)
             ->addHeader('x-api-key', $this->apiKey)
             ->addHeader('anthropic-version', '2023-06-01')
             ->addHeader('content-type', Client::CONTENT_TYPE_APPLICATION_JSON);

src/Agents/Adapters/Deepseek.php

@@ -40,25 +40,33 @@ class Deepseek extends Adapter
      */
     protected float $temperature;
 
+    /**
+     * @var int
+     */
+    protected int $timeout;
+
     /**
      * Create a new Deepseek adapter
      *
      * @param  string  $apiKey
      * @param  string  $model
      * @param  int  $maxTokens
      * @param  float  $temperature
+     * @param  int  $timeout
      *
      * @throws \Exception
      */
     public function __construct(
         string $apiKey,
         string $model = self::MODEL_DEEPSEEK_CHAT,
         int $maxTokens = 1024,
-        float $temperature = 1.0
+        float $temperature = 1.0,
+        int $timeout = 90
     ) {
         $this->apiKey = $apiKey;
         $this->maxTokens = $maxTokens;
         $this->temperature = $temperature;
+        $this->timeout = $timeout;
         $this->setModel($model);
     }
 
@@ -79,7 +87,7 @@ public function send(array $messages, ?callable $listener = null): Message
 
         $client = new Client();
         $client
-            ->setTimeout(90)
+            ->setTimeout($this->timeout)
             ->addHeader('authorization', 'Bearer '.$this->apiKey)
             ->addHeader('content-type', Client::CONTENT_TYPE_APPLICATION_JSON);
 

src/Agents/Adapters/OpenAI.php

@@ -55,6 +55,11 @@ class OpenAI extends Adapter
      */
     protected string $endpoint;
 
+    /**
+     * @var int
+     */
+    protected int $timeout;
+
     /**
      * Create a new OpenAI adapter
      *
@@ -63,6 +68,7 @@ class OpenAI extends Adapter
      * @param  int  $maxTokens
      * @param  float  $temperature
      * @param  string|null  $endpoint
+     * @param  int  $timeout
      *
      * @throws \Exception
      */
@@ -71,12 +77,14 @@ public function __construct(
         string $model = self::MODEL_GPT_3_5_TURBO,
         int $maxTokens = 1024,
         float $temperature = 1.0,
-        ?string $endpoint = null
+        ?string $endpoint = null,
+        int $timeout = 90
     ) {
         $this->apiKey = $apiKey;
         $this->maxTokens = $maxTokens;
         $this->temperature = $temperature;
         $this->endpoint = $endpoint ?? self::ENDPOINT;
+        $this->timeout = $timeout;
         $this->setModel($model);
     }
 
@@ -97,7 +105,7 @@ public function send(array $messages, ?callable $listener = null): Message
 
         $client = new Client();
         $client
-            ->setTimeout(90)
+            ->setTimeout($this->timeout)
             ->addHeader('authorization', 'Bearer '.$this->apiKey)
             ->addHeader('content-type', Client::CONTENT_TYPE_APPLICATION_JSON);
 

src/Agents/Adapters/Perplexity.php

@@ -42,6 +42,7 @@ class Perplexity extends OpenAI
      * @param  int  $maxTokens
      * @param  float  $temperature
      * @param  string|null  $endpoint
+     * @param  int  $timeout
      *
      * @throws \Exception
      */
@@ -50,14 +51,16 @@ public function __construct(
         string $model = self::MODEL_SONAR,
         int $maxTokens = 1024,
         float $temperature = 1.0,
-        ?string $endpoint = null
+        ?string $endpoint = null,
+        int $timeout = 90
     ) {
         parent::__construct(
             $apiKey,
             $model,
             $maxTokens,
             $temperature,
-            $endpoint ?? self::ENDPOINT
+            $endpoint ?? self::ENDPOINT,
+            $timeout
         );
     }
 

src/Agents/Adapters/XAI.php

@@ -27,6 +27,7 @@ class XAI extends OpenAI
      * @param  int  $maxTokens
      * @param  float  $temperature
      * @param  string|null  $endpoint
+     * @param  int  $timeout
      *
      * @throws \Exception
      */
@@ -35,14 +36,16 @@ public function __construct(
         string $model = self::MODEL_GROK_2_LATEST,
         int $maxTokens = 1024,
         float $temperature = 1.0,
-        ?string $endpoint = null
+        ?string $endpoint = null,
+        int $timeout = 90
     ) {
         parent::__construct(
             $apiKey,
             $model,
             $maxTokens,
             $temperature,
-            $endpoint ?? self::ENDPOINT
+            $endpoint ?? self::ENDPOINT,
+            $timeout
         );
     }