nova-prompt-optimizer 1.1.0

A Python SDK for optimizing prompts for Nova.

Nova Prompt Optimizer

A Python SDK for optimizing prompts for Amazon Nova and other models deployed on AWS.

📚 Table of contents

• Installation
• Pre-Requisites
• What's New
• Quick Start: Facility Support Analyzer Dataset
• Quick Start: SageMaker Endpoints
• Core Concepts
  • Input Adapters
    • 1. Prompt Adapter
    • 2. Inference Adapter
    • 3. Dataset Adapter
    • 4. Metric Adapter
    • 5. Optimization Adapter
  • Optimizers
    • NovaPromptOptimizer
  • Evaluator
• Advanced Features
  • Separate Inference Adapters
  • SageMaker Endpoint Support
• Optimization Recommendations
• Preview Status
• Interaction with AWS Bedrock
• Acknowledgements

Installation

Install the library using

pip3 install nova-prompt-optimizer

Pre-Requisites

Setup your AWS Access Keys:

To execute the SDK, you will need AWS credentials configured. Take a look at the AWS CLI configuration documentation for details on the various ways to configure credentials. An easy way to try out the SDK is to populate the following environment variables with your AWS API credentials. Take a look at this guide for Authenticating with short-term credentials for the AWS CLI

export AWS_ACCESS_KEY_ID="..."
export AWS_SECRET_ACCESS_KEY="..."

To enable Nova model access:

1. Go to the Amazon Bedrock Model Access page
2. Click "Manage model access"
3. Choose Amazon as provider and Nova models
4. Click "Request access"
5. Wait for approval (instant in most cases)

🎉 What's New

SageMaker Endpoint Support

Optimize prompts for models deployed on Amazon SageMaker! The optimizer now supports:

• SageMaker endpoints as task models for optimization
• Automatic Bedrock integration for meta-prompting with Nova 2.0 Lite
• OpenAI-compatible payload format for SageMaker endpoints

Separate Inference Adapters

Use different models for different optimization phases:

• Meta-prompting: Automatically uses Bedrock with Nova 2.0 Lite (or specify your own)
• Task optimization: Use any supported backend (Bedrock, SageMaker, etc.)

Example: Optimize for SageMaker

from amzn_nova_prompt_optimizer.core.inference import SageMakerInferenceAdapter
from amzn_nova_prompt_optimizer.core.optimizers import NovaPromptOptimizer

# Your SageMaker endpoint
sagemaker_adapter = SageMakerInferenceAdapter(
    endpoint_name="my-model-endpoint",
    region_name="us-west-2"
)

# Optimizer automatically uses Bedrock Nova 2.0 Lite for meta-prompting
optimizer = NovaPromptOptimizer(
    prompt_adapter=prompt_adapter,
    inference_adapter=sagemaker_adapter,  # Your SageMaker endpoint
    dataset_adapter=dataset_adapter,
    metric_adapter=metric_adapter
)

optimized_prompt = optimizer.optimize(mode="lite")

See the Quick Start for SageMaker section below for a complete example.

🏁 Quick Start

Facility Support Analyzer Dataset

The Facility Support Analyzer dataset consists of emails that are to be classified based on category, urgency and sentiment.

Please see the samples folder for example notebooks of how to optimize a prompt in the scenario where a user prompt template is to be optimized and the scenario where a user and system prompt is to be optimized together

🚀 Quick Start: SageMaker Endpoints

Optimize prompts for models deployed on Amazon SageMaker in just a few steps:

1. Install and Configure

pip install nova-prompt-optimizer
export AWS_ACCESS_KEY_ID="..."
export AWS_SECRET_ACCESS_KEY="..."

2. Prepare Your Dataset

{"input": "What is machine learning?", "answer": "Machine learning is..."}
{"input": "Explain neural networks", "answer": "Neural networks are..."}

3. Run Optimization

from amzn_nova_prompt_optimizer.core.inference import SageMakerInferenceAdapter
from amzn_nova_prompt_optimizer.core.input_adapters.dataset_adapter import JSONDatasetAdapter
from amzn_nova_prompt_optimizer.core.input_adapters.prompt_adapter import TextPromptAdapter
from amzn_nova_prompt_optimizer.core.input_adapters.metric_adapter import MetricAdapter
from amzn_nova_prompt_optimizer.core.optimizers import NovaPromptOptimizer

# Define your metric
class ExactMatchMetric(MetricAdapter):
    def apply(self, prediction: str, ground_truth: str) -> float:
        return 1.0 if prediction.strip() == ground_truth.strip() else 0.0
    
    def batch_apply(self, predictions: list, ground_truths: list) -> float:
        matches = sum(self.apply(p, g) for p, g in zip(predictions, ground_truths))
        return matches / len(predictions) if predictions else 0.0

# Setup SageMaker adapter for your endpoint
sagemaker_adapter = SageMakerInferenceAdapter(
    endpoint_name="YOUR-ENDPOINT-NAME",
    region_name="us-west-2"
)

# Load dataset
dataset_adapter = JSONDatasetAdapter(
    input_columns={"input"},
    output_columns={"answer"}
)
dataset_adapter.adapt("your_dataset.jsonl")
train_set, test_set = dataset_adapter.split(0.5)

# Setup initial prompt
prompt_adapter = TextPromptAdapter()
prompt_adapter.set_system_prompt(content="You are a helpful assistant.")
prompt_adapter.set_user_prompt(
    content="Answer this question: {{input}}",
    variables={"input"}
)
prompt_adapter.adapt()

# Setup metric
metric_adapter = ExactMatchMetric()

# Create optimizer (automatically uses Bedrock Nova 2.0 Lite for meta-prompting)
optimizer = NovaPromptOptimizer(
    prompt_adapter=prompt_adapter,
    inference_adapter=sagemaker_adapter,  # Your SageMaker endpoint
    dataset_adapter=dataset_adapter,
    metric_adapter=metric_adapter
)

# Run optimization
optimized_prompt = optimizer.optimize(mode="lite")

# Save results
optimized_prompt.save("optimized_prompts/")

What happens during optimization:

1. Meta-Prompting Phase: Uses Bedrock with Nova 2.0 Lite to generate an improved prompt structure (~30 seconds)
2. Task Optimization Phase: Tests multiple prompt variations on your SageMaker endpoint (~5-15 minutes)

For more details, see the SageMaker Quick Start Guide.

Core Concepts

Input Adapters

We break each component of Prompt Optimization into Adapters providing a modular approach to Prompt Optimization.

1. Prompt Adapter

Responsibility: Ability to load prompts from different formats and store them in the standardized format (JSON)

Sample Prompt Adapter Initialization

from amzn_nova_prompt_optimizer.core.input_adapters.prompt_adapter import TextPromptAdapter

prompt_adapter = TextPromptAdapter()

prompt_adapter.set_system_prompt(file_path="prompt/sys_prompt.txt", variables={"foo"})

prompt_adapter.set_user_prompt(content="You are a .....", variables={"bar"})

prompt_adapter.adapt()

Supported Prompt Adapters: TextPromptAdapter

Learn More about the Prompt Adapter here

2. Inference Adapter

Responsibility: Ability to call an inference backend for the models e.g. Bedrock, SageMaker, etc.

Sample use of Bedrock Inference Adapter

from amzn_nova_prompt_optimizer.core.inference import BedrockInferenceAdapter

inference_adapter = BedrockInferenceAdapter(region_name="us-east-1")

Sample use of SageMaker Inference Adapter

from amzn_nova_prompt_optimizer.core.inference import SageMakerInferenceAdapter

inference_adapter = SageMakerInferenceAdapter(
    endpoint_name="my-model-endpoint",
    region_name="us-west-2"
)

You can pass rate_limit into constructor of InferenceAdapter to limit the max TPS of calls to avoid throttle. Default to 2 if not set.

from amzn_nova_prompt_optimizer.core.inference import BedrockInferenceAdapter

inference_adapter = BedrockInferenceAdapter(region_name="us-east-1", rate_limit=10) # Max 10 TPS

Supported Inference Adapters:

• BedrockInferenceAdapter - For Amazon Bedrock models
• SageMakerInferenceAdapter - For SageMaker endpoints (OpenAI-compatible format)

Core Functions

Call the model using the parameters

# Call the model with the passed parameters
inference_output = inference_adapter.call_model(
    model_id: str, 
    system_prompt: str, 
    messages: List[Dict[str, str]], 
    inf_config: Dict[str, Any]
)

Test endpoint connectivity (SageMaker)

# Test if endpoint is accessible
if inference_adapter.test_connection():
    print("✓ Endpoint is ready")

The Inference Adapter accepts the system_prompt as a string.

The input to the model as a list of User/Assistant turns (messages). e.g. [{"user": "foo"}, {"assistant": "bar"}, {"user": "What comes next?"}]

3. Dataset Adapter

Responsibility: Ability to read/write datasets from different formats. Uses an intermediary standardized format when communicating with other adapters. It can also read list of JSON object. It can also create Train/Test splits (with stratify capability if set).

Requirements: Currently, you can only provide a singleton set as output column.

Sample Dataset Adapter Initialization

# Example Usage
from amzn_nova_prompt_optimizer.core.input_adapters.dataset_adapter import JSONDatasetAdapter

input_columns = {"input"}
output_columns = {"answer"}

dataset_adapter = JSONDatasetAdapter(input_columns, output_columns)

# Adapt
dataset_adapter.adapt(data_source="sample_data.jsonl")

# Split
train, test = dataset_adapter.split(0.5)

Supported Dataset Adapters: JSONDatasetAdapter, CSVDatasetAdapter

Learn More about the Dataset Adapter here

4. Metric Adapter

Responsibility: Ability to load custom metrics and apply them on inference output and ground truth

Metric Adapter Class

class MetricAdapter():
    def apply(self, y_pred: Any, y_true: Any) -> float:
        pass

    def batch_apply(self, y_preds: List[Any], y_trues: List[Any]) -> float:
        pass

Sample Custom Metric Adapter Initialization Let's create a Custom metric adapter that parses the inference output for the answer between <answer> </answer> tags and then performs an exact match metric.

from amzn_nova_prompt_optimizer.core.input_adapters.metric_adapter import MetricAdapter
from typing import List, Any, Dict
import re
import json

class CustomMetric(MetricAdapter):
    def _parse_answer(self, model_output):
        # Parse Answer between tags
        match = re.search(r"<answer>(.*?)</answer>", model_output)
        if not match:
            return "Choice not found"
        return match.group(1).lower().strip()

    def _calculate_metrics(self, y_pred: Any, y_true: Any) -> Dict:
        # Peform Exact Match
        pred = self._parse_answer(y_pred)
        ground_truth = self._parse_answer(y_true)
        return int(pred == ground_truth)

    def apply(self, y_pred: Any, y_true: Any):
        # Apply to one row of the dataset
        return self._calculate_metrics(y_pred, y_true)

    def batch_apply(self, y_preds: List[Any], y_trues: List[Any]):
        # Apply to the whole dataset
        evals = []
        for y_pred, y_true in zip(y_preds, y_trues):
            evals.append(self.apply(y_pred, y_true))
        return sum(evals)/len(evals)

metric_adapter = CustomMetric()

Core Functions Apply the metric on a prediction and ground_truth one row at a time

y_pred = "The question asks ...... <answer>3</answer>"
y_true = "<answer>3</answer>"

# Apply the metric on a prediction and ground_truth
score = metric_adapter.apply(y_pred, y_true)

# score = 1

Apply the metric on a list of prediction and ground_truth i.e. for the dataset

y_preds = ["The question asks ...... <answer>3</answer>", "The question asks ...... <answer>5</answer>"]
y_trues = ["<answer>3</answer>", "<answer>4</answer>"]

# Apply the metric on a list of prediction and ground_truth
aggregeate_score = metric_adapter.batch_apply(y_preds, y_trues)

# aggregeate_score = 0.5

5. Optimization Adapter

Responsibility: Load Optimizer, Prompt Adapter, and Optionally Dataset Adapter, Metric Adapter, and Inference Adapter. Perform Optimization and ability to create a Prompt Adapter with the Optimized Prompt.

Sample Optimization Initialization

from amzn_nova_prompt_optimizer.core.optimizers import NovaPromptOptimizer

nova_prompt_optimizer = NovaPromptOptimizer(prompt_adapter=prompt_adapter, inference_adapter=inference_adapter, dataset_adapter=train_dataset_adapter, metric_adapter=metric_adapter)

optimized_prompt_adapter = nova_prompt_optimizer.optimize(mode="lite")

We can take a look more deeply into the optimizers in the next section.

Optimizers

NovaPromptOptimizer

NovaPromptOptimizer is a combination of Meta Prompting using the Nova Guide on prompting and DSPy's MIPROv2 Optimizer using Nova Prompting Tips. NovaPromptOptimizer first runs a meta prompter to identify system instructions and user template from the prompt adapter. Then MIPROv2 is run on top of this to optimize system instructions and identify few-shot samples that need to be added. The few shot samples are added as converse format so they are added as User/Assistant turns.

Requirements: NovaPromptOptimizer requires Prompt Adapter, Dataset Adapter, Metric Adapter and Inference Adapter.

Optimization Example

from amzn_nova_prompt_optimizer.core.optimizers import NovaPromptOptimizer

nova_prompt_optimizer = NovaPromptOptimizer(prompt_adapter=prompt_adapter, inference_adapter=inference_adapter, dataset_adapter=train_dataset_adapter, metric_adapter=metric_adapter)

optimized_prompt_adapter = nova_prompt_optimizer.optimize(mode="lite")

NovaPromptOptimizer uses Nova 2.0 Lite for Meta Prompting and then uses MIPROv2 with 20 candidates and 30 trials. The task model depends on the mode it's set at.

Automatic Meta-Prompting with Bedrock: When you don't provide a meta_prompt_inference_adapter, NovaPromptOptimizer automatically creates a BedrockInferenceAdapter with Nova 2.0 Lite for the meta-prompting phase. This means you can use any inference adapter (including SageMaker) for your task model, and the optimizer will handle meta-prompting with Bedrock automatically.

Optimization Modes:

Mode	Meta-Prompt Model	Task Model	Use Case
micro	Nova 2.0 Lite	Nova Micro	Fast, cost-effective
lite	Nova 2.0 Lite	Nova Lite	Balanced (default)
pro	Nova 2.0 Lite	Nova Pro	High quality
lite-2	Nova 2.0 Lite	Nova 2.0 Lite	Maximum quality

You can specify enable_json_fallback=False to disable the behavior that MIPROv2 will fallback to use JSONAdapter to parse LM model output. This will force MIPROv2 use structured output (pydantic model) to parse LM output.

Custom Mode: You could also define a custom mode and pass your own parameter values to NovaPromptOptimizer

from amzn_nova_prompt_optimizer.core.optimizers import NovaPromptOptimizer

nova_prompt_optimizer = NovaPromptOptimizer(prompt_adapter=prompt_adapter, inference_adapter=inference_adapter, dataset_adapter=train_dataset_adapter, metric_adapter=metric_adapter)

optimized_prompt_adapter = nova_prompt_optimizer.optimize(mode="custom", custom_params={"task_model_id": "us.amazon.nova-pro-v1:0",
    "num_candidates": 10,
    "num_trials": 15,
    "max_bootstrapped_demos": 5,
    "max_labeled_demos": 0
})

Learn More about the Optimizers here

Evaluator

The SDK also provides a way to baseline prompts and provide evaluation scores. The evaluator has the aggregate_score and scores function.

Initialization Example

from amzn_nova_prompt_optimizer.core.evaluation import Evaluator

evaluator = Evaluator(nova_mp_optimized_prompt_adapter, test_dataset_adapter, metric_adapter, inference_adapter)

nova_mp_score = evaluator.aggregate_score(model_id="us.amazon.nova-lite-v1:0")

Core Functions

Runs Batch evaluation on the dataset using the batch_apply function of the metric

# Uses Batch Apply
aggregate_score = evaluator.aggregate_score(model_id)

Runs evaluation on the dataset a row at a time and returns the eval results as a whole.

# Uses Apply metric. Returns a list of scores.
scores = evaluator.score(model_id)

Save the eval results.

# Save the eval results
evaluator.save("eval_results.jsonl")

Note: You may come across the below warning. This is when prompt variables are missing from the prompt, the inference runner under the evaluator appends them to the end of the prompt for continuity

WARNING amzn_nova_prompt_optimizer.core.inference: Warn: Prompt Variables not found in User Prompt, injecting them at the end of the prompt

Advanced Features

Separate Inference Adapters

Use different inference adapters for meta-prompting and task optimization phases. This is particularly useful when optimizing prompts for SageMaker endpoints while using Bedrock for meta-prompting.

Example:

from amzn_nova_prompt_optimizer.core.inference import (
    BedrockInferenceAdapter,
    SageMakerInferenceAdapter
)
from amzn_nova_prompt_optimizer.core.optimizers import NovaPromptOptimizer

# Bedrock for meta-prompting (generates optimized prompts)
meta_adapter = BedrockInferenceAdapter(region_name="us-east-1")

# SageMaker for task model (the model being optimized)
task_adapter = SageMakerInferenceAdapter(
    endpoint_name="my-endpoint",
    region_name="us-west-2"
)

# Create optimizer with separate adapters
optimizer = NovaPromptOptimizer(
    prompt_adapter=prompt_adapter,
    inference_adapter=task_adapter,           # For task optimization
    dataset_adapter=dataset_adapter,
    metric_adapter=metric_adapter,
    meta_prompt_inference_adapter=meta_adapter  # For meta-prompting
)

optimized_prompt = optimizer.optimize(mode="lite")

Benefits:

• Use the best model for each optimization phase
• Optimize SageMaker endpoints with Bedrock intelligence
• Cross-region support
• Independent rate limiting per adapter

SageMaker Endpoint Support

The SDK now supports optimizing prompts for models deployed on Amazon SageMaker. SageMaker endpoints must use an OpenAI-compatible message format:

Required Payload Format:

{
  "messages": [
    {"role": "system", "content": "You are helpful"},
    {"role": "user", "content": "Hello"}
  ],
  "max_tokens": 1000,
  "temperature": 0.7,
  "top_p": 0.9,
  "top_k": 50
}

Expected Response Format:

{
  "choices": [
    {
      "message": {
        "role": "assistant",
        "content": "Hello! How can I help you?"
      }
    }
  ]
}

Testing Your Endpoint:

from amzn_nova_prompt_optimizer.core.inference import SageMakerInferenceAdapter

adapter = SageMakerInferenceAdapter(
    endpoint_name="my-endpoint",
    region_name="us-west-2"
)

# Test connectivity
if adapter.test_connection():
    print("✓ Endpoint is ready for optimization")
else:
    print("✗ Endpoint connection failed")

Optimization Recommendations

1. Provide representative real-world evaluation sets and split them into training and testing sets. Ensure dataset is balanced on output label when splitting train and test sets.
2. For evaluation sets, the ground truth column should be as close to the inference output as possible. e.g. If the inference output is {"answer": "POSITIVE"} ground truth should also be in the same format {"answer": "POSITIVE"}
3. For NovaPromptOptimizer, choose the mode (mode= "lite-2" | ""pro" | "lite" | "micro") based on your Nova Model of choice. By default, we use "pro".
4. The apply function of the evaluation metric should return a numerical value between 0 and 1 for NovaPromptOptimizer or MIPROv2.

⚠️ Preview Status

NovaPromptOptimizer is currently in public preview. During this period:

• SDK functionality might change as we support more use cases.
• We welcome feedback and contributions

Interaction with AWS Bedrock

NovaPromptOptimizer only uses AWS Bedrock for Model Calls. Bedrock by default disables model invocation logging. Data is not retained in the user's AWS account unless the user enables Bedrock Model Invocation logging. Data storage by Amazon Bedrock is independent of the use of the SDK.

Please refer to Amazon Bedrock's Data Protection Guide for additional guidance

Acknowledgements

• Special acknowledgment to DSPy – your innovations continue to inspire us.