third-party-mirrors/ollama
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

server: add "Cache-Control: max-age=0" response header

Browse Source
This commit is contained in:
Justin Cranford 2024-09-29 04:11:13 -04:00
parent 03608cb46e
commit 5f4ebeba95
1 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
server/routes.go
View File

@ -1092,6 +1092,13 @@ func allowedHostsMiddleware(addr net.Addr) gin.HandlerFunc {
} }
} }
func CacheControlMiddleware() gin.HandlerFunc {
return func(c *gin.Context) {
c.Header("Cache-Control", "max-age=0")
c.Next()
}
}
func (s *Server) GenerateRoutes() http.Handler { func (s *Server) GenerateRoutes() http.Handler {
config := cors.DefaultConfig() config := cors.DefaultConfig()
config.AllowWildcard = true config.AllowWildcard = true
@ -1105,6 +1112,7 @@ func (s *Server) GenerateRoutes() http.Handler {
r := gin.Default() r := gin.Default()
r.Use( r.Use(
CacheControlMiddleware(),
cors.New(config), cors.New(config),
allowedHostsMiddleware(s.addr), allowedHostsMiddleware(s.addr),
) )