ChatGPT/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-10-14 06:20:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

Support custom channel now

Browse Source
This commit is contained in:
JustSong
2023-04-23 20:35:49 +08:00
parent 57379d8e92
commit b3be4d8f85
8 changed files with 55 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
middleware/distributor.go
View File

@@ -63,6 +63,9 @@ func Distribute() func(c *gin.Context) {
}
c.Set("channel", channel.Type)
c.Request.Header.Set("Authorization", fmt.Sprintf("Bearer %s", channel.Key))
if channel.Type == common.ChannelTypeCustom {
c.Set("base_url", channel.BaseURL)
}
c.Next()
}
}