nirbarazida
/
promptfoo
mirror of https://github.com/promptfoo/promptfoo


  
1

	
2

	
3

	
4

	
5

	
6

	
7

	
8

	
9

	
10

	
11

	
12

	
13

	
14

	
15

	
16

	
17

	
18

	
19

	
20

	
21

	
22

	
23

	
24

	
25

	
26

	
27

	
28

	
29

	
            # yaml-language-server: $schema=https://promptfoo.dev/config-schema.json
description: 'AWS SageMaker JumpStart models evaluation'

prompts:
  - 'Generate a creative name for a coffee shop that specializes in {{flavor}} coffee.'
  - 'What are some potential names for a coffee shop that specializes in {{flavor}} coffee?'
  - 'Write a short story about a robot that becomes self-aware.'

providers:
  - id: sagemaker:jumpstart:your-jumpstart-endpoint
    label: 'JumpStart Llama 3.2 (8B)'
    config:
      region: us-west-2
      modelType: jumpstart # Use the dedicated jumpstart model type
      maxTokens: 256 # Specifies max_new_tokens in parameters
      temperature: 0.7 # Controls randomness
      topP: 0.9 # Controls diversity
      contentType: 'application/json'
      acceptType: 'application/json'
      responseFormat:
        path: 'json.generated_text' # Extract this field from the response

tests:
  - vars:
      flavor: caramel
  - vars:
      flavor: pumpkin spice
  - vars:
      flavor: lavender