Skip to content

Commit 8ff5ef6

Browse files
committed
bumb: test fix
1 parent db94c4a commit 8ff5ef6

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

e2e/inference_test.go

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -120,7 +120,10 @@ func TestE2E_Inference(t *testing.T) {
120120
"schema": map[string]any{
121121
"type": "object",
122122
"properties": map[string]any{
123-
"answer": map[string]any{"type": "string"},
123+
"answer": map[string]any{
124+
"type": "string",
125+
"enum": []string{"yes"},
126+
},
124127
},
125128
"required": []string{"answer"},
126129
"additionalProperties": false,
@@ -151,6 +154,9 @@ func TestE2E_Inference(t *testing.T) {
151154
if output.Answer == "" {
152155
t.Fatalf("answer field is empty in output_text=%q", resp.OutputText)
153156
}
157+
if output.Answer != "yes" {
158+
t.Fatalf("answer = %q, want yes (output_text=%q)", output.Answer, resp.OutputText)
159+
}
154160
})
155161

156162
t.Run("AnthropicMessages", func(t *testing.T) {

0 commit comments

Comments
 (0)