Added code for backend glue
This commit is contained in:
41
backend/internal/glue/linearize.go
Normal file
41
backend/internal/glue/linearize.go
Normal file
@@ -0,0 +1,41 @@
|
||||
package glue
|
||||
|
||||
import (
|
||||
"context"
|
||||
"mind/internal/models"
|
||||
)
|
||||
|
||||
type Linearized struct {
|
||||
Nodes []models.Node `json:"nodes"`
|
||||
Text string `json:"text"`
|
||||
}
|
||||
|
||||
func (g *Glue) LinearizeByBranch(ctx context.Context, convID int64, branchName string) (Linearized, error) {
|
||||
b, err := g.repo.GetBranch(ctx, convID, branchName)
|
||||
if err != nil { return Linearized{}, err }
|
||||
return g.linearizeFromHead(ctx, b.HeadNodeID)
|
||||
}
|
||||
|
||||
func (g *Glue) linearizeFromHead(ctx context.Context, head int64) (Linearized, error) {
|
||||
var seq []models.Node
|
||||
// Walk parents by latest-created parent until none
|
||||
curID := head
|
||||
for {
|
||||
n, err := g.repo.GetNode(ctx, curID)
|
||||
if err != nil { return Linearized{}, err }
|
||||
seq = append(seq, n)
|
||||
p, ok, err := g.repo.LatestParent(ctx, curID)
|
||||
if err != nil { return Linearized{}, err }
|
||||
if !ok { break }
|
||||
curID = p.ID
|
||||
}
|
||||
// Reverse seq to root→head, and stitch text
|
||||
for i, j := 0, len(seq)-1; i < j; i, j = i+1, j-1 { seq[i], seq[j] = seq[j], seq[i] }
|
||||
var txt string
|
||||
for _, n := range seq {
|
||||
role := "[user]"
|
||||
if n.AuthorKind == "assistant" { role = "[assistant]" }
|
||||
txt += role + "\n" + n.Content + "\n\n"
|
||||
}
|
||||
return Linearized{Nodes: seq, Text: txt}, nil
|
||||
}
|
||||
Reference in New Issue
Block a user