casbin
/
casibase
1
0
Fork 1
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix GetMessageAnswer() errors

This commit is contained in:
Yang Luo 2023-09-10 00:39:50 +08:00
parent 2c8f6ac553
commit 405eceb738
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
controllers/message.go
View File

@ -77,7 +77,7 @@ func (c *ApiController) GetMessageAnswer() {
message, err := object.GetMessage(id) message, err := object.GetMessage(id)
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseErrorStream(err.Error())
return return
} }
@ -94,7 +94,7 @@ func (c *ApiController) GetMessageAnswer() {
chatId := util.GetIdFromOwnerAndName("admin", message.Chat) chatId := util.GetIdFromOwnerAndName("admin", message.Chat)
chat, err := object.GetChat(chatId) chat, err := object.GetChat(chatId)
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseErrorStream(err.Error())
return return
} }
@ -116,13 +116,13 @@ func (c *ApiController) GetMessageAnswer() {
modelProviderObj, err := getModelProviderFromContext(chat.Owner, chat.User2) modelProviderObj, err := getModelProviderFromContext(chat.Owner, chat.User2)
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseErrorStream(err.Error())
return return
} }
embeddingProviderObj, err := getEmbeddingProviderFromContext(chat.Owner, chat.User2) embeddingProviderObj, err := getEmbeddingProviderFromContext(chat.Owner, chat.User2)
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseErrorStream(err.Error())
return return
} }
@ -156,7 +156,7 @@ func (c *ApiController) GetMessageAnswer() {
event := fmt.Sprintf("event: end\ndata: %s\n\n", "end") event := fmt.Sprintf("event: end\ndata: %s\n\n", "end")
_, err = c.Ctx.ResponseWriter.Write([]byte(event)) _, err = c.Ctx.ResponseWriter.Write([]byte(event))
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseErrorStream(err.Error())
return return
} }
@ -165,7 +165,7 @@ func (c *ApiController) GetMessageAnswer() {
message.Text = answer message.Text = answer
_, err = object.UpdateMessage(message.GetId(), message) _, err = object.UpdateMessage(message.GetId(), message)
if err != nil { if err != nil {
c.ResponseError(err.Error()) c.ResponseErrorStream(err.Error())
return return
} }
} }