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 42d50c06a3
6 changed files with 142 additions and 24 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)