server
/
new-api
1
0
Rozštěpit 0
Zdrojový kód Úkoly 0 Požadavky na natažení 0 Vydání 0 Wiki Aktivita
Nelze vybrat více než 25 témat Téma musí začínat písmenem nebo číslem, může obsahovat pomlčky („-“) a může být dlouhé až 35 znaků.
280 Revize
2 Větve
352 MiB
 
 
 
Strom: 56400b6c61
Větve Značky
${ item.name }
Vytvořit větev ${ searchTerm }
z „56400b6c61“
${ noResults }
new-api/relay/relay_adaptor.go

172 řádky
6.1 KiB
Surový Blame Historie 

  1. package relay

  2. 

  3. import (

  4. 	"strconv"

  5. 

  6. 	"github.com/QuantumNous/new-api/constant"

  7. 	"github.com/QuantumNous/new-api/relay/channel"

  8. 	"github.com/QuantumNous/new-api/relay/channel/ali"

  9. 	"github.com/QuantumNous/new-api/relay/channel/aws"

  10. 	"github.com/QuantumNous/new-api/relay/channel/baidu"

  11. 	"github.com/QuantumNous/new-api/relay/channel/baidu_v2"

  12. 	"github.com/QuantumNous/new-api/relay/channel/claude"

  13. 	"github.com/QuantumNous/new-api/relay/channel/cloudflare"

  14. 	"github.com/QuantumNous/new-api/relay/channel/codex"

  15. 	"github.com/QuantumNous/new-api/relay/channel/cohere"

  16. 	"github.com/QuantumNous/new-api/relay/channel/coze"

  17. 	"github.com/QuantumNous/new-api/relay/channel/deepseek"

  18. 	"github.com/QuantumNous/new-api/relay/channel/dify"

  19. 	"github.com/QuantumNous/new-api/relay/channel/gemini"

  20. 	"github.com/QuantumNous/new-api/relay/channel/jimeng"

  21. 	"github.com/QuantumNous/new-api/relay/channel/jina"

  22. 	"github.com/QuantumNous/new-api/relay/channel/minimax"

  23. 	"github.com/QuantumNous/new-api/relay/channel/mistral"

  24. 	"github.com/QuantumNous/new-api/relay/channel/mokaai"

  25. 	"github.com/QuantumNous/new-api/relay/channel/moonshot"

  26. 	"github.com/QuantumNous/new-api/relay/channel/ollama"

  27. 	"github.com/QuantumNous/new-api/relay/channel/openai"

  28. 	"github.com/QuantumNous/new-api/relay/channel/palm"

  29. 	"github.com/QuantumNous/new-api/relay/channel/perplexity"

  30. 	"github.com/QuantumNous/new-api/relay/channel/replicate"

  31. 	"github.com/QuantumNous/new-api/relay/channel/siliconflow"

  32. 	"github.com/QuantumNous/new-api/relay/channel/submodel"

  33. 	taskali "github.com/QuantumNous/new-api/relay/channel/task/ali"

  34. 	taskdoubao "github.com/QuantumNous/new-api/relay/channel/task/doubao"

  35. 	taskdoubaoaiping "github.com/QuantumNous/new-api/relay/channel/task/doubao_aiping"

  36. 	taskGemini "github.com/QuantumNous/new-api/relay/channel/task/gemini"

  37. 	"github.com/QuantumNous/new-api/relay/channel/task/hailuo"

  38. 	taskjimeng "github.com/QuantumNous/new-api/relay/channel/task/jimeng"

  39. 	"github.com/QuantumNous/new-api/relay/channel/task/kling"

  40. 	klingaiping "github.com/QuantumNous/new-api/relay/channel/task/kling/aiping"

  41. 	tasksora "github.com/QuantumNous/new-api/relay/channel/task/sora"

  42. 	"github.com/QuantumNous/new-api/relay/channel/task/suno"

  43. 	taskvertex "github.com/QuantumNous/new-api/relay/channel/task/vertex"

  44. 	taskVidu "github.com/QuantumNous/new-api/relay/channel/task/vidu"

  45. 	"github.com/QuantumNous/new-api/relay/channel/tencent"

  46. 	"github.com/QuantumNous/new-api/relay/channel/vertex"

  47. 	"github.com/QuantumNous/new-api/relay/channel/volcengine"

  48. 	"github.com/QuantumNous/new-api/relay/channel/xai"

  49. 	"github.com/QuantumNous/new-api/relay/channel/xunfei"

  50. 	"github.com/QuantumNous/new-api/relay/channel/zhipu"

  51. 	"github.com/QuantumNous/new-api/relay/channel/zhipu_4v"

  52. 	"github.com/gin-gonic/gin"

  53. )

  54. 

  55. func GetAdaptor(apiType int) channel.Adaptor {

  56. 	switch apiType {

  57. 	case constant.APITypeAli:

  58. 		return &ali.Adaptor{}

  59. 	case constant.APITypeAnthropic:

  60. 		return &claude.Adaptor{}

  61. 	case constant.APITypeBaidu:

  62. 		return &baidu.Adaptor{}

  63. 	case constant.APITypeGemini:

  64. 		return &gemini.Adaptor{}

  65. 	case constant.APITypeOpenAI:

  66. 		return &openai.Adaptor{}

  67. 	case constant.APITypePaLM:

  68. 		return &palm.Adaptor{}

  69. 	case constant.APITypeTencent:

  70. 		return &tencent.Adaptor{}

  71. 	case constant.APITypeXunfei:

  72. 		return &xunfei.Adaptor{}

  73. 	case constant.APITypeZhipu:

  74. 		return &zhipu.Adaptor{}

  75. 	case constant.APITypeZhipuV4:

  76. 		return &zhipu_4v.Adaptor{}

  77. 	case constant.APITypeOllama:

  78. 		return &ollama.Adaptor{}

  79. 	case constant.APITypePerplexity:

  80. 		return &perplexity.Adaptor{}

  81. 	case constant.APITypeAws:

  82. 		return &aws.Adaptor{}

  83. 	case constant.APITypeCohere:

  84. 		return &cohere.Adaptor{}

  85. 	case constant.APITypeDify:

  86. 		return &dify.Adaptor{}

  87. 	case constant.APITypeJina:

  88. 		return &jina.Adaptor{}

  89. 	case constant.APITypeCloudflare:

  90. 		return &cloudflare.Adaptor{}

  91. 	case constant.APITypeSiliconFlow:

  92. 		return &siliconflow.Adaptor{}

  93. 	case constant.APITypeVertexAi:

  94. 		return &vertex.Adaptor{}

  95. 	case constant.APITypeMistral:

  96. 		return &mistral.Adaptor{}

  97. 	case constant.APITypeDeepSeek:

  98. 		return &deepseek.Adaptor{}

  99. 	case constant.APITypeMokaAI:

  100. 		return &mokaai.Adaptor{}

  101. 	case constant.APITypeVolcEngine:

  102. 		return &volcengine.Adaptor{}

  103. 	case constant.APITypeBaiduV2:

  104. 		return &baidu_v2.Adaptor{}

  105. 	case constant.APITypeOpenRouter:

  106. 		return &openai.Adaptor{}

  107. 	case constant.APITypeXinference:

  108. 		return &openai.Adaptor{}

  109. 	case constant.APITypeXai:

  110. 		return &xai.Adaptor{}

  111. 	case constant.APITypeCoze:

  112. 		return &coze.Adaptor{}

  113. 	case constant.APITypeJimeng:

  114. 		return &jimeng.Adaptor{}

  115. 	case constant.APITypeMoonshot:

  116. 		return &moonshot.Adaptor{} // Moonshot uses Claude API

  117. 	case constant.APITypeSubmodel:

  118. 		return &submodel.Adaptor{}

  119. 	case constant.APITypeMiniMax:

  120. 		return &minimax.Adaptor{}

  121. 	case constant.APITypeReplicate:

  122. 		return &replicate.Adaptor{}

  123. 	case constant.APITypeCodex:

  124. 		return &codex.Adaptor{}

  125. 	}

  126. 	return nil

  127. }

  128. 

  129. func GetTaskPlatform(c *gin.Context) constant.TaskPlatform {

  130. 	channelType := c.GetInt("channel_type")

  131. 	if channelType > 0 {

  132. 		return constant.TaskPlatform(strconv.Itoa(channelType))

  133. 	}

  134. 	return constant.TaskPlatform(c.GetString("platform"))

  135. }

  136. 

  137. func GetTaskAdaptor(platform constant.TaskPlatform) channel.TaskAdaptor {

  138. 	switch platform {

  139. 	//case constant.APITypeAIProxyLibrary:

  140. 	//	return &aiproxy.Adaptor{}

  141. 	case constant.TaskPlatformSuno:

  142. 		return &suno.TaskAdaptor{}

  143. 	}

  144. 	if channelType, err := strconv.ParseInt(string(platform), 10, 64); err == nil {

  145. 		switch channelType {

  146. 		case constant.ChannelTypeAli:

  147. 			return &taskali.TaskAdaptor{}

  148. 		case constant.ChannelTypeKling:

  149. 			return &kling.TaskAdaptor{}

  150. 		case constant.ChannelTypeJimeng:

  151. 			return &taskjimeng.TaskAdaptor{}

  152. 		case constant.ChannelTypeVertexAi:

  153. 			return &taskvertex.TaskAdaptor{}

  154. 		case constant.ChannelTypeVidu:

  155. 			return &taskVidu.TaskAdaptor{}

  156. 		case constant.ChannelTypeDoubaoVideo, constant.ChannelTypeVolcEngine:

  157. 			return &taskdoubao.TaskAdaptor{}

  158. 		case constant.ChannelTypeDoubaoVideoCompatibleAiping:

  159. 			return &taskdoubaoaiping.TaskAdaptor{}

  160. 		case constant.ChannelTypeKlingAiping:

  161. 			return &klingaiping.TaskAdaptor{}

  162. 		case constant.ChannelTypeSora, constant.ChannelTypeOpenAI:

  163. 			return &tasksora.TaskAdaptor{}

  164. 		case constant.ChannelTypeGemini:

  165. 			return &taskGemini.TaskAdaptor{}

  166. 		case constant.ChannelTypeMiniMax:

  167. 			return &hailuo.TaskAdaptor{}

  168. 		}

  169. 	}

  170. 	return nil

  171. }