fix tests

This commit is contained in:
Jeffrey Morgan
2023-12-13 14:42:30 -05:00
parent bbd41494bf
commit 4a1abfe4fa
2 changed files with 4 additions and 3 deletions

View File

@@ -81,7 +81,7 @@ func TestChat(t *testing.T) {
Template: tt.template,
}
t.Run(tt.name, func(t *testing.T) {
got, err := m.ChatPrompt(tt.msgs)
got, _, err := m.ChatPrompt(tt.msgs)
if tt.wantErr != "" {
if err == nil {
t.Errorf("ChatPrompt() expected error, got nil")