Method APIs

module/ai/iml.go:803–930

(ctx context.Context, keyword string, providerId string, start int64, end int64, page int, pageSize int, sortCondition string, asc bool, models []string, serviceIds []string)

}

func (i *imlAIApiModule) APIs(ctx context.Context, keyword string, providerId string, start int64, end int64, page int, pageSize int, sortCondition string, asc bool, models []string, serviceIds []string) ([]*ai_dto.APIItem, *ai_dto.Condition, int64, error) {
	modelItems := make([]*ai_dto.BasicInfo, 0)
	if providerId == ai_provider_local.ProviderLocal {
		items, err := i.aiLocalModelService.Search(ctx, "", nil, "update_at desc")
		if err != nil {
			return nil, nil, 0, err
		}
		modelItems = utils.SliceToSlice(items, func(e *ai_local.LocalModel) *ai_dto.BasicInfo {
			return &ai_dto.BasicInfo{
				Id:   e.Id,
				Name: e.Name,
			}
		})
	} else {
		p, has := model_runtime.GetProvider(providerId)
		if !has {
			return nil, nil, 0, fmt.Errorf("ai provider not found")
		}
		modelItems = utils.SliceToSlice(p.Models(), func(e model_runtime.IModel) *ai_dto.BasicInfo {
			return &ai_dto.BasicInfo{
				Id:   e.ID(),
				Name: e.ID(),
			}
		})
	}

	sortRule := "desc"
	if asc {
		sortRule = "asc"
	}
	services, err := i.serviceService.ServiceListByKind(ctx, service.AIService)
	if err != nil {
		return nil, nil, 0, err
	}
	serviceItems := make([]*ai_dto.BasicInfo, 0, len(services))
	serviceTeamMap := make(map[string]string)
	for _, s := range services {
		serviceItems = append(serviceItems, &ai_dto.BasicInfo{
			Id:   s.Id,
			Name: s.Name,
		})
		serviceTeamMap[s.Id] = s.Team

	}

	condition := &ai_dto.Condition{Services: serviceItems, Models: modelItems}
	switch sortCondition {
	default:
		w := map[string]interface{}{
			"provider": providerId,
		}
		if len(models) > 0 {
			w["model"] = models
		}
		if len(serviceIds) > 0 {
			w["service"] = serviceIds
		}
		apis, err := i.aiAPIService.Search(ctx, keyword, w, "update_at desc")