Fix Repo

mili-tan · Jun 30, 2024 · 9284a3f · 9284a3f
1 parent 34c1b81
commit 9284a3f
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/Onllama.Tiny/FormRegistryInfo.cs b/Onllama.Tiny/FormRegistryInfo.cs
@@ -13,19 +13,23 @@ public FormRegistryInfo(string tags)
             InitializeComponent();
             this.Text += @" - " + tags;
 
-            var model = tags;
+            var repo = tags;
             var version = "latest";
             if (tags.Contains(":"))
             {
-                model = tags.Split(':').First();
+                repo = tags.Split(':').First();
                 version = tags.Split(':').Last();
             }
+            if (!repo.Contains("/"))
+            {
+                repo = "library/" + repo;
+            }
 
             try
             {
                 using var httpClient = new HttpClient();
                 using var request = new HttpRequestMessage(new HttpMethod("GET"),
-                    $"https://registry.ollama.ai/v2/library/{model}/manifests/{version}");
+                    $"https://registry.ollama.ai/v2/{repo}/manifests/{version}");
                 request.Headers.TryAddWithoutValidation("Accept", "application/vnd.docker.distribution.manifest.v2+json");
 
                 var response = JsonNode.Parse(httpClient.SendAsync(request).Result.Content.ReadAsStringAsync().Result);
@@ -40,17 +44,17 @@ public FormRegistryInfo(string tags)
                     else if (layer["mediaType"].ToString() == "application/vnd.ollama.image.template")
                     {
                         inputTemplate.Text = new HttpClient().GetStringAsync(
-                            $"https://registry.ollama.ai/v2/library/{model}/blobs/{layer["digest"]}").Result;
+                            $"https://registry.ollama.ai/v2/{repo}/blobs/{layer["digest"]}").Result;
                     }
                     else if (layer["mediaType"].ToString() == "application/vnd.ollama.image.license")
                     {
                         inputLicense.Text = new HttpClient().GetStringAsync(
-                            $"https://registry.ollama.ai/v2/library/{model}/blobs/{layer["digest"]}").Result;
+                            $"https://registry.ollama.ai/v2/{repo}/blobs/{layer["digest"]}").Result;
                     }
                     else if (layer["mediaType"].ToString() == "application/vnd.ollama.image.params")
                     {
                         inputParameters.Text += new HttpClient()
-                            .GetStringAsync($"https://registry.ollama.ai/v2/library/{model}/blobs/{layer["digest"]}")
+                            .GetStringAsync($"https://registry.ollama.ai/v2/{repo}/blobs/{layer["digest"]}")
                             .Result;
                     }
                 });