#163: Covered config to model building

pkg/routers/config_test.go

@@ -3,6 +3,8 @@ package routers
 import (
 	"testing"
 
+	"glide/pkg/providers/cohere"
+
 	"github.com/stretchr/testify/require"
 	"glide/pkg/providers"
 	"glide/pkg/providers/clients"
@@ -70,6 +72,49 @@ func TestRouterConfig_BuildModels(t *testing.T) {
 	require.IsType(t, &routing.LeastLatencyRouting{}, routers[1].chatRouting)
 }
 
+func TestRouterConfig_BuildModelsPerType(t *testing.T) {
+	tel := telemetry.NewTelemetryMock()
+	openAIParams := openai.DefaultParams()
+	cohereParams := cohere.DefaultParams()
+
+	cfg := LangRouterConfig{
+		ID:              "first_router",
+		Enabled:         true,
+		RoutingStrategy: routing.Priority,
+		Retry:           retry.DefaultExpRetryConfig(),
+		Models: []providers.LangModelConfig{
+			{
+				ID:          "first_model",
+				Enabled:     true,
+				Client:      clients.DefaultClientConfig(),
+				ErrorBudget: health.DefaultErrorBudget(),
+				Latency:     latency.DefaultConfig(),
+				OpenAI: &openai.Config{
+					APIKey:        "ABC",
+					DefaultParams: &openAIParams,
+				},
+			},
+			{
+				ID:          "second_model",
+				Enabled:     true,
+				Client:      clients.DefaultClientConfig(),
+				ErrorBudget: health.DefaultErrorBudget(),
+				Latency:     latency.DefaultConfig(),
+				Cohere: &cohere.Config{
+					APIKey:        "ABC",
+					DefaultParams: &cohereParams,
+				},
+			},
+		},
+	}
+
+	chatModels, streamChatModels, err := cfg.BuildModels(tel)
+
+	require.Len(t, chatModels, 2)
+	require.Len(t, streamChatModels, 1)
+	require.NoError(t, err)
+}
+
 func TestRouterConfig_InvalidSetups(t *testing.T) {
 	defaultParams := openai.DefaultParams()