36 lines
926 B
Go
36 lines
926 B
Go
package util
|
|
|
|
import (
|
|
"context"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
// GetServerName 获取服务名称
|
|
func GetServerName(ctx context.Context) string {
|
|
return g.Cfg().MustGet(ctx, "server.name", "").String()
|
|
}
|
|
|
|
// GetServerPort 从配置获取服务端口
|
|
func GetServerPort(ctx context.Context) string {
|
|
address := g.Cfg().MustGet(ctx, "server.address", ":8080").String()
|
|
// address 格式如 ":3009"，去掉冒号
|
|
if strings.HasPrefix(address, ":") {
|
|
return address[1:]
|
|
}
|
|
return "8080"
|
|
}
|
|
|
|
// GetModelPrompt 获取请求模型的提示词
|
|
func GetModelPrompt(ctx context.Context, modelType int) string {
|
|
key := "modelPrompts.types." + gconv.String(modelType)
|
|
return g.Cfg().MustGet(ctx, key, "").String()
|
|
}
|
|
|
|
// GetBuildPrompt 获取节点构建提示词
|
|
func GetBuildPrompt(ctx context.Context) string {
|
|
return g.Cfg().MustGet(ctx, "nodePrompts", "").String()
|
|
}
|