Method UpdateProviderConfig

module/ai/iml.go:569–703

(ctx context.Context, id string, input *ai_dto.UpdateConfig)

}

func (i *imlProviderModule) UpdateProviderConfig(ctx context.Context, id string, input *ai_dto.UpdateConfig) error {
	p, has := model_runtime.GetProvider(id)
	if !has {
		return fmt.Errorf("ai provider not found")
	}

	return i.transaction.Transaction(ctx, func(ctx context.Context) error {
		info, err := i.providerService.Get(ctx, id)
		if err != nil {
			if !errors.Is(err, gorm.ErrRecordNotFound) {
				return err
			}
			if input.DefaultLLM == "" {
				defaultLLM, has := p.DefaultModel(model_runtime.ModelTypeLLM)
				if !has {
					return fmt.Errorf("ai provider default llm not found")
				}
				input.DefaultLLM = defaultLLM.ID()
			}
			info = &ai.Provider{
				Id:         id,
				Name:       p.Name(),
				DefaultLLM: input.DefaultLLM,
				Config:     input.Config,
			}
			err = i.providerService.Create(ctx, &ai.CreateProvider{
				Id:         info.Id,
				Name:       info.Name,
				DefaultLLM: input.DefaultLLM,
				Config:     input.Config,
			})
			if err != nil {
				return err
			}
		}
		model, has := p.GetModel(input.DefaultLLM)
		if !has {
			return fmt.Errorf("ai provider model not found")
		}
		err = p.Check(input.Config)
		if err != nil {
			return err
		}
		input.Config, err = p.GenConfig(input.Config, info.Config)
		if err != nil {
			return err
		}
		if input.DefaultLLM != "" {
			if defaultLLM, has := p.GetModel(input.DefaultLLM); has {
				p.SetDefaultModel(model_runtime.ModelTypeLLM, defaultLLM)
			}
		}
		status := 0
		if input.Enable != nil && *input.Enable {
			status = 1
		}
		pInfo := &ai.SetProvider{
			Name:       &info.Name,