Name and order the view code

[onechat] / view_layout.go

diff --git a/layout.go b/view_layout.go
similarity index 74%
rename from layout.go
rename to view_layout.go
index c5e9ac3a5a812cc74fae830970408bcae9f56faf..0d4e7560581a5d554932bf46280f344aa5b0d75b 100644 (file)
--- a/layout.go
+++ b/view_layout.go
@@ -10,36 +10,14 @@ import (
        "strings"
 )
 
-// layoutFileName is the file of the view layout in the directory of an agent.
-const layoutFileName = "layout.json"
-
-// viewLayout splits the view of a turn request into user messages that only
-// grow between batches. The view goes out as the same messages as in the
-// request before, while their lines give the same text, and the new lines
-// follow as one new message. So the end of the view of each earlier turn stays
-// a message end. On the ChatGPT backend, the first request of a turn with this
-// layout read the view of the turn before from the prompt cache in each of 10
-// turns, and with the view in one message it missed once in 5.
-//
-// The first message starts with "<chat>\n". The closing tag goes into the
-// state message after the view, so that no view message changes when lines
-// come. The layout keeps the line count and a hash of the text of each message
-// in layout.json, so it outlives a restart.
-type viewLayout struct {
-       path     string
-       Messages []layoutMessage `json:"messages"`
-}
-
-// layoutMessage is one view message: its number of lines and the hash of its
-// text.
-type layoutMessage struct {
-       Lines int    `json:"lines"`
-       Hash  string `json:"hash"`
-}
-
-// loadLayout reads the layout at path. A missing file gives an empty layout.
-func loadLayout(path string) (*viewLayout, error) {
-       layout := &viewLayout{
+// viewLayoutFileName is the file of the view layout in the directory of an
+// agent.
+const viewLayoutFileName = "layout.json"
+
+// loadViewLayout reads the layout at path. A missing file gives an empty
+// layout.
+func loadViewLayout(path string) (*ViewLayout, error) {
+       layout := &ViewLayout{
                path: path,
        }
 
@@ -60,12 +38,29 @@ func loadLayout(path string) (*viewLayout, error) {
        return layout, nil
 }
 
+// ViewLayout splits the view of a turn request into user messages that only
+// grow between batches. The view goes out as the same messages as in the
+// request before, while their lines give the same text, and the new lines
+// follow as one new message. So the end of the view of each earlier turn stays
+// a message end. On the ChatGPT backend, the first request of a turn with this
+// layout read the view of the turn before from the prompt cache in each of 10
+// turns, and with the view in one message it missed once in 5.
+//
+// The first message starts with "<chat>\n". The closing tag goes into the
+// state message after the view, so that no view message changes when lines
+// come. The layout keeps the line count and a hash of the text of each message
+// in layout.json, so it outlives a restart.
+type ViewLayout struct {
+       path     string
+       Messages []ViewLayoutMessage `json:"messages"`
+}
+
 // Split returns the view messages for lines and saves the layout. It keeps the
 // earlier messages up to the first one whose text differs, and puts the rest
 // of the lines into one new message.
-func (l *viewLayout) Split(lines []string) ([]string, error) {
+func (l *ViewLayout) Split(lines []string) ([]string, error) {
        var texts []string
-       var kept []layoutMessage
+       var kept []ViewLayoutMessage
        start := 0
        for _, message := range l.Messages {
                if start+message.Lines > len(lines) {
@@ -73,7 +68,7 @@ func (l *viewLayout) Split(lines []string) ([]string, error) {
                }
 
                text := viewMessage(len(texts) == 0, lines[start:start+message.Lines])
-               if textHash(text) != message.Hash {
+               if viewTextHash(text) != message.Hash {
                        break
                }
 
@@ -85,9 +80,9 @@ func (l *viewLayout) Split(lines []string) ([]string, error) {
        if start < len(lines) || len(texts) == 0 {
                text := viewMessage(len(texts) == 0, lines[start:])
                texts = append(texts, text)
-               kept = append(kept, layoutMessage{
+               kept = append(kept, ViewLayoutMessage{
                        Lines: len(lines) - start,
-                       Hash:  textHash(text),
+                       Hash:  viewTextHash(text),
                })
        }
 
@@ -105,6 +100,13 @@ func (l *viewLayout) Split(lines []string) ([]string, error) {
        return texts, nil
 }
 
+// ViewLayoutMessage is one view message: its number of lines and the hash of
+// its text.
+type ViewLayoutMessage struct {
+       Lines int    `json:"lines"`
+       Hash  string `json:"hash"`
+}
+
 // viewMessage returns the text of a view message with lines. The first one
 // opens the chat.
 func viewMessage(first bool, lines []string) string {
@@ -116,8 +118,8 @@ func viewMessage(first bool, lines []string) string {
        return text
 }
 
-// textHash returns a short hash of text.
-func textHash(text string) string {
+// viewTextHash returns a short hash of text.
+func viewTextHash(text string) string {
        sum := sha256.Sum256([]byte(text))
        return hex.EncodeToString(sum[:16])
 }