feat(agent): 云端 Agent 运行时 - 完善 Pi 桥接、上下文压力与临时故障重试

This commit is contained in:
ddcat666 committed 2026-10-04 02:20:30 +08:00
1 parent 0030f4da9b
commit 76ca0fd247
32 files changed
+1663 -1485

No files matched your search

@@ -1,6 +1,11 @@
package app
import "testing"
import (
"strings"
"testing"
"infinite-canvas/backend/internal/model"
)
// 运行时会话消息(user/assistant/toolResult + 内容块)必须转换成供应商层能接受的规范请求,
// 否则第一步模型调用就会因“缺少工具选择”等校验失败。
@@ -10,12 +15,15 @@ func TestCanonicalFromRuntimeMessagesPassesProviderValidation(t *testing.T) {
{"role": "assistant", "content": []interface{}{
map[string]interface{}{"type": "thinking", "thinking": "先看看画布"},
map[string]interface{}{"type": "text", "text": "我先读取画布。"},
map[string]interface{}{"type": "toolCall", "id": "call-1", "name": "canvas_snapshot", "arguments": map[string]interface{}{}},
map[string]interface{}{"type": "toolCall", "id": "call-1", "itemId": "fc-1", "name": "canvas_snapshot", "arguments": map[string]interface{}{}},
}},
{"role": "toolResult", "toolCallId": "call-1", "toolName": "canvas_snapshot", "content": []interface{}{map[string]interface{}{"type": "text", "text": "{\"nodes\":0}"}}, "isError": false},
}
tools := runtimeToolSchemas([]map[string]any{{"name": "canvas_snapshot", "description": "获取画布快照", "parameters": map[string]any{"type": "object", "properties": map[string]any{}}}})
canonical := canonicalFromRuntimeMessages(messages, tools, "你是画布助手")
canonical, err := canonicalFromRuntimeMessages(messages, tools, "你是画布助手")
if err != nil {
t.Fatal(err)
}
if canonical.ToolChoice != "auto" {
t.Fatalf("tool choice = %v", canonical.ToolChoice)
}
@@ -25,6 +33,10 @@ func TestCanonicalFromRuntimeMessagesPassesProviderValidation(t *testing.T) {
if stringField(canonical.Messages[3], "role") != "tool" || stringField(canonical.Messages[3], "tool_call_id") != "call-1" {
t.Fatalf("tool result not converted: %#v", canonical.Messages[3])
}
calls := canonicalAgentToolCalls(canonical.Messages[2]["tool_calls"])
if len(calls) != 1 || stringField(calls[0], "item_id") != "fc-1" {
t.Fatalf("runtime Responses item id not preserved: %#v", calls)
}
if _, err := expandCanonicalAgentRequest(&canonical, providerConfig{}, false); err != nil {
t.Fatalf("provider validation rejected runtime transcript: %v", err)
}
@@ -41,3 +53,59 @@ func TestRuntimeToolCallsDecodeArguments(t *testing.T) {
t.Fatalf("unexpected runtime call: %#v", out[0])
}
}
func TestValidateRuntimeMessagesForModelRejectsOrphanToolResult(t *testing.T) {
messages := []map[string]any{{"role": "user", "content": "继续"}, {"role": "toolResult", "toolCallId": "missing", "content": "{}"}}
if err := validateRuntimeMessagesForModel(messages); err == nil || !strings.Contains(err.Error(), "missing") {
t.Fatalf("orphan tool result error = %v", err)
}
}
func TestValidateRuntimeMessagesForModelRejectsIncompleteToolTurn(t *testing.T) {
messages := []map[string]any{{"role": "user", "content": "读取"}, {"role": "assistant", "content": []interface{}{map[string]interface{}{"type": "toolCall", "id": "call-2", "name": "canvas_get_state", "arguments": map[string]interface{}{}}}}}
if err := validateRuntimeMessagesForModel(messages); err == nil || !strings.Contains(err.Error(), "call-2") {
t.Fatalf("incomplete tool turn error = %v", err)
}
}
func TestValidateRuntimeMessagesForModelAllowsMultipleToolCalls(t *testing.T) {
messages := []map[string]any{
{"role": "user", "content": "读取画布和当前选区"},
{"role": "assistant", "content": []interface{}{
map[string]interface{}{"type": "toolCall", "id": "call-state", "name": "canvas_get_state", "arguments": map[string]interface{}{}},
map[string]interface{}{"type": "toolCall", "id": "call-selection", "name": "canvas_get_selection", "arguments": map[string]interface{}{}},
}},
{"role": "toolResult", "toolCallId": "call-selection", "content": "{}"},
{"role": "toolResult", "toolCallId": "call-state", "content": "{}"},
}
if err := validateRuntimeMessagesForModel(messages); err != nil {
t.Fatalf("multiple tool calls should be accepted: %v", err)
}
}
func TestValidateCanonicalAgentHistoryRejectsOrphanToolResult(t *testing.T) {
request := canonicalAgentRequest{
Messages: []map[string]interface{}{
{"role": "user", "content": "继续"},
{"role": "tool", "tool_call_id": "missing", "content": "{}"},
},
ToolChoice: "auto",
}
if _, err := expandCanonicalAgentRequest(&request, providerConfig{}, false); err == nil || !strings.Contains(err.Error(), "missing") {
t.Fatalf("orphan canonical tool result error = %v", err)
}
}
func TestValidateCanonicalAgentHistoryRequiresResponsesItemID(t *testing.T) {
request := canonicalAgentRequest{
Messages: []map[string]interface{}{
{"role": "user", "content": "读取"},
{"role": "assistant", "content": "", "tool_calls": []map[string]interface{}{{"id": "call-3", "function": map[string]interface{}{"name": "canvas_get_state", "arguments": "{}"}}}},
{"role": "tool", "tool_call_id": "call-3", "content": "{}"},
},
ToolChoice: "auto",
}
if _, err := expandCanonicalAgentRequest(&request, providerConfig{InterfaceType: string(model.ChannelInterfaceOpenAIResponse)}, false); err == nil || !strings.Contains(err.Error(), "call-3") {
t.Fatalf("missing Responses item id error = %v", err)
}
}