Skip to content

Commit b51e7f6

Browse files
committed
mark chat context as cleared on launch
1 parent 16688e2 commit b51e7f6

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

Diff for: LlamaChat/model/chat/ChatModels.swift

+7
Original file line numberDiff line numberDiff line change
@@ -83,6 +83,13 @@ class ChatModel: ObservableObject {
8383
self.source = source
8484
self.messagesModel = messagesModel
8585
messages = messagesModel.loadMessages(from: source)
86+
87+
// By definition we clear the context on each launch because we don't persist session state.
88+
if messages.count > 0 && !(messages.last?.messageType.isClearedContext ?? false) {
89+
let clearedContextMessage = ClearedContextMessage(sendDate: Date())
90+
messages.append(clearedContextMessage)
91+
messagesModel.append(message: clearedContextMessage, in: source)
92+
}
8693
}
8794

8895
func send(message: StaticMessage) {

Diff for: LlamaChat/model/chat/messages/Message.swift

+9
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,15 @@ enum MessageType: Int {
1212
case message = 1
1313
// Implement this as a message just to make this easier
1414
case clearedContext = 2
15+
16+
var isClearedContext: Bool {
17+
switch self {
18+
case .message:
19+
return false
20+
case .clearedContext:
21+
return true
22+
}
23+
}
1524
}
1625

1726
protocol Message {

0 commit comments

Comments
 (0)