Method SimpleConfiguredProviders

module/ai/iml.go:341–413

(ctx context.Context, all bool)

}

func (i *imlProviderModule) SimpleConfiguredProviders(ctx context.Context, all bool) ([]*ai_dto.SimpleProviderItem, *ai_dto.BackupProvider, error) {
	list, err := i.providerService.List(ctx)
	if err != nil {
		return nil, nil, err
	}

	items := make([]*ai_dto.SimpleProviderItem, 0, len(list))

	healthProvider := make(map[string]struct{})
	if all {
		healthProvider[ai_provider_local.ProviderLocal] = struct{}{}
		items = append(items, &ai_dto.SimpleProviderItem{
			Id:            ai_provider_local.ProviderLocal,
			Name:          ai_provider_local.ProviderLocal,
			Logo:          ai_provider_local.LocalSvg,
			Configured:    true,
			DefaultConfig: "",
			Status:        ai_dto.ProviderEnabled,
			Type:          "local",
		})
	}

	var backup *ai_dto.BackupProvider
	for _, l := range list {
		p, has := model_runtime.GetProvider(l.Id)
		if !has {
			continue
		}
		model, has := p.GetModel(l.DefaultLLM)
		if !has {
			model, has = p.DefaultModel(model_runtime.ModelTypeLLM)
			if !has {
				continue
			}
		}
		item := &ai_dto.SimpleProviderItem{
			Id:            l.Id,
			Name:          l.Name,
			Logo:          p.Logo(),
			DefaultConfig: p.DefaultConfig(),
			Status:        ai_dto.ToProviderStatus(l.Status),
			Configured:    true,
			Model: &ai_dto.BasicInfo{
				Id:   model.ID(),
				Name: model.ID(),
			},
		}
		if item.Status == ai_dto.ProviderEnabled {
			healthProvider[l.Id] = struct{}{}
		}
		items = append(items, item)
	}

	aiBalanceItems, err := i.aiBalanceService.Search(ctx, "", nil, "priority asc")
	if err != nil {
		return nil, nil, err
	}
	for _, item := range aiBalanceItems {