Function insertChatModelConfigForTest

coderd/database/querier_test.go:10858–10893

(
	ctx context.Context,
	t testing.TB,
	store database.Store,
	params database.InsertChatModelConfigParams,
)

}

func insertChatModelConfigForTest(
	ctx context.Context,
	t testing.TB,
	store database.Store,
	params database.InsertChatModelConfigParams,
) (database.ChatModelConfig, error) {
	t.Helper()
	if params.AIProviderID.Valid {
		return store.InsertChatModelConfig(ctx, params)
	}
	providerName := params.Provider
	if providerName == "" {
		providerName = "openai"
		params.Provider = providerName
	}
	providers, err := store.GetAIProviders(ctx, database.GetAIProvidersParams{IncludeDisabled: true})
	if err != nil {
		return database.ChatModelConfig{}, err
	}
	var provider database.AIProvider
	for _, candidate := range providers {
		if candidate.Type != database.AIProviderType(providerName) {
			continue
		}
		if provider.ID == uuid.Nil || candidate.CreatedAt.After(provider.CreatedAt) {
			provider = candidate
		}
	}
	if provider.ID == uuid.Nil {
		provider = dbgen.AIProvider(t, store, database.AIProvider{
			Type: database.AIProviderType(providerName),
		})
	}
	params.AIProviderID = uuid.NullUUID{UUID: provider.ID, Valid: true}
	return store.InsertChatModelConfig(ctx, params)
}

func TestInsertChatMessages(t *testing.T) {
	t.Parallel()