#43: Fixed tests

EinStack · Jan 9, 2024 · e2df092 · e2df092
1 parent 7407148
commit e2df092
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/routers/router_test.go b/pkg/routers/router_test.go
@@ -46,7 +46,7 @@ func TestLangRouter_Priority_PickFistHealthy(t *testing.T) {
 	for i := 0; i < 2; i++ {
 		resp, err := router.Chat(ctx, req)
 
-		require.Equal(t, "first", resp.Model)
+		require.Equal(t, "first", resp.ModelID)
 		require.Equal(t, "test_router", resp.RouterID)
 		require.NoError(t, err)
 	}