Method UpdateProviderModel

module/ai-model/iml.go:59–100

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

}

func (i *imlProviderModelModule) UpdateProviderModel(ctx *gin.Context, provider string, input *model_dto.EditModel) error {
	p, has := model_runtime.GetProvider(provider)
	if !has {
		return fmt.Errorf("ai provider not found")
	}
	// check provider exist
	_, err := i.providerService.Get(ctx, provider)
	if err != nil {
		return err
	}
	modelInfo, _ := i.providerModelService.Get(ctx, input.Id)
	if modelInfo == nil || modelInfo.Provider != provider {
		return fmt.Errorf("model not found")
	}
	return i.transaction.Transaction(ctx, func(ctx context.Context) error {
		if err = i.providerModelService.Save(ctx, input.Id, &ai_model.Model{
			AccessConfiguration: &input.AccessConfiguration,
			ModelParameters:     &input.ModelParameters,
		}); err != nil {
			return err
		}

		// update provider model
		iModel, err := model_runtime.NewCustomizeModel(input.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(input.Id, iModel)
		return nil
	})

	return nil
}

func (i *imlProviderModelModule) DeleteProviderModel(ctx *gin.Context, provider string, id string) error {
	p, has := model_runtime.GetProvider(provider)