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

	
30

	
31

	
32

	
33

	
34

	
35

	
36

	
37

	
38

	
39

	
40

	
41

	
42

	
43

	
44

	
45

	
46

	
47

	
48

	
49

	
50

	
            # yaml-language-server: $schema=https://promptfoo.dev/config-schema.json
description: 'Comparing multiple SageMaker model endpoints'

prompts:
  - 'Write a tweet about {{topic}}'
  - 'Write a short blog introduction about {{topic}}'
  - 'List 3 key benefits of {{topic}}'

providers:
  - id: sagemaker:openai:your-openai-endpoint
    label: 'OpenAI-compatible model'
    config:
      region: us-east-1
      modelType: openai
      temperature: 0.7
      maxTokens: 256

  - id: sagemaker:anthropic:your-claude-endpoint
    label: 'Claude-compatible model'
    config:
      region: us-east-1
      modelType: anthropic
      temperature: 0.7
      maxTokens: 256

  - id: sagemaker:llama:your-llama-endpoint
    label: 'Llama-compatible model'
    config:
      region: us-east-1
      modelType: llama
      temperature: 0.7
      maxTokens: 256

  - id: sagemaker:jumpstart:your-jumpstart-endpoint
    label: 'JumpStart model'
    config:
      region: us-east-1
      modelType: jumpstart
      temperature: 0.7
      maxTokens: 256
      responseFormat:
        path: 'json.generated_text'

tests:
  - vars:
      topic: sustainable packaging
  - vars:
      topic: artificial intelligence in healthcare
  - vars:
      topic: remote work policies