Skip to content

phsl / new-api

Go to a project
Commit af2389ca by CaIon
Options

fix: api type error

parent 62672090
Showing with 9 additions and 0 deletions
relay/constant/api_type.go
...@@ -16,6 +16,7 @@ const ( ...@@ -16,6 +16,7 @@ const (
APITypeTencent APITypeTencent
APITypeGemini APITypeGemini
APITypeZhipu_v4 APITypeZhipu_v4
APITypeOllama
APITypePerplexity APITypePerplexity
APITypeDummy // this one is only for count, do not add any channel after this APITypeDummy // this one is only for count, do not add any channel after this
...@@ -44,6 +45,8 @@ func ChannelType2APIType(channelType int) int { ...@@ -44,6 +45,8 @@ func ChannelType2APIType(channelType int) int {
apiType = APITypeGemini apiType = APITypeGemini
case common.ChannelTypeZhipu_v4: case common.ChannelTypeZhipu_v4:
apiType = APITypeZhipu_v4 apiType = APITypeZhipu_v4
case common.ChannelTypeOllama:
apiType = APITypeOllama
case common.ChannelTypePerplexity: case common.ChannelTypePerplexity:
apiType = APITypePerplexity apiType = APITypePerplexity
} }
......
relay/relay_adaptor.go
...@@ -6,8 +6,10 @@ import ( ...@@ -6,8 +6,10 @@ import (
"one-api/relay/channel/baidu" "one-api/relay/channel/baidu"
"one-api/relay/channel/claude" "one-api/relay/channel/claude"
"one-api/relay/channel/gemini" "one-api/relay/channel/gemini"
"one-api/relay/channel/ollama"
"one-api/relay/channel/openai" "one-api/relay/channel/openai"
"one-api/relay/channel/palm" "one-api/relay/channel/palm"
"one-api/relay/channel/perplexity"
"one-api/relay/channel/tencent" "one-api/relay/channel/tencent"
"one-api/relay/channel/xunfei" "one-api/relay/channel/xunfei"
"one-api/relay/channel/zhipu" "one-api/relay/channel/zhipu"
...@@ -39,6 +41,10 @@ func GetAdaptor(apiType int) channel.Adaptor { ...@@ -39,6 +41,10 @@ func GetAdaptor(apiType int) channel.Adaptor {
return &zhipu.Adaptor{} return &zhipu.Adaptor{}
case constant.APITypeZhipu_v4: case constant.APITypeZhipu_v4:
return &zhipu_4v.Adaptor{} return &zhipu_4v.Adaptor{}
case constant.APITypeOllama:
return &ollama.Adaptor{}
case constant.APITypePerplexity:
return &perplexity.Adaptor{}
} }
return nil return nil
} }
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment