[Platform][Bedrock] Use dedicated model class

Author: OskarStark

src/Bridge/Bedrock/Anthropic/Claude.php

@@ -0,0 +1,53 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Symfony\AI\Platform\Bridge\Bedrock\Anthropic;
+
+use Symfony\AI\Platform\Capability;
+use Symfony\AI\Platform\Model;
+
+/**
+ * @author Christopher Hertel <[email protected]>
+ */
+class Claude extends Model
+{
+    public const HAIKU_3 = 'claude-3-haiku-20240307';
+    public const HAIKU_35 = 'claude-3-5-haiku-latest';
+    public const SONNET_3 = 'claude-3-sonnet-20240229';
+    public const SONNET_35 = 'claude-3-5-sonnet-latest';
+    public const SONNET_37 = 'claude-3-7-sonnet-latest';
+    public const SONNET_4 = 'claude-sonnet-4-20250514';
+    public const SONNET_4_0 = 'claude-sonnet-4-0';
+    public const OPUS_3 = 'claude-3-opus-20240229';
+    public const OPUS_4 = 'claude-opus-4-20250514';
+    public const OPUS_4_0 = 'claude-opus-4-0';
+    public const OPUS_4_1 = 'claude-opus-4-1';
+
+    /**
+     * @param array<string, mixed> $options The default options for the model usage
+     */
+    public function __construct(string $name, array $options = [])
+    {
+        $capabilities = [
+            Capability::INPUT_MESSAGES,
+            Capability::INPUT_IMAGE,
+            Capability::OUTPUT_TEXT,
+            Capability::OUTPUT_STREAMING,
+            Capability::TOOL_CALLING,
+        ];
+
+        if (!isset($options['max_tokens'])) {
+            $options['max_tokens'] = 1000;
+        }
+
+        parent::__construct($name, $capabilities, $options);
+    }
+}

src/Bridge/Bedrock/Anthropic/ClaudeModelClient.php

@@ -14,7 +14,6 @@
 use AsyncAws\BedrockRuntime\BedrockRuntimeClient;
 use AsyncAws\BedrockRuntime\Input\InvokeModelRequest;
 use AsyncAws\BedrockRuntime\Result\InvokeModelResponse;
-use Symfony\AI\Platform\Bridge\Anthropic\Claude;
 use Symfony\AI\Platform\Bridge\Bedrock\RawBedrockResult;
 use Symfony\AI\Platform\Exception\RuntimeException;
 use Symfony\AI\Platform\Model;

src/Bridge/Bedrock/Anthropic/ClaudeResultConverter.php

@@ -11,7 +11,6 @@
 
 namespace Symfony\AI\Platform\Bridge\Bedrock\Anthropic;
 
-use Symfony\AI\Platform\Bridge\Anthropic\Claude;
 use Symfony\AI\Platform\Bridge\Bedrock\RawBedrockResult;
 use Symfony\AI\Platform\Exception\RuntimeException;
 use Symfony\AI\Platform\Model;

tests/Bridge/Bedrock/Anthropic/ClaudeResultConverterTest.php

@@ -15,7 +15,7 @@
 use AsyncAws\Core\Test\ResultMockFactory;
 use PHPUnit\Framework\Attributes\TestDox;
 use PHPUnit\Framework\TestCase;
-use Symfony\AI\Platform\Bridge\Anthropic\Claude;
+use Symfony\AI\Platform\Bridge\Bedrock\Anthropic\Claude;
 use Symfony\AI\Platform\Bridge\Bedrock\Anthropic\ClaudeResultConverter;
 use Symfony\AI\Platform\Bridge\Bedrock\RawBedrockResult;
 use Symfony\AI\Platform\Exception\RuntimeException;

tests/Bridge/Bedrock/Anthropic/ClaudeTest.php

@@ -0,0 +1,45 @@
+<?php
+
+/*
+ * This file is part of the Symfony package.
+ *
+ * (c) Fabien Potencier <[email protected]>
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace Bridge\Bedrock\Anthropic;
+
+use PHPUnit\Framework\TestCase;
+use Symfony\AI\Platform\Bridge\Anthropic\Claude;
+
+/**
+ * @author Oskar Stark <[email protected]>
+ */
+final class ClaudeTest extends TestCase
+{
+    public function testItCreatesClaudeWithDefaultSettings()
+    {
+        $claude = new Claude(Claude::SONNET_35);
+
+        $this->assertSame(Claude::SONNET_35, $claude->getName());
+        $this->assertSame(['max_tokens' => 1000], $claude->getOptions());
+    }
+
+    public function testItCreatesClaudeWithCustomSettingsIncludingMaxTokens()
+    {
+        $claude = new Claude(Claude::SONNET_35, ['temperature' => 0.5, 'max_tokens' => 2000]);
+
+        $this->assertSame(Claude::SONNET_35, $claude->getName());
+        $this->assertSame(['temperature' => 0.5, 'max_tokens' => 2000], $claude->getOptions());
+    }
+
+    public function testItCreatesClaudeWithCustomSettingsWithoutMaxTokens()
+    {
+        $claude = new Claude(Claude::SONNET_35, ['temperature' => 0.5]);
+
+        $this->assertSame(Claude::SONNET_35, $claude->getName());
+        $this->assertSame(['temperature' => 0.5, 'max_tokens' => 1000], $claude->getOptions());
+    }
+}