feat: Added $outputDimensionality param to embedContent method

google-gemini-php · Dec 29, 2024 · 7ba35de · 7ba35de
1 parent 5053e41
commit 7ba35de
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/src/Contracts/Resources/EmbeddingModalContract.php b/src/Contracts/Resources/EmbeddingModalContract.php
@@ -16,7 +16,7 @@ interface EmbeddingModalContract
     /**
      * @param  string|Blob|array<string|Blob>|Content  $content
      */
-    public function embedContent(string|Blob|array|Content $content, ?TaskType $taskType = null, ?string $title = null): EmbedContentResponse;
+    public function embedContent(string|Blob|array|Content $content, ?TaskType $taskType = null, ?string $title = null, ?int $outputDimensionality = null): EmbedContentResponse;
 
     /**
      * Generates multiple embedding vectors from the input Content which consists of a batch of strings represented as EmbedContentRequest objects.

diff --git a/src/Requests/GenerativeModel/EmbedContentRequest.php b/src/Requests/GenerativeModel/EmbedContentRequest.php
@@ -24,7 +24,8 @@ public function __construct(
         protected readonly string $model,
         protected readonly string|Blob|array|Content $part,
         protected readonly ?TaskType $taskType = null,
-        protected readonly ?string $title = null
+        protected readonly ?string $title = null,
+        protected readonly ?int $outputDimensionality = null
     ) {}
 
     public function resolveEndpoint(): string
@@ -43,6 +44,7 @@ protected function defaultBody(): array
             'content' => Content::parse(part: $this->part)->toArray(),
             'taskType' => $this->taskType?->value,
             'title' => $this->title,
+            'outputDimensionality' => $this->outputDimensionality,
         ];
     }
 }
diff --git a/src/Resources/EmbeddingModel.php b/src/Resources/EmbeddingModel.php
@@ -37,7 +37,7 @@ public function __construct(
      *
      * @param  string|Blob|array<string|Blob>|Content  $content
      */
-    public function embedContent(string|Blob|array|Content $content, ?TaskType $taskType = null, ?string $title = null): EmbedContentResponse
+    public function embedContent(string|Blob|array|Content $content, ?TaskType $taskType = null, ?string $title = null, ?int $outputDimensionality = null): EmbedContentResponse
     {
         /** @var ResponseDTO<array{ embedding: array{ values: array<float> } }> $response */
         $response = $this->transporter->request(
@@ -46,6 +46,7 @@ public function embedContent(string|Blob|array|Content $content, ?TaskType $task
                 part: $content,
                 taskType: $taskType,
                 title: $title,
+                outputDimensionality: $outputDimensionality
             )
         );
 

diff --git a/src/Testing/Resources/EmbeddingModelTestResource.php b/src/Testing/Resources/EmbeddingModelTestResource.php
@@ -23,7 +23,7 @@ protected function resource(): string
         return EmbeddingModel::class;
     }
 
-    public function embedContent(Blob|Content|array|string $content, ?TaskType $taskType = null, ?string $title = null): EmbedContentResponse
+    public function embedContent(Blob|Content|array|string $content, ?TaskType $taskType = null, ?string $title = null, ?int $outputDimensionality = null): EmbedContentResponse
     {
         return $this->record(method: __FUNCTION__, args: func_get_args(), model: $this->model);
     }