fix: ChatGPT messages history for stream

This commit is contained in:
0xJacky 2023-12-15 21:52:41 +08:00
parent 2649b710bb
commit 6c8bfd267a
No known key found for this signature in database
GPG key ID: B6E4A6E4A561BAF0
3 changed files with 24 additions and 19 deletions

View file

@ -99,6 +99,18 @@ func GetStream(c *gin.Context) {
enabled = false
}
g := query.ChatGPTLog
chatgpt, err := g.Where(g.Name.Eq(path)).FirstOrCreate()
if err != nil {
api.ErrHandler(c, err)
return
}
if chatgpt.Content == nil {
chatgpt.Content = make([]openai.ChatCompletionMessage, 0)
}
s := query.Stream
stream, err := s.Where(s.Path.Eq(path)).FirstOrInit()
@ -120,7 +132,7 @@ func GetStream(c *gin.Context) {
Enabled: enabled,
Name: name,
Config: string(origContent),
ChatGPTMessages: stream.ChatGPTMessages,
ChatGPTMessages: chatgpt.Content,
})
return
}
@ -133,8 +145,6 @@ func GetStream(c *gin.Context) {
return
}
c.Set("maybe_error", "nginx_config_syntax_error")
c.JSON(http.StatusOK, Stream{
ModifiedAt: file.ModTime(),
Advanced: stream.Advanced,
@ -142,7 +152,7 @@ func GetStream(c *gin.Context) {
Name: name,
Config: nginxConfig.FmtCode(),
Tokenized: nginxConfig,
ChatGPTMessages: stream.ChatGPTMessages,
ChatGPTMessages: chatgpt.Content,
})
}