mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-11 02:15:48 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

enhance: check whether openai.token is empty

Browse source
This commit is contained in:
0xJacky 2023-03-22 09:18:56 +08:00
parent 94408cd21f
commit e11bc16842
No known key found for this signature in database
GPG key ID: B6E4A6E4A561BAF0
1 changed files with 12 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

16
server/api/openai.go
View file

@ -10,7 +10,6 @@ import (
"github.com/pkg/errors"
"github.com/sashabaranov/go-openai"
"io"
"log"
"net/http"
"net/url"
"os"
@ -39,7 +38,17 @@ func MakeChatCompletionRequest(c *gin.Context) {
c.Writer.Header().Set("Cache-Control", "no-cache")
c.Writer.Header().Set("Connection", "keep-alive")
c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
log.Println(settings.OpenAISettings.Token)
if settings.OpenAISettings.Token == "" {
c.Stream(func(w io.Writer) bool {
c.SSEvent("message", gin.H{
"type": "error",
"content": "[Error] OpenAI token is empty",
})
return false
})
return
}
config := openai.DefaultConfig(settings.OpenAISettings.Token)
@ -104,9 +113,8 @@ func MakeChatCompletionRequest(c *gin.Context) {
return
}
// Send SSE to client
message := fmt.Sprintf("%s", response.Choices[0].Delta.Content)
fmt.Printf("%s", response.Choices[0].Delta.Content)
fmt.Printf("%s", message)
_ = os.Stdout.Sync()
msgChan <- message