| 1 | package agent |
| 2 | |
| 3 | import ( |
| 4 | "context" |
| 5 | "reflect" |
| 6 | "strings" |
| 7 | "testing" |
| 8 | |
| 9 | "reasonix/internal/event" |
| 10 | "reasonix/internal/provider" |
| 11 | "reasonix/internal/tool" |
| 12 | ) |
| 13 | |
| 14 | func TestNormalizeModelRequestMessagesDoesNotMutateInput(t *testing.T) { |
| 15 | const createdAt = int64(99) |
| 16 | const storedContent = "task\n<execution-policy>local policy</execution-policy>" |
| 17 | msgs := []provider.Message{ |
| 18 | {Role: provider.RoleSystem, Content: "system"}, |
| 19 | {Role: provider.RoleUser, Content: storedContent, CreatedAt: createdAt}, |
| 20 | } |
| 21 | wantStored := append([]provider.Message(nil), msgs...) |
| 22 | a := New(nil, tool.NewRegistry(), NewSession("system"), Options{}, event.Discard) |
| 23 | |
| 24 | got := a.normalizeModelRequestMessages(msgs) |
| 25 | if got[1].CreatedAt != 0 || got[1].Content != "task" { |
| 26 | t.Fatalf("provider message = %+v, want stripped local metadata", got[1]) |
| 27 | } |
| 28 | if !reflect.DeepEqual(msgs, wantStored) { |
| 29 | t.Fatalf("normalization mutated its input: got=%+v want=%+v", msgs, wantStored) |
| 30 | } |
| 31 | } |
| 32 | |
| 33 | func TestPressurePruneNeverPromotesRawToolContent(t *testing.T) { |
| 34 | const rawOnlySentinel = "RAW-ONLY-PRESSURE-SENTINEL" |
| 35 | bounded := strings.Repeat("b", toolPruneThresholdRunes+1) |
| 36 | raw := strings.Repeat("r", 100) + rawOnlySentinel + strings.Repeat("r", toolPruneThresholdRunes+1) |
| 37 | sess := &Session{Messages: []provider.Message{ |
| 38 | {Role: provider.RoleSystem, Content: "system"}, |
| 39 | {Role: provider.RoleAssistant, ToolCalls: []provider.ToolCall{{ID: "call-1", Name: "read_file", Arguments: `{}`}}}, |
| 40 | {Role: provider.RoleTool, ToolCallID: "call-1", Name: "read_file", Content: bounded, RawContent: raw}, |
| 41 | }} |
| 42 | a := New(nil, tool.NewRegistry(), sess, Options{}, event.Discard) |
| 43 | |
| 44 | a.sess.compactionRunMu.Lock() |
| 45 | advanced, err := a.pruneToolResultsToProjectionLocked(context.Background(), CompactionTriggerPressure) |
| 46 | a.sess.compactionRunMu.Unlock() |
| 47 | if err != nil || !advanced { |
| 48 | t.Fatalf("pressure prune advanced=%v err=%v", advanced, err) |
| 49 | } |
| 50 | visible := modelInputMessages(a.modelVisibleMessages()) |
| 51 | if strings.Contains(joinContents(visible), rawOnlySentinel) { |
| 52 | t.Fatal("pressure projection promoted local RawContent") |
| 53 | } |
| 54 | if got := visible[len(visible)-1]; !strings.Contains(got.Content, toolPruneMarker) || got.RawContent != "" { |
| 55 | t.Fatalf("provider tool result = %+v, want pruned bounded Content without RawContent", got) |
| 56 | } |
| 57 | stored := sess.Snapshot()[2] |
| 58 | if stored.Content != bounded || stored.RawContent != raw { |
| 59 | t.Fatal("pressure prune modified canonical tool content") |
| 60 | } |
| 61 | } |
| 62 |