claude2 接口调用案例

try:
    import tiktoken
    import boto3
    import json
except:
    raise Exception("Error, need: pip install boto3 tiktoken")


apiKey = {
    "ApiKey": "You key",
    "ApkSecret": "you Secret"
}
# use local config
client_obj = boto3.client(
            service_name='bedrock-runtime',
            region_name="us-east-1",
            aws_access_key_id=apiKey['ApiKey'],
            aws_secret_access_key=apiKey['ApkSecret']
        )

body = json.dumps({
    "prompt": "\n\nHuman: 1+1=?\n\nAssistant:",
    "max_tokens_to_sample": 300,
    "temperature": 0.1, # 热度
    "top_p": 0.9,
})

modelId = 'anthropic.claude-v2:1'
accept = 'application/json'
contentType = 'application/json'
response = client_obj.invoke_model(body=body, modelId=modelId, accept=accept, contentType=contentType)
response_body = json.loads(response.get('body').read())
print(response_body)
# text
print(response_body.get('completion'))