Check if remote model exists on server

ai_models/__main__.py

@@ -216,7 +216,7 @@ def _main(argv):
         parser.add_argument(
             "model",
             metavar="MODEL",
-            choices=available_models(),
+            choices=available_models() if "--remote" not in argv else None,
             help="The model to run",
         )
 

ai_models/remote/model.py

@@ -1,4 +1,6 @@
+import logging
 import os
+import sys
 import tempfile
 from functools import cached_property
 
@@ -7,6 +9,8 @@
 from ..model import Model
 from .api import RemoteAPI
 
+LOG = logging.getLogger(__name__)
+
 
 class RemoteModel(Model):
     def __init__(self, **kwargs):
@@ -18,6 +22,13 @@ def __init__(self, **kwargs):
         self._param = {}
         self.api = RemoteAPI()
 
+        if self.model not in self.api.models():
+            LOG.error(f"Model '{self.model}' not available on remote server.")
+            LOG.error(
+                "Rerun the command with --models --remote to list available remote models."
+            )
+            sys.exit(1)
+
         self.load_parameters()
 
         super().__init__(**self.cfg)