Skip to content

phsl / new-api

Go to a project
Commit 9a6be430 by JustSong
Options

fix: use channel type to determine api type (close #321)

parent 30cb385b
Showing with 5 additions and 4 deletions
controller/relay-text.go
...@@ -85,13 +85,14 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode { ...@@ -85,13 +85,14 @@ func relayTextHelper(c *gin.Context, relayMode int) *OpenAIErrorWithStatusCode {
} }
} }
apiType := APITypeOpenAI apiType := APITypeOpenAI
if strings.HasPrefix(textRequest.Model, "claude") { switch channelType {
case common.ChannelTypeAnthropic:
apiType = APITypeClaude apiType = APITypeClaude
} else if strings.HasPrefix(textRequest.Model, "ERNIE") { case common.ChannelTypeBaidu:
apiType = APITypeBaidu apiType = APITypeBaidu
} else if strings.HasPrefix(textRequest.Model, "PaLM") { case common.ChannelTypePaLM:
apiType = APITypePaLM apiType = APITypePaLM
} else if strings.HasPrefix(textRequest.Model, "chatglm_") { case common.ChannelTypeZhipu:
apiType = APITypeZhipu apiType = APITypeZhipu
} }
baseURL := common.ChannelBaseURLs[channelType] baseURL := common.ChannelBaseURLs[channelType]
......
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