From c98669436742f79f8ffaa796ff94ee2c7f17201a Mon Sep 17 00:00:00 2001 From: Michael Yang Date: Thu, 21 Sep 2023 16:35:38 -0700 Subject: [PATCH] fix HEAD / request HEAD request should respond like their GET counterparts except without a response body. --- server/routes.go | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/server/routes.go b/server/routes.go index c00bad65..f86ffd07 100644 --- a/server/routes.go +++ b/server/routes.go @@ -543,12 +543,11 @@ func Serve(ln net.Listener, allowOrigins []string) error { }, ) - r.GET("/", func(c *gin.Context) { - c.String(http.StatusOK, "Ollama is running") - }) - r.HEAD("/", func(c *gin.Context) { - c.Status(http.StatusOK) - }) + for _, method := range []string{http.MethodGet, http.MethodHead} { + r.Handle(method, "/", func(c *gin.Context) { + c.String(http.StatusOK, "Ollama is running") + }) + } r.POST("/api/pull", PullModelHandler) r.POST("/api/generate", GenerateHandler)