mirrors/nginx-ui
1
0
Fork 0
mirror of https://github.com/0xJacky/nginx-ui.git synced 2025-05-12 02:45:49 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 6

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/pkg/errors"
"github.com/sashabaranov/go-openai" "github.com/sashabaranov/go-openai"
"io" "io"
"log"
"net/http" "net/http"
"net/url" "net/url"
"os" "os"
@ -39,7 +38,17 @@ func MakeChatCompletionRequest(c *gin.Context) {
c.Writer.Header().Set("Cache-Control", "no-cache") c.Writer.Header().Set("Cache-Control", "no-cache")
c.Writer.Header().Set("Connection", "keep-alive") c.Writer.Header().Set("Connection", "keep-alive")
c.Writer.Header().Set("Access-Control-Allow-Origin", "*") 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) config := openai.DefaultConfig(settings.OpenAISettings.Token)
@ -104,9 +113,8 @@ func MakeChatCompletionRequest(c *gin.Context) {
return return
} }
// Send SSE to client
message := fmt.Sprintf("%s", response.Choices[0].Delta.Content) message := fmt.Sprintf("%s", response.Choices[0].Delta.Content)
fmt.Printf("%s", response.Choices[0].Delta.Content) fmt.Printf("%s", message)
_ = os.Stdout.Sync() _ = os.Stdout.Sync()
msgChan <- message msgChan <- message