Show an input prompt while the agent is idle

[onechat] / tree.go

diff --git a/tree.go b/tree.go
index 83ecda3d5a8c4885c100591852ec2c30c46e733e..69558c54ca07197e5e09147092caa7a47127efdf 100644 (file)
--- a/tree.go
+++ b/tree.go
@@ -124,6 +124,7 @@ func (t *Tree) Close() error {
 
        for level, index := range t.levels {
                err = errors.Join(err, index.Close())
 
        for level, index := range t.levels {
                err = errors.Join(err, index.Close())

+

                delete(t.levels, level)
        }
 
                delete(t.levels, level)
        }
 


@@ -178,6 +179,7 @@ func (t *Tree) Node(level, index int) (Node, bool, error) {
        }
 
        var node Node
        }
 
        var node Node

+

        err = json.Unmarshal(line, &node)
        if err != nil {
                return Node{}, false, fmt.Errorf("decode node (%d, %d): %w", level, index, err)
        err = json.Unmarshal(line, &node)
        if err != nil {
                return Node{}, false, fmt.Errorf("decode node (%d, %d): %w", level, index, err)


@@ -212,6 +214,7 @@ func (t *Tree) Keys() ([]NodeKey, error) {
 
 func (t *Tree) open(ctx context.Context, total int) error {
        var err error
 
 func (t *Tree) open(ctx context.Context, total int) error {
        var err error

+

        t.order, err = openRecords(filepath.Join(t.dir, "tree.idx"), treeRecordBytes)
        if err != nil {
                return err
        t.order, err = openRecords(filepath.Join(t.dir, "tree.idx"), treeRecordBytes)
        if err != nil {
                return err


@@ -245,6 +248,7 @@ func (t *Tree) open(ctx context.Context, total int) error {
 
        return t.journal.Scan(ctx, start, func(line []byte, place Position) error {
                var node Node
 
        return t.journal.Scan(ctx, start, func(line []byte, place Position) error {
                var node Node

+

                err := json.Unmarshal(line, &node)
                if err != nil {
                        return fmt.Errorf("decode tree node: %w", err)
                err := json.Unmarshal(line, &node)
                if err != nil {
                        return fmt.Errorf("decode tree node: %w", err)