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

// AI

parent 93a9a810
...@@ -39,7 +39,7 @@ class YHAIHistoryMessage: SmartCodable { ...@@ -39,7 +39,7 @@ class YHAIHistoryMessage: SmartCodable {
body.type = self.type body.type = self.type
body.cardsInfo = self.cardsInfo body.cardsInfo = self.cardsInfo
msg.body = body msg.body = body
msg.setCompleted() msg.setDone()
msg.updateBodyToData() msg.updateBodyToData()
return msg return msg
} }
......
...@@ -49,15 +49,14 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -49,15 +49,14 @@ class YHAIChatMessage: CustomStringConvertible {
} }
func isTextCompleted() -> Bool { func isTextCompleted() -> Bool {
if self.body.contentType == YHAIMessageType.text.rawValue && self.body.status == "completed" && self.body.type == "answer" {
if self.body.contentType == YHAIMessageType.text.rawValue && self.body.status == "done" && self.body.type == "answer" {
return true return true
} }
return false return false
} }
func setCompleted() { func setDone() {
self.body.status = "completed" self.isDone = true
} }
func isTextMessage() -> Bool { func isTextMessage() -> Bool {
...@@ -76,7 +75,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -76,7 +75,7 @@ class YHAIChatMessage: CustomStringConvertible {
let body = YHAIMessageBody() let body = YHAIMessageBody()
body.contentType = 1 body.contentType = 1
body.contentText = text body.contentText = text
question.setCompleted() question.setDone()
question.body = body question.body = body
question.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description question.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
question.updateBodyToData() question.updateBodyToData()
...@@ -88,7 +87,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -88,7 +87,7 @@ class YHAIChatMessage: CustomStringConvertible {
message.isSelf = false message.isSelf = false
let body = YHAIMessageBody() let body = YHAIMessageBody()
body.contentType = YHAIMessageType.thinking.rawValue body.contentType = YHAIMessageType.thinking.rawValue
message.setCompleted() message.setDone()
message.body = body message.body = body
message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
message.updateBodyToData() message.updateBodyToData()
...@@ -102,7 +101,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -102,7 +101,7 @@ class YHAIChatMessage: CustomStringConvertible {
let body = YHAIMessageBody() let body = YHAIMessageBody()
body.contentType = YHAIMessageType.text.rawValue body.contentType = YHAIMessageType.text.rawValue
body.contentText = text body.contentText = text
message.setCompleted() message.setDone()
message.body = body message.body = body
message.updateBodyToData() message.updateBodyToData()
return message return message
...@@ -113,7 +112,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -113,7 +112,7 @@ class YHAIChatMessage: CustomStringConvertible {
message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
message.isSelf = false message.isSelf = false
let body = YHAIMessageBody() let body = YHAIMessageBody()
message.setCompleted() message.setDone()
body.contentType = YHAIMessageType.picture.rawValue body.contentType = YHAIMessageType.picture.rawValue
let imgInfo = YHAIImageInfo() let imgInfo = YHAIImageInfo()
imgInfo.imageUrl = url 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