Commit 4b4152ad authored by Steven杜宇's avatar Steven杜宇

// AI

parent 47076ef5
...@@ -221,11 +221,11 @@ class YHAIRequestManager: NSObject { ...@@ -221,11 +221,11 @@ class YHAIRequestManager: NSObject {
if sessionDone { if sessionDone {
print("SESSION DONE") print("SESSION DONE")
} }
if receiveMessage.isFinishd() { if receiveMessage.isCompleted() {
// 一段话结束需要重新生成uuid 来 // 一段话结束需要重新生成uuid 来
self.uuid = UUID().uuidString + NSDate().timeIntervalSince1970.description self.uuid = UUID().uuidString + NSDate().timeIntervalSince1970.description
} }
if !receiveMessage.body.isStart() && !receiveMessage.isFinishd() { if !receiveMessage.body.isStart() && !receiveMessage.isCompleted() {
completion?(receiveMessage, sessionDone) completion?(receiveMessage, sessionDone)
} }
} }
......
...@@ -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.setFinished() msg.setCompleted()
msg.updateBodyToData() msg.updateBodyToData()
return msg return msg
} }
......
...@@ -30,27 +30,26 @@ enum YHAIMessageType: Int { ...@@ -30,27 +30,26 @@ enum YHAIMessageType: Int {
class YHAIChatMessage: CustomStringConvertible { class YHAIChatMessage: CustomStringConvertible {
let completeText = "conversation.chat.completed"
var id: String = "" var id: String = ""
var event: String = "" var event: String = ""
var data: String = "" var data: String = ""
var dataDict: [String : Any] = [:] var dataDict: [String : Any] = [:]
var messageId: String = "" var messageId: String = ""
var isSelf: Bool = false var isSelf: Bool = false
var isDone: Bool = false var isDone: Bool = false
var body = YHAIMessageBody() var body = YHAIMessageBody()
func isFinishd() -> Bool { func isCompleted() -> Bool {
if event.contains("conversation.message.completed") { if event.contains(completeText) {
return true return true
} }
return false return false
} }
func setFinished() { func setCompleted() {
event = "conversation.message.completed" event = self.completeText
} }
func isTextMessage() -> Bool { func isTextMessage() -> Bool {
...@@ -69,7 +68,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -69,7 +68,7 @@ class YHAIChatMessage: CustomStringConvertible {
let body = YHAIMessageBody() let body = YHAIMessageBody()
body.contentType = 1 body.contentType = 1
body.contentText = text body.contentText = text
question.setFinished() question.setCompleted()
question.body = body question.body = body
question.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description question.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
question.updateBodyToData() question.updateBodyToData()
...@@ -81,7 +80,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -81,7 +80,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.setFinished() message.setCompleted()
message.body = body message.body = body
message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description message.messageId = UUID().uuidString + NSDate().timeIntervalSince1970.description
message.updateBodyToData() message.updateBodyToData()
...@@ -95,7 +94,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -95,7 +94,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.setFinished() message.setCompleted()
message.body = body message.body = body
message.updateBodyToData() message.updateBodyToData()
return message return message
...@@ -106,7 +105,7 @@ class YHAIChatMessage: CustomStringConvertible { ...@@ -106,7 +105,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.setFinished() message.setCompleted()
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