Method AddProvider

module/ai/iml.go:203–231

(ctx context.Context, input *ai_dto.NewProvider)

}

func (i *imlProviderModule) AddProvider(ctx context.Context, input *ai_dto.NewProvider) (*ai_dto.SimpleProvider, error) {
	_, has := model_runtime.GetProvider(strings.ToLower(input.Name))
	if has {
		return nil, fmt.Errorf("provider `%s` duplicate", input.Name)
	}
	// uuid = name
	if has := i.providerService.CheckUuidDuplicate(ctx, input.Name); has {
		return nil, fmt.Errorf("provider `%s` duplicate", input.Name)
	}
	config, defaultLLM := "{\"base_url\": \"\", \"api_key\": \"\"}", ""
	if err := i.providerService.Create(ctx, &ai.CreateProvider{
		Id:         input.Name,
		Name:       input.Name,
		DefaultLLM: defaultLLM,
		Config:     config,
		Type:       1,
	}); err != nil {
		return nil, err
	}
	// register provider
	iProvider, _ := model_runtime.NewCustomizeProvider(input.Name, input.Name, []model_runtime.IModel{}, "", "")
	model_runtime.Register(input.Name, iProvider)
	return &ai_dto.SimpleProvider{
		Id:            input.Name,
		Name:          input.Name,
		DefaultConfig: config,
		Logo:          iProvider.Logo(),
	}, nil
}

func (i *imlProviderModule) SimpleProvider(ctx context.Context, id string) (*ai_dto.SimpleProvider, error) {
	p, has := model_runtime.GetProvider(id)