baalajimaestro/ollama
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

fix HEAD / request

Browse source 
HEAD request should respond like their GET counterparts except without a
response body.
This commit is contained in:
Michael Yang 2023-09-21 16:35:38 -07:00
parent 058d0cd04b
commit c986694367
1 changed files with 5 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
server/routes.go
View file

@ -543,12 +543,11 @@ func Serve(ln net.Listener, allowOrigins []string) error {
},
)
r.GET("/", func(c *gin.Context) {
for _, method := range []string{http.MethodGet, http.MethodHead} {
r.Handle(method, "/", func(c *gin.Context) {
c.String(http.StatusOK, "Ollama is running")
})
r.HEAD("/", func(c *gin.Context) {
c.Status(http.StatusOK)
})
}
r.POST("/api/pull", PullModelHandler)
r.POST("/api/generate", GenerateHandler)