import boto3

import json

# ここを書き換えてモデルを切り替える

MODEL_ID = ‘apac.anthropic.claude-3-haiku-20240307-v1:0’

#MODEL_ID = ‘apac.anthropic.claude-sonnet-4-20250514-v1:0’

# AIに食わせる文言(自由に変えてOK)

INPUT_TEXT = “こんにちは”

# Claudeへの指示(自由に変えてOK)

PROMPT = “一言で返してください。”

def call_bedrock(prompt, content):

    bedrock = boto3.client(‘bedrock-runtime’, region_name=’ap-northeast-1′)

    response = bedrock.invoke_model(

        modelId=MODEL_ID,

        contentType=’application/json’,

        body=json.dumps({

            “anthropic_version”: “bedrock-2023-05-31”,

            “max_tokens”: 500,

            “temperature”: 0.7,

            “messages”: [

                {

                    “role”: “user”,

                    “content”: prompt + content

                }

            ]

        })

    )

    result = json.loads(response[‘body’].read())

    return result[‘content’][0][‘text’]

if __name__ == “__main__”:

    print(f”モデル: {MODEL_ID}”)

    print(f”入力:\n{INPUT_TEXT}”)

    print(“-” * 40)

    output = call_bedrock(PROMPT, INPUT_TEXT)

    print(f”出力:\n{output}”)

タイトルとURLをコピーしました