@infrab4a/connect-nestjs 1.5.2 → 1.5.3-beta.0

package/index.cjs.d.ts

@@ -1 +1 @@
-export * from "./src/index";
+export * from "./src\\index";

package/index.cjs.js

@@ -408,6 +408,27 @@ exports.DiscoveryEngineVertexAdapter = class DiscoveryEngineVertexAdapter {
         };
         await this.documentClient.deleteDocument(request);
     }
+    async bulkProducts(products) {
+        try {
+            const batchSize = 100;
+            let total = 0;
+            for (let index = 0; index < products.length; index += batchSize) {
+                const bulkProducts = products.slice(index, index + batchSize);
+                const [operation] = await this.documentClient.importDocuments({
+                    parent: `projects/${this.config.projectId}/locations/${this.config.location}/collections/default_collection/dataStores/${this.config.dataStoreId}/branches/default_branch`,
+                    inlineSource: {
+                        documents: bulkProducts,
+                    },
+                });
+                total += bulkProducts.length;
+                const [response] = await operation.promise();
+                console.log(`Imported: ${total} documents`, ((total / products.length) * 100).toFixed(2) + '%', `Erros: `, response.errorSamples ? response.errorSamples : []);
+            }
+        }
+        catch (error) {
+            console.log(error);
+        }
+    }
 };
 exports.DiscoveryEngineVertexAdapter = __decorate([
     common.Injectable(),

package/index.esm.js

@@ -404,6 +404,27 @@ let DiscoveryEngineVertexAdapter = class DiscoveryEngineVertexAdapter {
         };
         await this.documentClient.deleteDocument(request);
     }
+    async bulkProducts(products) {
+        try {
+            const batchSize = 100;
+            let total = 0;
+            for (let index = 0; index < products.length; index += batchSize) {
+                const bulkProducts = products.slice(index, index + batchSize);
+                const [operation] = await this.documentClient.importDocuments({
+                    parent: `projects/${this.config.projectId}/locations/${this.config.location}/collections/default_collection/dataStores/${this.config.dataStoreId}/branches/default_branch`,
+                    inlineSource: {
+                        documents: bulkProducts,
+                    },
+                });
+                total += bulkProducts.length;
+                const [response] = await operation.promise();
+                console.log(`Imported: ${total} documents`, ((total / products.length) * 100).toFixed(2) + '%', `Erros: `, response.errorSamples ? response.errorSamples : []);
+            }
+        }
+        catch (error) {
+            console.log(error);
+        }
+    }
 };
 DiscoveryEngineVertexAdapter = __decorate([
     Injectable(),

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@infrab4a/connect-nestjs",
-  "version": "1.5.2",
+  "version": "1.5.3-beta.0",
   "publishConfig": {
     "registry": "https://registry.npmjs.org"
   },
@@ -9,7 +9,7 @@
     "url": "https://github.com/B4AGroup/b4a-firebase-libs"
   },
   "peerDependencies": {
-    "@infrab4a/connect": "^4.15.1",
+    "@infrab4a/connect": "^4.15.2-beta.0",
     "@nestjs/common": "^10.3.3",
     "@nestjs/core": "^10.3.3",
     "firebase-admin": "^12.0.0"

package/src/infra/vertex-ai/adapters/discovery-engine-adapter.d.ts

@@ -1,4 +1,4 @@
-import { ProductHasuraGraphQL, ProductSearch, VertexSearchAdapter } from '@infrab4a/connect';
+import { ProductBulk, ProductHasuraGraphQL, ProductSearch, VertexSearchAdapter } from '@infrab4a/connect';
 import { VertexConfig } from '../types';
 export declare class DiscoveryEngineVertexAdapter implements VertexSearchAdapter<ProductSearch> {
     private config;
@@ -13,4 +13,5 @@ export declare class DiscoveryEngineVertexAdapter implements VertexSearchAdapter
     query(term: string, total: number, gender?: string): Promise<ProductSearch[]>;
     get(id: string): Promise<any>;
     delete(id: string): Promise<void>;
+    bulkProducts(products: ProductBulk[]): Promise<void>;
 }