Method createAIService

controller/service/iml.go:505–653

(ctx *gin.Context, teamID string, input *service_dto.CreateService)

}

func (i *imlServiceController) createAIService(ctx *gin.Context, teamID string, input *service_dto.CreateService) (*service_dto.Service, error) {
	if input.Provider == nil {
		return nil, fmt.Errorf("provider is required")
	}

	if input.Id == "" {
		input.Id = uuid.New().String()
	}
	if input.Prefix == "" {
		if len(input.Id) < 9 {
			input.Prefix = input.Id
		} else {
			input.Prefix = input.Id[:8]
		}
	}
	modelId := ""
	modelCfg := ""
	modelType := "online"
	if input.Model != nil {
		modelId = *input.Model
	}
	if *input.Provider == ai_provider_local.ProviderLocal {
		modelType = "local"
		if modelId == "" {
			list, err := i.aiLocalModel.SimpleList(ctx)
			if err != nil {
				return nil, err
			}
			if len(list) == 0 {
				return nil, fmt.Errorf("no local model")
			}
			modelId = list[0].Id
		}
		modelCfg = ai_provider_local.LocalConfig
	} else {
		pv, err := i.providerModule.Provider(ctx, *input.Provider)
		if err != nil {
			return nil, err
		}
		p, has := model_runtime.GetProvider(*input.Provider)
		if !has {
			return nil, fmt.Errorf("provider not found")
		}
		if modelId == "" {
			modelId = pv.DefaultLLM
		}
		m, has := p.GetModel(modelId)
		if !has {
			return nil, fmt.Errorf("model %s not found", pv.DefaultLLM)
		}
		modelId = m.Name()
		modelCfg = m.DefaultConfig()
	}

	var info *service_dto.Service
	err := i.transaction.Transaction(ctx, func(txCtx context.Context) error {
		var err error
		info, err = i.module.Create(ctx, teamID, input)