Unverified Commit 631cfd9e authored by Blake Mizerany's avatar Blake Mizerany Committed by GitHub
Browse files

types/model: remove knowledge of digest (#5500)

This was leading to ambiguity and confusion in ollama.com, and is not
used anywhere in ollama at the moment. Once manifests are addressable by
digest, we can add this back in, and in a way that is more tailored to
the concept of addressing a manifest by digest.
parent 78fb33dd
...@@ -91,7 +91,6 @@ type Name struct { ...@@ -91,7 +91,6 @@ type Name struct {
Namespace string Namespace string
Model string Model string
Tag string Tag string
RawDigest string
} }
// ParseName parses and assembles a Name from a name string. The // ParseName parses and assembles a Name from a name string. The
...@@ -143,11 +142,6 @@ func ParseNameBare(s string) Name { ...@@ -143,11 +142,6 @@ func ParseNameBare(s string) Name {
var n Name var n Name
var promised bool var promised bool
s, n.RawDigest, promised = cutLast(s, "@")
if promised && n.RawDigest == "" {
n.RawDigest = MissingPart
}
// "/" is an illegal tag character, so we can use it to split the host // "/" is an illegal tag character, so we can use it to split the host
if strings.LastIndex(s, ":") > strings.LastIndex(s, "/") { if strings.LastIndex(s, ":") > strings.LastIndex(s, "/") {
s, n.Tag, _ = cutPromised(s, ":") s, n.Tag, _ = cutPromised(s, ":")
...@@ -222,10 +216,6 @@ func (n Name) String() string { ...@@ -222,10 +216,6 @@ func (n Name) String() string {
b.WriteByte(':') b.WriteByte(':')
b.WriteString(n.Tag) b.WriteString(n.Tag)
} }
if n.RawDigest != "" {
b.WriteByte('@')
b.WriteString(n.RawDigest)
}
return b.String() return b.String()
} }
...@@ -250,16 +240,18 @@ func (n Name) DisplayShortest() string { ...@@ -250,16 +240,18 @@ func (n Name) DisplayShortest() string {
return sb.String() return sb.String()
} }
func IsValidNamespace(namespace string) bool { // IsValidNamespace reports whether the provided string is a valid
return isValidPart(kindNamespace, namespace) // namespace.
func IsValidNamespace(s string) bool {
return isValidPart(kindNamespace, s)
} }
// IsValid reports whether all parts of the name are present and valid. The // IsValid reports whether all parts of the name are present and valid. The
// digest is a special case, and is checked for validity only if present. // digest is a special case, and is checked for validity only if present.
//
// Note: The digest check has been removed as is planned to be added back in
// at a later time.
func (n Name) IsValid() bool { func (n Name) IsValid() bool {
if n.RawDigest != "" && !isValidPart(kindDigest, n.RawDigest) {
return false
}
return n.IsFullyQualified() return n.IsFullyQualified()
} }
......
...@@ -122,21 +122,6 @@ func TestParseNameParts(t *testing.T) { ...@@ -122,21 +122,6 @@ func TestParseNameParts(t *testing.T) {
}, },
wantFilepath: filepath.Join(part350, part80, part80, part80), wantFilepath: filepath.Join(part350, part80, part80, part80),
}, },
{
in: "@digest",
want: Name{
RawDigest: "digest",
},
wantValidDigest: false,
},
{
in: "model@sha256:123",
want: Name{
Model: "model",
RawDigest: "sha256:123",
},
wantValidDigest: true,
},
} }
for _, tt := range cases { for _, tt := range cases {
...@@ -160,22 +145,18 @@ var testCases = map[string]bool{ // name -> valid ...@@ -160,22 +145,18 @@ var testCases = map[string]bool{ // name -> valid
"_why/_the/_lucky:_stiff": true, "_why/_the/_lucky:_stiff": true,
// minimal // minimal
"h/n/m:t@d": true, "h/n/m:t": true,
"host/namespace/model:tag": true, "host/namespace/model:tag": true,
"host/namespace/model": false, "host/namespace/model": false,
"namespace/model": false, "namespace/model": false,
"model": false, "model": false,
"@sha256-1000000000000000000000000000000000000000000000000000000000000000": false,
"model@sha256-1000000000000000000000000000000000000000000000000000000000000000": false,
"model@sha256:1000000000000000000000000000000000000000000000000000000000000000": false,
// long (but valid) // long (but valid)
part80 + "/" + part80 + "/" + part80 + ":" + part80: true, part80 + "/" + part80 + "/" + part80 + ":" + part80: true,
part350 + "/" + part80 + "/" + part80 + ":" + part80: true, part350 + "/" + part80 + "/" + part80 + ":" + part80: true,
"h/nn/mm:t@sha256-1000000000000000000000000000000000000000000000000000000000000000": true, // bare minimum part sizes "h/nn/mm:t": true, // bare minimum part sizes
"h/nn/mm:t@sha256:1000000000000000000000000000000000000000000000000000000000000000": true, // bare minimum part sizes
// unqualified // unqualified
"m": false, "m": false,
...@@ -196,11 +177,10 @@ var testCases = map[string]bool{ // name -> valid ...@@ -196,11 +177,10 @@ var testCases = map[string]bool{ // name -> valid
"@": false, "@": false,
// not starting with alphanum // not starting with alphanum
"-hh/nn/mm:tt@dd": false, "-hh/nn/mm:tt": false,
"hh/-nn/mm:tt@dd": false, "hh/-nn/mm:tt": false,
"hh/nn/-mm:tt@dd": false, "hh/nn/-mm:tt": false,
"hh/nn/mm:-tt@dd": false, "hh/nn/mm:-tt": false,
"hh/nn/mm:tt@-dd": false,
// hosts // hosts
"host:https/namespace/model:tag": true, "host:https/namespace/model:tag": true,
...@@ -334,7 +314,7 @@ func FuzzName(f *testing.F) { ...@@ -334,7 +314,7 @@ func FuzzName(f *testing.F) {
f.Fuzz(func(t *testing.T, s string) { f.Fuzz(func(t *testing.T, s string) {
n := ParseNameBare(s) n := ParseNameBare(s)
if n.IsValid() { if n.IsValid() {
parts := [...]string{n.Host, n.Namespace, n.Model, n.Tag, n.RawDigest} parts := [...]string{n.Host, n.Namespace, n.Model, n.Tag}
for _, part := range parts { for _, part := range parts {
if part == ".." { if part == ".." {
t.Errorf("unexpected .. as valid part") t.Errorf("unexpected .. as valid part")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment