Fix chat in GetAnswer()

JiawenZou · Mar 17, 2024 · 9953f8f · 9953f8f
1 parent c25466e
commit 9953f8f
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/controllers/message_answer.go b/controllers/message_answer.go
@@ -299,6 +299,8 @@ func (c *ApiController) GetAnswer() {
 		Text:         question,
 	}
 
+	questionMessage.Currency = modelResult.Currency
+
 	_, err = object.AddMessage(questionMessage)
 	if err != nil {
 		c.ResponseError(err.Error())
@@ -327,5 +329,20 @@ func (c *ApiController) GetAnswer() {
 		return
 	}
 
+	chat.TokenCount += answerMessage.TokenCount
+	chat.Price += answerMessage.Price
+	if chat.Currency == "" {
+		chat.Currency = answerMessage.Currency
+	}
+
+	chat.UpdatedTime = util.GetCurrentTime()
+	chat.MessageCount += 2
+
+	_, err = object.UpdateChat(chat.GetId(), chat)
+	if err != nil {
+		c.ResponseOk(err.Error())
+		return
+	}
+
 	c.ResponseOk(answer)
 }