@infrab4a/connect-nestjs 1.4.0-beta.2 → 1.4.0-beta.4

package/index.cjs.js

@@ -5,6 +5,8 @@ Object.defineProperty(exports, '__esModule', { value: true });
 var elasticsearch = require('@elastic/elasticsearch');
 var connect = require('@infrab4a/connect');
 var common = require('@nestjs/common');
+var discoveryengine = require('@google-cloud/discoveryengine');
+var v1beta = require('@google-cloud/discoveryengine/build/src/v1beta');
 var nestjsFirebase = require('nestjs-firebase');
 
 const ES_CONFIG = 'ES_CONFIG';
@@ -13,6 +15,8 @@ const HASURA_OPTIONS = 'HASURA_OPTIONS';
 
 const FIREBASE_STORAGE = 'FIREBASE_STORAGE';
 
+const VERTEX_CONFIG = 'VERTEX_CONFIG';
+
 /******************************************************************************
 Copyright (c) Microsoft Corporation.
 
@@ -286,6 +290,138 @@ class ConnectFirestoreService {
     }
 }
 
+exports.DiscoveryEngineVertexAdapter = class DiscoveryEngineVertexAdapter {
+    constructor(configuration) {
+        this.config = configuration;
+        this.parentEngine = `projects/${this.config.projectId}/locations/${this.config.location}/collections/default_collection/engines/${this.config.dataStoreId}`;
+        this.parentDataStore = `projects/${this.config.projectId}/locations/${this.config.location}/collections/default_collection/dataStores/${this.config.dataStoreId}/branches/default_branch/documents`;
+        this.searchClient = new v1beta.SearchServiceClient({
+            apiEndpoint: configuration.apiEndpoint,
+            credentials: configuration.credentials,
+        });
+        this.documentClient = new discoveryengine.DocumentServiceClient({
+            apiEndpoint: configuration.apiEndpoint,
+            credentials: configuration.credentials,
+        });
+        this.servingConfig = this.searchClient.projectLocationCollectionDataStoreServingConfigPath(configuration.projectId, configuration.location, configuration.collectionId, configuration.dataStoreId, configuration.servingConfigId);
+    }
+    async save(product) {
+        const productSearch = this.productMapper(product);
+        const request = {
+            parent: this.parentEngine,
+            allowMissing: true,
+            document: {
+                name: this.parentDataStore + `/${product.id}`,
+                jsonData: JSON.stringify(productSearch),
+                data: 'jsonData',
+            },
+        };
+        const response = await this.documentClient.updateDocument(request);
+        return response;
+    }
+    async update(id, data) {
+        const productSearch = this.productMapper(data);
+        const request = {
+            parent: this.parentEngine,
+            allowMissing: true,
+            document: {
+                name: this.parentDataStore + `/${id}`,
+                jsonData: JSON.stringify(productSearch),
+                data: 'jsonData',
+            },
+        };
+        const response = await this.documentClient.updateDocument(request);
+        return response;
+    }
+    async query(term, total, gender) {
+        console.log('adapter query', term, total, gender);
+        const defaultFilter = gender ? `gender: ANY(${gender}, "unisex")` : '';
+        const response = (await this.searchClient.search({
+            pageSize: total,
+            query: term,
+            filter: defaultFilter,
+            languageCode: 'pt-BR',
+            servingConfig: this.servingConfig,
+            relevanceThreshold: 'MEDIUM',
+        }, {
+            autoPaginate: false,
+        }));
+        return this.resultProductMapper(response);
+    }
+    async get(id) {
+        const request = {
+            name: this.parentDataStore + `/${id}`,
+        };
+        const response = await this.documentClient.getDocument(request);
+        const document = JSON.parse(response[0].jsonData);
+        return document;
+    }
+    async delete(id) {
+        const request = {
+            name: this.parentDataStore + `/${id}`,
+        };
+        await this.documentClient.deleteDocument(request);
+    }
+    productMapper(product) {
+        var _a, _b;
+        const image = product.miniatures && product.miniatures.length ? product.miniatures[0] : null;
+        return {
+            id: product.id.toString(),
+            miniatures: image,
+            icon: image,
+            name: product.name,
+            brand: ((_a = product.brand) === null || _a === void 0 ? void 0 : _a.toString().trim()) ? (_b = product.brand) === null || _b === void 0 ? void 0 : _b.toString().trim() : null,
+            ean: product.EAN,
+            sku: product.sku,
+            slug: product.slug,
+            published: product.published,
+            gender: product.gender,
+            stock: product.stock.quantity,
+            rating: product.rate,
+            shoppingCount: product.shoppingCount,
+            description: product.description.description,
+            fullPrice: product.fullPrice,
+            price: product.price.price,
+            subscriberPrice: product.subscriberPrice,
+            outlet: product.outlet,
+            createdAt: product.createdAt,
+        };
+    }
+    resultProductMapper(result) {
+        return result[0].length
+            ? result[0].map((result) => {
+                var _a, _b, _c, _d, _e, _f, _g;
+                return {
+                    id: (_a = result.document) === null || _a === void 0 ? void 0 : _a.structData.fields.id.stringValue,
+                    name: (_b = result.document) === null || _b === void 0 ? void 0 : _b.structData.fields.name.stringValue,
+                    brand: (_c = result.document) === null || _c === void 0 ? void 0 : _c.structData.fields.brand.stringValue,
+                    stock: (_e = (_d = result.document) === null || _d === void 0 ? void 0 : _d.structData.fields.stock) === null || _e === void 0 ? void 0 : _e.numberValue,
+                    rating: (_g = (_f = result.document) === null || _f === void 0 ? void 0 : _f.structData.fields.rating) === null || _g === void 0 ? void 0 : _g.stringValue,
+                    // gender: result.document.structData.fields.gender.stringValue,
+                    // createdAt: result.document.structData.fields.createdAt.stringValue,
+                    // description: result.document.structData.fields.gender.stringValue,
+                    // ean: result.document.structData.fields.gender.stringValue,
+                    // fullPrice: result.document.structData.fields.gender.stringValue,
+                    // icon: result.document.structData.fields.gender.stringValue,
+                    // outlet: result.document.structData.fields.gender.stringValue,
+                    // slug: result.document.structData.fields.gender.stringValue,
+                    // price: result.document.structData.fields.gender.stringValue,
+                    // published: result.document.structData.fields.gender.stringValue,
+                    // shoppingCount: result.document.structData.fields.gender.stringValue,
+                    // sku: result.document.structData.fields.gender.stringValue,
+                    // subscriberPrice: result.document.structData.fields.gender.stringValue,
+                    // miniatures: result.document.structData.fields.gender.stringValue,
+                };
+            })
+            : [];
+    }
+};
+exports.DiscoveryEngineVertexAdapter = __decorate([
+    common.Injectable(),
+    __param(0, common.Inject(VERTEX_CONFIG)),
+    __metadata("design:paramtypes", [Object])
+], exports.DiscoveryEngineVertexAdapter);
+
 var NestElasticSeachModule_1;
 exports.NestElasticSeachModule = NestElasticSeachModule_1 = class NestElasticSeachModule {
     static initializeApp(options) {
@@ -741,6 +877,30 @@ exports.NestHasuraGraphQLModule = NestHasuraGraphQLModule_1 = __decorate([
     })
 ], exports.NestHasuraGraphQLModule);
 
+var NestVertexSeachModule_1;
+let NestVertexSeachModule = NestVertexSeachModule_1 = class NestVertexSeachModule {
+    static initializeApp(options) {
+        return {
+            module: NestVertexSeachModule_1,
+            providers: [{ provide: VERTEX_CONFIG, useValue: options }],
+            exports: [VERTEX_CONFIG],
+        };
+    }
+};
+NestVertexSeachModule = NestVertexSeachModule_1 = __decorate([
+    common.Module({
+        providers: [
+            exports.DiscoveryEngineVertexAdapter,
+            {
+                provide: 'ProductsVertexIndex',
+                useFactory: (adapter) => new connect.ProductsVertexSearch(adapter),
+                inject: [exports.DiscoveryEngineVertexAdapter],
+            },
+        ],
+        exports: [connect.ProductsVertexSearch],
+    })
+], NestVertexSeachModule);
+
 var NestConnectModule_1;
 exports.NestConnectModule = NestConnectModule_1 = class NestConnectModule {
     static initializeApp(options) {
@@ -750,11 +910,13 @@ exports.NestConnectModule = NestConnectModule_1 = class NestConnectModule {
                 ...(options.firebase ? [exports.NestFirestoreModule.initializeApp({ firebase: options.firebase })] : []),
                 ...(connect.isNil(options === null || options === void 0 ? void 0 : options.hasura) ? [] : [exports.NestHasuraGraphQLModule.initializeApp(options.hasura)]),
                 ...(connect.isNil(options === null || options === void 0 ? void 0 : options.elasticSearch) ? [] : [exports.NestElasticSeachModule.initializeApp(options.elasticSearch)]),
+                ...(connect.isNil(options === null || options === void 0 ? void 0 : options.vertex) ? [] : [NestVertexSeachModule.initializeApp(options.vertex)]),
             ],
             exports: [
                 ...(options.firebase ? [exports.NestFirestoreModule] : []),
                 ...(connect.isNil(options === null || options === void 0 ? void 0 : options.hasura) ? [] : [exports.NestHasuraGraphQLModule]),
                 ...(connect.isNil(options === null || options === void 0 ? void 0 : options.elasticSearch) ? [] : [exports.NestElasticSeachModule]),
+                ...(connect.isNil(options === null || options === void 0 ? void 0 : options.vertex) ? [] : [NestVertexSeachModule]),
             ],
         };
     }
@@ -769,3 +931,4 @@ exports.ConnectFirestoreService = ConnectFirestoreService;
 exports.ES_CONFIG = ES_CONFIG;
 exports.FIREBASE_STORAGE = FIREBASE_STORAGE;
 exports.HASURA_OPTIONS = HASURA_OPTIONS;
+exports.VERTEX_CONFIG = VERTEX_CONFIG;

package/index.esm.js

@@ -1,6 +1,8 @@
 import { Client } from '@elastic/elasticsearch';
-import { DebugHelper, isEmpty, NotFoundError, ProductsIndex, UserBeautyProfileFirestoreRepository, Buy2WinFirestoreRepository, CategoryFirestoreRepository, CheckoutFirestoreRepository, CheckoutSubscriptionFirestoreRepository, CouponFirestoreRepository, CampaignHashtagFirestoreRepository, CampaignDashboardFirestoreRepository, SubscriptionEditionFirestoreRepository, HomeFirestoreRepository, LeadFirestoreRepository, LegacyOrderFirestoreRepository, ShopMenuFirestoreRepository, OrderFirestoreRepository, PaymentFirestoreRepository, ProductFirestoreRepository, ShopSettingsFirestoreRepository, SubscriptionPaymentFirestoreRepository, SubscriptionPlanFirestoreRepository, SubscriptionProductFirestoreRepository, SubscriptionFirestoreRepository, UserFirestoreRepository, UserAddressFirestoreRepository, UserPaymentMethodFirestoreRepository, SubscriptionMaterializationFirestoreRepository, SubscriptionSummaryFirestoreRepository, ProductVariantFirestoreRepository, OrderBlockedFirestoreRepository, CategoryHasuraGraphQLRepository, ProductHasuraGraphQLRepository, CategoryFilterHasuraGraphQLRepository, ProductReviewsHasuraGraphQLRepository, VariantHasuraGraphQLRepository, ProductStockNotificationHasuraGraphQLRepository, FilterOptionHasuraGraphQLRepository, FilterHasuraGraphQLRepository, CategoryCollectionChildrenHasuraGraphQLRepository, WishlistHasuraGraphQLRepository, isNil } from '@infrab4a/connect';
+import { DebugHelper, isEmpty, NotFoundError, ProductsIndex, UserBeautyProfileFirestoreRepository, Buy2WinFirestoreRepository, CategoryFirestoreRepository, CheckoutFirestoreRepository, CheckoutSubscriptionFirestoreRepository, CouponFirestoreRepository, CampaignHashtagFirestoreRepository, CampaignDashboardFirestoreRepository, SubscriptionEditionFirestoreRepository, HomeFirestoreRepository, LeadFirestoreRepository, LegacyOrderFirestoreRepository, ShopMenuFirestoreRepository, OrderFirestoreRepository, PaymentFirestoreRepository, ProductFirestoreRepository, ShopSettingsFirestoreRepository, SubscriptionPaymentFirestoreRepository, SubscriptionPlanFirestoreRepository, SubscriptionProductFirestoreRepository, SubscriptionFirestoreRepository, UserFirestoreRepository, UserAddressFirestoreRepository, UserPaymentMethodFirestoreRepository, SubscriptionMaterializationFirestoreRepository, SubscriptionSummaryFirestoreRepository, ProductVariantFirestoreRepository, OrderBlockedFirestoreRepository, CategoryHasuraGraphQLRepository, ProductHasuraGraphQLRepository, CategoryFilterHasuraGraphQLRepository, ProductReviewsHasuraGraphQLRepository, VariantHasuraGraphQLRepository, ProductStockNotificationHasuraGraphQLRepository, FilterOptionHasuraGraphQLRepository, FilterHasuraGraphQLRepository, CategoryCollectionChildrenHasuraGraphQLRepository, WishlistHasuraGraphQLRepository, ProductsVertexSearch, isNil } from '@infrab4a/connect';
 import { Injectable, Inject, Module } from '@nestjs/common';
+import { DocumentServiceClient } from '@google-cloud/discoveryengine';
+import { SearchServiceClient } from '@google-cloud/discoveryengine/build/src/v1beta';
 import { FirebaseConstants, FirebaseModule } from 'nestjs-firebase';
 
 const ES_CONFIG = 'ES_CONFIG';
@@ -9,6 +11,8 @@ const HASURA_OPTIONS = 'HASURA_OPTIONS';
 
 const FIREBASE_STORAGE = 'FIREBASE_STORAGE';
 
+const VERTEX_CONFIG = 'VERTEX_CONFIG';
+
 /******************************************************************************
 Copyright (c) Microsoft Corporation.
 
@@ -282,6 +286,138 @@ class ConnectFirestoreService {
     }
 }
 
+let DiscoveryEngineVertexAdapter = class DiscoveryEngineVertexAdapter {
+    constructor(configuration) {
+        this.config = configuration;
+        this.parentEngine = `projects/${this.config.projectId}/locations/${this.config.location}/collections/default_collection/engines/${this.config.dataStoreId}`;
+        this.parentDataStore = `projects/${this.config.projectId}/locations/${this.config.location}/collections/default_collection/dataStores/${this.config.dataStoreId}/branches/default_branch/documents`;
+        this.searchClient = new SearchServiceClient({
+            apiEndpoint: configuration.apiEndpoint,
+            credentials: configuration.credentials,
+        });
+        this.documentClient = new DocumentServiceClient({
+            apiEndpoint: configuration.apiEndpoint,
+            credentials: configuration.credentials,
+        });
+        this.servingConfig = this.searchClient.projectLocationCollectionDataStoreServingConfigPath(configuration.projectId, configuration.location, configuration.collectionId, configuration.dataStoreId, configuration.servingConfigId);
+    }
+    async save(product) {
+        const productSearch = this.productMapper(product);
+        const request = {
+            parent: this.parentEngine,
+            allowMissing: true,
+            document: {
+                name: this.parentDataStore + `/${product.id}`,
+                jsonData: JSON.stringify(productSearch),
+                data: 'jsonData',
+            },
+        };
+        const response = await this.documentClient.updateDocument(request);
+        return response;
+    }
+    async update(id, data) {
+        const productSearch = this.productMapper(data);
+        const request = {
+            parent: this.parentEngine,
+            allowMissing: true,
+            document: {
+                name: this.parentDataStore + `/${id}`,
+                jsonData: JSON.stringify(productSearch),
+                data: 'jsonData',
+            },
+        };
+        const response = await this.documentClient.updateDocument(request);
+        return response;
+    }
+    async query(term, total, gender) {
+        console.log('adapter query', term, total, gender);
+        const defaultFilter = gender ? `gender: ANY(${gender}, "unisex")` : '';
+        const response = (await this.searchClient.search({
+            pageSize: total,
+            query: term,
+            filter: defaultFilter,
+            languageCode: 'pt-BR',
+            servingConfig: this.servingConfig,
+            relevanceThreshold: 'MEDIUM',
+        }, {
+            autoPaginate: false,
+        }));
+        return this.resultProductMapper(response);
+    }
+    async get(id) {
+        const request = {
+            name: this.parentDataStore + `/${id}`,
+        };
+        const response = await this.documentClient.getDocument(request);
+        const document = JSON.parse(response[0].jsonData);
+        return document;
+    }
+    async delete(id) {
+        const request = {
+            name: this.parentDataStore + `/${id}`,
+        };
+        await this.documentClient.deleteDocument(request);
+    }
+    productMapper(product) {
+        var _a, _b;
+        const image = product.miniatures && product.miniatures.length ? product.miniatures[0] : null;
+        return {
+            id: product.id.toString(),
+            miniatures: image,
+            icon: image,
+            name: product.name,
+            brand: ((_a = product.brand) === null || _a === void 0 ? void 0 : _a.toString().trim()) ? (_b = product.brand) === null || _b === void 0 ? void 0 : _b.toString().trim() : null,
+            ean: product.EAN,
+            sku: product.sku,
+            slug: product.slug,
+            published: product.published,
+            gender: product.gender,
+            stock: product.stock.quantity,
+            rating: product.rate,
+            shoppingCount: product.shoppingCount,
+            description: product.description.description,
+            fullPrice: product.fullPrice,
+            price: product.price.price,
+            subscriberPrice: product.subscriberPrice,
+            outlet: product.outlet,
+            createdAt: product.createdAt,
+        };
+    }
+    resultProductMapper(result) {
+        return result[0].length
+            ? result[0].map((result) => {
+                var _a, _b, _c, _d, _e, _f, _g;
+                return {
+                    id: (_a = result.document) === null || _a === void 0 ? void 0 : _a.structData.fields.id.stringValue,
+                    name: (_b = result.document) === null || _b === void 0 ? void 0 : _b.structData.fields.name.stringValue,
+                    brand: (_c = result.document) === null || _c === void 0 ? void 0 : _c.structData.fields.brand.stringValue,
+                    stock: (_e = (_d = result.document) === null || _d === void 0 ? void 0 : _d.structData.fields.stock) === null || _e === void 0 ? void 0 : _e.numberValue,
+                    rating: (_g = (_f = result.document) === null || _f === void 0 ? void 0 : _f.structData.fields.rating) === null || _g === void 0 ? void 0 : _g.stringValue,
+                    // gender: result.document.structData.fields.gender.stringValue,
+                    // createdAt: result.document.structData.fields.createdAt.stringValue,
+                    // description: result.document.structData.fields.gender.stringValue,
+                    // ean: result.document.structData.fields.gender.stringValue,
+                    // fullPrice: result.document.structData.fields.gender.stringValue,
+                    // icon: result.document.structData.fields.gender.stringValue,
+                    // outlet: result.document.structData.fields.gender.stringValue,
+                    // slug: result.document.structData.fields.gender.stringValue,
+                    // price: result.document.structData.fields.gender.stringValue,
+                    // published: result.document.structData.fields.gender.stringValue,
+                    // shoppingCount: result.document.structData.fields.gender.stringValue,
+                    // sku: result.document.structData.fields.gender.stringValue,
+                    // subscriberPrice: result.document.structData.fields.gender.stringValue,
+                    // miniatures: result.document.structData.fields.gender.stringValue,
+                };
+            })
+            : [];
+    }
+};
+DiscoveryEngineVertexAdapter = __decorate([
+    Injectable(),
+    __param(0, Inject(VERTEX_CONFIG)),
+    __metadata("design:paramtypes", [Object])
+], DiscoveryEngineVertexAdapter);
+
 var NestElasticSeachModule_1;
 let NestElasticSeachModule = NestElasticSeachModule_1 = class NestElasticSeachModule {
     static initializeApp(options) {
@@ -737,6 +873,30 @@ NestHasuraGraphQLModule = NestHasuraGraphQLModule_1 = __decorate([
     })
 ], NestHasuraGraphQLModule);
 
+var NestVertexSeachModule_1;
+let NestVertexSeachModule = NestVertexSeachModule_1 = class NestVertexSeachModule {
+    static initializeApp(options) {
+        return {
+            module: NestVertexSeachModule_1,
+            providers: [{ provide: VERTEX_CONFIG, useValue: options }],
+            exports: [VERTEX_CONFIG],
+        };
+    }
+};
+NestVertexSeachModule = NestVertexSeachModule_1 = __decorate([
+    Module({
+        providers: [
+            DiscoveryEngineVertexAdapter,
+            {
+                provide: 'ProductsVertexIndex',
+                useFactory: (adapter) => new ProductsVertexSearch(adapter),
+                inject: [DiscoveryEngineVertexAdapter],
+            },
+        ],
+        exports: [ProductsVertexSearch],
+    })
+], NestVertexSeachModule);
+
 var NestConnectModule_1;
 let NestConnectModule = NestConnectModule_1 = class NestConnectModule {
     static initializeApp(options) {
@@ -746,11 +906,13 @@ let NestConnectModule = NestConnectModule_1 = class NestConnectModule {
                 ...(options.firebase ? [NestFirestoreModule.initializeApp({ firebase: options.firebase })] : []),
                 ...(isNil(options === null || options === void 0 ? void 0 : options.hasura) ? [] : [NestHasuraGraphQLModule.initializeApp(options.hasura)]),
                 ...(isNil(options === null || options === void 0 ? void 0 : options.elasticSearch) ? [] : [NestElasticSeachModule.initializeApp(options.elasticSearch)]),
+                ...(isNil(options === null || options === void 0 ? void 0 : options.vertex) ? [] : [NestVertexSeachModule.initializeApp(options.vertex)]),
             ],
             exports: [
                 ...(options.firebase ? [NestFirestoreModule] : []),
                 ...(isNil(options === null || options === void 0 ? void 0 : options.hasura) ? [] : [NestHasuraGraphQLModule]),
                 ...(isNil(options === null || options === void 0 ? void 0 : options.elasticSearch) ? [] : [NestElasticSeachModule]),
+                ...(isNil(options === null || options === void 0 ? void 0 : options.vertex) ? [] : [NestVertexSeachModule]),
             ],
         };
     }
@@ -759,4 +921,4 @@ NestConnectModule = NestConnectModule_1 = __decorate([
     Module({})
 ], NestConnectModule);
 
-export { ConnectCollectionService, ConnectDocumentService, ConnectFirestoreService, ES_CONFIG, FIREBASE_STORAGE, HASURA_OPTIONS, NativeElasticSearchAdapter, NestConnectModule, NestElasticSeachModule, NestFirestoreModule, NestHasuraGraphQLModule, NestStorageModule };
+export { ConnectCollectionService, ConnectDocumentService, ConnectFirestoreService, DiscoveryEngineVertexAdapter, ES_CONFIG, FIREBASE_STORAGE, HASURA_OPTIONS, NativeElasticSearchAdapter, NestConnectModule, NestElasticSeachModule, NestFirestoreModule, NestHasuraGraphQLModule, NestStorageModule, VERTEX_CONFIG };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@infrab4a/connect-nestjs",
-  "version": "1.4.0-beta.2",
+  "version": "1.4.0-beta.4",
   "publishConfig": {
     "registry": "https://registry.npmjs.org"
   },
@@ -9,13 +9,14 @@
     "url": "https://github.com/B4AGroup/b4a-firebase-libs"
   },
   "peerDependencies": {
-    "@infrab4a/connect": "4.14.0-beta.2",
+    "@infrab4a/connect": "4.14.0-beta.4",
     "@nestjs/common": "^10.3.3",
     "@nestjs/core": "^10.3.3",
     "firebase-admin": "^12.0.0"
   },
   "dependencies": {
     "@elastic/elasticsearch": "^8.13.1",
+    "@google-cloud/discoveryengine": "^1.14.0",
     "nestjs-firebase": "^10.4.0"
   },
   "exports": {

package/src/consts/index.d.ts

@@ -1,3 +1,4 @@
 export * from './es-config.const';
 export * from './hasura-options.const';
 export * from './storage.const';
+export * from './vertex-config.const';

package/src/consts/vertex-config.const.d.ts

@@ -0,0 +1 @@
+export declare const VERTEX_CONFIG = "VERTEX_CONFIG";

package/src/infra/index.d.ts

@@ -1,2 +1,3 @@
 export * from './elasticsearch';
 export * from './firebase';
+export * from './vertex-ai';

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

@@ -0,0 +1,18 @@
+import { Product, ProductSearch, VertexSearchAdapter } from '@infrab4a/connect';
+import { VertexConfig } from '../types';
+export declare class DiscoveryEngineVertexAdapter implements VertexSearchAdapter<ProductSearch> {
+    private config;
+    private parentEngine;
+    private parentDataStore;
+    private servingConfig;
+    private searchClient;
+    private documentClient;
+    constructor(configuration: VertexConfig);
+    save(product: Product): Promise<[import("@google-cloud/discoveryengine/build/protos/protos").google.cloud.discoveryengine.v1.IDocument, import("@google-cloud/discoveryengine/build/protos/protos").google.cloud.discoveryengine.v1.IUpdateDocumentRequest, {}]>;
+    update(id: string, data: Product): Promise<[import("@google-cloud/discoveryengine/build/protos/protos").google.cloud.discoveryengine.v1.IDocument, import("@google-cloud/discoveryengine/build/protos/protos").google.cloud.discoveryengine.v1.IUpdateDocumentRequest, {}]>;
+    query(term: string, total: number, gender?: string): Promise<ProductSearch[]>;
+    get(id: string): Promise<any>;
+    delete(id: string): Promise<void>;
+    private productMapper;
+    private resultProductMapper;
+}

package/src/infra/vertex-ai/adapters/index.d.ts

@@ -0,0 +1 @@
+export * from './discovery-engine-adapter';

package/src/infra/vertex-ai/index.d.ts

@@ -0,0 +1,2 @@
+export * from './adapters';
+export * from './types';

package/src/infra/vertex-ai/types/index.d.ts

@@ -0,0 +1,2 @@
+export * from './vertex-config';
+export * from './vertex-search-result';

package/src/infra/vertex-ai/types/vertex-config.d.ts

@@ -0,0 +1,23 @@
+export type VertexConfig = {
+    credentials: {
+        type: string;
+        project_id: string;
+        private_key_id: string;
+        private_key: string;
+        client_email: string;
+        client_id: string;
+        auth_uri?: string;
+        token_uri?: string;
+        auth_provider_x509_cert_url?: string;
+        client_x509_cert_url?: string;
+        apiKey?: string;
+        authDomain?: string;
+    };
+    apiEndpoint: string;
+    projectId: string;
+    location: string;
+    dataStoreId: string;
+    collectionId: string;
+    engine: string;
+    servingConfigId: string;
+};

package/src/infra/vertex-ai/types/vertex-search-result.d.ts

@@ -0,0 +1,18 @@
+import { protos } from '@google-cloud/discoveryengine';
+export type VertexSearchResult<T> = [
+    VertexResult<T>[],
+    protos.google.cloud.discoveryengine.v1.ISearchRequest | null,
+    protos.google.cloud.discoveryengine.v1.ISearchResponse
+];
+type VertexResult<T> = {
+    id?: string | null;
+    document?: {
+        structData: {
+            fields: {
+                [P in keyof T]: any;
+            };
+        };
+    };
+    chunk?: protos.google.cloud.discoveryengine.v1.IChunk;
+};
+export {};

package/src/nest-connect.module.d.ts

@@ -1,5 +1,5 @@
 import { DynamicModule } from '@nestjs/common';
-import { ESClientOptions } from './infra';
+import { ESClientOptions, VertexConfig } from './infra';
 import { HasuraGraphQLOptions } from './nest-hasura-graphql.module';
 export declare class NestConnectModule {
     static initializeApp(options: {
@@ -8,5 +8,6 @@ export declare class NestConnectModule {
         };
         elasticSearch?: ESClientOptions;
         hasura: HasuraGraphQLOptions;
+        vertex: VertexConfig;
     }): DynamicModule;
 }

package/src/nest-vertex-ai-search.module.d.ts

@@ -0,0 +1,5 @@
+import { DynamicModule } from '@nestjs/common';
+import { VertexConfig } from './infra';
+export declare class NestVertexSeachModule {
+    static initializeApp(options: VertexConfig): DynamicModule;
+}