Method AddProviderModel

module/ai-model/iml.go:152–202

(ctx *gin.Context, provider string, input *model_dto.Model)

}

func (i *imlProviderModelModule) AddProviderModel(ctx *gin.Context, provider string, input *model_dto.Model) (*model_dto.SimpleModel, error) {
	p, has := model_runtime.GetProvider(provider)
	if !has {
		return nil, fmt.Errorf("ai provider not found")
	}
	// check model name duplicate
	if has := i.providerModelService.CheckNameDuplicate(ctx, provider, input.Name, ""); has {
		return nil, fmt.Errorf("model name: `%s` duplicate", input.Name)
	}
	// check provider model exist
	if _, has := p.GetModel(input.Name); has {
		return nil, fmt.Errorf("provider model already exist")
	}
	id := uuid.New().String()
	err := i.transaction.Transaction(ctx, func(ctx context.Context) error {
		typeValue := "chat"
		err := i.providerModelService.Save(ctx, id, &ai_model.Model{
			Name:                &input.Name,
			Type:                &typeValue,
			Provider:            &provider,
			AccessConfiguration: &input.AccessConfiguration,
			ModelParameters:     &input.ModelParameters,
		})
		if err != nil {
			return err
		}
		// update provider model
		iModel, err := model_runtime.NewCustomizeModel(id, input.Name, p.Logo(), input.AccessConfiguration, input.ModelParameters)
		if err != nil {
			return err
		}
		// 判断是否需要发布model
		if p.GetModelConfig().AccessConfigurationStatus {
			if err := i.syncGateway(ctx, cluster.DefaultClusterID, []*gateway.DynamicRelease{
				newModel(provider, input.Name, input.AccessConfiguration),
			}, true); err != nil {
				return err
			}
		}

		p.SetModel(id, iModel)
		return nil
	})
	if err != nil {
		return nil, err
	}
	return &model_dto.SimpleModel{
		Id:   id,
		Name: input.Name,
	}, nil
}

func newModel(provider string, model string, config string) *gateway.DynamicRelease {
	if config == "" {