Run the turns of an agent in a loop

[onechat] / memory.go

diff --git a/memory.go b/memory.go
index ca8277c629889037c17c4ef221bcf94c049f0e8c..d7fff4ee1d6e6fef5a7c84919dd8605ee5fd6beb 100644 (file)
--- a/memory.go
+++ b/memory.go
@@ -543,6 +543,39 @@ func (m *Memory) View() string {
        return m.chat.Render()
 }
 
        return m.chat.Render()
 }
 

+// ViewWithout returns the chat view without the leaf lines of the messages in
+// skip.
+func (m *Memory) ViewWithout(skip map[int]bool) string {
+       m.mutex.Lock()
+       defer m.mutex.Unlock()
+
+       return m.chat.RenderWithout(skip)
+}
+
+// Len returns the number of messages in the log.
+func (m *Memory) Len() int {
+       m.mutex.Lock()
+       defer m.mutex.Unlock()
+
+       return m.log.Len()
+}
+
+// Message reads the message with the given ID.
+func (m *Memory) Message(id int) (Message, error) {
+       m.mutex.Lock()
+       defer m.mutex.Unlock()
+
+       return m.log.Message(id)
+}
+
+// Node reads the node (l, i) if it is built.
+func (m *Memory) Node(level, index int) (Node, bool, error) {
+       m.mutex.Lock()
+       defer m.mutex.Unlock()
+
+       return m.tree.Node(level, index)
+}
+

 // Context returns the compaction view for the build of node (l, i). It ends
 // before the message of a leaf, or after the last message of a merge, and it
 // stops at the first line that is not built.
 // Context returns the compaction view for the build of node (l, i). It ends
 // before the message of a leaf, or after the last message of a merge, and it
 // stops at the first line that is not built.