0ct0pu5/siyuan
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

🎨 OpenAI API 接入增加环境变量 SIYUAN_OPENAI_API_BASE_URL

Browse source
This commit is contained in:
Liang Ding 2023-03-08 20:00:17 +08:00
parent 1c9e0356cc
commit a8129de5f5
No known key found for this signature in database
GPG key ID: 136F30F901A2231D
2 changed files with 18 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
kernel/model/ai.go
View file

@ -27,26 +27,20 @@ import (
)
func ChatGPT(msg string) (ret string) {
cloud := IsSubscriber()
if !cloud && !isOpenAIAPIEnabled() {
if !isOpenAIAPIEnabled() {
return
}
cloud = false
return chatGPT(msg, cloud)
return chatGPT(msg, false)
}
func ChatGPTWithAction(ids []string, action string) (ret string) {
cloud := IsSubscriber()
if !cloud && !isOpenAIAPIEnabled() {
if !isOpenAIAPIEnabled() {
return
}
cloud = false
msg := getBlocksContent(ids)
ret = chatGPTWithAction(msg, action, cloud)
ret = chatGPTWithAction(msg, action, false)
return
}

15
kernel/util/openai.go
View file

@ -34,6 +34,7 @@ var (
OpenAIAPITimeout = 30 * time.Second
OpenAIAPIProxy = ""
OpenAIAPIMaxTokens = 0
OpenAIAPIBaseURL = "https://api.openai.com/v1"
)
func ChatGPT(msg string, contextMsgs []string, c *gogpt.Client) (ret string, stop bool) {
@ -94,6 +95,8 @@ func NewOpenAIClient() *gogpt.Client {
config.HTTPClient = &http.Client{Transport: &http.Transport{Proxy: http.ProxyURL(proxyUrl)}}
}
}
config.BaseURL = OpenAIAPIBaseURL
return gogpt.NewClientWithConfig(config)
}
@ -124,5 +127,15 @@ func initOpenAI() {
}
}
logging.LogInfof("OpenAI API enabled [maxTokens=%d, timeout=%ds, proxy=%s]", OpenAIAPIMaxTokens, int(OpenAIAPITimeout.Seconds()), OpenAIAPIProxy)
baseURL := os.Getenv("SIYUAN_OPENAI_API_BASE_URL")
if "" != baseURL {
OpenAIAPIBaseURL = baseURL
}
logging.LogInfof("OpenAI API enabled\n"+
" baseURL=%s\n"+
" timeout=%ds\n"+
" proxy=%s\n"+
" maxTokens=%d",
OpenAIAPIBaseURL, int(OpenAIAPITimeout.Seconds()), OpenAIAPIProxy, OpenAIAPIMaxTokens)
}