Add Sub2API async image channel

.gitignore

@@ -32,6 +32,8 @@ data/
 .gomodcache/
 .gocache-temp
 .gopath
-.test
 token_estimator_test.go
+# Local copy of vendor API doc; not tracked (example URLs may trip GitHub push protection)
+Pingxingshijie-AI接口.md
+.test
 skills-lock.json

common/api_type.go

@@ -53,6 +53,15 @@ func ChannelType2APIType(channelType int) (int, bool) {
 		apiType = constant.APITypeMokaAI
 	case constant.ChannelTypeVolcEngine:
 		apiType = constant.APITypeVolcEngine
+	case constant.ChannelTypePingXingShiJie:
+		// Text/embed/rerank/image sync: same HTTP paths as Volcengine Ark (/api/v3/*). Async tasks still use task/pingxingshijie.
+		apiType = constant.APITypeVolcEngine
+	case constant.ChannelTypeKieAI:
+		// KieAI currently exposes async Market jobs through task/kie; sync calls fall back to OpenAI-compatible handling when used.
+		apiType = constant.APITypeOpenAI
+	case constant.ChannelTypeSub2APIAsync:
+		// Sub2API-async is task-only for async image jobs; keep API type OpenAI-compatible for channel metadata.
+		apiType = constant.APITypeOpenAI
 	case constant.ChannelTypeBaiduV2:
 		apiType = constant.APITypeBaiduV2
 	case constant.ChannelTypeOpenRouter:

common/api_type_test.go

@@ -0,0 +1,17 @@
+package common
+
+import (
+	"testing"
+
+	"github.com/QuantumNous/new-api/constant"
+)
+
+func TestChannelType2APITypeMapsSub2APIAsyncToOpenAI(t *testing.T) {
+	apiType, ok := ChannelType2APIType(constant.ChannelTypeSub2APIAsync)
+	if !ok {
+		t.Fatal("expected Sub2API-async channel type to be known")
+	}
+	if apiType != constant.APITypeOpenAI {
+		t.Fatalf("api type = %d, want %d", apiType, constant.APITypeOpenAI)
+	}
+}

constant/channel.go

@@ -55,6 +55,10 @@ const (
 	ChannelTypeSora           = 55
 	ChannelTypeReplicate      = 56
 	ChannelTypeCodex          = 57
+	ChannelTypePingXingShiJie = 58
+	ChannelTypeKieAI          = 59
+	ChannelTypeFalAI          = 60
+	ChannelTypeSub2APIAsync   = 61
 	ChannelTypeDummy          // this one is only for count, do not add any channel after this
 
 )
@@ -118,6 +122,10 @@ var ChannelBaseURLs = []string{
 	"https://api.openai.com",                    //55
 	"https://api.replicate.com",                 //56
 	"https://chatgpt.com",                       //57
+	"https://api.pingxingshijie.cn",             //58 PingXingShiJie
+	"https://api.kie.ai",                        //59 KieAI
+	"https://queue.fal.run",                     //60 FalAI
+	"https://api.sub2api.com",                   //61 Sub2API-async
 }
 
 var ChannelTypeNames = map[int]string{
@@ -175,6 +183,10 @@ var ChannelTypeNames = map[int]string{
 	ChannelTypeSora:           "Sora",
 	ChannelTypeReplicate:      "Replicate",
 	ChannelTypeCodex:          "Codex",
+	ChannelTypePingXingShiJie: "PingXingShiJie",
+	ChannelTypeKieAI:          "KieAI",
+	ChannelTypeFalAI:          "FalAI",
+	ChannelTypeSub2APIAsync:   "Sub2API-async",
 }
 
 func GetChannelTypeName(channelType int) string {

controller/channel-test.go

@@ -122,6 +122,11 @@ func testChannel(channel *model.Channel, testModel string, endpointType string,
 			requestPath = "/v1/images/generations"
 		}
 
+		// PingXingShiJie: sync image models use OpenAI image relay; text uses chat (Volc-style upstream via adaptor mapping).
+		if channel.Type == constant.ChannelTypePingXingShiJie && strings.Contains(strings.ToLower(testModel), "seedream") {
+			requestPath = "/v1/images/generations"
+		}
+
 		// responses-only models
 		if strings.Contains(strings.ToLower(testModel), "codex") {
 			requestPath = "/v1/responses"
@@ -135,6 +140,10 @@ func testChannel(channel *model.Channel, testModel string, endpointType string,
 	if strings.HasPrefix(requestPath, "/v1/responses/compact") {
 		testModel = ratio_setting.WithCompactModelSuffix(testModel)
 	}
+	effectiveEndpointType := endpointType
+	if effectiveEndpointType == "" && requestPath == "/v1/images/generations" {
+		effectiveEndpointType = string(constant.EndpointTypeImageGeneration)
+	}
 
 	c.Request = &http.Request{
 		Method: "POST",
@@ -219,7 +228,7 @@ func testChannel(channel *model.Channel, testModel string, endpointType string,
 		}
 	}
 
-	request := buildTestRequest(testModel, endpointType, channel, isStream)
+	request := buildTestRequest(testModel, effectiveEndpointType, channel, isStream)
 
 	info, err := relaycommon.GenRelayInfo(c, relayFormat, request, nil)
 

controller/model.go

@@ -14,6 +14,9 @@ import (
 	"github.com/QuantumNous/new-api/relay/channel/lingyiwanwu"
 	"github.com/QuantumNous/new-api/relay/channel/minimax"
 	"github.com/QuantumNous/new-api/relay/channel/moonshot"
+	taskkie "github.com/QuantumNous/new-api/relay/channel/task/kie"
+	taskpxsj "github.com/QuantumNous/new-api/relay/channel/task/pingxingshijie"
+	tasksub2apiasync "github.com/QuantumNous/new-api/relay/channel/task/sub2api_async"
 	relaycommon "github.com/QuantumNous/new-api/relay/common"
 	"github.com/QuantumNous/new-api/relay/helper"
 	"github.com/QuantumNous/new-api/service"
@@ -79,6 +82,14 @@ func init() {
 			OwnedBy: minimax.ChannelName,
 		})
 	}
+	for _, modelName := range taskkie.ModelList {
+		openAIModels = append(openAIModels, dto.OpenAIModels{
+			Id:      modelName,
+			Object:  "model",
+			Created: 1626777600,
+			OwnedBy: taskkie.ChannelName,
+		})
+	}
 	for modelName, _ := range constant.MidjourneyModel2Action {
 		openAIModels = append(openAIModels, dto.OpenAIModels{
 			Id:      modelName,
@@ -93,6 +104,18 @@ func init() {
 	}
 	channelId2Models = make(map[int][]string)
 	for i := 1; i <= constant.ChannelTypeDummy; i++ {
+		if i == constant.ChannelTypePingXingShiJie {
+			channelId2Models[i] = append([]string(nil), taskpxsj.ModelList...)
+			continue
+		}
+		if i == constant.ChannelTypeKieAI {
+			channelId2Models[i] = append([]string(nil), taskkie.ModelList...)
+			continue
+		}
+		if i == constant.ChannelTypeSub2APIAsync {
+			channelId2Models[i] = append([]string(nil), tasksub2apiasync.ModelList...)
+			continue
+		}
 		apiType, success := common.ChannelType2APIType(i)
 		if !success || apiType == constant.APITypeAIProxyLibrary {
 			continue

controller/model_list_test.go

@@ -12,6 +12,8 @@ import (
 	"github.com/QuantumNous/new-api/constant"
 	"github.com/QuantumNous/new-api/dto"
 	"github.com/QuantumNous/new-api/model"
+	taskkie "github.com/QuantumNous/new-api/relay/channel/task/kie"
+	tasksub2apiasync "github.com/QuantumNous/new-api/relay/channel/task/sub2api_async"
 	"github.com/QuantumNous/new-api/setting/config"
 	"github.com/QuantumNous/new-api/setting/operation_setting"
 	"github.com/gin-gonic/gin"
@@ -210,6 +212,18 @@ func TestListModelsIncludesTieredBillingModel(t *testing.T) {
 	require.Empty(t, missingExprPricing.BillingExpr)
 }
 
+func TestKieModelsAppearInGlobalModelRegistry(t *testing.T) {
+	for _, modelName := range taskkie.ModelList {
+		aiModel, ok := openAIModelsMap[modelName]
+		require.True(t, ok, "missing Kie model %s", modelName)
+		require.Equal(t, taskkie.ChannelName, aiModel.OwnedBy)
+	}
+}
+
+func TestSub2APIAsyncModelsAppearInChannelModelRegistry(t *testing.T) {
+	require.Equal(t, tasksub2apiasync.ModelList, channelId2Models[constant.ChannelTypeSub2APIAsync])
+}
+
 func TestListModelsTokenLimitIncludesTieredBillingModel(t *testing.T) {
 	withSelfUseModeDisabled(t)
 	withTieredBillingConfig(t, map[string]string{

controller/relay.go

@@ -17,6 +17,7 @@ import (
 	"github.com/QuantumNous/new-api/model"
 	perfmetrics "github.com/QuantumNous/new-api/pkg/perf_metrics"
 	"github.com/QuantumNous/new-api/relay"
+	taskpxsj "github.com/QuantumNous/new-api/relay/channel/task/pingxingshijie"
 	relaycommon "github.com/QuantumNous/new-api/relay/common"
 	relayconstant "github.com/QuantumNous/new-api/relay/constant"
 	"github.com/QuantumNous/new-api/relay/helper"
@@ -578,6 +579,7 @@ func RelayTask(c *gin.Context) {
 
 		task := model.InitTask(result.Platform, relayInfo)
 		task.PrivateData.UpstreamTaskID = result.UpstreamTaskID
+		task.PrivateData.UpstreamKind = taskpxsj.UpstreamKindFromPath(c.Request.URL.Path)
 		task.PrivateData.BillingSource = relayInfo.BillingSource
 		task.PrivateData.SubscriptionId = relayInfo.SubscriptionId
 		task.PrivateData.TokenId = relayInfo.TokenId
@@ -594,6 +596,8 @@ func RelayTask(c *gin.Context) {
 		task.Action = relayInfo.Action
 		if insertErr := task.Insert(); insertErr != nil {
 			common.SysError("insert task error: " + insertErr.Error())
+		} else if relayInfo.TaskRelayInfo != nil && relayInfo.AfterTaskInserted != nil {
+			relayInfo.AfterTaskInserted(task.ID)
 		}
 	}