Include model version in metadata request

ai_models/remote/api.py

@@ -124,11 +124,15 @@ def run(self, cfg: dict):
 
         LOG.debug("Result written to %s", self.output_file)
 
-    def metadata(self, model, param) -> dict:
+    def metadata(self, model, model_version, param) -> dict:
         if isinstance(param, str):
-            return self._request(requests.get, f"metadata/{model}/{param}")
+            return self._request(
+                requests.get, f"metadata/{model}/{model_version}/{param}"
+            )
         elif isinstance(param, (list, dict)):
-            return self._request(requests.post, f"metadata/{model}", json=param)
+            return self._request(
+                requests.post, f"metadata/{model}/{model_version}", json=param
+            )
         else:
             raise ValueError("param must be a string, list, or dict with 'param' key.")
 

ai_models/remote/model.py

@@ -19,6 +19,7 @@ def __init__(self, **kwargs):
         self.cfg["assets_extra_dir"] = None
 
         self.model = self.cfg["model"]
+        self.model_version = self.cfg.get("model_version", "latest")
         self._param = {}
         self.api = RemoteAPI()
 
@@ -61,6 +62,7 @@ def patch_retrieve_request(self, request):
     def load_parameters(self):
         params = self.api.metadata(
             self.model,
+            self.model_version,
             [
                 "expver",
                 "version",
@@ -80,7 +82,8 @@ def get_parameter(self, name):
         if (param := self._param.get(name)) is not None:
             return param
 
-        self._param.update(self.api.metadata(self.model, name))
+        _param = self.api.metadata(self.model, self.model_version, name)
+        self._param.update(_param)
 
         return self._param.get(name)