プロンプトを最適化する

AWS マネジメントコンソールでプレイグラウンドまたはプロンプト管理を使用して、プロンプトを最適化できます。プロンプトを最適化する前に、モデルを選択する必要があります。プロンプトは、選択したモデルに合わせて最適化されます。

プロンプトを最適化するには、Amazon Bedrock エージェントのランタイムエンドポイントを使用して OptimizePrompt リクエストを送信します。input オブジェクトで最適化するプロンプトを指定し、targetModelId フィールドで最適化の対象となるモデルを指定します。

レスポンスストリームは以下のイベントを返します。

次のコードサンプルを実行して、プロンプトを最適化します。

import boto3

# Set values here
TARGET_MODEL_ID = "anthropic.claude-3-sonnet-20240229-v1:0" # Model to optimize for. For model IDs, see https://docs.aws.amazon.com/bedrock/latest/userguide/model-ids.html
PROMPT = "Please summarize this text: " # Prompt to optimize

def get_input(prompt):
    return {
        "textPrompt": {
            "text": prompt
        }
    }
 
def handle_response_stream(response):
    try:
        event_stream = response['optimizedPrompt']
        for event in event_stream:
            if 'optimizedPromptEvent' in event:
                print("========================== OPTIMIZED PROMPT ======================\n")
                optimized_prompt = event['optimizedPromptEvent']
                print(optimized_prompt)
            else:
                print("========================= ANALYZE PROMPT =======================\n")
                analyze_prompt = event['analyzePromptEvent']
                print(analyze_prompt)
    except Exception as e:
        raise e
 
 
if __name__ == '__main__':
    client = boto3.client('bedrock-agent-runtime')
    try:
        response = client.optimize_prompt(
            input=get_input(PROMPT),
            targetModelId=TARGET_MODEL_ID
        )
        print("Request ID:", response.get("ResponseMetadata").get("RequestId"))
        print("========================== INPUT PROMPT ======================\n")
        print(PROMPT)
        handle_response_stream(response)
    except Exception as e:
        raise e