feat: support stepfun's models

This commit is contained in:
JustSong 2024-04-05 12:32:05 +08:00
parent 0bb7db0b44
commit 5b41f57423
6 changed files with 21 additions and 0 deletions

View File

@ -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 模式**,可以通过流式传输实现打字机效果。

View File

@ -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 (

View File

@ -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
} }

View File

@ -0,0 +1,7 @@
package stepfun
var ModelList = []string{
"step-1-32k",
"step-1v-32k",
"step-1-200k",
}

View File

@ -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: '自定义渠道',

View File

@ -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' },