DBT/clawgo
1
0
Fork 0
mirror of https://github.com/YspCoder/clawgo.git synced 2026-04-14 22:09:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

Slim subagent runtime surface and remove legacy interfaces

Browse Source
This commit is contained in:
LPF
2026-03-17 13:41:12 +08:00
parent 341e578c9f
commit 0674d85ae1
76 changed files with 778 additions and 8782 deletions
Download Patch File Download Diff File Expand all files Collapse all files

105
pkg/agent/session_planner_test.go
View File

@@ -1,13 +1,8 @@
package agent
import (
"fmt"
"os"
"path/filepath"
"strings"
"testing"
"github.com/YspCoder/clawgo/pkg/ekg"
)
func TestSummarizePlannedTaskProgressBodyPreservesUsefulLines(t *testing.T) {
@@ -27,75 +22,6 @@ func TestSummarizePlannedTaskProgressBodyPreservesUsefulLines(t *testing.T) {
}
}
func TestEKGHintForTaskRequiresStrongMatchAndStaysCompact(t *testing.T) {
t.Parallel()
workspace := t.TempDir()
memoryDir := filepath.Join(workspace, "memory")
if err := os.MkdirAll(memoryDir, 0o755); err != nil {
t.Fatalf("mkdir memory: %v", err)
}
logText := "open /srv/app/config.yaml: permission denied after deploy 42"
taskAudit := []string{
fmt.Sprintf(`{"task_id":"task-1","status":"error","source":"planner","channel":"chat","input_preview":"check nginx logs quickly","log":"%s"}`, logText),
fmt.Sprintf(`{"task_id":"task-2","status":"error","source":"planner","channel":"chat","input_preview":"deploy config service restart on cluster","log":"%s"}`, logText),
}
if err := os.WriteFile(filepath.Join(memoryDir, "task-audit.jsonl"), []byte(strings.Join(taskAudit, "\n")+"\n"), 0o644); err != nil {
t.Fatalf("write task audit: %v", err)
}
errSig := ekg.NormalizeErrorSignature(logText)
ekgEvents := []string{
fmt.Sprintf(`{"task_id":"task-2","status":"error","errsig":"%s","log":"%s"}`, errSig, logText),
fmt.Sprintf(`{"task_id":"task-2","status":"error","errsig":"%s","log":"%s"}`, errSig, logText),
fmt.Sprintf(`{"task_id":"task-2","status":"error","errsig":"%s","log":"%s"}`, errSig, logText),
}
if err := os.WriteFile(filepath.Join(memoryDir, "ekg-events.jsonl"), []byte(strings.Join(ekgEvents, "\n")+"\n"), 0o644); err != nil {
t.Fatalf("write ekg events: %v", err)
}
al := &AgentLoop{workspace: workspace, ekg: ekg.New(workspace)}
hint := al.ekgHintForTask(plannedTask{Content: "deploy config service restart after rollout"})
if hint == "" {
t.Fatalf("expected compact ekg hint")
}
if !strings.Contains(hint, "repeat_errsig=") || !strings.Contains(hint, "backoff=300s") {
t.Fatalf("expected compact fields, got: %s", hint)
}
if strings.Contains(strings.ToLower(hint), "last error") || strings.Contains(hint, logText) {
t.Fatalf("expected raw error log to be omitted, got: %s", hint)
}
}
func TestEKGHintForTaskSkipsWeakTaskMatch(t *testing.T) {
t.Parallel()
workspace := t.TempDir()
memoryDir := filepath.Join(workspace, "memory")
if err := os.MkdirAll(memoryDir, 0o755); err != nil {
t.Fatalf("mkdir memory: %v", err)
}
logText := "dial tcp 10.0.0.8:443: i/o timeout"
taskAudit := `{"task_id":"task-3","status":"error","source":"planner","channel":"chat","input_preview":"investigate cache timeout","log":"dial tcp 10.0.0.8:443: i/o timeout"}`
if err := os.WriteFile(filepath.Join(memoryDir, "task-audit.jsonl"), []byte(taskAudit+"\n"), 0o644); err != nil {
t.Fatalf("write task audit: %v", err)
}
errSig := ekg.NormalizeErrorSignature(logText)
ekgEvents := []string{
fmt.Sprintf(`{"task_id":"task-3","status":"error","errsig":"%s","log":"%s"}`, errSig, logText),
fmt.Sprintf(`{"task_id":"task-3","status":"error","errsig":"%s","log":"%s"}`, errSig, logText),
fmt.Sprintf(`{"task_id":"task-3","status":"error","errsig":"%s","log":"%s"}`, errSig, logText),
}
if err := os.WriteFile(filepath.Join(memoryDir, "ekg-events.jsonl"), []byte(strings.Join(ekgEvents, "\n")+"\n"), 0o644); err != nil {
t.Fatalf("write ekg events: %v", err)
}
al := &AgentLoop{workspace: workspace, ekg: ekg.New(workspace)}
hint := al.ekgHintForTask(plannedTask{Content: "cache rebuild"})
if hint != "" {
t.Fatalf("expected weak match to skip ekg hint, got: %s", hint)
}
}
func TestCompactMemoryHintDropsVerboseScaffolding(t *testing.T) {
t.Parallel()
@@ -116,29 +42,28 @@ func TestDedupeTaskPromptHintsDropsRepeatedContext(t *testing.T) {
t.Parallel()
seen := map[string]struct{}{}
first := taskPromptHints{ekg: "repeat_errsig=x; backoff=300s", memory: "src=memory/a.md#L1-L2 | restart service"}
second := taskPromptHints{ekg: "repeat_errsig=x; backoff=300s", memory: "src=memory/a.md#L1-L2 | restart service"}
first := taskPromptHints{memory: "src=memory/a.md#L1-L2 | restart service"}
second := taskPromptHints{memory: "src=memory/a.md#L1-L2 | restart service"}
dedupeTaskPromptHints(&first, seen)
dedupeTaskPromptHints(&second, seen)
if first.ekg == "" || first.memory == "" {
if first.memory == "" {
t.Fatalf("expected first hint set to remain: %+v", first)
}
if second.ekg != "" || second.memory != "" {
if second.memory != "" {
t.Fatalf("expected repeated hints removed: %+v", second)
}
}
func TestRenderTaskPromptWithHintsKeepsCompactShape(t *testing.T) {
func TestBuildPlannedTaskPromptKeepsCompactShape(t *testing.T) {
t.Parallel()
got := renderTaskPromptWithHints("deploy config service", taskPromptHints{
ekg: "repeat_errsig=perm; backoff=300s",
got := buildPlannedTaskPrompt("deploy config service", taskPromptHints{
memory: "src=memory/x.md#L1-L2 | restart after change",
})
if !strings.Contains(got, "Task Context:\nEKG: repeat_errsig=perm; backoff=300s\nMemory: src=memory/x.md#L1-L2 | restart after change\nTask:\ndeploy config service") {
t.Fatalf("unexpected prompt shape: %s", got)
if !strings.Contains(got.content, "Task Context:\nMemory: src=memory/x.md#L1-L2 | restart after change\nTask:\ndeploy config service") {
t.Fatalf("unexpected prompt shape: %+v", got)
}
}
@@ -146,29 +71,24 @@ func TestBuildPlannedTaskPromptTracksExtraChars(t *testing.T) {
t.Parallel()
prompt := buildPlannedTaskPrompt("deploy config service", taskPromptHints{
ekg: "repeat_errsig=perm; backoff=300s",
memory: "src=memory/x.md#L1-L2 | restart after change",
})
if prompt.content == "" {
t.Fatalf("expected prompt content")
}
if prompt.extraChars <= 0 || prompt.ekgChars <= 0 || prompt.memoryChars <= 0 {
if prompt.extraChars <= 0 || prompt.memoryChars <= 0 {
t.Fatalf("expected prompt stats populated: %+v", prompt)
}
}
func TestApplyPromptBudgetPrefersEKGOverMemory(t *testing.T) {
func TestApplyPromptBudgetDropsMemoryWhenOverBudget(t *testing.T) {
t.Parallel()
hints := taskPromptHints{
ekg: "repeat_errsig=perm; backoff=300s",
memory: "src=memory/x.md#L1-L2 | restart after change",
}
remaining := estimateHintChars(hints) - len(hints.memory)
remaining := estimateHintChars(hints) - 1
applyPromptBudget(&hints, &remaining)
if hints.ekg == "" {
t.Fatalf("expected ekg retained")
}
if hints.memory != "" {
t.Fatalf("expected memory dropped under budget pressure: %+v", hints)
}
@@ -178,12 +98,11 @@ func TestApplyPromptBudgetDropsAllWhenBudgetExhausted(t *testing.T) {
t.Parallel()
hints := taskPromptHints{
ekg: "repeat_errsig=perm; backoff=300s",
memory: "src=memory/x.md#L1-L2 | restart after change",
}
remaining := 8
applyPromptBudget(&hints, &remaining)
if hints.ekg != "" || hints.memory != "" {
if hints.memory != "" {
t.Fatalf("expected all hints removed under tiny budget: %+v", hints)
}
}