Ecosystem/ai-gateway
5
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: remove rate limit for relay api

Browse Source
This commit is contained in:
JustSong 2023-04-26 21:50:09 +08:00
parent b9cc5dfa3f
commit fd149c242f
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/constants.go
View File

@ -65,7 +65,7 @@ var (
// All duration's unit is seconds // All duration's unit is seconds
// Shouldn't larger then RateLimitKeyExpirationDuration // Shouldn't larger then RateLimitKeyExpirationDuration
var ( var (
GlobalApiRateLimitNum = 60000 // TODO: temporary set to 60000 GlobalApiRateLimitNum = 180
GlobalApiRateLimitDuration int64 = 3 * 60 GlobalApiRateLimitDuration int64 = 3 * 60
GlobalWebRateLimitNum = 60 GlobalWebRateLimitNum = 60

4
router/relay-router.go
View File

@ -8,12 +8,12 @@ import (
func SetRelayRouter(router *gin.Engine) { func SetRelayRouter(router *gin.Engine) {
relayV1Router := router.Group("/v1") relayV1Router := router.Group("/v1")
relayV1Router.Use(middleware.GlobalAPIRateLimit(), middleware.TokenAuth(), middleware.Distribute()) relayV1Router.Use(middleware.TokenAuth(), middleware.Distribute())
{ {
relayV1Router.Any("/*path", controller.Relay) relayV1Router.Any("/*path", controller.Relay)
} }
relayDashboardRouter := router.Group("/dashboard") relayDashboardRouter := router.Group("/dashboard")
relayDashboardRouter.Use(middleware.GlobalAPIRateLimit(), middleware.TokenAuth(), middleware.Distribute()) relayDashboardRouter.Use(middleware.TokenAuth(), middleware.Distribute())
{ {
relayDashboardRouter.Any("/*path", controller.Relay) relayDashboardRouter.Any("/*path", controller.Relay)
} }