Skip to content

Commit 772fc17

Browse files
committed
fix message mapping for assistant messages
1 parent 8ac8422 commit 772fc17

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/Providers/Gemini/HandleChat.php

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
use GuzzleHttp\Promise\PromiseInterface;
88
use GuzzleHttp\RequestOptions;
99
use NeuronAI\Chat\Enums\MessageRole;
10+
use NeuronAI\Chat\Messages\AssistantMessage;
1011
use NeuronAI\Chat\Messages\Message;
1112
use NeuronAI\Chat\Messages\Usage;
1213
use Psr\Http\Message\ResponseInterface;
@@ -52,7 +53,7 @@ public function chatAsync(array $messages): PromiseInterface
5253
if (\array_key_exists('functionCall', $parts[0]) && !empty($parts[0]['functionCall'])) {
5354
$response = $this->createToolCallMessage($content);
5455
} else {
55-
$response = new Message(MessageRole::from($content['role']), $parts[0]['text'] ?? '');
56+
$response = new AssistantMessage($parts[0]['text'] ?? '');
5657
}
5758

5859
if (\array_key_exists('groundingMetadata', $result['candidates'][0])) {

0 commit comments

Comments
 (0)