Strona główna Odkrywaj Pomoc
Zaloguj się
Server
/
new-api
40
0
Forkuj 0
Pliki Problemy 0 Oczekujące zmiany 0 Wiki
Drzewo: 8af4e28f75
Gałęzie Tagi
new-api
/
relay
/
relay_adaptor.go

relay_adaptor.go 1.8 KB
Historia Czysty

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071 
  1. package relay
    2. 

  2. 

  3. import (
    4. 

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

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

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

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

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

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

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

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

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

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

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

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

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

  17. 	"one-api/relay/channel/task/suno"
    18. 

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

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

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

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

  22. 	"one-api/relay/constant"
    23. 

  23. )
    24. 

  24. 

  25. func GetAdaptor(apiType int) channel.Adaptor {
    26. 

  26. 	switch apiType {
    27. 

  27. 	//case constant.APITypeAIProxyLibrary:
    28. 

  28. 	//	return &aiproxy.Adaptor{}
    29. 

  29. 	case constant.APITypeAli:
    30. 

  30. 		return &ali.Adaptor{}
    31. 

  31. 	case constant.APITypeAnthropic:
    32. 

  32. 		return &claude.Adaptor{}
    33. 

  33. 	case constant.APITypeBaidu:
    34. 

  34. 		return &baidu.Adaptor{}
    35. 

  35. 	case constant.APITypeGemini:
    36. 

  36. 		return &gemini.Adaptor{}
    37. 

  37. 	case constant.APITypeOpenAI:
    38. 

  38. 		return &openai.Adaptor{}
    39. 

  39. 	case constant.APITypePaLM:
    40. 

  40. 		return &palm.Adaptor{}
    41. 

  41. 	case constant.APITypeTencent:
    42. 

  42. 		return &tencent.Adaptor{}
    43. 

  43. 	case constant.APITypeXunfei:
    44. 

  44. 		return &xunfei.Adaptor{}
    45. 

  45. 	case constant.APITypeZhipu:
    46. 

  46. 		return &zhipu.Adaptor{}
    47. 

  47. 	case constant.APITypeZhipuV4:
    48. 

  48. 		return &zhipu_4v.Adaptor{}
    49. 

  49. 	case constant.APITypeOllama:
    50. 

  50. 		return &ollama.Adaptor{}
    51. 

  51. 	case constant.APITypePerplexity:
    52. 

  52. 		return &perplexity.Adaptor{}
    53. 

  53. 	case constant.APITypeAws:
    54. 

  54. 		return &aws.Adaptor{}
    55. 

  55. 	case constant.APITypeCohere:
    56. 

  56. 		return &cohere.Adaptor{}
    57. 

  57. 	case constant.APITypeDify:
    58. 

  58. 		return &dify.Adaptor{}
    59. 

  59. 	}
    60. 

  60. 	return nil
    61. 

  61. }
    62. 

  62. 

  63. func GetTaskAdaptor(platform commonconstant.TaskPlatform) channel.TaskAdaptor {
    64. 

  64. 	switch platform {
    65. 

  65. 	//case constant.APITypeAIProxyLibrary:
    66. 

  66. 	//	return &aiproxy.Adaptor{}
    67. 

  67. 	case commonconstant.TaskPlatformSuno:
    68. 

  68. 		return &suno.TaskAdaptor{}
    69. 

  69. 	}
    70. 

  70. 	return nil
    71. 

  71. }
    72.