修复供应商修改配置失败的问题

This commit is contained in:
Liujian
2024-10-14 16:29:23 +08:00
parent f1052ae2f3
commit b04d06da4c
4 changed files with 20 additions and 37 deletions
@@ -38,4 +38,4 @@ provider_credential_schema:
placeholder:
zh_Hans: 在此输入您的 Secret Key
en_US: Enter your Secret Key
address: https://api.openai.com
address: https://aip.baidubce.com
+9 -1
View File
@@ -32,7 +32,15 @@ func (p ParamValidator) GenConfig(target map[string]interface{}, origin map[stri
if !ok || v == "******" {
continue
}
target[rule.Name] = origin[rule.Name]
tv, ok := target[rule.Name]
if !ok {
continue
}
v, ok = tv.(string)
if !ok || v == "******" {
target[rule.Name] = origin[rule.Name]
}
}
data, err := json.Marshal(target)
if err != nil {
-28
View File
@@ -4,7 +4,6 @@ import (
"encoding/json"
"fmt"
"net/url"
"strings"
yaml "gopkg.in/yaml.v3"
@@ -229,30 +228,3 @@ func (p *providerUri) Host() string {
func (p *providerUri) Path() string {
return p.path
}
func PartialMasking(origin string, begin int, length int) string {
target := strings.Builder{}
runes := []rune(origin)
size := len(runes)
if begin > size {
return origin
} else if length == -1 || begin+length > size {
for i := 0; i < begin; i++ {
target.WriteRune(runes[i])
}
for i := begin; i < size; i++ {
target.WriteRune('*')
}
} else {
for i := 0; i < begin; i++ {
target.WriteRune(runes[i])
}
for i := begin; i < begin+length; i++ {
target.WriteRune('*')
}
for i := begin + length; i < size; i++ {
target.WriteRune(runes[i])
}
}
return target.String()
}
+10 -7
View File
@@ -95,10 +95,7 @@ func (i *imlServiceController) CreateAIService(ctx *gin.Context, teamID string,
if input.Provider == nil {
return nil, fmt.Errorf("provider is required")
}
pv, err := i.providerModule.Provider(ctx, *input.Provider)
if err != nil {
return nil, err
}
if input.Id == "" {
input.Id = uuid.New().String()
}
@@ -113,8 +110,14 @@ func (i *imlServiceController) CreateAIService(ctx *gin.Context, teamID string,
if !has {
return nil, fmt.Errorf("provider not found")
}
m, has := p.DefaultModel(model_runtime.ModelTypeLLM)
if !has {
return nil, fmt.Errorf("provider default llm not found")
}
var info *service_dto.Service
err = i.transaction.Transaction(ctx, func(txCtx context.Context) error {
err := i.transaction.Transaction(ctx, func(txCtx context.Context) error {
var err error
info, err = i.module.Create(ctx, teamID, input)
if err != nil {
return err
@@ -143,8 +146,8 @@ func (i *imlServiceController) CreateAIService(ctx *gin.Context, teamID string,
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: pv.DefaultLLM,
Config: pv.DefaultLLMConfig,
Id: m.ID(),
Config: m.DefaultConfig(),
}
name := "Demo Translation API"
description := "A demo that shows you how to use a prompt to create a Translation API."