weichen
7710be2db4
Why: long multi-turn executions grow the JSONL transcript until they hit provider context limits (parity-matrix #11), and a crashed append left a partial line that made the whole session unloadable (parity-matrix #12 resume semantics). Evidence: go vet ./... clean; go test ./... 13 packages ok. Session cases: crash-residue tail skipped while mid-file corruption still errors, Stats line/byte counts, meta sidecar roundtrip (missing meta nil), Compact rewriting system+summary+keepLastN atomically with no tmp residue, no-op on short transcripts, summarizer error leaving the file untouched. Adapter case: threshold-triggered compaction runs before the loop and the summary marker reaches both the transcript and the model's context. Impact: session gains Stats/SaveMeta/ReadMeta/Compact; native.Options gains Compact *CompactConfig (default summarizer is one tool-less stream call; compaction failure logs and continues - optimization, never a silent failure); session meta sidecar records model/scope/ exec id per execution. Refs: docs/parity-matrix.md #11/#12 (Batch3 scope).
70 lines
2.1 KiB
Go
70 lines
2.1 KiB
Go
package native
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
"testing"
|
|
|
|
"agent-native/exec"
|
|
"agent-native/runtime/llm"
|
|
)
|
|
|
|
func TestBackendCompactBeforeRun(t *testing.T) {
|
|
dir := t.TempDir()
|
|
// 预置一份超阈值的长转录（系统消息 + 12 条历史）。
|
|
transcript := filepath.Join(dir, ".agentx", "transcript.jsonl")
|
|
if err := os.MkdirAll(filepath.Dir(transcript), 0o755); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
var b strings.Builder
|
|
b.WriteString(`{"role":"system","content":"sys"}` + "\n")
|
|
for i := 0; i < 12; i++ {
|
|
b.WriteString(`{"role":"user","content":"` + strings.Repeat("x", 400) + `"}` + "\n")
|
|
}
|
|
if err := os.WriteFile(transcript, []byte(b.String()), 0o644); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
var sawSummary bool
|
|
// 注入式捕获：验证 Run 收到的消息里有摘要标记。
|
|
bk2 := New(Options{
|
|
Model: "fake",
|
|
Stream: captureFirstStream(&sawSummary),
|
|
Compact: &CompactConfig{ThresholdChars: 1000, KeepLastN: 3, Summarizer: func(ctx context.Context, messages []llm.Message) (string, error) {
|
|
return "预置历史摘要", nil
|
|
}},
|
|
})
|
|
res, err := bk2.Execute(context.Background(), exec.Request{
|
|
WorkDir: dir, Prompt: "继续", Targets: []string{"output.md"},
|
|
ToolProfile: exec.ToolProfileReasoning,
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if res.Success && !sawSummary {
|
|
t.Fatal("compaction should have run before the loop")
|
|
}
|
|
data, _ := os.ReadFile(transcript)
|
|
if !strings.Contains(string(data), "[conversation summary]") {
|
|
t.Fatalf("transcript should contain the summary marker, got:\n%.200s", data)
|
|
}
|
|
}
|
|
|
|
func captureFirstStream(saw *bool) llm.StreamFunc {
|
|
return func(ctx context.Context, model string, messages []llm.Message, decls []llm.ToolDecl, opts llm.StreamOptions) <-chan llm.Event {
|
|
for _, m := range messages {
|
|
if strings.Contains(m.Content, "[conversation summary]") {
|
|
*saw = true
|
|
}
|
|
}
|
|
ch := make(chan llm.Event, 4)
|
|
go func() {
|
|
defer close(ch)
|
|
ch <- llm.Event{Kind: llm.KindMessageEnd, Message: llm.Message{Role: llm.RoleAssistant, Content: "压缩后续写完成", StopReason: llm.StopEnd}}
|
|
}()
|
|
return ch
|
|
}
|
|
}
|