Amazon Bedrock で一般提供になった OpenAI GPT-6 Astra を試してみた

import json
import os

import httpx
from botocore.auth import SigV4Auth
from botocore.awsrequest import AWSRequest
from botocore.credentials import Credentials
from openai import OpenAI

REGION = "us-west-2"
SIGNING_SERVICE = "bedrock"
MODEL_ID = "global.openai.gpt-6-astra"

credentials = Credentials(
    os.environ["AWS_ACCESS_KEY_ID"],
    os.environ["AWS_SECRET_ACCESS_KEY"],
    os.environ.get("AWS_SESSION_TOKEN"),
)

class SigV4Signer(httpx.Auth):
    requires_request_body = True

    def auth_flow(self, request):
        aws_request = AWSRequest(
            method=request.method,
            url=str(request.url),
            data=request.content,
            headers={"content-type": request.headers.get("content-type", "application/json")},
        )
        SigV4Auth(credentials, SIGNING_SERVICE, REGION).add_auth(aws_request)
        for key, value in aws_request.headers.items():
            request.headers[key] = value
        yield request

client = OpenAI(
    api_key="unused-sigv4",
    base_url=f"https://bedrock-runtime.{REGION}.amazonaws.com/openai/v1",
    http_client=httpx.Client(auth=SigV4Signer(), timeout=120.0),
)

response = client.responses.create(
    model=MODEL_ID,
    input="Can you explain the features of Amazon Bedrock?",
)
print("=== output_text ===")
print(response.output_text)
print("=== usage ===")
print(json.dumps(response.usage.model_dump(), indent=2))

=== output_text ===
**Amazon Bedrock is AWS’s managed platform for building generative AI applications.** It gives you access to foundation models and tools for connecting them to your data, automating workflows, and adding security controls—without having to manage the underlying model-serving infrastructure for its standard managed offerings.

Here are its main features:

### 1. Access to multiple foundation models
...

=== usage ===
{
  "input_tokens": 16,
  "output_tokens": 1010,
  "total_tokens": 1026
}

REGION = "us-west-2"
SIGNING_SERVICE = "bedrock-mantle"
MODEL_ID = "openai.gpt-6-astra"

client = OpenAI(
    api_key="unused-sigv4",
    base_url=f"https://bedrock-mantle.{REGION}.api.aws/openai/v1",
    http_client=httpx.Client(auth=SigV4Signer(), timeout=120.0),
)

response = client.responses.create(
    model=MODEL_ID,
    input="Can you explain the features of Amazon Bedrock?",
)
print(response.output_text)

**Amazon Bedrock is AWS’s fully managed service for building generative AI applications.** It lets you access foundation models, connect them to your data, and deploy AI-powered workflows without managing the underlying model-serving infrastructure.

Its main features include:

### 1. Access to multiple foundation models
...