@dofe/infra-vector 0.1.1

package/dist/embedding.service.d.ts

@@ -0,0 +1,54 @@
+/**
+ * Embedding Service
+ * 文本嵌入服务，用于生成向量嵌入
+ */
+import { OnModuleInit } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
+import { Logger } from 'winston';
+import { HttpService } from '@nestjs/axios';
+export declare class EmbeddingService implements OnModuleInit {
+    private readonly logger;
+    private readonly configService;
+    private readonly httpService;
+    private config;
+    constructor(logger: Logger, configService: ConfigService, httpService: HttpService);
+    onModuleInit(): void;
+    /**
+     * 生成单个文本的嵌入向量
+     */
+    embed(text: string): Promise<number[]>;
+    /**
+     * 批量生成嵌入向量
+     */
+    embedBatch(texts: string[]): Promise<number[][]>;
+    /**
+     * 为查询生成嵌入（带前缀）
+     */
+    embedQuery(query: string): Promise<number[]>;
+    /**
+     * 为文档生成嵌入（带前缀）
+     */
+    embedDocument(document: string): Promise<number[]>;
+    /**
+     * 批量为文档生成嵌入
+     */
+    embedDocuments(documents: string[]): Promise<number[][]>;
+    /**
+     * 计算两个向量的余弦相似度
+     */
+    cosineSimilarity(a: number[], b: number[]): number;
+    /**
+     * 计算 MMR（最大边际相关性）多样性
+     */
+    mmr(queryEmbedding: number[], candidates: Array<{
+        embedding?: number[];
+        id: string;
+    }>, options?: {
+        lambda?: number;
+        topK?: number;
+    }): string[];
+    private buildUrl;
+    private buildHeaders;
+    private addPrefix;
+}
+//# sourceMappingURL=embedding.service.d.ts.map

package/dist/embedding.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.service.d.ts","sourceRoot":"","sources":["../src/embedding.service.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAsB,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAClE,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAE/C,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAM5C,qBACa,gBAAiB,YAAW,YAAY;IAIhB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACxD,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,WAAW;IAL9B,OAAO,CAAC,MAAM,CAAkB;gBAGoB,MAAM,EAAE,MAAM,EAC/C,aAAa,EAAE,aAAa,EAC5B,WAAW,EAAE,WAAW;IAG3C,YAAY;IAsDZ;;OAEG;IACG,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAK5C;;OAEG;IACG,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAmFtD;;OAEG;IACG,UAAU,CAAC,KAAK,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAKlD;;OAEG;IACG,aAAa,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IAKxD;;OAEG;IACG,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC;IAO9D;;OAEG;IACH,gBAAgB,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM;IAmBlD;;OAEG;IACH,GAAG,CACD,cAAc,EAAE,MAAM,EAAE,EACxB,UAAU,EAAE,KAAK,CAAC;QAAE,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;QAAC,EAAE,EAAE,MAAM,CAAA;KAAE,CAAC,EACvD,OAAO,GAAE;QAAE,MAAM,CAAC,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAO,GAC/C,MAAM,EAAE;IA8DX,OAAO,CAAC,QAAQ;IAQhB,OAAO,CAAC,YAAY;IAOpB,OAAO,CAAC,SAAS;CAMlB"}

package/dist/embedding.service.js

@@ -0,0 +1,254 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.EmbeddingService = void 0;
+/**
+ * Embedding Service
+ * 文本嵌入服务，用于生成向量嵌入
+ */
+const common_1 = require("@nestjs/common");
+const config_1 = require("@nestjs/config");
+const nest_winston_1 = require("nest-winston");
+const winston_1 = require("winston");
+const axios_1 = require("@nestjs/axios");
+const rxjs_1 = require("rxjs");
+const infra_common_1 = require("@dofe/infra-common");
+let EmbeddingService = class EmbeddingService {
+    logger;
+    configService;
+    httpService;
+    config;
+    constructor(logger, configService, httpService) {
+        this.logger = logger;
+        this.configService = configService;
+        this.httpService = httpService;
+    }
+    onModuleInit() {
+        const keysConfig = (0, infra_common_1.getKeysConfig)();
+        const embeddingKeys = keysConfig?.embedding;
+        // keys/config.json 优先，其次才 fallback env（兼容旧配置）
+        this.config = {
+            baseUrl: embeddingKeys?.baseUrl ??
+                this.configService.get('EMBEDDING_BASE_URL', this.configService.get('OPENAI_API_BASE', 'https://api.openai.com/v1')),
+            apiKey: embeddingKeys?.apiKey ??
+                this.configService.get('EMBEDDING_API_KEY', this.configService.get('OPENAI_API_KEY')),
+            model: embeddingKeys?.model ??
+                this.configService.get('EMBEDDING_MODEL', 'text-embedding-3-small'),
+            dimensions: embeddingKeys?.dimensions ??
+                this.configService.get('EMBEDDING_DIMENSIONS'),
+            timeoutMs: embeddingKeys?.timeoutMs ??
+                this.configService.get('EMBEDDING_TIMEOUT_MS', 20000),
+            queryPrefix: embeddingKeys?.queryPrefix ??
+                this.configService.get('EMBEDDING_QUERY_PREFIX', 'query:'),
+            documentPrefix: embeddingKeys?.documentPrefix ??
+                this.configService.get('EMBEDDING_DOCUMENT_PREFIX', 'passage:'),
+        };
+        this.logger.info('[Embedding] Service initialized', {
+            baseUrl: this.config.baseUrl ? '[configured]' : '[not configured]',
+            model: this.config.model,
+            dimensions: this.config.dimensions,
+        });
+    }
+    // ============================================================================
+    // Public API
+    // ============================================================================
+    /**
+     * 生成单个文本的嵌入向量
+     */
+    async embed(text) {
+        const embeddings = await this.embedBatch([text]);
+        return embeddings[0] ?? [];
+    }
+    /**
+     * 批量生成嵌入向量
+     */
+    async embedBatch(texts) {
+        if (texts.length === 0) {
+            return [];
+        }
+        if (!this.config.baseUrl || !this.config.apiKey) {
+            this.logger.warn('[Embedding] Service not configured, returning empty embeddings');
+            return texts.map(() => []);
+        }
+        const url = this.buildUrl('/embeddings');
+        const headers = this.buildHeaders();
+        // 过滤空文本
+        const validTexts = texts.filter((t) => t && t.trim().length > 0);
+        if (validTexts.length === 0) {
+            return texts.map(() => []);
+        }
+        try {
+            const response = await (0, rxjs_1.firstValueFrom)(this.httpService
+                .post(url, {
+                model: this.config.model,
+                input: validTexts,
+                ...(this.config.dimensions
+                    ? { dimensions: this.config.dimensions }
+                    : {}),
+            }, { headers })
+                .pipe((0, rxjs_1.timeout)(this.config.timeoutMs), (0, rxjs_1.retry)({
+                count: 2,
+                delay: 500,
+            })));
+            const data = response.data?.data ?? [];
+            // 按索引排序并提取嵌入
+            const sortedEmbeddings = data
+                .sort((a, b) => (a.index ?? 0) - (b.index ?? 0))
+                .map((item) => item.embedding ?? []);
+            // 如果请求数量和返回数量不匹配，用空数组填充
+            const result = [];
+            let dataIndex = 0;
+            for (const text of texts) {
+                if (text && text.trim().length > 0) {
+                    result.push(sortedEmbeddings[dataIndex] ?? []);
+                    dataIndex++;
+                }
+                else {
+                    result.push([]);
+                }
+            }
+            this.logger.debug('[Embedding] Generated embeddings', {
+                count: result.filter((e) => e.length > 0).length,
+                total: texts.length,
+            });
+            return result;
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            this.logger.error('[Embedding] Failed to generate embeddings', {
+                error: errorMessage,
+                textsCount: texts.length,
+            });
+            // 返回空嵌入而不是抛出错误
+            return texts.map(() => []);
+        }
+    }
+    /**
+     * 为查询生成嵌入（带前缀）
+     */
+    async embedQuery(query) {
+        const prefixedQuery = this.addPrefix(query, this.config.queryPrefix);
+        return this.embed(prefixedQuery);
+    }
+    /**
+     * 为文档生成嵌入（带前缀）
+     */
+    async embedDocument(document) {
+        const prefixedDoc = this.addPrefix(document, this.config.documentPrefix);
+        return this.embed(prefixedDoc);
+    }
+    /**
+     * 批量为文档生成嵌入
+     */
+    async embedDocuments(documents) {
+        const prefixedDocs = documents.map((doc) => this.addPrefix(doc, this.config.documentPrefix));
+        return this.embedBatch(prefixedDocs);
+    }
+    /**
+     * 计算两个向量的余弦相似度
+     */
+    cosineSimilarity(a, b) {
+        if (a.length !== b.length || a.length === 0) {
+            return 0;
+        }
+        let dotProduct = 0;
+        let normA = 0;
+        let normB = 0;
+        for (let i = 0; i < a.length; i++) {
+            dotProduct += a[i] * b[i];
+            normA += a[i] * a[i];
+            normB += b[i] * b[i];
+        }
+        const denominator = Math.sqrt(normA) * Math.sqrt(normB);
+        return denominator === 0 ? 0 : dotProduct / denominator;
+    }
+    /**
+     * 计算 MMR（最大边际相关性）多样性
+     */
+    mmr(queryEmbedding, candidates, options = {}) {
+        const { lambda = 0.7, topK = 10 } = options;
+        if (candidates.length === 0) {
+            return [];
+        }
+        const selected = [];
+        const remaining = [...candidates];
+        const selectedEmbeddings = [];
+        while (selected.length < topK && remaining.length > 0) {
+            let bestScore = -Infinity;
+            let bestIndex = -1;
+            for (let i = 0; i < remaining.length; i++) {
+                const candidate = remaining[i];
+                if (!candidate.embedding || candidate.embedding.length === 0) {
+                    continue;
+                }
+                // 与查询的相似度
+                const querySimilarity = this.cosineSimilarity(queryEmbedding, candidate.embedding);
+                // 与已选择文档的最大相似度
+                let maxSelectedSimilarity = 0;
+                for (const selectedEmb of selectedEmbeddings) {
+                    const sim = this.cosineSimilarity(candidate.embedding, selectedEmb);
+                    maxSelectedSimilarity = Math.max(maxSelectedSimilarity, sim);
+                }
+                // MMR 分数
+                const mmrScore = lambda * querySimilarity - (1 - lambda) * maxSelectedSimilarity;
+                if (mmrScore > bestScore) {
+                    bestScore = mmrScore;
+                    bestIndex = i;
+                }
+            }
+            if (bestIndex >= 0) {
+                const best = remaining.splice(bestIndex, 1)[0];
+                selected.push(best.id);
+                if (best.embedding) {
+                    selectedEmbeddings.push(best.embedding);
+                }
+            }
+            else {
+                break;
+            }
+        }
+        return selected;
+    }
+    // ============================================================================
+    // Private Methods
+    // ============================================================================
+    buildUrl(path) {
+        const baseUrl = this.config.baseUrl.endsWith('/')
+            ? this.config.baseUrl.slice(0, -1)
+            : this.config.baseUrl;
+        const normalizedPath = path.startsWith('/') ? path : `/${path}`;
+        return `${baseUrl}${normalizedPath}`;
+    }
+    buildHeaders() {
+        return {
+            'Content-Type': 'application/json',
+            Authorization: `Bearer ${this.config.apiKey}`,
+        };
+    }
+    addPrefix(text, prefix) {
+        if (!prefix || text.startsWith(prefix)) {
+            return text;
+        }
+        return `${prefix}${text}`;
+    }
+};
+exports.EmbeddingService = EmbeddingService;
+exports.EmbeddingService = EmbeddingService = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Inject)(nest_winston_1.WINSTON_MODULE_PROVIDER)),
+    __metadata("design:paramtypes", [winston_1.Logger,
+        config_1.ConfigService,
+        axios_1.HttpService])
+], EmbeddingService);
+//# sourceMappingURL=embedding.service.js.map

package/dist/embedding.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"embedding.service.js","sourceRoot":"","sources":["../src/embedding.service.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA;;;GAGG;AACH,2CAAkE;AAClE,2CAA+C;AAC/C,+CAAuD;AACvD,qCAAiC;AACjC,yCAA4C;AAC5C,+BAAsD;AACtD,qDAAmD;AAK5C,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAIyB;IACjC;IACA;IALX,MAAM,CAAkB;IAEhC,YACoD,MAAc,EAC/C,aAA4B,EAC5B,WAAwB;QAFS,WAAM,GAAN,MAAM,CAAQ;QAC/C,kBAAa,GAAb,aAAa,CAAe;QAC5B,gBAAW,GAAX,WAAW,CAAa;IACxC,CAAC;IAEJ,YAAY;QACV,MAAM,UAAU,GAAG,IAAA,4BAAa,GAAE,CAAC;QACnC,MAAM,aAAa,GAAG,UAAU,EAAE,SAErB,CAAC;QAEd,8CAA8C;QAC9C,IAAI,CAAC,MAAM,GAAG;YACZ,OAAO,EACL,aAAa,EAAE,OAAO;gBACtB,IAAI,CAAC,aAAa,CAAC,GAAG,CACpB,oBAAoB,EACpB,IAAI,CAAC,aAAa,CAAC,GAAG,CACpB,iBAAiB,EACjB,2BAA2B,CAC5B,CACF;YACH,MAAM,EACJ,aAAa,EAAE,MAAM;gBACrB,IAAI,CAAC,aAAa,CAAC,GAAG,CACpB,mBAAmB,EACnB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAS,gBAAgB,CAAC,CACjD;YACH,KAAK,EACH,aAAa,EAAE,KAAK;gBACpB,IAAI,CAAC,aAAa,CAAC,GAAG,CACpB,iBAAiB,EACjB,wBAAwB,CACzB;YACH,UAAU,EACR,aAAa,EAAE,UAAU;gBACzB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAS,sBAAsB,CAAC;YACxD,SAAS,EACP,aAAa,EAAE,SAAS;gBACxB,IAAI,CAAC,aAAa,CAAC,GAAG,CAAS,sBAAsB,EAAE,KAAK,CAAC;YAC/D,WAAW,EACT,aAAa,EAAE,WAAW;gBAC1B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAS,wBAAwB,EAAE,QAAQ,CAAC;YACpE,cAAc,EACZ,aAAa,EAAE,cAAc;gBAC7B,IAAI,CAAC,aAAa,CAAC,GAAG,CAAS,2BAA2B,EAAE,UAAU,CAAC;SAC1E,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,iCAAiC,EAAE;YAClD,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,kBAAkB;YAClE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;YACxB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;SACnC,CAAC,CAAC;IACL,CAAC;IAED,+EAA+E;IAC/E,aAAa;IACb,+EAA+E;IAE/E;;OAEG;IACH,KAAK,CAAC,KAAK,CAAC,IAAY;QACtB,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACjD,OAAO,UAAU,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC7B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,KAAe;QAC9B,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC;YAChD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,gEAAgE,CACjE,CAAC;YACF,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC;QAED,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;QACzC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QAEpC,QAAQ;QACR,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACjE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC;QAED,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAA,qBAAc,EACnC,IAAI,CAAC,WAAW;iBACb,IAAI,CAGH,GAAG,EACH;gBACE,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBACxB,KAAK,EAAE,UAAU;gBACjB,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,UAAU;oBACxB,CAAC,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE;oBACxC,CAAC,CAAC,EAAE,CAAC;aACR,EACD,EAAE,OAAO,EAAE,CACZ;iBACA,IAAI,CACH,IAAA,cAAO,EAAC,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,EAC9B,IAAA,YAAK,EAAC;gBACJ,KAAK,EAAE,CAAC;gBACR,KAAK,EAAE,GAAG;aACX,CAAC,CACH,CACJ,CAAC;YAEF,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,IAAI,IAAI,EAAE,CAAC;YAEvC,aAAa;YACb,MAAM,gBAAgB,GAAG,IAAI;iBAC1B,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,CAAC;iBAC/C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC;YAEvC,wBAAwB;YACxB,MAAM,MAAM,GAAe,EAAE,CAAC;YAC9B,IAAI,SAAS,GAAG,CAAC,CAAC;YAClB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,IAAI,IAAI,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACnC,MAAM,CAAC,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC/C,SAAS,EAAE,CAAC;gBACd,CAAC;qBAAM,CAAC;oBACN,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE;gBACpD,KAAK,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,MAAM;gBAChD,KAAK,EAAE,KAAK,CAAC,MAAM;aACpB,CAAC,CAAC;YAEH,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAChB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,2CAA2C,EAAE;gBAC7D,KAAK,EAAE,YAAY;gBACnB,UAAU,EAAE,KAAK,CAAC,MAAM;aACzB,CAAC,CAAC;YACH,eAAe;YACf,OAAO,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,KAAa;QAC5B,MAAM,aAAa,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACrE,OAAO,IAAI,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,aAAa,CAAC,QAAgB;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC;QACzE,OAAO,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,SAAmB;QACtC,MAAM,YAAY,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACzC,IAAI,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,CAChD,CAAC;QACF,OAAO,IAAI,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,gBAAgB,CAAC,CAAW,EAAE,CAAW;QACvC,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,CAAC,MAAM,IAAI,CAAC,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5C,OAAO,CAAC,CAAC;QACX,CAAC;QAED,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,KAAK,GAAG,CAAC,CAAC;QAEd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAClC,UAAU,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;YACrB,KAAK,IAAI,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QACvB,CAAC;QAED,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACxD,OAAO,WAAW,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,GAAG,WAAW,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,GAAG,CACD,cAAwB,EACxB,UAAuD,EACvD,UAA8C,EAAE;QAEhD,MAAM,EAAE,MAAM,GAAG,GAAG,EAAE,IAAI,GAAG,EAAE,EAAE,GAAG,OAAO,CAAC;QAE5C,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC5B,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,QAAQ,GAAa,EAAE,CAAC;QAC9B,MAAM,SAAS,GAAG,CAAC,GAAG,UAAU,CAAC,CAAC;QAClC,MAAM,kBAAkB,GAAe,EAAE,CAAC;QAE1C,OAAO,QAAQ,CAAC,MAAM,GAAG,IAAI,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtD,IAAI,SAAS,GAAG,CAAC,QAAQ,CAAC;YAC1B,IAAI,SAAS,GAAG,CAAC,CAAC,CAAC;YAEnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;gBAC1C,MAAM,SAAS,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;gBAC/B,IAAI,CAAC,SAAS,CAAC,SAAS,IAAI,SAAS,CAAC,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7D,SAAS;gBACX,CAAC;gBAED,UAAU;gBACV,MAAM,eAAe,GAAG,IAAI,CAAC,gBAAgB,CAC3C,cAAc,EACd,SAAS,CAAC,SAAS,CACpB,CAAC;gBAEF,eAAe;gBACf,IAAI,qBAAqB,GAAG,CAAC,CAAC;gBAC9B,KAAK,MAAM,WAAW,IAAI,kBAAkB,EAAE,CAAC;oBAC7C,MAAM,GAAG,GAAG,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,SAAS,EAAE,WAAW,CAAC,CAAC;oBACpE,qBAAqB,GAAG,IAAI,CAAC,GAAG,CAAC,qBAAqB,EAAE,GAAG,CAAC,CAAC;gBAC/D,CAAC;gBAED,SAAS;gBACT,MAAM,QAAQ,GACZ,MAAM,GAAG,eAAe,GAAG,CAAC,CAAC,GAAG,MAAM,CAAC,GAAG,qBAAqB,CAAC;gBAElE,IAAI,QAAQ,GAAG,SAAS,EAAE,CAAC;oBACzB,SAAS,GAAG,QAAQ,CAAC;oBACrB,SAAS,GAAG,CAAC,CAAC;gBAChB,CAAC;YACH,CAAC;YAED,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;gBACnB,MAAM,IAAI,GAAG,SAAS,CAAC,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;gBAC/C,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACvB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;oBACnB,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBAC1C,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,MAAM;YACR,CAAC;QACH,CAAC;QAED,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,+EAA+E;IAC/E,kBAAkB;IAClB,+EAA+E;IAEvE,QAAQ,CAAC,IAAY;QAC3B,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAC;YAC/C,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;YAClC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;QACxB,MAAM,cAAc,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;QAChE,OAAO,GAAG,OAAO,GAAG,cAAc,EAAE,CAAC;IACvC,CAAC;IAEO,YAAY;QAClB,OAAO;YACL,cAAc,EAAE,kBAAkB;YAClC,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE;SAC9C,CAAC;IACJ,CAAC;IAEO,SAAS,CAAC,IAAY,EAAE,MAAc;QAC5C,IAAI,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,GAAG,MAAM,GAAG,IAAI,EAAE,CAAC;IAC5B,CAAC;CACF,CAAA;AAvSY,4CAAgB;2BAAhB,gBAAgB;IAD5B,IAAA,mBAAU,GAAE;IAKR,WAAA,IAAA,eAAM,EAAC,sCAAuB,CAAC,CAAA;qCAA0B,gBAAM;QAChC,sBAAa;QACf,mBAAW;GANhC,gBAAgB,CAuS5B"}

package/dist/index.d.ts

@@ -0,0 +1,9 @@
+/**
+ * Vector 模块导出
+ */
+export { VectorModule } from './vector.module';
+export * from './vikingdb-client.service';
+export * from './embedding.service';
+export * from './knowledge-base-vikingdb.client';
+export * from './vikingdb.types';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,YAAY,EAAE,MAAM,iBAAiB,CAAC;AAC/C,cAAc,2BAA2B,CAAC;AAC1C,cAAc,qBAAqB,CAAC;AACpC,cAAc,kCAAkC,CAAC;AACjD,cAAc,kBAAkB,CAAC"}

package/dist/index.js

@@ -0,0 +1,27 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorModule = void 0;
+/**
+ * Vector 模块导出
+ */
+var vector_module_1 = require("./vector.module");
+Object.defineProperty(exports, "VectorModule", { enumerable: true, get: function () { return vector_module_1.VectorModule; } });
+__exportStar(require("./vikingdb-client.service"), exports);
+__exportStar(require("./embedding.service"), exports);
+__exportStar(require("./knowledge-base-vikingdb.client"), exports);
+__exportStar(require("./vikingdb.types"), exports);
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA;;GAEG;AACH,iDAA+C;AAAtC,6GAAA,YAAY,OAAA;AACrB,4DAA0C;AAC1C,sDAAoC;AACpC,mEAAiD;AACjD,mDAAiC"}

package/dist/knowledge-base-vikingdb.client.d.ts

@@ -0,0 +1,29 @@
+import type { Logger } from 'winston';
+import { VikingDbClientService } from './vikingdb-client.service';
+export interface KnowledgeBaseVectorDocument {
+    documentId: string;
+    knowledgeBaseId: string;
+    tenantId: string;
+    title: string;
+    content: string;
+    summary?: string | null;
+    sourceType: string;
+    sourceUrl?: string | null;
+    mimeType?: string | null;
+    fileExtension?: string | null;
+    embedding: number[];
+}
+export declare class KnowledgeBaseVikingdbClient {
+    private readonly vikingDbClientService;
+    private readonly logger;
+    constructor(vikingDbClientService: VikingDbClientService, logger: Logger);
+    upsertDocument(document: KnowledgeBaseVectorDocument): Promise<void>;
+    deleteDocument(documentId: string): Promise<void>;
+    /**
+     * 批量更新知识库下所有文档的 bots 字段
+     * @param documentIds 该知识库下所有文档的 ID 列表
+     * @param bots 关联的员工 ID 数组
+     */
+    updateKnowledgeBaseBots(documentIds: string[], bots: string[]): Promise<void>;
+}
+//# sourceMappingURL=knowledge-base-vikingdb.client.d.ts.map

package/dist/knowledge-base-vikingdb.client.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-base-vikingdb.client.d.ts","sourceRoot":"","sources":["../src/knowledge-base-vikingdb.client.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACtC,OAAO,EAAE,qBAAqB,EAAE,MAAM,2BAA2B,CAAC;AAElE,MAAM,WAAW,2BAA2B;IAC1C,UAAU,EAAE,MAAM,CAAC;IACnB,eAAe,EAAE,MAAM,CAAC;IACxB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,SAAS,EAAE,MAAM,EAAE,CAAC;CACrB;AAED,qBACa,2BAA2B;IAGpC,OAAO,CAAC,QAAQ,CAAC,qBAAqB;IACL,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADvC,qBAAqB,EAAE,qBAAqB,EACX,MAAM,EAAE,MAAM;IAG5D,cAAc,CAAC,QAAQ,EAAE,2BAA2B,GAAG,OAAO,CAAC,IAAI,CAAC;IA2BpE,cAAc,CAAC,UAAU,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIvD;;;;OAIG;IACG,uBAAuB,CAC3B,WAAW,EAAE,MAAM,EAAE,EACrB,IAAI,EAAE,MAAM,EAAE,GACb,OAAO,CAAC,IAAI,CAAC;CA2BjB"}

package/dist/knowledge-base-vikingdb.client.js

@@ -0,0 +1,85 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.KnowledgeBaseVikingdbClient = void 0;
+const common_1 = require("@nestjs/common");
+const nest_winston_1 = require("nest-winston");
+const vikingdb_client_service_1 = require("./vikingdb-client.service");
+let KnowledgeBaseVikingdbClient = class KnowledgeBaseVikingdbClient {
+    vikingDbClientService;
+    logger;
+    constructor(vikingDbClientService, logger) {
+        this.vikingDbClientService = vikingDbClientService;
+        this.logger = logger;
+    }
+    async upsertDocument(document) {
+        await this.vikingDbClientService.upsert([
+            {
+                id: document.documentId,
+                text: document.content,
+                summary: document.summary ?? undefined,
+                tags: [document.sourceType],
+                taskType: 'knowledge-base',
+                embedding: document.embedding,
+                scope: {
+                    tenantId: document.tenantId,
+                    scopeType: 'knowledge-base',
+                    scopeId: document.knowledgeBaseId,
+                },
+                metadata: {
+                    knowledgeBaseId: document.knowledgeBaseId,
+                    documentId: document.documentId,
+                    title: document.title,
+                    sourceType: document.sourceType,
+                    sourceUrl: document.sourceUrl ?? null,
+                    mimeType: document.mimeType ?? null,
+                    fileExtension: document.fileExtension ?? null,
+                },
+            },
+        ]);
+    }
+    async deleteDocument(documentId) {
+        await this.vikingDbClientService.delete({ ids: [documentId] });
+    }
+    /**
+     * 批量更新知识库下所有文档的 bots 字段
+     * @param documentIds 该知识库下所有文档的 ID 列表
+     * @param bots 关联的员工 ID 数组
+     */
+    async updateKnowledgeBaseBots(documentIds, bots) {
+        if (documentIds.length === 0) {
+            this.logger.info('[KnowledgeBaseVikingdbClient] No documents to update');
+            return;
+        }
+        this.logger.info('[KnowledgeBaseVikingdbClient] Updating bots for documents', {
+            documentCount: documentIds.length,
+            bots,
+        });
+        // 批量更新所有文档的 bots 字段
+        const updateDocs = documentIds.map((docId) => ({
+            id: docId,
+            bots,
+        }));
+        await this.vikingDbClientService.update(updateDocs);
+        this.logger.info('[KnowledgeBaseVikingdbClient] Update completed for documents', { count: documentIds.length });
+    }
+};
+exports.KnowledgeBaseVikingdbClient = KnowledgeBaseVikingdbClient;
+exports.KnowledgeBaseVikingdbClient = KnowledgeBaseVikingdbClient = __decorate([
+    (0, common_1.Injectable)(),
+    __param(0, (0, common_1.Inject)((0, common_1.forwardRef)(() => vikingdb_client_service_1.VikingDbClientService))),
+    __param(1, (0, common_1.Inject)(nest_winston_1.WINSTON_MODULE_PROVIDER)),
+    __metadata("design:paramtypes", [vikingdb_client_service_1.VikingDbClientService, Function])
+], KnowledgeBaseVikingdbClient);
+//# sourceMappingURL=knowledge-base-vikingdb.client.js.map

package/dist/knowledge-base-vikingdb.client.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"knowledge-base-vikingdb.client.js","sourceRoot":"","sources":["../src/knowledge-base-vikingdb.client.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAgE;AAChE,+CAAuD;AAEvD,uEAAkE;AAiB3D,IAAM,2BAA2B,GAAjC,MAAM,2BAA2B;IAGnB;IACiC;IAHpD,YAEmB,qBAA4C,EACX,MAAc;QAD/C,0BAAqB,GAArB,qBAAqB,CAAuB;QACX,WAAM,GAAN,MAAM,CAAQ;IAC/D,CAAC;IAEJ,KAAK,CAAC,cAAc,CAAC,QAAqC;QACxD,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC;YACtC;gBACE,EAAE,EAAE,QAAQ,CAAC,UAAU;gBACvB,IAAI,EAAE,QAAQ,CAAC,OAAO;gBACtB,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,SAAS;gBACtC,IAAI,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC;gBAC3B,QAAQ,EAAE,gBAAgB;gBAC1B,SAAS,EAAE,QAAQ,CAAC,SAAS;gBAC7B,KAAK,EAAE;oBACL,QAAQ,EAAE,QAAQ,CAAC,QAAQ;oBAC3B,SAAS,EAAE,gBAAgB;oBAC3B,OAAO,EAAE,QAAQ,CAAC,eAAe;iBAClC;gBACD,QAAQ,EAAE;oBACR,eAAe,EAAE,QAAQ,CAAC,eAAe;oBACzC,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,KAAK,EAAE,QAAQ,CAAC,KAAK;oBACrB,UAAU,EAAE,QAAQ,CAAC,UAAU;oBAC/B,SAAS,EAAE,QAAQ,CAAC,SAAS,IAAI,IAAI;oBACrC,QAAQ,EAAE,QAAQ,CAAC,QAAQ,IAAI,IAAI;oBACnC,aAAa,EAAE,QAAQ,CAAC,aAAa,IAAI,IAAI;iBAC9C;aACK;SACT,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,UAAkB;QACrC,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,EAAE,GAAG,EAAE,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,uBAAuB,CAC3B,WAAqB,EACrB,IAAc;QAEd,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sDAAsD,CAAC,CAAC;YACzE,OAAO;QACT,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,2DAA2D,EAC3D;YACE,aAAa,EAAE,WAAW,CAAC,MAAM;YACjC,IAAI;SACL,CACF,CAAC;QAEF,oBAAoB;QACpB,MAAM,UAAU,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;YAC7C,EAAE,EAAE,KAAK;YACT,IAAI;SACL,CAAC,CAAC,CAAC;QAEJ,MAAM,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;QAEpD,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,8DAA8D,EAC9D,EAAE,KAAK,EAAE,WAAW,CAAC,MAAM,EAAE,CAC9B,CAAC;IACJ,CAAC;CACF,CAAA;AAzEY,kEAA2B;sCAA3B,2BAA2B;IADvC,IAAA,mBAAU,GAAE;IAGR,WAAA,IAAA,eAAM,EAAC,IAAA,mBAAU,EAAC,GAAG,EAAE,CAAC,+CAAqB,CAAC,CAAC,CAAA;IAE/C,WAAA,IAAA,eAAM,EAAC,sCAAuB,CAAC,CAAA;qCADQ,+CAAqB;GAHpD,2BAA2B,CAyEvC"}

package/dist/vector.module.d.ts

@@ -0,0 +1,3 @@
+export declare class VectorModule {
+}
+//# sourceMappingURL=vector.module.d.ts.map

package/dist/vector.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector.module.d.ts","sourceRoot":"","sources":["../src/vector.module.ts"],"names":[],"mappings":"AAWA,qBAmBa,YAAY;CAAG"}

package/dist/vector.module.js

@@ -0,0 +1,44 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VectorModule = void 0;
+/**
+ * Vector Module
+ * 向量服务模块，包含 VikingDB 客户端和 Embedding 服务
+ */
+const common_1 = require("@nestjs/common");
+const axios_1 = require("@nestjs/axios");
+const vikingdb_client_service_1 = require("./vikingdb-client.service");
+const embedding_service_1 = require("./embedding.service");
+const knowledge_base_vikingdb_client_1 = require("./knowledge-base-vikingdb.client");
+const config_1 = require("@nestjs/config");
+let VectorModule = class VectorModule {
+};
+exports.VectorModule = VectorModule;
+exports.VectorModule = VectorModule = __decorate([
+    (0, common_1.Module)({
+        imports: [
+            config_1.ConfigModule,
+            axios_1.HttpModule.register({
+                timeout: 30000,
+                maxRedirects: 5,
+            }),
+        ],
+        providers: [
+            vikingdb_client_service_1.VikingDbClientService,
+            embedding_service_1.EmbeddingService,
+            knowledge_base_vikingdb_client_1.KnowledgeBaseVikingdbClient,
+        ],
+        exports: [
+            vikingdb_client_service_1.VikingDbClientService,
+            embedding_service_1.EmbeddingService,
+            knowledge_base_vikingdb_client_1.KnowledgeBaseVikingdbClient,
+        ],
+    })
+], VectorModule);
+//# sourceMappingURL=vector.module.js.map

package/dist/vector.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"vector.module.js","sourceRoot":"","sources":["../src/vector.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA;;;GAGG;AACH,2CAAoD;AACpD,yCAA2C;AAC3C,uEAAkE;AAClE,2DAAuD;AACvD,qFAA+E;AAC/E,2CAA8C;AAqBvC,IAAM,YAAY,GAAlB,MAAM,YAAY;CAAG,CAAA;AAAf,oCAAY;uBAAZ,YAAY;IAnBxB,IAAA,eAAM,EAAC;QACN,OAAO,EAAE;YACP,qBAAY;YACZ,kBAAU,CAAC,QAAQ,CAAC;gBAClB,OAAO,EAAE,KAAK;gBACd,YAAY,EAAE,CAAC;aAChB,CAAC;SACH;QACD,SAAS,EAAE;YACT,+CAAqB;YACrB,oCAAgB;YAChB,4DAA2B;SAC5B;QACD,OAAO,EAAE;YACP,+CAAqB;YACrB,oCAAgB;YAChB,4DAA2B;SAC5B;KACF,CAAC;GACW,YAAY,CAAG"}