File size: 5,065 Bytes
b13e256 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 | import os
AG = "/tmp/k12/src/agent.go"
WS = "/tmp/k12/src/web_stream.go"
MG = "/tmp/k12/src/main.go"
def rd(p): return open(p).read()
def wr(p, s): open(p, "w").write(s)
def enclosing(code, idx):
st = code.rfind("\nfunc ", 0, idx)
en = code.find("\nfunc ", idx + 10)
return code[st:(en if en > 0 else len(code))]
ALC = '''type antiLoopState struct {
\tm map[string]int
}
var antiLoopMap sync.Map
// antiLoopCheck returns (blocked, n). Tracks the last two distinct call
// signatures per key, so both period-1 (A,A,A) and period-2 (A,B,A,B) loops
// reach the block threshold. A third distinct signature resets the tracker.
func antiLoopCheck(key, sig string) (bool, int) {
\tany, _ := antiLoopMap.Load(key)
\tif any == nil {
\t\ts := &antiLoopState{m: make(map[string]int)}
\t\tantiLoopMap.Store(key, s)
\t\tany = s
\t}
\ts := any.(*antiLoopState)
\tif _, seen := s.m[sig]; !seen && len(s.m) >= 2 {
\t\ts.m = make(map[string]int)
\t}
\ts.m[sig]++
\treturn s.m[sig] >= 3, s.m[sig]
}
'''
def block_tmpl(ind, key):
return (
ind + 'alSig, _ := json.Marshal(args)\n'
+ ind + 'if blocked, aln := antiLoopCheck(' + key + ', tc.Function.Name+"|"+truncate(string(alSig), 400)); blocked {\n'
+ ind + '\tlog.Printf("[ANTI-LOOP] blocked repeat #%d %s", aln, truncate(tc.Function.Name, 40))\n'
+ ind + '\tblockMsg := "HARD SYSTEM BLOCK: this exact tool call has already been executed " + fmt.Sprintf("%d", aln) + " times in a row with unchanged context. The path is DEAD. You MUST now either (a) switch to a completely different tool/source/approach, or (b) if your evidence is sufficient, output the final answer on the last line in the format: FINAL ANSWER: <answer>. Do NOT repeat this call."\n'
+ ind + '\thistory = append(history, LLMMessage{Role: "tool", Content: blockMsg, ToolCallID: tc.ID})\n'
+ ind + '\tcontinue\n'
+ ind + '}\n')
def ws_tmpl(ind):
return (
ind + '// [ANTI-LOOP FIX 2026-09-22]\n'
+ ind + 'alSig, _ := json.Marshal(args)\n'
+ ind + 'if blocked, aln := antiLoopCheck(conv.ID, tc.Function.Name+"|"+truncate(string(alSig), 400)); blocked {\n'
+ ind + '\tlog.Printf("[ANTI-LOOP] blocked repeat #%d %s", aln, truncate(tc.Function.Name, 40))\n'
+ ind + '\tblockMsg := "HARD SYSTEM BLOCK: this exact tool call has already been executed " + fmt.Sprintf("%d", aln) + " times in a row with unchanged context. The path is DEAD. You MUST now either (a) switch to a completely different tool/source/approach, or (b) if your evidence is sufficient, output the final answer on the last line in the format: FINAL ANSWER: <answer>. Do NOT repeat this call."\n'
+ ind + '\thistory = append(history, LLMMessage{Role: "tool", Content: blockMsg, ToolCallID: tc.ID})\n'
+ ind + '\tconv.Messages = append(conv.Messages, WebChatMessage{Role: "tool", ToolName: tc.Function.Name, ToolArgs: args, ToolResult: blockMsg, ToolOK: false, ToolCallID: tc.ID, Timestamp: nextConvTimestamp(conv)})\n'
+ ind + '\tcontinue\n'
+ ind + '}\n')
# 1) agent.go defs
code = rd(AG)
assert "func antiLoopCheck" not in code, "agent.go not clean"
anchor = "func (a *Agent) SyncLLMFromConfig() {"
assert anchor in code
code = code.replace(anchor, ALC + "\n" + anchor, 1)
wr(AG, code)
print("ALC_DEFS_INJECTED")
# 2) agent.go sites — FRESH find each iteration
code = rd(AG)
needle = "result := a.toolCtx.Execute(tc.Function.Name, args)"
prev = 0
for n in range(2):
pos = code.find(needle, prev)
assert pos > 0, "ag site %d missing" % (n + 1)
prev = pos + 800
ls = code.rfind("\n", 0, pos) + 1
le = code.find("\n", pos)
line = code[ls:le]
ind = line[:len(line) - len(line.lstrip("\t"))]
fn_scope = enclosing(code, pos)
has_sl = "sourceLabel" in fn_scope
key = ('"SYNC|"+sourceLabel' if n == 1 else "sourceLabel") if has_sl else ('"SYNC"' if n == 1 else '"AG"')
print("AG_SITE%d ind=%d key=%s sl=%s" % (n + 1, len(ind), key, has_sl))
code = code[:ls] + block_tmpl(ind, key) + code[ls:]
wr(AG, code)
# 3) web_stream.go sites — FRESH find each iteration
code = rd(WS)
needle = "result := loopToolCtx.Execute(tc.Function.Name, args)"
assert "antiLoopCheck(" not in code
assert code.count("nextConvTimestamp(") >= 2
prev = 0
for n in range(2):
pos = code.find(needle, prev)
assert pos > 0, "ws site %d missing" % (n + 1)
prev = pos + 800
ls = code.rfind("\n", 0, pos) + 1
le = code.find("\n", pos)
line = code[ls:le]
ind = line[:len(line) - len(line.lstrip("\t"))]
print("WS_SITE%d ind=%d" % (n + 1, len(ind)))
code = code[:ls] + ws_tmpl(ind) + code[ls:]
wr(WS, code)
print("WS_PATCH_DONE")
# 4) main.go debug.Stack
code = rd(MG)
if "debug.Stack()" not in code:
for old in ['PANIC in ws.Start(): %v", r)', 'PANIC in agent.Run(): %v", r)', 'PANIC in restarted agent.Run(): %v", r)']:
code = code.replace(old, old.replace('%v", r)', '%v\\n%s", r, debug.Stack())'))
if '"runtime/debug"' not in code:
code = code.replace("import (", 'import (\n\t"runtime/debug"', 1)
wr(MG, code)
print("MAIN_STACKED")
else:
print("MAIN_ALREADY")
print("PATCH_V2_ALL_DONE")
|