最速(?)最短(?) Amazon Bedrockの動かし方はこれだ

お知らせ

過去にQiitaに投稿した内容のアーカイブです。

boto3は1.28.57でBedrockに対応しました。

pip install -q -U boto3

bedrock と bedrock-runtime があるので注意。invoke_modelはbedrock-runtime

import boto3
import json

client = boto3.client('bedrock-runtime')

prompt = '''
Human: こんにちは！
Assistant:
'''

body = {
    'prompt': prompt,
    'max_tokens_to_sample': 300,
}

response = client.invoke_model(
    modelId="anthropic.claude-v2",
    body=json.dumps(body)
)

body = response['body'].readlines()

completion = json.loads(body[0].decode('utf-8'))['completion']

print(completion)

回答

 はい、こんにちは。よろしくお願いします。