bedrock_text.py

import boto3
import json

def lambda_handler(event, context):
    bedrock = boto3.client(service_name='bedrock-runtime')
    body = json.loads(event["body"])
    message = body["message"]
    
    body = json.dumps({
        "prompt": f"\n\nHuman:{message}\n\nAssistant:",
        "max_tokens_to_sample": 300,
        "temperature": 0.1,
        "top_p": 0.9,
    })
    
    modelId = 'anthropic.claude-v2'
    accept = 'application/json'
    contentType = 'application/json'
    
    response = bedrock.invoke_model(body=body, modelId=modelId, accept=accept, contentType=contentType)
    
    response_body = json.loads(response.get('body').read())
    answer = response_body.get('completion').strip()

    return {
        'statusCode': 200,
        'body': json.dumps({"answer": answer})
    }