mirror of
https://github.com/esengine/DeepSeek-Reasonix.git
synced 2026-11-07 02:36:01 -10:00
72 lines
2.8 KiB
Go
72 lines
2.8 KiB
Go
package agent
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"reasonix/internal/contract/provider"
|
|
)
|
|
|
|
// foldFixture is a transcript shaped like a real tool-driven run: fat tool-call
|
|
// arguments, reasoning on the assistant turns that carry them, and the decision
|
|
// receipts an approval-bearing session accumulates — the records that decide
|
|
// whether provider.ModelMessages can hand back its input or has to copy.
|
|
func foldFixture(turns int, cjk, receipts bool) []provider.Message {
|
|
body := strings.Repeat("a plain english sentence about the work. ", 30)
|
|
if cjk {
|
|
body = strings.Repeat("这是一段关于当前工作的中文说明，包含具体的路径与结论。", 30)
|
|
}
|
|
msgs := []provider.Message{{Role: provider.RoleSystem, Content: strings.Repeat("standing instruction. ", 200)}}
|
|
for i := range turns {
|
|
id := fmt.Sprintf("call_%d", i)
|
|
asst := provider.Message{
|
|
Role: provider.RoleAssistant,
|
|
Content: body,
|
|
ReasoningContent: strings.Repeat("weighing the options here. ", 60),
|
|
ToolCalls: []provider.ToolCall{{
|
|
ID: id, Name: "edit_file",
|
|
Arguments: `{"path":"internal/x.go","old":"` + body + `","new":"` + body + `"}`,
|
|
}},
|
|
}
|
|
if receipts && i%5 == 0 {
|
|
asst.DecisionReceipts = []*provider.DecisionReceipt{{ID: "r", Kind: "approval", Outcome: "approved"}}
|
|
}
|
|
msgs = append(msgs,
|
|
provider.Message{Role: provider.RoleUser, Content: "continue with the next step"},
|
|
asst,
|
|
provider.Message{Role: provider.RoleTool, ToolCallID: id, Name: "edit_file", Content: body},
|
|
)
|
|
}
|
|
return msgs
|
|
}
|
|
|
|
// remeasuredTailStart carries one measurement forward instead of re-measuring
|
|
// each candidate tail whole. That is only sound while a request's shape is the
|
|
// sum of its messages' shapes — for every suffix, on a transcript carrying the
|
|
// records ModelMessages strips, under every replay policy a provider declares.
|
|
func TestProjectedMessageShapesSumToTheWholeSliceMeasurement(t *testing.T) {
|
|
msgs := append(foldFixture(40, true, true),
|
|
provider.Message{Role: provider.RoleAssistant, Content: "local", LocalOnly: true},
|
|
provider.Message{Role: provider.RoleAssistant, Content: "shown", ProviderContent: "sent to the provider instead"},
|
|
provider.Message{Role: provider.RoleUser, Content: "raw", RawContent: "unprojected"},
|
|
)
|
|
for _, policy := range []provider.SharedWindowInputPolicy{
|
|
{},
|
|
{ReplaysOrdinaryReasoning: true},
|
|
{ReplaysResponsesItems: true},
|
|
} {
|
|
for start := range len(msgs) + 1 {
|
|
want := requestCalibrationShapeWithPolicy(
|
|
provider.Request{Messages: provider.ModelMessages(msgs[start:])}, policy)
|
|
var got requestCalibrationShape
|
|
for _, m := range msgs[start:] {
|
|
got = got.plus(projectedMessageCalibrationShape(m, policy))
|
|
}
|
|
if got != want {
|
|
t.Fatalf("policy %+v start %d: summed %+v, whole-slice %+v", policy, start, got, want)
|
|
}
|
|
}
|
|
}
|
|
}
|