third-party-mirrors/ollama
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

registry: use exact match on path

Browse Source
This commit is contained in:
Blake Mizerany 2024-03-31 15:01:26 -07:00
parent 60ef0e6b4a
commit c0eddb10fd
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
registry/server.go
View File

@ -8,7 +8,6 @@ import (
"errors"
"log"
"net/http"
"strings"
"time"
"bllamo.com/client/ollama"
@ -35,13 +34,14 @@ func (s *Server) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
func (s *Server) serveHTTP(w http.ResponseWriter, r *http.Request) error {
switch {
case strings.HasPrefix(r.URL.Path, "/v1/push"):
switch r.URL.Path {
case "/v1/push":
return s.handlePush(w, r)
case strings.HasPrefix(r.URL.Path, "/v1/pull"):
case "/v1/pull":
return s.handlePull(w, r)
default:
return oweb.ErrNotFound
}
return oweb.ErrNotFound
}
func (s *Server) handlePush(w http.ResponseWriter, r *http.Request) error {