Skip to content

Commit 734c965

Browse files
authored
fix: remove add channel split key (#619)
1 parent d44d862 commit 734c965

3 files changed

Lines changed: 19 additions & 37 deletions

File tree

core/controller/channel.go

Lines changed: 4 additions & 35 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@ import (
66
"net/http"
77
"slices"
88
"strconv"
9-
"strings"
109
"time"
1110

1211
"github.com/bytedance/sonic"
@@ -166,13 +165,13 @@ func AddChannels(c *gin.Context) {
166165

167166
_channels := make([]*model.Channel, 0, len(channels))
168167
for _, channel := range channels {
169-
channels, err := channel.ToChannels()
168+
channel, err := channel.ToChannel()
170169
if err != nil {
171170
middleware.ErrorResponse(c, http.StatusBadRequest, err.Error())
172171
return
173172
}
174173

175-
_channels = append(_channels, channels...)
174+
_channels = append(_channels, channel)
176175
}
177176

178177
err = model.BatchInsertChannels(_channels)
@@ -332,36 +331,6 @@ func (r *AddChannelRequest) ToChannel() (*model.Channel, error) {
332331
}, nil
333332
}
334333

335-
func (r *AddChannelRequest) ToChannels() ([]*model.Channel, error) {
336-
keys := strings.Split(r.Key, "\n")
337-
338-
channels := make([]*model.Channel, 0, len(keys))
339-
for _, key := range keys {
340-
if key == "" {
341-
continue
342-
}
343-
344-
c, err := r.ToChannel()
345-
if err != nil {
346-
return nil, err
347-
}
348-
349-
c.Key = key
350-
channels = append(channels, c)
351-
}
352-
353-
if len(channels) == 0 {
354-
ch, err := r.ToChannel()
355-
if err != nil {
356-
return nil, err
357-
}
358-
359-
return []*model.Channel{ch}, nil
360-
}
361-
362-
return channels, nil
363-
}
364-
365334
// AddChannel godoc
366335
//
367336
// @Summary Add a single channel
@@ -382,13 +351,13 @@ func AddChannel(c *gin.Context) {
382351
return
383352
}
384353

385-
channels, err := channel.ToChannels()
354+
ch, err := channel.ToChannel()
386355
if err != nil {
387356
middleware.ErrorResponse(c, http.StatusBadRequest, err.Error())
388357
return
389358
}
390359

391-
err = model.BatchInsertChannels(channels)
360+
err = model.BatchInsertChannels([]*model.Channel{ch})
392361
if err != nil {
393362
middleware.ErrorResponse(c, http.StatusInternalServerError, err.Error())
394363
return

core/controller/channel_test.go

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,19 @@ import (
1212
"github.com/stretchr/testify/require"
1313
)
1414

15+
func TestAddChannelRequestToChannelPreservesNewlinesInKey(t *testing.T) {
16+
const key = "first-key\nsecond-key"
17+
18+
channel, err := (&AddChannelRequest{
19+
Type: model.ChannelTypeOpenAI,
20+
Name: "channel",
21+
Key: key,
22+
}).ToChannel()
23+
24+
require.NoError(t, err)
25+
require.Equal(t, key, channel.Key)
26+
}
27+
1528
func TestRunAutoTestBannedModelsHonorsConcurrencyLimit(t *testing.T) {
1629
const (
1730
concurrency = 7

core/controller/import.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,12 +154,12 @@ func AddOneAPIChannel(ch OneAPIChannel) error {
154154
add.BaseURL += "/v1"
155155
}
156156

157-
chs, err := add.ToChannels()
157+
channel, err := add.ToChannel()
158158
if err != nil {
159159
return err
160160
}
161161

162-
return model.BatchInsertChannels(chs)
162+
return model.BatchInsertChannels([]*model.Channel{channel})
163163
}
164164

165165
// ImportChannelFromOneAPI godoc

0 commit comments

Comments
 (0)