Optimisation d’une invite

Vous pouvez optimiser une invite en utilisant un terrain de jeu ou en utilisant la gestion des invites dans la AWS Management Console. Vous devez sélectionner un modèle avant de pouvoir optimiser une invite. L’invite est optimisée pour le modèle que vous choisissez.

Pour optimiser une invite, envoyez une OptimizePromptdemande avec un point de terminaison Agents for Amazon Bedrock Runtime. Indiquez l’invite à optimiser dans l’objet input et spécifiez le modèle pour l’optimiser dans le champ targetModelId.

Le flux de réponse renvoie les événements suivants :

Exécutez l’exemple de code suivant pour optimiser une invite :

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