From 765994362cf2860b9d137d113f23590ba8b2cf16 Mon Sep 17 00:00:00 2001
From: Bruce MacDonald <brucewmacdonald@gmail.com>
Date: Tue, 1 Aug 2023 14:50:38 -0400
Subject: [PATCH] use head to check heartbeat

---
 api/client.go    | 2 +-
 server/routes.go | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/api/client.go b/api/client.go
index ed1053e8..7753eb91 100644
--- a/api/client.go
+++ b/api/client.go
@@ -225,7 +225,7 @@ func (c *Client) Delete(ctx context.Context, req *DeleteRequest) error {
 }
 
 func (c *Client) Heartbeat(ctx context.Context) error {
-	if err := c.do(ctx, http.MethodGet, "/", nil, nil); err != nil {
+	if err := c.do(ctx, http.MethodHead, "/", nil, nil); err != nil {
 		return err
 	}
 	return nil
diff --git a/server/routes.go b/server/routes.go
index cc9df958..108c0743 100644
--- a/server/routes.go
+++ b/server/routes.go
@@ -319,6 +319,9 @@ func Serve(ln net.Listener) error {
 	r.GET("/", 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)