feat: support stepfun's models
This commit is contained in:
parent
0bb7db0b44
commit
5b41f57423
@ -81,6 +81,7 @@ _✨ 通过标准的 OpenAI API 格式访问所有的大模型,开箱即用
|
|||||||
+ [x] [Groq](https://wow.groq.com/)
|
+ [x] [Groq](https://wow.groq.com/)
|
||||||
+ [x] [Ollama](https://github.com/ollama/ollama)
|
+ [x] [Ollama](https://github.com/ollama/ollama)
|
||||||
+ [x] [零一万物](https://platform.lingyiwanwu.com/)
|
+ [x] [零一万物](https://platform.lingyiwanwu.com/)
|
||||||
|
+ [x] [阶跃星辰](https://platform.stepfun.com/)
|
||||||
2. 支持配置镜像以及众多[第三方代理服务](https://iamazing.cn/page/openai-api-third-party-services)。
|
2. 支持配置镜像以及众多[第三方代理服务](https://iamazing.cn/page/openai-api-third-party-services)。
|
||||||
3. 支持通过**负载均衡**的方式访问多个渠道。
|
3. 支持通过**负载均衡**的方式访问多个渠道。
|
||||||
4. 支持 **stream 模式**,可以通过流式传输实现打字机效果。
|
4. 支持 **stream 模式**,可以通过流式传输实现打字机效果。
|
||||||
|
@ -71,6 +71,7 @@ const (
|
|||||||
ChannelTypeGroq
|
ChannelTypeGroq
|
||||||
ChannelTypeOllama
|
ChannelTypeOllama
|
||||||
ChannelTypeLingYiWanWu
|
ChannelTypeLingYiWanWu
|
||||||
|
ChannelTypeStepFun
|
||||||
|
|
||||||
ChannelTypeDummy
|
ChannelTypeDummy
|
||||||
)
|
)
|
||||||
@ -108,6 +109,7 @@ var ChannelBaseURLs = []string{
|
|||||||
"https://api.groq.com/openai", // 29
|
"https://api.groq.com/openai", // 29
|
||||||
"http://localhost:11434", // 30
|
"http://localhost:11434", // 30
|
||||||
"https://api.lingyiwanwu.com", // 31
|
"https://api.lingyiwanwu.com", // 31
|
||||||
|
"https://api.stepfun.com", // 32
|
||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
|
@ -9,6 +9,7 @@ import (
|
|||||||
"github.com/songquanpeng/one-api/relay/channel/minimax"
|
"github.com/songquanpeng/one-api/relay/channel/minimax"
|
||||||
"github.com/songquanpeng/one-api/relay/channel/mistral"
|
"github.com/songquanpeng/one-api/relay/channel/mistral"
|
||||||
"github.com/songquanpeng/one-api/relay/channel/moonshot"
|
"github.com/songquanpeng/one-api/relay/channel/moonshot"
|
||||||
|
"github.com/songquanpeng/one-api/relay/channel/stepfun"
|
||||||
)
|
)
|
||||||
|
|
||||||
var CompatibleChannels = []int{
|
var CompatibleChannels = []int{
|
||||||
@ -20,6 +21,7 @@ var CompatibleChannels = []int{
|
|||||||
common.ChannelTypeMistral,
|
common.ChannelTypeMistral,
|
||||||
common.ChannelTypeGroq,
|
common.ChannelTypeGroq,
|
||||||
common.ChannelTypeLingYiWanWu,
|
common.ChannelTypeLingYiWanWu,
|
||||||
|
common.ChannelTypeStepFun,
|
||||||
}
|
}
|
||||||
|
|
||||||
func GetCompatibleChannelMeta(channelType int) (string, []string) {
|
func GetCompatibleChannelMeta(channelType int) (string, []string) {
|
||||||
@ -40,6 +42,8 @@ func GetCompatibleChannelMeta(channelType int) (string, []string) {
|
|||||||
return "groq", groq.ModelList
|
return "groq", groq.ModelList
|
||||||
case common.ChannelTypeLingYiWanWu:
|
case common.ChannelTypeLingYiWanWu:
|
||||||
return "lingyiwanwu", lingyiwanwu.ModelList
|
return "lingyiwanwu", lingyiwanwu.ModelList
|
||||||
|
case common.ChannelTypeStepFun:
|
||||||
|
return "stepfun", stepfun.ModelList
|
||||||
default:
|
default:
|
||||||
return "openai", ModelList
|
return "openai", ModelList
|
||||||
}
|
}
|
||||||
|
7
relay/channel/stepfun/constants.go
Normal file
7
relay/channel/stepfun/constants.go
Normal file
@ -0,0 +1,7 @@
|
|||||||
|
package stepfun
|
||||||
|
|
||||||
|
var ModelList = []string{
|
||||||
|
"step-1-32k",
|
||||||
|
"step-1v-32k",
|
||||||
|
"step-1-200k",
|
||||||
|
}
|
@ -107,6 +107,12 @@ export const CHANNEL_OPTIONS = {
|
|||||||
value: 31,
|
value: 31,
|
||||||
color: 'primary'
|
color: 'primary'
|
||||||
},
|
},
|
||||||
|
32: {
|
||||||
|
key: 32,
|
||||||
|
text: '阶跃星辰',
|
||||||
|
value: 32,
|
||||||
|
color: 'primary'
|
||||||
|
},
|
||||||
8: {
|
8: {
|
||||||
key: 8,
|
key: 8,
|
||||||
text: '自定义渠道',
|
text: '自定义渠道',
|
||||||
|
@ -17,6 +17,7 @@ export const CHANNEL_OPTIONS = [
|
|||||||
{ key: 29, text: 'Groq', value: 29, color: 'orange' },
|
{ key: 29, text: 'Groq', value: 29, color: 'orange' },
|
||||||
{ key: 30, text: 'Ollama', value: 30, color: 'black' },
|
{ key: 30, text: 'Ollama', value: 30, color: 'black' },
|
||||||
{ key: 31, text: '零一万物', value: 31, color: 'green' },
|
{ key: 31, text: '零一万物', value: 31, color: 'green' },
|
||||||
|
{ key: 31, text: '阶跃星辰', value: 32, color: 'blue' },
|
||||||
{ key: 8, text: '自定义渠道', value: 8, color: 'pink' },
|
{ key: 8, text: '自定义渠道', value: 8, color: 'pink' },
|
||||||
{ key: 22, text: '知识库:FastGPT', value: 22, color: 'blue' },
|
{ key: 22, text: '知识库:FastGPT', value: 22, color: 'blue' },
|
||||||
{ key: 21, text: '知识库:AI Proxy', value: 21, color: 'purple' },
|
{ key: 21, text: '知识库:AI Proxy', value: 21, color: 'purple' },
|
||||||
|
Loading…
Reference in New Issue
Block a user