third/APIPark
1
0
Fork 0
mirror of https://github.com/APIParkLab/APIPark.git synced 2026-06-04 10:13:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix bug

Browse Source
This commit is contained in:
Liujian
2025-03-14 15:51:33 +08:00
parent 33c61f6131
commit bc85e59352
5 changed files with 33 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ai-provider/model-runtime/model-providers/bailian/bailian.yaml
+1 -1
View File
@@ -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
+21 -21
View File
@@ -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{
{
Key: "source_lang",
Description: "",
Require: true,
},
{
Key: "target_lang",
Description: "",
Require: true,
},
{
Key: "text",
Description: "",
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}}",
//Variables: []*ai_api_dto.AiPromptVariable{
// {
// Key: "source_lang",
// Description: "",
// Require: true,
// },
// {
// Key: "target_lang",
// Description: "",
// Require: true,
// },
// {
// Key: "text",
// Description: "",
// 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}}",
}
aiModel := &ai_api_dto.AiModel{
Id: m.ID(),
Config: m.DefaultConfig(),
Provider: providerId,
}
name := "Demo Translation API"
description := "A demo that shows you how to use a prompt to create a Translation API."
name := "Demo Chat 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
+8 -2
View File
@@ -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
+2 -1
View File
@@ -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
+1 -1
View File
@@ -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 {