使用 Amazon Bedrock 上的 Claude 3 将架构图转换为 CDK/Terraform 代码

import base64import jsonimport os
import boto3import clickfrom botocore.exceptions import ClientError
def call_claude_multi_model(bedrock_runtime, model_id, input_text, image, max_tokens):    """    Streams the response from a multimodal prompt.    Args:        bedrock_runtime: The Amazon Bedrock boto3 client.        model_id (str): The model ID to use.        input_text (str) : The prompt text        image (str) : The path to  an image that you want in the prompt.        max_tokens (int) : The maximum  number of tokens to generate.    Returns:        None.    """
    with open(image, "rb") as image_file:        encoded_string = base64.b64encode(image_file.read())
    body = json.dumps(        {            "anthropic_version": "bedrock-2023-05-31",            "max_tokens": max_tokens,            "messages": [                {                    "role": "user",                    "content": [                        {"type": "text", "text": input_text},                        {                            "type": "image",                            "source": {                                "type": "base64",                                "media_type": "image/jpeg",                                "data": encoded_string.decode("utf-8"),                            },                        },                    ],                }            ],        }    )
    response = bedrock_runtime.invoke_model_with_response_stream(        body=body, modelId=model_id    )
    for event in response.get("body"):        chunk = json.loads(event["chunk"]["bytes"])
        if chunk["type"] == "content_block_delta":            if chunk["delta"]["type"] == "text_delta":                print(chunk["delta"]["text"], end="")
@click.command()@click.option("--image_path", prompt="path to image", help="Image you want to parse")def main(image_path):    """    Entrypoint for Anthropic Claude Sonnet multimodal prompt example.    """
    model_id = "anthropic.claude-3-sonnet-20240229-v1:0"    input_text = """You are a AWS solution architect, The image provided is an architecture diagram. Use two heading to explain below.1. Explain the technical data flow in detail.2. Provide cdk typescript code to implement using aws-cdk-lib
Do not:1. use preambles.2. make assumptions."""    max_tokens = 4000
    try:
        bedrock_runtime = boto3.client("bedrock-runtime")
        call_claude_multi_model(            bedrock_runtime,            model_id,            input_text,            os.path.abspath(image_path),            max_tokens,        )
    except ClientError as err:        message = err.response["Error"]["Message"]        logger.error("A client error occurred: %s", message)        print("A client error occured: " + format(message))
if __name__ == "__main__":    main()

import * as cdk from 'aws-cdk-lib';import * as lambda from 'aws-cdk-lib/aws-lambda';import * as s3 from 'aws-cdk-lib/aws-s3';import * as dynamodb from 'aws-cdk-lib/aws-dynamodb';import * as lambdaEventSources from 'aws-cdk-lib/aws-lambda-event-sources';
const app = new cdk.App();const stack = new cdk.Stack(app, 'DataPipelineStack');
// Create the S3 bucketconst sourceBucket = new s3.Bucket(stack, 'SourceBucket');
// Create the DynamoDB tableconst eventsTable = new dynamodb.Table(stack, 'StoreEventsDB', {  partitionKey: { name: 'id', type: dynamodb.AttributeType.STRING },  billingMode: dynamodb.BillingMode.PAY_PER_REQUEST,});
// Create the Lambda functionconst lambdaExtractor = new lambda.Function(stack, 'LambdaExtractor', {  runtime: lambda.Runtime.NODEJS_14_X,  code: lambda.Code.fromAsset('path/to/lambda/code'),  handler: 'index.handler',  environment: {    EVENTS_TABLE_NAME: eventsTable.tableName,  },});
// Grant permissions for the Lambda function to access the DynamoDB tableeventsTable.grantWriteData(lambdaExtractor);
// Create the S3 event source and associate it with the Lambda functionlambdaExtractor.addEventSource(  new lambdaEventSources.S3EventSource(sourceBucket, {    events: [s3.EventType.OBJECT_CREATED, s3.EventType.OBJECT_REMOVED],    recursive: true,  }));