Domovská stránka Prehľadávať Pomoc
Registrovať Prihlásiť sa
AI
/
new-api
zrkadlo https://github.com/QuantumNous/new-api.git
2
0
Fork 0
Súbory Issues 0 Wiki
Branch: refactor_error
Branche Tagy
new-api
/
relay
/
relay_adaptor.go

relay_adaptor.go 3.1 KB
Permanentný odkaz História Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112 
  1. package relay
    2. 

  2. 

  3. import (
    4. 

  4. 	"one-api/constant"
    5. 

  5. 	commonconstant "one-api/constant"
    6. 

  6. 	"one-api/relay/channel"
    7. 

  7. 	"one-api/relay/channel/ali"
    8. 

  8. 	"one-api/relay/channel/aws"
    9. 

  9. 	"one-api/relay/channel/baidu"
    10. 

  10. 	"one-api/relay/channel/baidu_v2"
    11. 

  11. 	"one-api/relay/channel/claude"
    12. 

  12. 	"one-api/relay/channel/cloudflare"
    13. 

  13. 	"one-api/relay/channel/cohere"
    14. 

  14. 	"one-api/relay/channel/coze"
    15. 

  15. 	"one-api/relay/channel/deepseek"
    16. 

  16. 	"one-api/relay/channel/dify"
    17. 

  17. 	"one-api/relay/channel/gemini"
    18. 

  18. 	"one-api/relay/channel/jina"
    19. 

  19. 	"one-api/relay/channel/mistral"
    20. 

  20. 	"one-api/relay/channel/mokaai"
    21. 

  21. 	"one-api/relay/channel/ollama"
    22. 

  22. 	"one-api/relay/channel/openai"
    23. 

  23. 	"one-api/relay/channel/palm"
    24. 

  24. 	"one-api/relay/channel/perplexity"
    25. 

  25. 	"one-api/relay/channel/siliconflow"
    26. 

  26. 	"one-api/relay/channel/task/jimeng"
    27. 

  27. 	"one-api/relay/channel/task/kling"
    28. 

  28. 	"one-api/relay/channel/task/suno"
    29. 

  29. 	"one-api/relay/channel/tencent"
    30. 

  30. 	"one-api/relay/channel/vertex"
    31. 

  31. 	"one-api/relay/channel/volcengine"
    32. 

  32. 	"one-api/relay/channel/xai"
    33. 

  33. 	"one-api/relay/channel/xunfei"
    34. 

  34. 	"one-api/relay/channel/zhipu"
    35. 

  35. 	"one-api/relay/channel/zhipu_4v"
    36. 

  36. )
    37. 

  37. 

  38. func GetAdaptor(apiType int) channel.Adaptor {
    39. 

  39. 	switch apiType {
    40. 

  40. 	case constant.APITypeAli:
    41. 

  41. 		return &ali.Adaptor{}
    42. 

  42. 	case constant.APITypeAnthropic:
    43. 

  43. 		return &claude.Adaptor{}
    44. 

  44. 	case constant.APITypeBaidu:
    45. 

  45. 		return &baidu.Adaptor{}
    46. 

  46. 	case constant.APITypeGemini:
    47. 

  47. 		return &gemini.Adaptor{}
    48. 

  48. 	case constant.APITypeOpenAI:
    49. 

  49. 		return &openai.Adaptor{}
    50. 

  50. 	case constant.APITypePaLM:
    51. 

  51. 		return &palm.Adaptor{}
    52. 

  52. 	case constant.APITypeTencent:
    53. 

  53. 		return &tencent.Adaptor{}
    54. 

  54. 	case constant.APITypeXunfei:
    55. 

  55. 		return &xunfei.Adaptor{}
    56. 

  56. 	case constant.APITypeZhipu:
    57. 

  57. 		return &zhipu.Adaptor{}
    58. 

  58. 	case constant.APITypeZhipuV4:
    59. 

  59. 		return &zhipu_4v.Adaptor{}
    60. 

  60. 	case constant.APITypeOllama:
    61. 

  61. 		return &ollama.Adaptor{}
    62. 

  62. 	case constant.APITypePerplexity:
    63. 

  63. 		return &perplexity.Adaptor{}
    64. 

  64. 	case constant.APITypeAws:
    65. 

  65. 		return &aws.Adaptor{}
    66. 

  66. 	case constant.APITypeCohere:
    67. 

  67. 		return &cohere.Adaptor{}
    68. 

  68. 	case constant.APITypeDify:
    69. 

  69. 		return &dify.Adaptor{}
    70. 

  70. 	case constant.APITypeJina:
    71. 

  71. 		return &jina.Adaptor{}
    72. 

  72. 	case constant.APITypeCloudflare:
    73. 

  73. 		return &cloudflare.Adaptor{}
    74. 

  74. 	case constant.APITypeSiliconFlow:
    75. 

  75. 		return &siliconflow.Adaptor{}
    76. 

  76. 	case constant.APITypeVertexAi:
    77. 

  77. 		return &vertex.Adaptor{}
    78. 

  78. 	case constant.APITypeMistral:
    79. 

  79. 		return &mistral.Adaptor{}
    80. 

  80. 	case constant.APITypeDeepSeek:
    81. 

  81. 		return &deepseek.Adaptor{}
    82. 

  82. 	case constant.APITypeMokaAI:
    83. 

  83. 		return &mokaai.Adaptor{}
    84. 

  84. 	case constant.APITypeVolcEngine:
    85. 

  85. 		return &volcengine.Adaptor{}
    86. 

  86. 	case constant.APITypeBaiduV2:
    87. 

  87. 		return &baidu_v2.Adaptor{}
    88. 

  88. 	case constant.APITypeOpenRouter:
    89. 

  89. 		return &openai.Adaptor{}
    90. 

  90. 	case constant.APITypeXinference:
    91. 

  91. 		return &openai.Adaptor{}
    92. 

  92. 	case constant.APITypeXai:
    93. 

  93. 		return &xai.Adaptor{}
    94. 

  94. 	case constant.APITypeCoze:
    95. 

  95. 		return &coze.Adaptor{}
    96. 

  96. 	}
    97. 

  97. 	return nil
    98. 

  98. }
    99. 

  99. 

  100. func GetTaskAdaptor(platform commonconstant.TaskPlatform) channel.TaskAdaptor {
    101. 

  101. 	switch platform {
    102. 

  102. 	//case constant.APITypeAIProxyLibrary:
    103. 

  103. 	//	return &aiproxy.Adaptor{}
    104. 

  104. 	case commonconstant.TaskPlatformSuno:
    105. 

  105. 		return &suno.TaskAdaptor{}
    106. 

  106. 	case commonconstant.TaskPlatformKling:
    107. 

  107. 		return &kling.TaskAdaptor{}
    108. 

  108. 	case commonconstant.TaskPlatformJimeng:
    109. 

  109. 		return &jimeng.TaskAdaptor{}
    110. 

  110. 	}
    111. 

  111. 	return nil
    112. 

  112. }
    113.