mirror of
https://github.com/QuantumNous/new-api.git
synced 2026-09-10 22:20:25 +00:00
32 lines
951 B
Go
32 lines
951 B
Go
package plugins_test
|
|
|
|
import "testing"
|
|
|
|
func TestDoubaoResponsesProtocol(t *testing.T) {
|
|
testVideoResponsesProtocol(t, videoResponsesTestCase{
|
|
pluginKey: "doubao",
|
|
model: "doubao-seedance-2-0-260128",
|
|
requestBody: map[string]any{
|
|
"model": "doubao-seedance-2-0-260128",
|
|
"input": []any{map[string]any{"role": "user", "content": []any{
|
|
map[string]any{"type": "input_text", "text": "a running fox"},
|
|
map[string]any{"type": "input_image", "image_url": "https://cdn.example/frame.png"},
|
|
}}},
|
|
"seconds": 6,
|
|
"size": "1920x1080",
|
|
},
|
|
wantAction: "image_to_video",
|
|
wantRequest: map[string]any{
|
|
"model": "doubao-seedance-2-0-260128",
|
|
"prompt": "a running fox",
|
|
"images": []any{"https://cdn.example/frame.png"},
|
|
"seconds": float64(6),
|
|
"metadata": map[string]any{
|
|
"resolution": "1080p",
|
|
},
|
|
},
|
|
wantUsageKeys: []string{"resolution", "tokens", "video_input"},
|
|
wantVendorName: "doubao",
|
|
})
|
|
}
|