refactor: adjust model_runtime structure of provider model

module/ai-model/iml.go

@@ -78,19 +78,25 @@ func (i *imlProviderModelModule) UpdateProviderModel(ctx *gin.Context, provider
 }
 
 func (i *imlProviderModelModule) DeleteProviderModel(ctx *gin.Context, provider string, id string) error {
+	p, has := model_runtime.GetProvider(provider)
 	// check provider exist
 	providerInfo, err := i.providerService.Get(ctx, provider)
 	if err != nil {
 		return err
 	}
-	if providerInfo == nil {
+	if providerInfo == nil || !has {
 		return fmt.Errorf("provider not found")
 	}
 	modelInfo, _ := i.providerModelService.Get(ctx, id)
 	if modelInfo == nil || modelInfo.Provider != provider {
 		return fmt.Errorf("model not found")
 	}
-	return i.providerModelService.Delete(ctx, id)
+	if err := i.providerModelService.Delete(ctx, id); err != nil {
+		return err
+	}
+	p.RemoveModel(id)
+
+	return nil
 }
 
 func (i *imlProviderModelModule) AddProviderModel(ctx *gin.Context, provider string, input *model_dto.Model) (*model_dto.SimpleModel, error) {

module/ai/iml.go

@@ -200,12 +200,12 @@ func (i *imlProviderModule) AddProvider(ctx context.Context, input *ai_dto.NewPr
 	}
 	id := uuid.New().String()
 	config, defaultLLM := "{\"api_endpoint_url\": \"http://127.0.0.1\", \"api_key\": \"\"}", ""
-	typeValue := 1
-	if err := i.providerService.Save(ctx, id, &ai.SetProvider{
-		Name:       &input.Name,
-		DefaultLLM: &defaultLLM,
-		Config:     &config,
-		Type:       &typeValue,
+	if err := i.providerService.Create(ctx, &ai.CreateProvider{
+		Id:         id,
+		Name:       input.Name,
+		DefaultLLM: defaultLLM,
+		Config:     config,
+		Type:       1,
 	}); err != nil {
 		return nil, err
 	}
@@ -213,8 +213,10 @@ func (i *imlProviderModule) AddProvider(ctx context.Context, input *ai_dto.NewPr
 	iProvider, _ := model_runtime.NewCustomizeProvider(id, input.Name, []model_runtime.IModel{}, "", "")
 	model_runtime.Register(id, iProvider)
 	return &ai_dto.SimpleProvider{
-		Id:   id,
-		Name: input.Name,
+		Id:            id,
+		Name:          input.Name,
+		DefaultConfig: config,
+		Logo:          model_runtime.GetCustomizeLogo(),
 	}, nil
 }