@digipair/skill-vespa 0.1.4 → 0.1.6

package/index.cjs.js

@@ -34643,28 +34643,27 @@ let VespaService = class VespaService {
             ...parents,
             ...chunks.flat()
         ];
-        const embeddings = await this.embeddings(result.map((document)=>document.content));
-        return result.map((document, index)=>_extends({}, document, {
-                content_embedding: embeddings[index]
-            }));
+        return result;
     }
     async embedding(text) {
-        const { HuggingFaceTransformersEmbeddings } = await eval(`import('@langchain/community/embeddings/hf_transformers')`);
-        const model = new HuggingFaceTransformersEmbeddings({
-            modelName: 'Xenova/multilingual-e5-large'
-        });
-        return await model.embedQuery(text);
-    }
-    async embeddings(texts) {
-        return Promise.all(texts.map((text)=>this.embedding(text)));
+        if (this.embeddingsModel === null) {
+            const { HuggingFaceTransformersEmbeddings } = await eval(`import('@langchain/community/embeddings/hf_transformers')`);
+            this.embeddingsModel = new HuggingFaceTransformersEmbeddings({
+                modelName: 'Xenova/multilingual-e5-large'
+            });
+        }
+        return await this.embeddingsModel.embedQuery(text);
     }
     async pushDocuments(baseUrl, collection, documents) {
         const results = [];
         for (const document of documents){
+            const content_embedding = await this.embedding(document.content);
             const response = await fetch(`${baseUrl}/document/v1/Digipair_default/${collection}/docid/${document.uuid}`, {
                 method: 'POST',
                 body: JSON.stringify({
-                    fields: _extends({}, document)
+                    fields: _extends({}, document, {
+                        content_embedding
+                    })
                 }),
                 headers: {
                     'Content-Type': 'application/json'
@@ -34724,6 +34723,9 @@ let VespaService = class VespaService {
         const results = await this.prepareDocuments(documents);
         return await this.pushDocuments(baseUrl, collection, results);
     }
+    constructor(){
+        this.embeddingsModel = null;
+    }
 };
 const find = (params, pinsSettingsList, context)=>new VespaService().find(params, pinsSettingsList, context);
 const search = (params, pinsSettingsList, context)=>new VespaService().search(params, pinsSettingsList, context);

package/index.esm.js

@@ -34639,28 +34639,27 @@ let VespaService = class VespaService {
             ...parents,
             ...chunks.flat()
         ];
-        const embeddings = await this.embeddings(result.map((document)=>document.content));
-        return result.map((document, index)=>_extends({}, document, {
-                content_embedding: embeddings[index]
-            }));
+        return result;
     }
     async embedding(text) {
-        const { HuggingFaceTransformersEmbeddings } = await eval(`import('@langchain/community/embeddings/hf_transformers')`);
-        const model = new HuggingFaceTransformersEmbeddings({
-            modelName: 'Xenova/multilingual-e5-large'
-        });
-        return await model.embedQuery(text);
-    }
-    async embeddings(texts) {
-        return Promise.all(texts.map((text)=>this.embedding(text)));
+        if (this.embeddingsModel === null) {
+            const { HuggingFaceTransformersEmbeddings } = await eval(`import('@langchain/community/embeddings/hf_transformers')`);
+            this.embeddingsModel = new HuggingFaceTransformersEmbeddings({
+                modelName: 'Xenova/multilingual-e5-large'
+            });
+        }
+        return await this.embeddingsModel.embedQuery(text);
     }
     async pushDocuments(baseUrl, collection, documents) {
         const results = [];
         for (const document of documents){
+            const content_embedding = await this.embedding(document.content);
             const response = await fetch(`${baseUrl}/document/v1/Digipair_default/${collection}/docid/${document.uuid}`, {
                 method: 'POST',
                 body: JSON.stringify({
-                    fields: _extends({}, document)
+                    fields: _extends({}, document, {
+                        content_embedding
+                    })
                 }),
                 headers: {
                     'Content-Type': 'application/json'
@@ -34720,6 +34719,9 @@ let VespaService = class VespaService {
         const results = await this.prepareDocuments(documents);
         return await this.pushDocuments(baseUrl, collection, results);
     }
+    constructor(){
+        this.embeddingsModel = null;
+    }
 };
 const find = (params, pinsSettingsList, context)=>new VespaService().find(params, pinsSettingsList, context);
 const search = (params, pinsSettingsList, context)=>new VespaService().search(params, pinsSettingsList, context);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@digipair/skill-vespa",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "dependencies": {},
   "main": "./index.cjs.js",
   "module": "./index.esm.js"