From 759dea466bd07916d02e0a719b703313f6dbf119 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 28 Jan 2025 14:21:13 +0000
Subject: [PATCH] Added template support for fastembed

---
 .../components/embeddings-fastembed.jsonnet   | 43 +++++++++++++++++++
 1 file changed, 43 insertions(+)
 create mode 100644 templates/components/embeddings-fastembed.jsonnet

diff --git a/templates/components/embeddings-fastembed.jsonnet b/templates/components/embeddings-fastembed.jsonnet
new file mode 100644
index 00000000..a515617f
--- /dev/null
+++ b/templates/components/embeddings-fastembed.jsonnet
@@ -0,0 +1,43 @@
+local base = import "base/base.jsonnet";
+local images = import "values/images.jsonnet";
+local url = import "values/url.jsonnet";
+local prompts = import "prompts/mixtral.jsonnet";
+
+{
+
+    "embeddings-model":: "sentence-transformers/all-MiniLM-L6-v2",
+
+    embeddings +: {
+    
+        create:: function(engine)
+
+            local container =
+                engine.container("embeddings")
+                    .with_image(images.trustgraph)
+                    .with_command([
+                        "embeddings-fastembed",
+                        "-p",
+                        url.pulsar,
+                        "-m",
+                        $["embeddings-model"],
+                    ])
+                    .with_limits("1.0", "400M")
+                    .with_reservations("0.5", "400M");
+
+            local containerSet = engine.containers(
+                "embeddings", [ container ]
+            );
+
+            local service =
+                engine.internalService(containerSet)
+                .with_port(8000, 8000, "metrics");
+
+            engine.resources([
+                containerSet,
+                service,
+            ])
+
+    },
+
+}
+