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

	
51

	
52

	
53

	
54

	
55

	
56

	
57

	
58

	
59

	
60

	
61

	
62

	
63

	
64

	
65

	
66

	
67

	
68

	
69

	
70

	
71

	
72

	
73

	
74

	
75

	
76

	
77

	
78

	
79

	
80

	
81

	
82

	
83

	
84

	
85

	
86

	
87

	
88

	
89

	
90

	
91

	
92

	
93

	
94

	
95

	
96

	
97

	
98

	
99

	
100

	
101

	
102

	
103

	
104

	
105

	
106

	
107

	
            prompts:
  - 'What is the color of {{topic}}? ONLY answer using the "color_json" tool.'

providers:
  - id: bedrock:amazon.nova-micro-v1:0
    config:
      region: 'us-east-1'
      inferenceConfig:
        temperature: 0.7
        max_new_tokens: 256
      toolConfig:
        toolChoice:
          auto: {}
        tools:
          - toolSpec:
              name: color_json
              description: 'Outputs the color of a thing'
              inputSchema:
                json:
                  type: object
                  properties:
                    name:
                      type: string
                      description: 'The name of the thing'
                    color:
                      type: string
                      description: 'The color of the thing'
                  required:
                    - name
                    - color
  - id: bedrock:amazon.nova-lite-v1:0
    config:
      region: 'us-east-1'
      inferenceConfig:
        temperature: 0.7
        max_new_tokens: 256
      toolConfig:
        toolChoice:
          auto: {}
        tools:
          - toolSpec:
              name: color_json
              description: 'Outputs the color of a thing'
              inputSchema:
                json:
                  type: object
                  properties:
                    name:
                      type: string
                      description: 'The name of the thing'
                    color:
                      type: string
                      description: 'The color of the thing'
                  required:
                    - name
                    - color
  - id: bedrock:amazon.nova-pro-v1:0
    config:
      region: 'us-east-1'
      inferenceConfig:
        temperature: 0.7
        max_new_tokens: 256
      toolConfig:
        toolChoice:
          auto: {}
        tools:
          - toolSpec:
              name: color_json
              description: 'Outputs the color of a thing'
              inputSchema:
                json:
                  type: object
                  properties:
                    name:
                      type: string
                      description: 'The name of the thing'
                    color:
                      type: string
                      description: 'The color of the thing'
                  required:
                    - name
                    - color

defaultTest:
  options:
    transform: 'JSON.parse(output).input.color'
tests:
  - vars:
      topic: sky
    assert:
      - type: equals
        value: blue
  - vars:
      topic: ocean
    assert:
      - type: equals
        value: blue
  - vars:
      topic: banana
    assert:
      - type: equals
        value: yellow
  - vars:
      topic: grass
    assert:
      - type: equals
        value: green