Ver Fonte

fix: api type error

CaIon há 1 ano atrás
pai
commit
4c4e087060
2 ficheiros alterados com 9 adições e 0 exclusões
  1. 3 0
      relay/constant/api_type.go
  2. 6 0
      relay/relay_adaptor.go

+ 3 - 0
relay/constant/api_type.go

@@ -16,6 +16,7 @@ const (
 	APITypeTencent
 	APITypeGemini
 	APITypeZhipu_v4
+	APITypeOllama
 	APITypePerplexity
 
 	APITypeDummy // this one is only for count, do not add any channel after this
@@ -44,6 +45,8 @@ func ChannelType2APIType(channelType int) int {
 		apiType = APITypeGemini
 	case common.ChannelTypeZhipu_v4:
 		apiType = APITypeZhipu_v4
+	case common.ChannelTypeOllama:
+		apiType = APITypeOllama
 	case common.ChannelTypePerplexity:
 		apiType = APITypePerplexity
 	}

+ 6 - 0
relay/relay_adaptor.go

@@ -6,8 +6,10 @@ import (
 	"one-api/relay/channel/baidu"
 	"one-api/relay/channel/claude"
 	"one-api/relay/channel/gemini"
+	"one-api/relay/channel/ollama"
 	"one-api/relay/channel/openai"
 	"one-api/relay/channel/palm"
+	"one-api/relay/channel/perplexity"
 	"one-api/relay/channel/tencent"
 	"one-api/relay/channel/xunfei"
 	"one-api/relay/channel/zhipu"
@@ -39,6 +41,10 @@ func GetAdaptor(apiType int) channel.Adaptor {
 		return &zhipu.Adaptor{}
 	case constant.APITypeZhipu_v4:
 		return &zhipu_4v.Adaptor{}
+	case constant.APITypeOllama:
+		return &ollama.Adaptor{}
+	case constant.APITypePerplexity:
+		return &perplexity.Adaptor{}
 	}
 	return nil
 }