Input and Result unit tests for BedrockRuntime InvokeModel operation

Author: Victor Thuillier

src/Service/BedrockRuntime/tests/Unit/Input/InvokeModelRequestTest.php

@@ -9,28 +9,29 @@ class InvokeModelRequestTest extends TestCase
 {
     public function testRequest(): void
     {
-        self::fail('Not implemented');
-
         $input = new InvokeModelRequest([
-            'body' => 'change me',
-            'contentType' => 'change me',
-            'accept' => 'change me',
-            'modelId' => 'change me',
-            'trace' => 'change me',
-            'guardrailIdentifier' => 'change me',
-            'guardrailVersion' => 'change me',
-            'performanceConfigLatency' => 'change me',
+            'body' => '{"anthropic_version":"bedrock-2023-05-31","max_tokens":4096,"messages":[{"role":"user","content":[{"type":"text","text":"Write me a love poem."}]}],"temperature":1}',
+            'contentType' => 'application/json',
+            'accept' => 'application/json',
+            'modelId' => 'us.anthropic.claude-3-7-sonnet-20250219-v1:0',
+            'trace' => 'DISABLED',
+            'guardrailIdentifier' => 'arn:aws:bedrock:eu-west-1:965624758642:guardrail/azertyuiopqs',
+            'guardrailVersion' => 'DRAFT',
+            'performanceConfigLatency' => 'standard',
         ]);
 
         // see https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_InvokeModel.html
         $expected = '
-            POST / HTTP/1.0
+            POST /model/us.anthropic.claude-3-7-sonnet-20250219-v1:0/invoke HTTP/1.0
             Content-Type: application/json
+            Accept: application/json
+            X-Amzn-Bedrock-GuardrailIdentifier: arn:aws:bedrock:eu-west-1:965624758642:guardrail/azertyuiopqs
+            X-Amzn-Bedrock-GuardrailVersion: DRAFT
+            X-Amzn-Bedrock-PerformanceConfig-Latency: standard
+            X-Amzn-Bedrock-Trace: DISABLED
 
-            {
-            "change": "it"
-        }
-                ';
+            {"anthropic_version":"bedrock-2023-05-31","max_tokens":4096,"messages":[{"role":"user","content":[{"type":"text","text":"Write me a love poem."}]}],"temperature":1}
+        ';
 
         self::assertRequestEqualsHttpRequest($expected, $input->request());
     }

src/Service/BedrockRuntime/tests/Unit/Result/InvokeModelResponseTest.php

@@ -13,18 +13,17 @@ class InvokeModelResponseTest extends TestCase
 {
     public function testInvokeModelResponse(): void
     {
-        self::fail('Not implemented');
-
         // see https://docs.aws.amazon.com/bedrock/latest/APIReference/API_runtime_InvokeModel.html
-        $response = new SimpleMockedResponse('{
-            "change": "it"
-        }');
+        $response = new SimpleMockedResponse(
+            '{"id":"msg_bdrk_01SK3CsA23VBbZig4HqemAoo","type":"message","role":"assistant","model":"claude-3-7-sonnet-20250219","content":[{"type":"text","text":"# Whispers of the Heart\n\nIn the quiet moments between breaths,\nI find my thoughts wandering to you,\nLike stars that navigate sailors home,\nYour smile guides me through.\n\nThe universe conspired in silent ways,\nTo weave our paths into one,\nNow each sunrise holds new promise,\nA dance we\'ve just begun.\n\nYour laughter echoes in my dreams,\nA melody I\'ve known all along,\nMy heart recognizes yours completely,\nLike remembering a forgotten song.\n\nI love you not just for who you are,\nBut for who I become when we\'re together,\nA better version, a truer self,\nA love to cherish forever."}],"stop_reason":"end_turn","stop_sequence":null,"usage":{"input_tokens":13,"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"output_tokens":153}}',
+            ['Content-type' => 'Application/json', 'X-Amzn-Bedrock-PerformanceConfig-Latency' => 'standard']
+        );
 
         $client = new MockHttpClient($response);
         $result = new InvokeModelResponse(new Response($client->request('POST', 'http://localhost'), $client, new NullLogger()));
 
-        // self::assertTODO(expected, $result->getBody());
-        self::assertSame('changeIt', $result->getContentType());
-        self::assertSame('changeIt', $result->getPerformanceConfigLatency());
+        self::assertSame('{"id":"msg_bdrk_01SK3CsA23VBbZig4HqemAoo","type":"message","role":"assistant","model":"claude-3-7-sonnet-20250219","content":[{"type":"text","text":"# Whispers of the Heart\n\nIn the quiet moments between breaths,\nI find my thoughts wandering to you,\nLike stars that navigate sailors home,\nYour smile guides me through.\n\nThe universe conspired in silent ways,\nTo weave our paths into one,\nNow each sunrise holds new promise,\nA dance we\'ve just begun.\n\nYour laughter echoes in my dreams,\nA melody I\'ve known all along,\nMy heart recognizes yours completely,\nLike remembering a forgotten song.\n\nI love you not just for who you are,\nBut for who I become when we\'re together,\nA better version, a truer self,\nA love to cherish forever."}],"stop_reason":"end_turn","stop_sequence":null,"usage":{"input_tokens":13,"cache_creation_input_tokens":0,"cache_read_input_tokens":0,"output_tokens":153}}', $result->getBody());
+        self::assertSame('application/json', $result->getContentType());
+        self::assertSame('standard', $result->getPerformanceConfigLatency());
     }
 }