fix tests

This commit is contained in:
Michael Yang
2024-02-01 11:48:11 -08:00
parent d125510b4b
commit e49dc9f3d8
2 changed files with 28 additions and 8 deletions

View File

@@ -455,7 +455,8 @@ func Test_ChatPrompt(t *testing.T) {
NumCtx: tt.numCtx,
},
}
got, err := trimmedPrompt(context.Background(), tt.chat, m)
// TODO: add tests for trimming images
got, _, err := trimmedPrompt(context.Background(), tt.chat, m)
if tt.wantErr != "" {
if err == nil {
t.Errorf("ChatPrompt() expected error, got nil")