X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/f5c601f1801aaa276f8c335fcb51c373edc0b50c..9301f77a5709fe7daea3a71f8dd072c6cd7df4fe:/compactor.go diff --git a/compactor.go b/compactor.go index e101428..64ff1fd 100644 --- a/compactor.go +++ b/compactor.go @@ -56,12 +56,12 @@ func newCompactor(client *ModelClient, instructions func() (string, error), mode func (c *Compactor) Summarize(ctx context.Context, task Task) (string, error) { text, err := compactorTask(task) if err != nil { - return "", err + return "", fmt.Errorf("prepare compaction task: %w", err) } instructions, err := c.instructions() if err != nil { - return "", err + return "", fmt.Errorf("read compaction instructions: %w", err) } source := task.Context + "\n" + text @@ -110,7 +110,7 @@ func (c *Compactor) Summarize(ctx context.Context, task Task) (string, error) { } if len(lines) == 0 { - return "", err + return "", fmt.Errorf("summarize node: %w", err) } return compactorShortest(lines), nil @@ -125,6 +125,7 @@ func compactorTask(task Task) (string, error) { %s `, task.Index, nodeBytes, compactorRuler, task.Input[0]) + if len(task.Images) > 0 { text += "\nKeep each [image FILE] name and describe its kind, visible text, amounts, names, and context." }