X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/c3c7bc90eb73e83f48fbeac6be149ade95a726c0..59948b44c0fe2165b90e3aa0d18646f6c3d74af3:/memory.go

diff --git a/memory.go b/memory.go
index a240b6c..1b734c0 100644
--- a/memory.go
+++ b/memory.go
@@ -485,9 +485,9 @@ func (m *Memory) dispatch() {
 }
 
 // addMessage builds the leaf of message when it needs no model call, or else
-// adds the leaf to the queue. Then it appends the leaf to the views. A crash can
-// stop an append after the leaf build and before the view save, so a leaf that
-// is built already stays as it is, and restoreQueue settles its parent.
+// adds the leaf to the queue. Then it appends the leaf to the views. A crash
+// can stop an append after the leaf build and before the view save, so a leaf
+// that is built already stays as it is, and restoreQueue settles its parent.
 func (m *Memory) addMessage(message Message) error {
 	built, err := m.tree.Built(0, message.ID)
 	if err != nil {
@@ -607,6 +607,7 @@ func (m *Memory) task(key NodeKey) (Task, error) {
 		}
 
 		task.Input = []string{leafSource(message)}
+		task.Images = imageFiles(message.Text)
 
 		return task, nil
 	}
@@ -781,12 +782,14 @@ func (m *Memory) buildBuiltByModel(key NodeKey) (bool, error) {
 
 // Task is the build of the node (Level, Index). Context is the compaction view
 // for the build. Input holds the source "kind: text" of a leaf, or the two
-// child lines of a merge with their newlines as spaces.
+// child lines of a merge with their newlines as spaces. Images holds the leaf
+// image filenames and is empty for merges.
 type Task struct {
 	Level   int
 	Index   int
 	Context string
 	Input   []string
+	Images  []string
 }
 
 // MemoryViewFile is the content of view.json.