Move push code to transport python

Signed-off-by: Daniel J Walsh <[email protected]>

ramalama.py

@@ -18,6 +18,9 @@ def main(args):
             ramalama.usage()
 
         while len(args) > 0:
+            if args[0] == "-h" or args[0] == "--help":
+                ramalama.usage()
+
             if args[0] == "--dryrun":
                 args.pop(0)
                 dryrun = True

ramalama/cli.py

@@ -209,11 +209,12 @@ def push_cli(store, args, port):
     target = args.pop(0)
     if model.startswith("oci://"):
         return oci.push(store, model, target)
+    if model.startswith("huggingface://"):
+        return huggingface.push(store, model, target)
+    if model.startswith("ollama://"):
+        return ollama.push(store, model, target)
 
-    # TODO: Additional repository types can be added here, e.g., Ollama, HuggingFace, etc.
-    else:
-        raise NotImplementedError(
-            f"Unsupported repository type for model: {model}")
+    raise KeyError(f"Unsupported transport model must have ollama://, oci://, or huggingface:// prefix: {model}")
 
 
 def run_cli(store, args, port):

ramalama/huggingface.py

@@ -30,3 +30,6 @@ def pull(model, ramalama_store):
     run_cmd(["ln", "-sf", relative_target_path, symlink_path])
 
     return symlink_path
+
+def push(store, model, target):
+    raise NotImplementedError("ramalama push not implemented for huggingface transport")

ramalama/ollama.py

@@ -92,3 +92,6 @@ def init_pull(repos, manifests, accept, registry_head, model_name, model_tag, mo
                   symlink_path)
 
     return symlink_path
+
+def push(store, model, target):
+    raise NotImplementedError("ramalama push not implemented for ollama transport")