62 lines
1.3 KiB
Go
62 lines
1.3 KiB
Go
|
package ai
|
||
|
|
|
||
|
|
import (
|
||
|
|
"apigo.cc/ai/ai/llm"
|
||
|
|
_ "apigo.cc/ai/ai/llm/openai"
|
||
|
|
_ "apigo.cc/ai/ai/llm/zhipu"
|
||
|
|
"github.com/ssgo/config"
|
||
|
|
"github.com/ssgo/u"
|
||
|
|
"os"
|
||
|
|
)
|
||
|
|
|
||
|
|
var confAes = u.NewAes([]byte("?GQ$0K0GgLdO=f+~L68PLm$uhKr4'=tV"), []byte("VFs7@sK61cj^f?HZ"))
|
||
|
|
var keysIsSet = false
|
||
|
|
var isInit = false
|
||
|
|
|
||
|
|
func SetSSKey(key, iv []byte) {
|
||
|
|
if !keysIsSet {
|
||
|
|
confAes = u.NewAes(key, iv)
|
||
|
|
keysIsSet = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func Init() {
|
||
|
|
InitFrom("")
|
||
|
|
}
|
||
|
|
|
||
|
|
func InitFrom(filePath string) {
|
||
|
|
if !isInit {
|
||
|
|
isInit = true
|
||
|
|
list := map[string]*struct {
|
||
|
|
Endpoint string
|
||
|
|
ApiKey string
|
||
|
|
DefaultChatModelConfig llm.ChatConfig
|
||
|
|
Llm string
|
||
|
|
}{}
|
||
|
|
savedPath := ""
|
||
|
|
if filePath != "" {
|
||
|
|
curPath, _ := os.Getwd()
|
||
|
|
if curPath != filePath {
|
||
|
|
savedPath = curPath
|
||
|
|
_ = os.Chdir(filePath)
|
||
|
|
config.ResetConfigEnv()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
_ = config.LoadConfig("llm", &list)
|
||
|
|
if savedPath != "" {
|
||
|
|
_ = os.Chdir(savedPath)
|
||
|
|
}
|
||
|
|
for name, llmConf := range list {
|
||
|
|
if llmConf.Llm == "" {
|
||
|
|
llmConf.Llm = name
|
||
|
|
}
|
||
|
|
llmConf.ApiKey = confAes.DecryptUrlBase64ToString(llmConf.ApiKey)
|
||
|
|
llm.Create(name, llmConf.Llm, llm.Config{
|
||
|
|
Endpoint: llmConf.Endpoint,
|
||
|
|
ApiKey: llmConf.ApiKey,
|
||
|
|
ChatConfig: llmConf.DefaultChatModelConfig,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|