Added system instructions and tools to stream

google-gemini-php · Jan 10, 2025 · 2fea930 · 2fea930
1 parent 9c39211
commit 2fea930
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 1 deletion.
diff --git a/src/Requests/GenerativeModel/StreamGenerateContentRequest.php b/src/Requests/GenerativeModel/StreamGenerateContentRequest.php
@@ -9,6 +9,8 @@
 use Gemini\Data\Content;
 use Gemini\Data\GenerationConfig;
 use Gemini\Data\SafetySetting;
+use Gemini\Data\Tool;
+use Gemini\Data\ToolConfig;
 use Gemini\Enums\Method;
 use Gemini\Foundation\Request;
 use Gemini\Requests\Concerns\HasJsonBody;
@@ -26,12 +28,16 @@ class StreamGenerateContentRequest extends Request
     /**
      * @param  array<string|Blob|array<string|Blob>|Content>  $parts
      * @param  array<SafetySetting>  $safetySettings
+     * @param  array<Tool>  $tools
      */
     public function __construct(
         protected readonly string $model,
         protected readonly array $parts,
         protected readonly array $safetySettings = [],
-        protected readonly ?GenerationConfig $generationConfig = null
+        protected readonly ?GenerationConfig $generationConfig = null,
+        protected readonly ?Content $systemInstruction = null,
+        protected readonly array $tools = [],
+        protected readonly ?ToolConfig $toolConfig = null,
     ) {}
 
     public function resolveEndpoint(): string
@@ -56,6 +62,12 @@ protected function defaultBody(): array
                 $this->safetySettings ?? []
             ),
             'generationConfig' => $this->generationConfig?->toArray(),
+            'systemInstruction' => $this->systemInstruction?->toArray(),
+            'tools' => array_map(
+                static fn (Tool $tool): array => $tool->toArray(),
+                $this->tools ?? []
+            ),
+            'toolConfig' => $this->toolConfig?->toArray(),
         ];
     }
 }
diff --git a/src/Resources/GenerativeModel.php b/src/Resources/GenerativeModel.php
@@ -133,6 +133,9 @@ public function streamGenerateContent(string|Blob|array|Content ...$parts): Stre
                 parts: $parts,
                 safetySettings: $this->safetySettings,
                 generationConfig: $this->generationConfig,
+                systemInstruction: $this->systemInstruction,
+                tools: $this->tools,
+                toolConfig: $this->toolConfig,
             )
         );