Merge pull request #51 from deepseek-php/feat-support-pass-max-tokens-option

feat / support pass max tokens option

Author: omaralalwi

README-AR.md

@@ -100,6 +100,7 @@ $response = $client
     ->withModel(Models::CODER->value)
     ->withStream()
     ->withTemperature(1.2)
+    ->setMaxTokens(8192)
     ->query('Explain quantum computing in simple terms')
     ->run();
 

README-CN.md

@@ -99,6 +99,7 @@ $response = $client
     ->withModel(Models::CODER->value)
     ->withStream()
     ->withTemperature(1.2)
+    ->setMaxTokens(8192)
     ->query('Explain quantum computing in simple terms')
     ->run();
 

README.md

@@ -99,6 +99,7 @@ $response = $client
     ->withModel(Models::CODER->value)
     ->withStream()
     ->setTemperature(1.2)
+    ->setMaxTokens(8192)
     ->query('Explain quantum computing in simple terms')
     ->run();
 

composer.json

@@ -49,7 +49,7 @@
             "role": "creator"
         }
     ],
-    "version": "2.0.4",
+    "version": "2.0.5",
     "require": {
         "php": "^8.1.0",
         "nyholm/psr7": "^1.8",

src/DeepSeekClient.php

@@ -49,6 +49,7 @@ class DeepSeekClient implements ClientContract
     protected bool $stream;
 
     protected float $temperature;
+    protected int $maxTokens;
 
     /**
      * response result contract
@@ -79,6 +80,7 @@ public function __construct(ClientInterface $httpClient)
         $this->requestMethod = 'POST';
         $this->endpointSuffixes = EndpointSuffixes::CHAT->value;
         $this->temperature = (float) TemperatureValues::GENERAL_CONVERSATION->value;
+        $this->maxTokens = (int) TemperatureValues::MAX_TOKENS->value;
         $this->tools = null;
     }
 
@@ -89,6 +91,7 @@ public function run(): string
             QueryFlags::MODEL->value    => $this->model,
             QueryFlags::STREAM->value   => $this->stream,
             QueryFlags::TEMPERATURE->value   => $this->temperature,
+            QueryFlags::MAX_TOKENS->value   => $this->maxTokens,
             QueryFlags::TOOLS->value  => $this->tools,
         ];
 
@@ -183,6 +186,12 @@ public function setTemperature(float $temperature): self
         return $this;
     }
 
+    public function setMaxTokens(int $maxTokens): self
+    {
+        $this->maxTokens = $maxTokens;
+        return $this;
+    }
+
     public function buildQuery(string $content, ?string $role = null): array
     {
         return [

src/Enums/Configs/TemperatureValues.php

@@ -12,4 +12,5 @@ enum TemperatureValues: string
     case TRANSLATION = "1.4";
     case CREATIVE_WRITING = "1.5";
     case POETRY = "1.6";
+    case MAX_TOKENS = "4096";
 }

src/Enums/Requests/QueryFlags.php

@@ -8,5 +8,6 @@ enum QueryFlags: string
     case MODEL = 'model';
     case STREAM = 'stream';
     case TEMPERATURE = 'temperature';
+    case MAX_TOKENS = 'max_tokens';
     case TOOLS = 'tools';
 }