Merge pull request #5 from aydinfatih/main

Added startChat method to ChatSessionContract
google-gemini-php · Feb 27, 2024 · 6e48284 · 6e48284
2 parents 1db9396 + 0060511
commit 6e48284
Show file tree

Hide file tree

Showing 4 changed files with 32 additions and 0 deletions.
diff --git a/src/Contracts/Resources/ChatSessionContract.php b/src/Contracts/Resources/ChatSessionContract.php
@@ -14,4 +14,9 @@ interface ChatSessionContract
      * @param  string|Blob|array<string|Blob>|Content  ...$parts
      */
     public function sendMessage(string|Blob|array|Content ...$parts): GenerateContentResponse;
+
+    /**
+     * @param  array<Content>  $history
+     */
+    public function startChat(array $history = []): ChatSessionContract;
 }
diff --git a/src/Resources/ChatSession.php b/src/Resources/ChatSession.php
@@ -42,4 +42,12 @@ public function sendMessage(string|Blob|array|Content ...$parts): GenerateConten
 
         return $response;
     }
+
+    /**
+     * @param  array<Content>  $history
+     */
+    public function startChat(array $history = []): ChatSession
+    {
+        return new self(model: $this->model, history: $history);
+    }
 }
diff --git a/src/Testing/Resources/ChatSessionTestResource.php b/src/Testing/Resources/ChatSessionTestResource.php
@@ -24,4 +24,12 @@ public function sendMessage(string|array|Blob|Content ...$parts): GenerateConten
     {
         return $this->record(method: __FUNCTION__, args: func_get_args(), model: $this->model);
     }
+
+    /**
+     * @param  array<Content>  $history
+     */
+    public function startChat(array $history = []): ChatSessionTestResource
+    {
+        return $this->record(method: __FUNCTION__, args: func_get_args(), model: $this->model);
+    }
 }
diff --git a/tests/Resources/ChatSession.php b/tests/Resources/ChatSession.php
@@ -1,6 +1,7 @@
 <?php
 
 use Gemini\Enums\Method;
+use Gemini\Enums\ModelType;
 use Gemini\Resources\ChatSession;
 use Gemini\Responses\GenerativeModel\GenerateContentResponse;
 
@@ -12,3 +13,13 @@
     expect($result)
         ->toBeInstanceOf(ChatSession::class);
 });
+
+test('start chat', function () {
+    $modelType = ModelType::GEMINI_PRO;
+    $client = mockClient(method: Method::POST, endpoint: "{$modelType->value}:generateContent", response: GenerateContentResponse::fake(), times: 0);
+
+    $result = $client->chat()->startChat();
+
+    expect($result)
+        ->toBeInstanceOf(ChatSession::class);
+});