fix: text generator message steps (#154)

Author: ChrisB-TL

Diff for: src/Structured/Generator.php

@@ -24,10 +24,10 @@ public function __construct(protected Provider $provider)
 
     public function generate(Request $request): Response
     {
-        $response = $this->sendProviderRequest($request);
-
         $this->messages = $request->messages;
 
+        $response = $this->sendProviderRequest($request);
+
         $this->responseBuilder->addStep(new Step(
             text: $response->text,
             object: $this->decodeObject($response->text),

Diff for: tests/Structured/GeneratorTest.php

@@ -172,7 +172,7 @@ public function toArray(): array
     });
 });
 
-test('it adds system message and user message to first step', function (): void {
+test('it adds system message user message and assistant message response to first step', function (): void {
     Prism::fake([
         new ProviderResponse(
             text: json_encode(['I am a string']),
@@ -192,6 +192,7 @@ public function toArray(): array
     $response = $request->generate();
 
     expect($response)->toBeInstanceOf(Response::class);
+    expect($response->steps[0]->messages)->toHaveCount(3);
 
     /** @var SystemMessage */
     $system_message = $response->steps[0]->messages[0];
@@ -206,4 +207,12 @@ public function toArray(): array
     expect($user_message)->toBeInstanceOf(UserMessage::class)
         ->and($user_message->text())
         ->toBe('User Prompt');
+
+    /** @var AssistantMessage */
+    $assistant_message = $response->steps[0]->messages[2];
+
+    expect($assistant_message)->toBeInstanceOf(AssistantMessage::class)
+        ->and($assistant_message->content)
+        ->toBe(json_encode(['I am a string']));
+
 });

Diff for: tests/Text/GeneratorTest.php

@@ -307,7 +307,7 @@
         );
 });
 
-test('it adds the system message and user message to first step', function (): void {
+test('it adds the system message user message and assistant response to first step', function (): void {
     $request = (new PendingRequest)
         ->using('test-provider', 'test-model')
         ->withSystemPrompt('System Prompt')
@@ -330,4 +330,11 @@
     expect($user_message)->toBeInstanceOf(UserMessage::class)
         ->and($user_message->text())
         ->toBe('User Prompt');
+
+    /** @var AssistantMessage */
+    $assistant_message = $response->steps[0]->messages[2];
+
+    expect($assistant_message)->toBeInstanceOf(AssistantMessage::class)
+        ->and($assistant_message->content)
+        ->toBe("I'm nyx!");
 });