Support ollama.ai

DaoCloud · Apr 1, 2024 · a89c2cb · a89c2cb
1 parent cdf47f4
commit a89c2cb
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmd/crproxy/main.go b/cmd/crproxy/main.go
@@ -82,12 +82,16 @@ func main() {
 		crproxy.WithMaxClientSizeForEachRegistry(16),
 		crproxy.WithDomainAlias(map[string]string{
 			"docker.io": "registry-1.docker.io",
+			"ollama.ai": "registry.ollama.ai",
 		}),
 		crproxy.WithPathInfoModifyFunc(func(info *crproxy.PathInfo) *crproxy.PathInfo {
 			// docker.io/busybox => docker.io/library/busybox
 			if info.Host == "registry-1.docker.io" && !strings.Contains(info.Image, "/") {
 				info.Image = "library/" + info.Image
 			}
+			if info.Host == "registry.ollama.ai" && !strings.Contains(info.Image, "/") {
+				info.Image = "library/" + info.Image
+			}
 			return info
 		}),
 		crproxy.WithDisableKeepAlives(disableKeepAlives),