Inicio Explorar Ayuda
Iniciar sesión
Server
/
new-api
40
0
Fork 0
Archivos Incidencias 0 Pull Requests 0 Wiki
Árbol: 126f04e08f
Ramas Etiquetas
new-api
/
relay
/
constant
/
api_type.go

api_type.go 2.0 KB
Histórico Raw

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 
  1. package constant
    2. 

  2. 

  3. import (
    4. 

  4. 	"one-api/common"
    5. 

  5. )
    6. 

  6. 

  7. const (
    8. 

  8. 	APITypeOpenAI = iota
    9. 

  9. 	APITypeAnthropic
    10. 

  10. 	APITypePaLM
    11. 

  11. 	APITypeBaidu
    12. 

  12. 	APITypeZhipu
    13. 

  13. 	APITypeAli
    14. 

  14. 	APITypeXunfei
    15. 

  15. 	APITypeAIProxyLibrary
    16. 

  16. 	APITypeTencent
    17. 

  17. 	APITypeGemini
    18. 

  18. 	APITypeZhipuV4
    19. 

  19. 	APITypeOllama
    20. 

  20. 	APITypePerplexity
    21. 

  21. 	APITypeAws
    22. 

  22. 	APITypeCohere
    23. 

  23. 	APITypeDify
    24. 

  24. 	APITypeJina
    25. 

  25. 	APITypeCloudflare
    26. 

  26. 	APITypeSiliconFlow
    27. 

  27. 	APITypeVertexAi
    28. 

  28. 	APITypeMistral
    29. 

  29. 	APITypeDeepSeek
    30. 

  30. 	APITypeMokaAI
    31. 

  31. 	APITypeDummy // this one is only for count, do not add any channel after this
    32. 

  32. )
    33. 

  33. 

  34. func ChannelType2APIType(channelType int) (int, bool) {
    35. 

  35. 	apiType := -1
    36. 

  36. 	switch channelType {
    37. 

  37. 	case common.ChannelTypeOpenAI:
    38. 

  38. 		apiType = APITypeOpenAI
    39. 

  39. 	case common.ChannelTypeAnthropic:
    40. 

  40. 		apiType = APITypeAnthropic
    41. 

  41. 	case common.ChannelTypeBaidu:
    42. 

  42. 		apiType = APITypeBaidu
    43. 

  43. 	case common.ChannelTypePaLM:
    44. 

  44. 		apiType = APITypePaLM
    45. 

  45. 	case common.ChannelTypeZhipu:
    46. 

  46. 		apiType = APITypeZhipu
    47. 

  47. 	case common.ChannelTypeAli:
    48. 

  48. 		apiType = APITypeAli
    49. 

  49. 	case common.ChannelTypeXunfei:
    50. 

  50. 		apiType = APITypeXunfei
    51. 

  51. 	case common.ChannelTypeAIProxyLibrary:
    52. 

  52. 		apiType = APITypeAIProxyLibrary
    53. 

  53. 	case common.ChannelTypeTencent:
    54. 

  54. 		apiType = APITypeTencent
    55. 

  55. 	case common.ChannelTypeGemini:
    56. 

  56. 		apiType = APITypeGemini
    57. 

  57. 	case common.ChannelTypeZhipu_v4:
    58. 

  58. 		apiType = APITypeZhipuV4
    59. 

  59. 	case common.ChannelTypeOllama:
    60. 

  60. 		apiType = APITypeOllama
    61. 

  61. 	case common.ChannelTypePerplexity:
    62. 

  62. 		apiType = APITypePerplexity
    63. 

  63. 	case common.ChannelTypeAws:
    64. 

  64. 		apiType = APITypeAws
    65. 

  65. 	case common.ChannelTypeCohere:
    66. 

  66. 		apiType = APITypeCohere
    67. 

  67. 	case common.ChannelTypeDify:
    68. 

  68. 		apiType = APITypeDify
    69. 

  69. 	case common.ChannelTypeJina:
    70. 

  70. 		apiType = APITypeJina
    71. 

  71. 	case common.ChannelCloudflare:
    72. 

  72. 		apiType = APITypeCloudflare
    73. 

  73. 	case common.ChannelTypeSiliconFlow:
    74. 

  74. 		apiType = APITypeSiliconFlow
    75. 

  75. 	case common.ChannelTypeVertexAi:
    76. 

  76. 		apiType = APITypeVertexAi
    77. 

  77. 	case common.ChannelTypeMistral:
    78. 

  78. 		apiType = APITypeMistral
    79. 

  79. 	case common.ChannelTypeDeepSeek:
    80. 

  80. 		apiType = APITypeDeepSeek
    81. 

  81. 	case common.ChannelTypeMokaAI:
    82. 

  82. 		apiType = APITypeMokaAI
    83. 

  83. 	}
    84. 

  84. 	if apiType == -1 {
    85. 

  85. 		return APITypeOpenAI, false
    86. 

  86. 	}
    87. 

  87. 	return apiType, true
    88. 

  88. }
    89.