fix: Change the implementation of retry from 307 redirect to Retry middleware

This commit is contained in:
AhhhLiu
2023-11-15 22:20:33 +08:00
parent 58bb3ab6f6
commit 3a443f0e47
6 changed files with 123 additions and 16 deletions

View File

@@ -17,6 +17,7 @@ func SetRelayRouter(router *gin.Engine) {
modelsRouter.GET("/:model", controller.RetrieveModel)
}
relayV1Router := router.Group("/v1")
relayV1Router.Use(middleware.RetryHandler(relayV1Router))
relayV1Router.Use(middleware.TokenAuth(), middleware.Distribute())
{
relayV1Router.POST("/completions", controller.Relay)