Parse https://ollama.com/library/ syntax

People search for ollama models using the web ui, this change
allows one to copy the url from the browser and for it to be
compatible with llama-run.

Signed-off-by: Eric Curtin <[email protected]>

Author: ericcurtin

examples/run/run.cpp

@@ -674,9 +674,9 @@ class LlamaData {
     }
 
     int github_dl(const std::string & model, const std::string & bn) {
-        std::string repository = model;
-        std::string branch     = "main";
-        size_t      at_pos     = model.find('@');
+        std::string  repository = model;
+        std::string  branch     = "main";
+        const size_t at_pos     = model.find('@');
         if (at_pos != std::string::npos) {
             repository = model.substr(0, at_pos);
             branch     = model.substr(at_pos + 1);
@@ -694,9 +694,9 @@ class LlamaData {
             return 1;
         }
 
-        const std::string org          = repo_parts[0];
-        const std::string project      = repo_parts[1];
-        std::string       project_path = repo_parts[2];
+        const std::string & org          = repo_parts[0];
+        const std::string & project      = repo_parts[1];
+        std::string         project_path = repo_parts[2];
         for (size_t i = 3; i < repo_parts.size(); ++i) {
             project_path += "/" + repo_parts[i];
         }
@@ -735,19 +735,20 @@ class LlamaData {
         }
 
         const std::string bn = basename(model_);
-        if (string_starts_with(model_, "hf://") || string_starts_with(model_, "huggingface://")) {
+        if (string_starts_with(model_, "hf://") || string_starts_with(model_, "huggingface://") ||
+            string_starts_with(model_, "hf.co/")) {
             rm_until_substring(model_, "://");
-            ret = huggingface_dl(model_, bn);
-        } else if (string_starts_with(model_, "hf.co/")) {
             rm_until_substring(model_, "hf.co/");
             ret = huggingface_dl(model_, bn);
-        } else if (string_starts_with(model_, "https://") || string_starts_with(model_, "http://")) {
+        } else if ((string_starts_with(model_, "https://") || string_starts_with(model_, "http://")) &&
+                   !string_starts_with(model_, "https://ollama.com/library/")) {
             ret = download(model_, bn, true);
         } else if (string_starts_with(model_, "github:") || string_starts_with(model_, "github://")) {
             rm_until_substring(model_, "github://");
             rm_until_substring(model_, "github:");
             ret = github_dl(model_, bn);
         } else {  // ollama:// or nothing
+            rm_until_substring(model_, "ollama.com/library/");
             rm_until_substring(model_, "://");
             ret = ollama_dl(model_, bn);
         }