79 lines
2.9 KiB
Go
79 lines
2.9 KiB
Go
package app
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"infinite-canvas/backend/internal/model"
|
|
)
|
|
|
|
const cloudAgentToolAttemptLimit = 3
|
|
|
|
type cloudAgentToolRepair struct {
|
|
GroupID string `json:"groupId"`
|
|
Attempt int `json:"attempt"`
|
|
}
|
|
|
|
func cloudAgentRetryReceipt(groupID string, attempt int, status string) map[string]any {
|
|
terminal := status == "exhausted"
|
|
severity := "warning"
|
|
if terminal {
|
|
severity = "error"
|
|
}
|
|
return map[string]any{
|
|
"groupId": groupID, "attempt": attempt, "maxAttempts": cloudAgentToolAttemptLimit,
|
|
"status": status, "severity": severity, "terminal": terminal,
|
|
}
|
|
}
|
|
|
|
// Count per tool, not per model step: intervening reads must not reset a failed
|
|
// write's allowance. Only explicitly typed argument errors and safe, pre-submission
|
|
// transient admission errors are repairable.
|
|
func cloudAgentTrackToolRepair(runID string, state *cloudAgentRuntime, call cloudAgentCall, result any, err error, payload map[string]any) bool {
|
|
previous, exists := state.ToolRepairs[call.Function.Name]
|
|
if err == nil {
|
|
if exists {
|
|
payload["retry"] = cloudAgentRetryReceipt(previous.GroupID, previous.Attempt, "recovered")
|
|
delete(state.ToolRepairs, call.Function.Name)
|
|
}
|
|
return false
|
|
}
|
|
detail, _ := result.(map[string]any)
|
|
var argumentErr *cloudAgentArgumentError
|
|
argumentRepairable := errors.As(err, &argumentErr)
|
|
var admissionErr *cloudAgentMediaAdmissionError
|
|
admissionRepairable := errors.As(err, &admissionErr) && admissionErr.Retryable && detail["phase"] == "admission" && detail["taskSubmitted"] != true
|
|
if (!argumentRepairable && !admissionRepairable) || detail["taskSubmitted"] == true || detail["phase"] == "completion" {
|
|
return false
|
|
}
|
|
if state.ToolRepairs == nil {
|
|
state.ToolRepairs = map[string]cloudAgentToolRepair{}
|
|
}
|
|
if !exists {
|
|
previous.GroupID = runID + ":repair:" + call.ID
|
|
}
|
|
previous.Attempt++
|
|
state.ToolRepairs[call.Function.Name] = previous
|
|
exhausted := previous.Attempt >= cloudAgentToolAttemptLimit
|
|
status := "retrying"
|
|
if exhausted {
|
|
status = "exhausted"
|
|
}
|
|
retry := cloudAgentRetryReceipt(previous.GroupID, previous.Attempt, status)
|
|
payload["retry"], detail["retry"] = retry, retry
|
|
return exhausted
|
|
}
|
|
|
|
func cloudAgentRecordToolResult(run *model.CloudAgentExecution, state *cloudAgentRuntime, call cloudAgentCall, result any, err error) {
|
|
if !cloudAgentToolResult(run.ID, state, call, result, err) {
|
|
return
|
|
}
|
|
run.Status = "failed"
|
|
run.FailureMessage = truncateRunes(fmt.Sprintf("自动纠正已尝试 %d 次，仍未完成：%s", cloudAgentToolAttemptLimit, cloudAgentSafeToolError(err)), 1000)
|
|
cloudAgentDropInterjections(run.ID, "本轮已结束："+truncateRunes(run.FailureMessage, 120), state)
|
|
state.event(run.ID, "run_failed", map[string]any{
|
|
"text": run.FailureMessage, "reason": "tool_retry_exhausted", "toolName": call.Function.Name,
|
|
"callId": call.ID, "attempts": cloudAgentToolAttemptLimit, "severity": "error", "terminal": true,
|
|
})
|
|
}
|