Merge pull request #312 from docker/alias-v1-engines-to-v1

Alias /engines/v1 to /v1

Author: ericcurtin

main.go

@@ -4,6 +4,7 @@ import (
 	"context"
 	"net"
 	"net/http"
+	"net/url"
 	"os"
 	"os/signal"
 	"path/filepath"
@@ -26,6 +27,27 @@ import (
 
 var log = logrus.New()
 
+// V1AliasHandler provides an alias from /v1/ to /engines/v1/ paths
+type V1AliasHandler struct {
+	scheduler http.Handler
+}
+
+func (h *V1AliasHandler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
+	// Modify the URL path to prepend /engines/ before /v1/
+	originalPath := r.URL.Path
+	newPath := inference.InferencePrefix + originalPath // originalPath is like "/v1/models", so result is "/engines/v1/models"
+
+	// Create a clone of the request with the modified path
+	r2 := new(http.Request)
+	*r2 = *r
+	r2.URL = new(url.URL)
+	*r2.URL = *r.URL
+	r2.URL.Path = newPath
+
+	// Pass the modified request to the scheduler
+	h.scheduler.ServeHTTP(w, r2)
+}
+
 func main() {
 	ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
 	defer cancel()
@@ -154,6 +176,8 @@ func main() {
 	router.Handle(inference.ModelsPrefix, modelManager)
 	router.Handle(inference.ModelsPrefix+"/", modelManager)
 	router.Handle(inference.InferencePrefix+"/", scheduler)
+	// Add /v1 as an alias for /engines/v1
+	router.Handle("/v1/", &V1AliasHandler{scheduler: scheduler})
 
 	// Add metrics endpoint if enabled
 	if os.Getenv("DISABLE_METRICS") != "1" {