xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-16 21:23:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ feat: Support channel setting chat only

Browse Source
This commit is contained in:
MartialBE
2024-05-15 17:51:49 +08:00
parent be874c330f
commit 2bc2dfca39
6 changed files with 70 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
relay/chat.go
View File

@@ -32,6 +32,10 @@ func (r *relayChat) setRequest() error {
return errors.New("max_tokens is invalid")
}
if r.chatRequest.Tools != nil {
r.c.Set("skip_only_chat", true)
}
r.originalModel = r.chatRequest.Model
return nil

11
relay/common.go
View File

@@ -101,7 +101,16 @@ func fetchChannelById(channelId int) (*model.Channel, error) {
func fetchChannelByModel(c *gin.Context, modelName string) (*model.Channel, error) {
group := c.GetString("group")
skip_channel_id := c.GetInt("skip_channel_id")
channel, err := model.ChannelGroup.Next(group, modelName, skip_channel_id)
skip_only_chat := c.GetBool("skip_only_chat")
var filters []model.ChannelsFilterFunc
if skip_only_chat {
filters = append(filters, model.FilterOnlyChat())
}
if skip_channel_id > 0 {
filters = append(filters, model.FilterChannelId(skip_channel_id))
}
channel, err := model.ChannelGroup.Next(group, modelName, filters...)
if err != nil {
message := fmt.Sprintf("当前分组 %s 下对于模型 %s 无可用渠道", group, modelName)
if channel != nil {