Method ConfiguredProviders

module/ai/iml.go:247–309

(ctx context.Context, keyword string)

}

func (i *imlProviderModule) ConfiguredProviders(ctx context.Context, keyword string) ([]*ai_dto.ConfiguredProviderItem, error) {
	// 获取已配置的AI服务商
	list, err := i.providerService.Search(ctx, keyword, nil, "update_at desc")
	if err != nil {
		return nil, fmt.Errorf("get provider list error:%v", err)
	}
	aiAPIMap, err := i.aiAPIService.CountMapByProvider(ctx, "", nil)
	if err != nil {
		return nil, fmt.Errorf("get ai api count error:%v", err)
	}
	keyMap, err := i.aiKeyService.CountMapByProvider(ctx, "", nil)
	if err != nil {
		return nil, fmt.Errorf("get ai key count error:%v", err)
	}
	providers := make([]*ai_dto.ConfiguredProviderItem, 0, len(list))
	for _, l := range list {
		// 检查是否有默认Key
		_, err = i.aiKeyService.DefaultKey(ctx, l.Id)
		if err != nil {
			if !errors.Is(err, gorm.ErrRecordNotFound) {
				return nil, err
			}
			err = i.aiKeyService.Create(ctx, &ai_key.Create{
				ID:         l.Id,
				Name:       l.Name,
				Config:     l.Config,
				Provider:   l.Id,
				Priority:   1,
				Status:     1,
				ExpireTime: 0,
				UseToken:   0,
				Default:    true,
			})
			if err != nil {
				return nil, fmt.Errorf("create default key error:%v", err)
			}
		}

		p, has := model_runtime.GetProvider(l.Id)
		if !has {
			continue
		}
		apiCount := aiAPIMap[l.Id]
		defaultLLMName := ""
		if defaultModel, has := p.GetModel(l.DefaultLLM); has {
			defaultLLMName = defaultModel.Name()
		}
		providers = append(providers, &ai_dto.ConfiguredProviderItem{
			Id:             l.Id,
			Name:           l.Name,
			Logo:           p.Logo(),
			DefaultLLM:     l.DefaultLLM,
			DefaultLLMName: defaultLLMName,
			Status:         ai_dto.ToProviderStatus(l.Status),
			APICount:       apiCount,
			KeyCount:       keyMap[l.Id],
			CanDelete:      apiCount < 1,
			ModelCount:     int64(len(p.Models())),