feat: publish v1.0.0-rc.36-custom.1
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
Publish Docker image (Multi-arch) / Build & push (amd64) (push) Failing after 9m20s
Release (Linux, macOS, Windows) / Linux Release (push) Failing after 55m7s
Publish Docker image (Multi-arch) / Build & push (arm64) (push) Canceled after 0s
Release (Linux, macOS, Windows) / macOS Release (push) Canceled after 0s
Release (Linux, macOS, Windows) / Windows Release (push) Canceled after 0s
Publish Docker image (Multi-arch) / Create multi-arch manifests (push) Canceled after 0s
This commit is contained in:
Codex committed
commit
bb7968b612
2381 files changed
+551494
No files matched your search
@@ -0,0 +1,252 @@
|
||||
package plugins_test
|
||||
|
||||
import (
|
||||
"testing"
|
||||
|
||||
"github.com/QuantumNous/new-api/common"
|
||||
"github.com/QuantumNous/new-api/pkg/jsplugin"
|
||||
builtinplugins "github.com/QuantumNous/new-api/plugins"
|
||||
"github.com/QuantumNous/new-api/relay"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestAlibabaResponsesProtocol(t *testing.T) {
|
||||
source, err := builtinplugins.Source("alibaba")
|
||||
require.NoError(t, err)
|
||||
registry := jsplugin.NewRegistry()
|
||||
plugin, err := registry.RegisterFactory(source, jsplugin.Options{Key: "alibaba"})
|
||||
require.NoError(t, err)
|
||||
|
||||
t.Run("claims every Ali model", func(t *testing.T) {
|
||||
for _, model := range plugin.Meta.Models {
|
||||
binding, found := registry.Generation().LookupEndpoint("POST", "/v1/responses", model)
|
||||
require.True(t, found, model)
|
||||
assert.Same(t, plugin, binding.Plugin)
|
||||
assert.Equal(t, "openai_responses", binding.Protocol)
|
||||
}
|
||||
})
|
||||
|
||||
t.Run("declares documented usage facts", func(t *testing.T) {
|
||||
require.Len(t, plugin.Meta.UsageSchema, 2)
|
||||
for _, key := range []string{"seconds", "resolution"} {
|
||||
schema, exists := plugin.Meta.UsageSchema[key]
|
||||
require.True(t, exists, key)
|
||||
assert.NotEmpty(t, schema.Description, key)
|
||||
}
|
||||
|
||||
value, callErr := plugin.Engine.Call(t.Context(), "extractUsage", map[string]any{
|
||||
"model": "wan2.5-i2v-preview",
|
||||
"upstreamModel": "wan2.5-i2v-preview",
|
||||
"usagePurpose": "facts",
|
||||
"requestBody": map[string]any{
|
||||
"model": "wan2.5-i2v-preview",
|
||||
"duration": 10,
|
||||
"size": "1080p",
|
||||
"image": "https://cdn.example/first.png",
|
||||
},
|
||||
})
|
||||
require.NoError(t, callErr)
|
||||
encoded, marshalErr := common.Marshal(value)
|
||||
require.NoError(t, marshalErr)
|
||||
var facts map[string]any
|
||||
require.NoError(t, common.Unmarshal(encoded, &facts))
|
||||
assert.Equal(t, map[string]any{"seconds": float64(10), "resolution": "1080P"}, facts)
|
||||
})
|
||||
|
||||
t.Run("parses text input and options", func(t *testing.T) {
|
||||
value, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "decodeRequest"}, map[string]any{
|
||||
"model": "wan2.7-t2v", "body": map[string]any{"kind": "json", "value": map[string]any{
|
||||
"model": "wan2.7-t2v",
|
||||
"input": "waves at sunset",
|
||||
"size": "1280*720",
|
||||
"duration": 6,
|
||||
"metadata": map[string]any{"parameters": map[string]any{"watermark": true}},
|
||||
}},
|
||||
"stream": false,
|
||||
})
|
||||
require.NoError(t, callErr)
|
||||
encoded, marshalErr := common.Marshal(value)
|
||||
require.NoError(t, marshalErr)
|
||||
var resolved map[string]any
|
||||
require.NoError(t, common.Unmarshal(encoded, &resolved))
|
||||
|
||||
assert.Equal(t, map[string]any{
|
||||
"kind": "submit",
|
||||
"model": "wan2.7-t2v",
|
||||
"action": "text_to_video",
|
||||
"requestBody": map[string]any{
|
||||
"model": "wan2.7-t2v",
|
||||
"prompt": "waves at sunset",
|
||||
"size": "1280*720",
|
||||
"duration": float64(6),
|
||||
"metadata": map[string]any{"parameters": map[string]any{"watermark": true}},
|
||||
},
|
||||
}, resolved)
|
||||
})
|
||||
|
||||
t.Run("parses multimodal image input", func(t *testing.T) {
|
||||
value, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "decodeRequest"}, map[string]any{
|
||||
"model": "wan2.7-i2v", "body": map[string]any{"kind": "json", "value": map[string]any{
|
||||
"model": "wan2.7-i2v",
|
||||
"input": []any{
|
||||
map[string]any{
|
||||
"role": "user",
|
||||
"content": []any{
|
||||
map[string]any{"type": "input_text", "text": "animate between frames"},
|
||||
map[string]any{"type": "input_image", "image_url": "https://cdn.example/first.png"},
|
||||
map[string]any{"type": "input_image", "image_url": map[string]any{"url": "https://cdn.example/last.png"}},
|
||||
},
|
||||
},
|
||||
},
|
||||
}},
|
||||
"stream": true,
|
||||
})
|
||||
require.NoError(t, callErr)
|
||||
encoded, marshalErr := common.Marshal(value)
|
||||
require.NoError(t, marshalErr)
|
||||
var resolved map[string]any
|
||||
require.NoError(t, common.Unmarshal(encoded, &resolved))
|
||||
|
||||
assert.Equal(t, "wan2.7-i2v", resolved["model"])
|
||||
assert.Equal(t, "image_to_video", resolved["action"])
|
||||
requestBody, ok := resolved["requestBody"].(map[string]any)
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, "animate between frames", requestBody["prompt"])
|
||||
assert.Equal(t, []any{"https://cdn.example/first.png", "https://cdn.example/last.png"}, requestBody["images"])
|
||||
})
|
||||
|
||||
t.Run("accepts image-only i2v input", func(t *testing.T) {
|
||||
value, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "decodeRequest"}, map[string]any{
|
||||
"model": "wan2.7-i2v", "body": map[string]any{"kind": "json", "value": map[string]any{
|
||||
"model": "wan2.7-i2v",
|
||||
"input": []any{
|
||||
map[string]any{"type": "input_image", "image_url": "https://cdn.example/first.png"},
|
||||
},
|
||||
}},
|
||||
"stream": false,
|
||||
})
|
||||
require.NoError(t, callErr)
|
||||
encoded, marshalErr := common.Marshal(value)
|
||||
require.NoError(t, marshalErr)
|
||||
var resolved map[string]any
|
||||
require.NoError(t, common.Unmarshal(encoded, &resolved))
|
||||
|
||||
assert.Equal(t, "image_to_video", resolved["action"])
|
||||
requestBody, ok := resolved["requestBody"].(map[string]any)
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, "", requestBody["prompt"])
|
||||
assert.Equal(t, []any{"https://cdn.example/first.png"}, requestBody["images"])
|
||||
})
|
||||
|
||||
t.Run("rejects a request without input text", func(t *testing.T) {
|
||||
_, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "decodeRequest"}, map[string]any{
|
||||
"model": "wan2.7-t2v", "body": map[string]any{"kind": "json", "value": map[string]any{"model": "wan2.7-t2v"}},
|
||||
"stream": false,
|
||||
})
|
||||
require.ErrorContains(t, callErr, "input is required")
|
||||
})
|
||||
|
||||
protocolContext := map[string]any{
|
||||
"requestBody": map[string]any{"model": "wan2.7-t2v"},
|
||||
"stream": true,
|
||||
"artifacts": map[string]any{
|
||||
"video": map[string]any{
|
||||
"key": "video",
|
||||
"type": "video",
|
||||
"mimeType": "video/mp4",
|
||||
"url": "https://gateway.example/v1/tasks/task_public/artifacts/video/content?access=host%2Bcapability%3D",
|
||||
},
|
||||
},
|
||||
}
|
||||
successTask := map[string]any{
|
||||
"task_id": "task_public",
|
||||
"status": "SUCCESS",
|
||||
"progress": "100%",
|
||||
"created_at": 10,
|
||||
"updated_at": 20,
|
||||
"data": map[string]any{
|
||||
"output": map[string]any{
|
||||
"video_url": "https://upstream.example/video.mp4?Expires=1&Signature=must-not-leak",
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
t.Run("renders stream semantics accepted by the host", func(t *testing.T) {
|
||||
progressValue, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "renderEvents"}, protocolContext, map[string]any{
|
||||
"task_id": "task_public",
|
||||
"status": "IN_PROGRESS",
|
||||
"progress": "42%",
|
||||
})
|
||||
require.NoError(t, callErr)
|
||||
progressResult, decodeErr := relay.DecodePluginProtocolEventResult(progressValue, relay.DefaultPluginProtocolLimits())
|
||||
require.NoError(t, decodeErr)
|
||||
require.Len(t, progressResult.Events, 1)
|
||||
require.NotNil(t, progressResult.Events[0].Progress)
|
||||
assert.Equal(t, float64(42), *progressResult.Events[0].Progress)
|
||||
assert.False(t, progressResult.Done)
|
||||
|
||||
value, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "renderEvents"}, protocolContext, successTask)
|
||||
require.NoError(t, callErr)
|
||||
result, decodeErr := relay.DecodePluginProtocolEventResult(value, relay.DefaultPluginProtocolLimits())
|
||||
require.NoError(t, decodeErr)
|
||||
require.Len(t, result.Events, 1)
|
||||
assert.Equal(t, "output", result.Events[0].Type)
|
||||
assert.True(t, result.Done)
|
||||
var text string
|
||||
require.NoError(t, common.Unmarshal(result.Events[0].Data, &text))
|
||||
assert.Equal(t, `<video controls src="https://gateway.example/v1/tasks/task_public/artifacts/video/content?access=host%2Bcapability%3D"></video>`, text)
|
||||
assert.NotContains(t, text, "upstream.example")
|
||||
|
||||
machine := relay.NewPluginResponsesMachine("task_public", "wan2.7-t2v", 10, relay.DefaultPluginProtocolLimits())
|
||||
_, machineErr := machine.CreatedEvent()
|
||||
require.NoError(t, machineErr)
|
||||
wireEvents, machineErr := machine.ApplyTick(result, "SUCCESS")
|
||||
require.NoError(t, machineErr)
|
||||
require.NotEmpty(t, wireEvents)
|
||||
assert.Equal(t, "response.completed", wireEvents[len(wireEvents)-1].Type)
|
||||
})
|
||||
|
||||
t.Run("renders a valid non-stream response", func(t *testing.T) {
|
||||
value, callErr := plugin.Engine.CallPath(t.Context(), "protocols", []string{"openai_responses", "renderFinal"}, protocolContext, successTask)
|
||||
require.NoError(t, callErr)
|
||||
machine := relay.NewPluginResponsesMachine("task_public", "wan2.7-t2v", 10, relay.DefaultPluginProtocolLimits())
|
||||
response, finalErr := machine.FinalResponse(value, "SUCCESS")
|
||||
require.NoError(t, finalErr)
|
||||
assert.Equal(t, "resp_public", response["id"])
|
||||
assert.Equal(t, "completed", response["status"])
|
||||
|
||||
output, ok := response["output"].([]any)
|
||||
require.True(t, ok)
|
||||
require.Len(t, output, 1)
|
||||
item, ok := output[0].(map[string]any)
|
||||
require.True(t, ok)
|
||||
content, ok := item["content"].([]any)
|
||||
require.True(t, ok)
|
||||
require.Len(t, content, 1)
|
||||
part, ok := content[0].(map[string]any)
|
||||
require.True(t, ok)
|
||||
text, ok := part["text"].(string)
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, `<video controls src="https://gateway.example/v1/tasks/task_public/artifacts/video/content?access=host%2Bcapability%3D"></video>`, text)
|
||||
assert.NotContains(t, text, "upstream.example")
|
||||
metadata, ok := response["metadata"].(map[string]string)
|
||||
require.True(t, ok)
|
||||
assert.Equal(t, "ali", metadata["vendor"])
|
||||
})
|
||||
|
||||
t.Run("does not fall back to the upstream URL when the host artifact is absent", func(t *testing.T) {
|
||||
_, callErr := plugin.Engine.CallPath(
|
||||
t.Context(),
|
||||
"protocols",
|
||||
[]string{"openai_responses", "renderFinal"},
|
||||
map[string]any{
|
||||
"requestBody": map[string]any{"model": "wan2.7-t2v"},
|
||||
"stream": false,
|
||||
},
|
||||
successTask,
|
||||
)
|
||||
require.ErrorContains(t, callErr, "video artifact is unavailable")
|
||||
assert.NotContains(t, callErr.Error(), "upstream.example")
|
||||
})
|
||||
}
|
||||
Reference in new issue
Block a user