1 files changed, 36 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ImportedModels.java b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ImportedModels.java
new file mode 100644
index 00000000000..d1837134691
--- /dev/null
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/expressiontransforms/ImportedModels.java
@@ -0,0 +1,36 @@
+// Copyright 2018 Yahoo Holdings. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.searchdefinition.expressiontransforms;
+
+import com.yahoo.searchlib.rankingexpression.integration.ml.ImportedModel;
+import com.yahoo.searchlib.rankingexpression.integration.ml.ModelImporter;
+
+import java.io.File;
+import java.util.HashMap;
+import java.util.Map;
+
+/**
+ * Lazily loaded models imported from the models/ directory in the application package
+ *
+ * @author bratseth
+ */
+class ImportedModels {
+
+    private final ModelImporter modelImporter;
+
+    /** The cache of already imported models */
+    private final Map<String, ImportedModel> importedModels = new HashMap<>();
+
+    ImportedModels(ModelImporter modelImporter) {
+        this.modelImporter = modelImporter;
+    }
+
+    /**
+     * Returns the model at the given location in the application package (lazily loaded),
+     *
+     * @throws IllegalArgumentException if the model cannot be loaded
+     */
+    public ImportedModel imported(String modelName, File modelDir) {
+        return importedModels.computeIfAbsent(modelName, __ -> modelImporter.importModel(modelName, modelDir));
+    }
+
+}