Handle unauthorized error

ekcasey · ekcasey · commit e6bb21191383 · 2025-04-17T17:52:34.000-04:00
Signed-off-by: Emily Casey &lt;emily.casey@docker.com&gt;
diff --git a/pkg/inference/models/manager.go b/pkg/inference/models/manager.go
@@ -368,6 +368,11 @@ func (m *Manager) handlePushModel(w http.ResponseWriter, r *http.Request, model
 			http.Error(w, "Model not found", http.StatusNotFound)
 			return
 		}
+		if errors.Is(err, distribution.ErrUnauthorized) {
+			m.log.Warnf("Failed to push model %q: %v", model, err)
+			http.Error(w, "Unauthorized", http.StatusUnauthorized)
+			return
+		}
 		http.Error(w, err.Error(), http.StatusInternalServerError)
 		return
 	}

Original file line number	Diff line number	Diff line change
`@@ -368,6 +368,11 @@ func (m Manager) handlePushModel(w http.ResponseWriter, r http.Request, model`
`368`	`368`	`http.Error(w, "Model not found", http.StatusNotFound)`
`369`	`369`	`return`
`370`	`370`	`}`
	`371`	`+ if errors.Is(err, distribution.ErrUnauthorized) {`
	`372`	`+ m.log.Warnf("Failed to push model %q: %v", model, err)`
	`373`	`+ http.Error(w, "Unauthorized", http.StatusUnauthorized)`
	`374`	`+ return`
	`375`	`+ }`
`371`	`376`	`http.Error(w, err.Error(), http.StatusInternalServerError)`
`372`	`377`	`return`
`373`	`378`	`}`