From 687f07bc103c8cc31b11ee375a1ab51e8807ad58 Mon Sep 17 00:00:00 2001 From: iszcz <74706321+iszcz@users.noreply.github.com> Date: Tue, 31 Dec 2024 12:49:13 +0800 Subject: [PATCH] Update channel-test.go --- controller/channel-test.go | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) 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 }