diff --git a/controller/channel-test.go b/controller/channel-test.go index 5e69f70b..fc7fec28 100644 --- a/controller/channel-test.go +++ b/controller/channel-test.go @@ -5,7 +5,6 @@ import ( "encoding/json" "errors" "fmt" - "github.com/bytedance/gopkg/util/gopool" "io" "math" "net/http" @@ -24,6 +23,8 @@ import ( "sync" "time" + "github.com/bytedance/gopkg/util/gopool" + "github.com/gin-gonic/gin" ) @@ -32,6 +33,9 @@ func testChannel(channel *model.Channel, testModel string) (err error, openAIErr if channel.Type == common.ChannelTypeMidjourney { return errors.New("midjourney channel test is not supported"), nil } + if channel.Type == common.ChannelTypeMidjourneyPlus { + return errors.New("midjourney plus channel test is not supported!!!"), nil + } if channel.Type == common.ChannelTypeSunoAPI { return errors.New("suno channel test is not supported"), nil }