tt/ttstd-infinite-canvas
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(cloud-agent): 云端 Agent - 完善运行时与资源持久化

Browse Source
This commit is contained in:
ddcat666
2026-09-26 17:05:36 +08:00
parent 48e6a705ef
commit 993580a7bd
40 changed files with 1739 additions and 440 deletions
Download Patch File Download Diff File Expand all files Collapse all files

61
backend/internal/app/cloud_agent_model_selection_test.go
View File

@@ -5,10 +5,23 @@ import (
"errors"
"strings"
"testing"
"time"
"gorm.io/gorm"
"infinite-canvas/backend/internal/model"
)
func cloudAgentProjectForDefaultModel(t *testing.T, db *gorm.DB, modelRef string) {
t.Helper()
project := &model.Project{ID: "agent-default-project", UserID: "user", Name: "Agent 默认模型项目", Type: "short-drama", Status: model.ProjectStatusActive, DefaultImageModel: modelRef, DefaultVideoModel: modelRef}
if err := db.Create(project).Error; err != nil {
t.Fatal(err)
}
if err := db.Model(&model.CanvasProject{}).Where("id = ? AND user_id = ?", "agent-canvas", "user").Update("project_id", project.ID).Error; err != nil {
t.Fatal(err)
}
}
func TestCloudAgentModelSelectionRejectsBeforeCanvasRead(t *testing.T) {
for _, tc := range []struct{ name, args, field, issue string }{
{"missing", `{}`, "logicalModelId", "required"},
@@ -62,6 +75,54 @@ func TestCloudAgentModelSelectionAcceptsOnlyCompleteSelections(t *testing.T) {
}
}
func TestCloudAgentProjectDefaultChannelModelFillsOnlyOmittedSelection(t *testing.T) {
s, db, _ := agentMediaFixture(t)
cloudAgentProjectForDefaultModel(t, db, "channel::seedance-test")
run := &model.CloudAgentExecution{ID: "run-default-channel", UserID: "user"}
state := &cloudAgentRuntime{Request: CloudAgentRequest{CanvasID: "agent-canvas"}}
args := cloudAgentMediaArgs{Mode: "video"}
used, err := s.applyCloudAgentProjectDefaultModel(run, state, `{"mode":"video"}`, &args)
if err != nil || !used || args.ChannelID != "channel" || args.ChannelModelKey != "seedance-test" || args.LogicalModelID != "" {
t.Fatalf("default channel selection = used:%v args:%+v err:%v", used, args, err)
}
args = cloudAgentMediaArgs{Mode: "video"}
used, err = s.applyCloudAgentProjectDefaultModel(run, state, `{"mode":"video","channelId":"","channelModelKey":""}`, &args)
if err != nil || used {
t.Fatalf("explicit empty selection was silently defaulted: used:%v args:%+v err:%v", used, args, err)
}
if err := validateCloudAgentModelSelection(`{"mode":"video","channelId":"","channelModelKey":""}`, args); err == nil {
t.Fatal("explicit empty selection unexpectedly passed validation")
}
}
func TestCloudAgentProjectDefaultLogicalModelFillsWhenAvailable(t *testing.T) {
s, db, _ := agentMediaFixture(t)
cloudAgentProjectForDefaultModel(t, db, "logical-image")
s.routeCatalogTTL = time.Hour
imageSpec := CapabilitySpec{Version: 1, Capability: "image"}
s.routeCatalog = &routeCatalogSnapshot{
LoadedAt: time.Now(), Ordered: []string{"logical-image"}, Models: map[string]cachedLogicalModel{
"logical-image": {
Model: model.LogicalModel{ID: "logical-image", Name: "默认逻辑图片", Capability: "image", Enabled: true, PricePolicy: "unified"},
ProductSpec: imageSpec,
Routes: []cachedLogicalRoute{{
Route: model.LogicalModelRoute{ID: "logical-image-route", Enabled: true, Weight: 1},
CapabilitySpec: imageSpec,
ChannelModel: model.ChannelModel{ID: "video-cm", ChannelID: "channel", ModelKey: "seedance-test", Capability: "image"},
}},
},
},
}
run := &model.CloudAgentExecution{ID: "run-default-logical", UserID: "user"}
state := &cloudAgentRuntime{Request: CloudAgentRequest{CanvasID: "agent-canvas"}}
args := cloudAgentMediaArgs{Mode: "image"}
used, err := s.applyCloudAgentProjectDefaultModel(run, state, `{"mode":"image"}`, &args)
if err != nil || !used || args.LogicalModelID != "logical-image" || args.ChannelID != "" || args.ChannelModelKey != "" {
t.Fatalf("default logical selection = used:%v args:%+v err:%v", used, args, err)
}
}
func TestCloudAgentMediaMalformedJSONKeepsStrictFeedback(t *testing.T) {
for _, raw := range []string{`{"unknown":"private-sentinel"}`, `{"channelId":3}`, `{} {}`, `null`} {
call := cloudAgentCall{}