xinyin025/new-api
1
0
Fork 0
mirror of https://github.com/linux-do/new-api.git synced 2025-11-14 01:43:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #137 from MapleEve/main

Browse Source 
feat: support 01.AI
This commit is contained in:
Calcium-Ion
2024-03-23 17:38:53 +08:00
committed by GitHub
parent 67332bc8df a825699e9a
commit a500097b36
8 changed files with 39 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
relay/channel/lingyiwanwu/constrants.go Normal file
View File

@@ -0,0 +1,9 @@
package lingyiwanwu
// https://platform.lingyiwanwu.com/docs
var ModelList = []string{
"yi-34b-chat-0205",
"yi-34b-chat-200k",
"yi-vl-plus",
}

3
relay/channel/openai/adaptor.go
View File

@@ -11,6 +11,7 @@ import (
"one-api/relay/channel"
"one-api/relay/channel/ai360"
"one-api/relay/channel/moonshot"
"one-api/relay/channel/lingyiwanwu"
relaycommon "one-api/relay/common"
"one-api/service"
"strings"
@@ -88,6 +89,8 @@ func (a *Adaptor) GetModelList() []string {
return ai360.ModelList
case common.ChannelTypeMoonshot:
return moonshot.ModelList
case common.ChannelTypeLingYiWanWu:
return lingyiwanwu.ModelList
default:
return ModelList
}