1 files changed, 79 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/container/xml/embedder/EmbedderConfig.java b/config-model/src/main/java/com/yahoo/vespa/model/container/xml/embedder/EmbedderConfig.java
new file mode 100644
index 00000000000..1558f01c231
--- /dev/null
+++ b/config-model/src/main/java/com/yahoo/vespa/model/container/xml/embedder/EmbedderConfig.java
@@ -0,0 +1,79 @@
+package com.yahoo.vespa.model.container.xml.embedder;
+
+import com.yahoo.config.model.deploy.DeployState;
+import org.w3c.dom.Element;
+import org.w3c.dom.Node;
+import org.w3c.dom.NodeList;
+
+/**
+ * Translates config in services.xml of the form
+ *
+ *     <embedder id="..." class="..." bundle="..." def="...">
+ *         <!-- options -->
+ *     </embedder>
+ *
+ * to component configuration of the form
+ *
+ *     <component id="..." class="..." bundle="...">
+ *         <config name=def>
+ *             <!-- options -->
+ *         </config>
+ *     </component>
+ *
+ * with some added interpretations based on recognizing the class.
+ *
+ * @author lesters
+ */
+public class EmbedderConfig {
+
+    static EmbedderConfigTransformer getEmbedderTransformer(Element spec, boolean hosted) {
+        String classId = getEmbedderClass(spec);
+        switch (classId) {
+            case "ai.vespa.embedding.BertBaseEmbedder": return new EmbedderConfigBertBaseTransformer(spec, hosted);
+        }
+        return new EmbedderConfigTransformer(spec, hosted);
+    }
+
+    static String modelIdToUrl(String id) {
+        switch (id) {
+            case "test-model-id":
+                return "test-model-url";
+            case "minilm-l6-v2":
+                return "https://data.vespa.oath.cloud/onnx_models/sentence_all_MiniLM_L6_v2.onnx";
+            case "bert-base-uncased":
+                return "https://data.vespa.oath.cloud/onnx_models/bert-base-uncased-vocab.txt";
+        }
+        throw new IllegalArgumentException("Unknown model id: '" + id + "'");
+    }
+
+    /**
+     * Transforms the <embedder ...> element to component configuration.
+     *
+     * @param deployState the deploy state - as config generation can depend on context
+     * @param embedderSpec the XML element containing the <embedder ...>
+     * @return a new XML element containting the <component ...> configuration
+     */
+    public static Element transform(DeployState deployState, Element embedderSpec) {
+        EmbedderConfigTransformer transformer = getEmbedderTransformer(embedderSpec, deployState.isHosted());
+        NodeList children = embedderSpec.getChildNodes();
+        for (int i = 0; i < children.getLength(); i++) {
+            Node child = children.item(i);
+            if (child instanceof Element) {
+                transformer.addOption((Element) child);
+            }
+        }
+        return transformer.createComponentConfig(deployState);
+    }
+
+    private static String getEmbedderClass(Element spec) {
+        if (spec.hasAttribute("class")) {
+            return spec.getAttribute("class");
+        }
+        if (spec.hasAttribute("id")) {
+            return spec.getAttribute("id");
+        }
+        throw new IllegalArgumentException("Embedder specification does not have a required class attribute");
+    }
+
+
+}