fix bug

ai-provider/model-runtime/model-providers/bailian/bailian.yaml

@@ -1,7 +1,7 @@
 provider: bailian
 label:
   zh_Hans: 阿里云百炼
-  en_US: bailian
+  en_US: BaiLian
 icon_small:
   en_US: icon_s_en.svg
 icon_large:

controller/system/iml.go

@@ -395,36 +395,36 @@ func (i *imlInitController) createAIService(ctx context.Context, teamID string,
 		if err != nil {
 			return err
 		}
-		path := fmt.Sprintf("/%s/demo_translation_api", strings.Trim(input.Prefix, "/"))
+		path := fmt.Sprintf("/%s/chat/completions", strings.Trim(input.Prefix, "/"))
 		timeout := 300000
 		retry := 0
 		aiPrompt := &ai_api_dto.AiPrompt{
-			Variables: []*ai_api_dto.AiPromptVariable{
+			//Variables: []*ai_api_dto.AiPromptVariable{
-				{
+			//	{
-					Key:         "source_lang",
+			//		Key:         "source_lang",
-					Description: "",
+			//		Description: "",
-					Require:     true,
+			//		Require:     true,
-				},
+			//	},
-				{
+			//	{
-					Key:         "target_lang",
+			//		Key:         "target_lang",
-					Description: "",
+			//		Description: "",
-					Require:     true,
+			//		Require:     true,
-				},
+			//	},
-				{
+			//	{
-					Key:         "text",
+			//		Key:         "text",
-					Description: "",
+			//		Description: "",
-					Require:     true,
+			//		Require:     true,
-				},
+			//	},
-			},
+			//},
-			Prompt: "You need to translate {{source_lang}} into {{target_lang}}, and the following is the content that needs to be translated.\n---\n{{text}}",
+			//Prompt: "You need to translate {{source_lang}} into {{target_lang}}, and the following is the content that needs to be translated.\n---\n{{text}}",
 		}
 		aiModel := &ai_api_dto.AiModel{
 			Id:       m.ID(),
 			Config:   m.DefaultConfig(),
 			Provider: providerId,
 		}
-		name := "Demo Translation API"
+		name := "Demo Chat API"
-		description := "A demo that shows you how to use a prompt to create a Translation API."
+		description := "A demo that shows you how to use a prompt to create a Chat API."
 		apiId := uuid.New().String()
 		err = i.aiAPIModule.Create(
 			ctx,

module/ai-balance/iml.go

@@ -268,7 +268,8 @@ func (i *imlBalanceModule) getLocalBalances(ctx context.Context, v string) ([]*g
 		var has bool
 		v, has = i.settingService.Get(ctx, "system.ai_model.ollama_address")
 		if !has {
-			return nil, fmt.Errorf("ollama address not found")
+			//return nil, fmt.Errorf("ollama address not found")
+			return nil, nil
 		}
 	}
 
@@ -294,7 +295,8 @@ func (i *imlBalanceModule) getBalances(ctx context.Context) ([]*gateway.DynamicR
 	}
 	v, has := i.settingService.Get(ctx, "system.ai_model.ollama_address")
 	if !has {
-		return nil, fmt.Errorf("ollama address not found")
+		//return nil, fmt.Errorf("ollama address not found")
+		return nil, nil
 	}
 	releases := make([]*gateway.DynamicRelease, 0, len(balances))
 	for _, item := range balances {
@@ -305,6 +307,10 @@ func (i *imlBalanceModule) getBalances(ctx context.Context) ([]*gateway.DynamicR
 				continue
 			}
 			base = fmt.Sprintf("%s://%s%s", p.URI().Scheme(), p.URI().Host(), p.URI().Path())
+		} else {
+			if v == "" {
+				continue
+			}
 		}
 		releases = append(releases, newRelease(item, base))
 	}

module/ai-local/iml.go

@@ -614,7 +614,8 @@ func (i *imlLocalModel) getLocalModels(ctx context.Context, v string) ([]*gatewa
 		var has bool
 		v, has = i.settingService.Get(ctx, "system.ai_model.ollama_address")
 		if !has {
-			return nil, errors.New("ollama_address not set")
+			//return nil, errors.New("ollama_address not set")
+			return nil, nil
 		}
 	}
 	provider := ai_provider_local.ProviderLocal

module/service/iml.go

@@ -473,7 +473,7 @@ func (i *imlServiceModule) Edit(ctx context.Context, id string, input *service_d
 
 			}
 		}
-		if input.ModelMapping != nil {
+		if input.ModelMapping != nil && *input.ModelMapping != "" {
 			m := make(map[string]string)
 			err = json.Unmarshal([]byte(*input.ModelMapping), &m)
 			if err != nil {