diff --git a/ramalama/cli.py b/ramalama/cli.py index 479eb15..c64f4a4 100644 --- a/ramalama/cli.py +++ b/ramalama/cli.py @@ -193,7 +193,7 @@ def login_parser(subparsers): def login_cli(args): transport = args.TRANSPORT if transport != "": - transport = os.getenv("RAMALAMA_TRANSPORT") + transport = os.getenv("RAMALAMA_TRANSPORT") + "://" model = New(str(transport)) return model.login(args) @@ -211,7 +211,7 @@ def logout_parser(subparsers): def logout_cli(args): transport = args.TRANSPORT if transport != "": - transport = os.getenv("RAMALAMA_TRANSPORT") + transport = os.getenv("RAMALAMA_TRANSPORT") + "://" model = New(str(transport)) return model.logout(args) @@ -679,11 +679,11 @@ def dry_run(args): def New(model): - if model.startswith("huggingface"): + if model.startswith("huggingface://") or model.startswith("hf://"): return Huggingface(model) - if model.startswith("ollama"): + if model.startswith("ollama://"): return Ollama(model) - if model.startswith("oci") | model.startswith("docker"): + if model.startswith("oci://") or model.startswith("docker://"): return OCI(model) transport = os.getenv("RAMALAMA_TRANSPORT") diff --git a/test/system/050-pull.bats b/test/system/050-pull.bats index 28cdb92..bcabd06 100644 --- a/test/system/050-pull.bats +++ b/test/system/050-pull.bats @@ -25,6 +25,11 @@ load helpers # bats test_tags=distro-integration @test "ramalama pull huggingface" { + run_ramalama pull hf://afrideva/Tiny-Vicuna-1B-GGUF/tiny-vicuna-1b.q2_k.gguf + run_ramalama list + is "$output" ".*afrideva/Tiny-Vicuna-1B-GGUF/tiny-vicuna-1b.q2_k" "image was actually pulled locally" + run_ramalama rm hf://afrideva/Tiny-Vicuna-1B-GGUF/tiny-vicuna-1b.q2_k.gguf + run_ramalama pull huggingface://afrideva/Tiny-Vicuna-1B-GGUF/tiny-vicuna-1b.q2_k.gguf run_ramalama list is "$output" ".*afrideva/Tiny-Vicuna-1B-GGUF/tiny-vicuna-1b.q2_k" "image was actually pulled locally"