server
/
new-api
1
0
포크 0
코드 이슈 0 풀 리퀘스트 0 릴리즈 0 위키 활동
25개 이상의 토픽을 선택하실 수 없습니다. Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
285 커밋
2 브랜치
352 MiB
 
 
 
트리: 678add297f
브랜치 태그
${ item.name }
${ searchTerm } 브랜치 생성
from '678add297f'
${ noResults }
new-api/relay/relay_adaptor.go

175 lines
6.3 KiB
Raw Blame 히스토리 

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

  54. )

  55. 

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

  57. 	switch apiType {

  58. 	case constant.APITypeAli:

  59. 		return &ali.Adaptor{}

  60. 	case constant.APITypeAnthropic:

  61. 		return &claude.Adaptor{}

  62. 	case constant.APITypeBaidu:

  63. 		return &baidu.Adaptor{}

  64. 	case constant.APITypeGemini:

  65. 		return &gemini.Adaptor{}

  66. 	case constant.APITypeOpenAI:

  67. 		return &openai.Adaptor{}

  68. 	case constant.APITypePaLM:

  69. 		return &palm.Adaptor{}

  70. 	case constant.APITypeTencent:

  71. 		return &tencent.Adaptor{}

  72. 	case constant.APITypeXunfei:

  73. 		return &xunfei.Adaptor{}

  74. 	case constant.APITypeZhipu:

  75. 		return &zhipu.Adaptor{}

  76. 	case constant.APITypeZhipuV4:

  77. 		return &zhipu_4v.Adaptor{}

  78. 	case constant.APITypeOllama:

  79. 		return &ollama.Adaptor{}

  80. 	case constant.APITypePerplexity:

  81. 		return &perplexity.Adaptor{}

  82. 	case constant.APITypeAws:

  83. 		return &aws.Adaptor{}

  84. 	case constant.APITypeCohere:

  85. 		return &cohere.Adaptor{}

  86. 	case constant.APITypeDify:

  87. 		return &dify.Adaptor{}

  88. 	case constant.APITypeJina:

  89. 		return &jina.Adaptor{}

  90. 	case constant.APITypeCloudflare:

  91. 		return &cloudflare.Adaptor{}

  92. 	case constant.APITypeSiliconFlow:

  93. 		return &siliconflow.Adaptor{}

  94. 	case constant.APITypeVertexAi:

  95. 		return &vertex.Adaptor{}

  96. 	case constant.APITypeMistral:

  97. 		return &mistral.Adaptor{}

  98. 	case constant.APITypeDeepSeek:

  99. 		return &deepseek.Adaptor{}

  100. 	case constant.APITypeMokaAI:

  101. 		return &mokaai.Adaptor{}

  102. 	case constant.APITypeVolcEngine:

  103. 		return &volcengine.Adaptor{}

  104. 	case constant.APITypeBaiduV2:

  105. 		return &baidu_v2.Adaptor{}

  106. 	case constant.APITypeOpenRouter:

  107. 		return &openai.Adaptor{}

  108. 	case constant.APITypeXinference:

  109. 		return &openai.Adaptor{}

  110. 	case constant.APITypeXai:

  111. 		return &xai.Adaptor{}

  112. 	case constant.APITypeCoze:

  113. 		return &coze.Adaptor{}

  114. 	case constant.APITypeJimeng:

  115. 		return &jimeng.Adaptor{}

  116. 	case constant.APITypeMoonshot:

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

  118. 	case constant.APITypeSubmodel:

  119. 		return &submodel.Adaptor{}

  120. 	case constant.APITypeMiniMax:

  121. 		return &minimax.Adaptor{}

  122. 	case constant.APITypeReplicate:

  123. 		return &replicate.Adaptor{}

  124. 	case constant.APITypeCodex:

  125. 		return &codex.Adaptor{}

  126. 	}

  127. 	return nil

  128. }

  129. 

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

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

  132. 	if channelType > 0 {

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

  134. 	}

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

  136. }

  137. 

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

  139. 	switch platform {

  140. 	//case constant.APITypeAIProxyLibrary:

  141. 	//	return &aiproxy.Adaptor{}

  142. 	case constant.TaskPlatformSuno:

  143. 		return &suno.TaskAdaptor{}

  144. 	}

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

  146. 		switch channelType {

  147. 		case constant.ChannelTypeAli:

  148. 			return &taskali.TaskAdaptor{}

  149. 		case constant.ChannelTypeKling:

  150. 			return &kling.TaskAdaptor{}

  151. 		case constant.ChannelTypeJimeng:

  152. 			return &taskjimeng.TaskAdaptor{}

  153. 		case constant.ChannelTypeVertexAi:

  154. 			return &taskvertex.TaskAdaptor{}

  155. 		case constant.ChannelTypeVidu:

  156. 			return &taskVidu.TaskAdaptor{}

  157. 		case constant.ChannelTypeDoubaoVideo, constant.ChannelTypeVolcEngine:

  158. 			return &taskdoubao.TaskAdaptor{}

  159. 		case constant.ChannelTypeDoubaoVideoCompatibleAiping:

  160. 			return &taskdoubaoaiping.TaskAdaptor{}

  161. 		case constant.ChannelTypeDoubaoVideoCompatibleTianyiYun:

  162. 			return &taskdoubaotianyiyun.TaskAdaptor{}

  163. 		case constant.ChannelTypeKlingAiping:

  164. 			return &klingaiping.TaskAdaptor{}

  165. 		case constant.ChannelTypeSora, constant.ChannelTypeOpenAI:

  166. 			return &tasksora.TaskAdaptor{}

  167. 		case constant.ChannelTypeGemini:

  168. 			return &taskGemini.TaskAdaptor{}

  169. 		case constant.ChannelTypeMiniMax:

  170. 			return &hailuo.TaskAdaptor{}

  171. 		}

  172. 	}

  173. 	return nil

  174. }