Amazon BedrockでxAIのGrok 4.3が利用可能になったのでIAM認証で試してみた

import time
from openai import OpenAI
from aws_bedrock_token_generator import provide_token

REGION = "us-west-2"
MODEL = "xai.grok-4.3"
BASE_URL = f"https://bedrock-mantle.{REGION}.api.aws/openai/v1"

client = OpenAI(
    base_url=BASE_URL,
    api_key=provide_token(region=REGION),
)

start = time.time()
response = client.chat.completions.create(
    model=MODEL,
    messages=[{"role": "user", "content": "Amazon Bedrockの特徴を3行で説明してください。"}],
)
elapsed = time.time() - start

print(f"Time: {elapsed:.1f}s")
print(f"Usage: in={response.usage.prompt_tokens}, out={response.usage.completion_tokens}")
print(response.choices[0].message.content)

Time: 1.5s
Usage: in=16, out=71

Amazon Bedrockは、AWSが提供するAIサービスで、様々な基盤モデルに簡単にアクセスできる。
開発者は、テキスト生成や画像生成、構造化データ抽出など、様々なタスクに活用できる。
低コストで高性能な処理を実現しながら、セキュリティとプライバシーに配慮した設計になっている。

import json

tools = [
    {
        "type": "function",
        "function": {
            "name": "get_weather",
            "description": "Get the current weather for a location",
            "parameters": {
                "type": "object",
                "properties": {
                    "location": {"type": "string", "description": "City name"},
                    "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]}
                },
                "required": ["location"]
            }
        }
    }
]

# Step 1: ユーザーメッセージ → モデルがtool callを返す
response1 = client.chat.completions.create(
    model=MODEL,
    messages=[{"role": "user", "content": "東京の天気を教えて"}],
    tools=tools,
)

tool_call = response1.choices[0].message.tool_calls[0]
print(f"finish_reason: {response1.choices[0].finish_reason}")
print(f"tool_call: {tool_call.function.name}({tool_call.function.arguments})")

finish_reason: tool_calls
tool_call: get_weather({"location":"東京"})

# Step 2: tool結果を返す → モデルが最終回答を生成
response2 = client.chat.completions.create(
    model=MODEL,
    messages=[
        {"role": "user", "content": "東京の天気を教えて"},
        response1.choices[0].message,
        {
            "role": "tool",
            "tool_call_id": tool_call.id,
            "content": json.dumps(
                {"location": "東京", "temperature": 28, "condition": "晴れ", "humidity": 65}
            ),
        },
    ],
    tools=tools,
)

print(f"finish_reason: {response2.choices[0].finish_reason}")
print(response2.choices[0].message.content)

finish_reason: stop
東京の現在の天気は、晴れです。

気温は 28℃ で、湿度65%と少し蒸し暑く感じるかもしれません。外に出る際は水分補給をこまめにどうぞ！