[AINode] Add a model ckpt path retrieve interface (#15689)

Author: CRZbulabula

iotdb-core/ainode/ainode/core/manager/model_manager.py

@@ -100,6 +100,18 @@ def load_model(self, model_id: str, acceleration: bool = False) -> Callable:
         logger.info(f"load model {model_id}")
         return self.model_storage.load_model(model_id, acceleration)
 
+    def get_ckpt_path(self, model_id: str) -> str:
+        """
+        Get the checkpoint path for a given model ID.
+
+        Args:
+            model_id (str): The ID of the model.
+
+        Returns:
+            str: The path to the checkpoint file for the model.
+        """
+        return self.model_storage.get_ckpt_path(model_id)
+
     @staticmethod
     def load_built_in_model(model_id: str, attributes: {}):
         model_id = model_id.lower()

iotdb-core/ainode/ainode/core/model/model_storage.py

@@ -119,3 +119,15 @@ def delete_model(self, model_id: str) -> None:
     def _remove_from_cache(self, file_path: str) -> None:
         if file_path in self._model_cache:
             del self._model_cache[file_path]
+
+    def get_ckpt_path(self, model_id: str) -> str:
+        """
+        Get the checkpoint path for a given model ID.
+
+        Args:
+            model_id (str): The ID of the model.
+
+        Returns:
+            str: The path to the checkpoint file for the model.
+        """
+        return os.path.join(self._model_dir, f"{model_id}")