Provide /healthz

google-gemini · Sep 9, 2024 · c821b12 · c821b12
1 parent c806a2c
commit c821b12
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/cmd/proxy-to-gemini/main.go b/cmd/proxy-to-gemini/main.go
@@ -53,13 +53,16 @@ func main() {
 	defer client.Close()
 
 	r := mux.NewRouter()
-	r.HandleFunc("/", indexHandler)
+	r.HandleFunc("/healthz", func(w http.ResponseWriter, r *http.Request) {
+		fmt.Fprint(w, "ok")
+	})
 	switch api {
 	case "openai":
 		openai.RegisterHandlers(r, client)
 	case "ollama":
 		ollama.RegisterHandlers(r, client)
 	}
+	r.HandleFunc("/", indexHandler)
 
 	log.Printf("Starting server on %v", hostport)
 	if err := http.ListenAndServe(hostport, r); err != nil {