types/model: accept former : as a separator in digest (ollama#3724)

This also converges the old sep `:` to the new sep `-`.
bean5 · Apr 18, 2024 · 0408205 · 0408205
1 parent 63a7edd
commit 0408205
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/types/model/digest.go b/types/model/digest.go
@@ -1,6 +1,7 @@
 package model
 
 import (
+	"fmt"
 	"log/slog"
 	"strings"
 	"unicode"
@@ -47,8 +48,11 @@ var (
 // Digest.
 func ParseDigest(s string) Digest {
 	typ, digest, ok := strings.Cut(s, "-")
+	if !ok {
+		typ, digest, ok = strings.Cut(s, ":")
+	}
 	if ok && isValidDigestType(typ) && isValidHex(digest) {
-		return Digest{s: s}
+		return Digest{s: fmt.Sprintf("%s-%s", typ, digest)}
 	}
 	return Digest{}
 }