more

Signed-off-by: bitliu <[email protected]>

Author: Xunzhuo

src/semantic-router/pkg/api/server_test.go

@@ -309,10 +309,17 @@ func TestOpenAIModelsEndpoint(t *testing.T) {
 				Name:    "primary",
 				Address: "127.0.0.1", // Changed from localhost to IP address
 				Port:    8000,
-				Models:  []string{"gpt-4o-mini", "llama-3.1-8b-instruct"},
 				Weight:  1,
 			},
 		},
+		ModelConfig: map[string]config.ModelParams{
+			"gpt-4o-mini": {
+				PreferredEndpoints: []string{"primary"},
+			},
+			"llama-3.1-8b-instruct": {
+				PreferredEndpoints: []string{"primary"},
+			},
+		},
 	}
 
 	apiServer := &ClassificationAPIServer{