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

// AI

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