fix: fix gemini panic

This commit is contained in:
CaIon committed 2024-01-07 21:27:28 +08:00
1 parent 171315e674
commit d9107379e9
3 files changed
+35 -1

No files matched your search

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