server
/
new-api
1
0
Fork 0
Code Issues 0 Pull Requests 0 Releases 0 Wiki Activity
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
269 Commits
2 Branches
352 MiB
 
 
 
Tree: 84d364d76a
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '84d364d76a'
${ noResults }
new-api/relay/relay_adaptor.go

169 lines
6.0 KiB
Raw Blame History 

  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. 	tasksora "github.com/QuantumNous/new-api/relay/channel/task/sora"

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

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

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

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

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

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

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

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

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

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

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

  52. )

  53. 

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

  55. 	switch apiType {

  56. 	case constant.APITypeAli:

  57. 		return &ali.Adaptor{}

  58. 	case constant.APITypeAnthropic:

  59. 		return &claude.Adaptor{}

  60. 	case constant.APITypeBaidu:

  61. 		return &baidu.Adaptor{}

  62. 	case constant.APITypeGemini:

  63. 		return &gemini.Adaptor{}

  64. 	case constant.APITypeOpenAI:

  65. 		return &openai.Adaptor{}

  66. 	case constant.APITypePaLM:

  67. 		return &palm.Adaptor{}

  68. 	case constant.APITypeTencent:

  69. 		return &tencent.Adaptor{}

  70. 	case constant.APITypeXunfei:

  71. 		return &xunfei.Adaptor{}

  72. 	case constant.APITypeZhipu:

  73. 		return &zhipu.Adaptor{}

  74. 	case constant.APITypeZhipuV4:

  75. 		return &zhipu_4v.Adaptor{}

  76. 	case constant.APITypeOllama:

  77. 		return &ollama.Adaptor{}

  78. 	case constant.APITypePerplexity:

  79. 		return &perplexity.Adaptor{}

  80. 	case constant.APITypeAws:

  81. 		return &aws.Adaptor{}

  82. 	case constant.APITypeCohere:

  83. 		return &cohere.Adaptor{}

  84. 	case constant.APITypeDify:

  85. 		return &dify.Adaptor{}

  86. 	case constant.APITypeJina:

  87. 		return &jina.Adaptor{}

  88. 	case constant.APITypeCloudflare:

  89. 		return &cloudflare.Adaptor{}

  90. 	case constant.APITypeSiliconFlow:

  91. 		return &siliconflow.Adaptor{}

  92. 	case constant.APITypeVertexAi:

  93. 		return &vertex.Adaptor{}

  94. 	case constant.APITypeMistral:

  95. 		return &mistral.Adaptor{}

  96. 	case constant.APITypeDeepSeek:

  97. 		return &deepseek.Adaptor{}

  98. 	case constant.APITypeMokaAI:

  99. 		return &mokaai.Adaptor{}

  100. 	case constant.APITypeVolcEngine:

  101. 		return &volcengine.Adaptor{}

  102. 	case constant.APITypeBaiduV2:

  103. 		return &baidu_v2.Adaptor{}

  104. 	case constant.APITypeOpenRouter:

  105. 		return &openai.Adaptor{}

  106. 	case constant.APITypeXinference:

  107. 		return &openai.Adaptor{}

  108. 	case constant.APITypeXai:

  109. 		return &xai.Adaptor{}

  110. 	case constant.APITypeCoze:

  111. 		return &coze.Adaptor{}

  112. 	case constant.APITypeJimeng:

  113. 		return &jimeng.Adaptor{}

  114. 	case constant.APITypeMoonshot:

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

  116. 	case constant.APITypeSubmodel:

  117. 		return &submodel.Adaptor{}

  118. 	case constant.APITypeMiniMax:

  119. 		return &minimax.Adaptor{}

  120. 	case constant.APITypeReplicate:

  121. 		return &replicate.Adaptor{}

  122. 	case constant.APITypeCodex:

  123. 		return &codex.Adaptor{}

  124. 	}

  125. 	return nil

  126. }

  127. 

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

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

  130. 	if channelType > 0 {

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

  132. 	}

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

  134. }

  135. 

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

  137. 	switch platform {

  138. 	//case constant.APITypeAIProxyLibrary:

  139. 	//	return &aiproxy.Adaptor{}

  140. 	case constant.TaskPlatformSuno:

  141. 		return &suno.TaskAdaptor{}

  142. 	}

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

  144. 		switch channelType {

  145. 		case constant.ChannelTypeAli:

  146. 			return &taskali.TaskAdaptor{}

  147. 		case constant.ChannelTypeKling:

  148. 			return &kling.TaskAdaptor{}

  149. 		case constant.ChannelTypeJimeng:

  150. 			return &taskjimeng.TaskAdaptor{}

  151. 		case constant.ChannelTypeVertexAi:

  152. 			return &taskvertex.TaskAdaptor{}

  153. 		case constant.ChannelTypeVidu:

  154. 			return &taskVidu.TaskAdaptor{}

  155. 		case constant.ChannelTypeDoubaoVideo, constant.ChannelTypeVolcEngine:

  156. 			return &taskdoubao.TaskAdaptor{}

  157. 		case constant.ChannelTypeDoubaoVideoCompatibleAiping:

  158. 			return &taskdoubaoaiping.TaskAdaptor{}

  159. 		case constant.ChannelTypeSora, constant.ChannelTypeOpenAI:

  160. 			return &tasksora.TaskAdaptor{}

  161. 		case constant.ChannelTypeGemini:

  162. 			return &taskGemini.TaskAdaptor{}

  163. 		case constant.ChannelTypeMiniMax:

  164. 			return &hailuo.TaskAdaptor{}

  165. 		}

  166. 	}

  167. 	return nil

  168. }