Commit 323f4644 authored by Steven杜宇's avatar Steven杜宇

// AI

parent 93a9a810
......@@ -39,7 +39,7 @@ class YHAIHistoryMessage: SmartCodable {
body.type = self.type
body.cardsInfo = self.cardsInfo
msg.body = body
msg.setCompleted()
msg.setDone()
msg.updateBodyToData()
return msg
}
......
......@@ -49,15 +49,14 @@ class YHAIChatMessage: CustomStringConvertible {
}
func isTextCompleted() -> Bool {
if self.body.contentType == YHAIMessageType.text.rawValue && self.body.status == "done" && self.body.type == "answer" {
if self.body.contentType == YHAIMessageType.text.rawValue && self.body.status == "completed" && self.body.type == "answer" {
return true
}
return false
}
func setCompleted() {
self.body.status = "completed"
func setDone() {
self.isDone = true
}
func isTextMessage() -> Bool {
......@@ -76,7 +75,7 @@ class YHAIChatMessage: CustomStringConvertible {
let body = YHAIMessageBody()
body.contentType = 1
body.contentText = text
question.setCompleted()
question.setDone()
question.body = body
question.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
question.updateBodyToData()
......@@ -88,7 +87,7 @@ class YHAIChatMessage: CustomStringConvertible {
message.isSelf = false
let body = YHAIMessageBody()
body.contentType = YHAIMessageType.thinking.rawValue
message.setCompleted()
message.setDone()
message.body = body
message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
message.updateBodyToData()
......@@ -102,7 +101,7 @@ class YHAIChatMessage: CustomStringConvertible {
let body = YHAIMessageBody()
body.contentType = YHAIMessageType.text.rawValue
body.contentText = text
message.setCompleted()
message.setDone()
message.body = body
message.updateBodyToData()
return message
......@@ -113,7 +112,7 @@ class YHAIChatMessage: CustomStringConvertible {
message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
message.isSelf = false
let body = YHAIMessageBody()
message.setCompleted()
message.setDone()
body.contentType = YHAIMessageType.picture.rawValue
let imgInfo = YHAIImageInfo()
imgInfo.imageUrl = url
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment