@dofe/infra-vector 0.1.1

package/dist/vikingdb-client.service.d.ts

@@ -0,0 +1,134 @@
+/**
+ * VikingDB Client Service
+ * VikingDB 向量数据库客户端服务
+ */
+import { OnModuleInit } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
+import { Logger } from 'winston';
+import { HttpService } from '@nestjs/axios';
+import type { MemoryDocument, RetrievedCandidate, DeleteOptions, ScopeMap, VolcengineFieldInfo } from './vikingdb.types';
+export declare class VikingDbClientService implements OnModuleInit {
+    private readonly logger;
+    private readonly configService;
+    private readonly httpService;
+    private config;
+    private volcService?;
+    constructor(logger: Logger, configService: ConfigService, httpService: HttpService);
+    onModuleInit(): void;
+    /**
+     * Volcengine: 获取 collection schema（字段列表）
+     */
+    getVolcengineCollectionSchema(): Promise<{
+        collectionName: string;
+        fields: Array<{
+            FieldName: string;
+            FieldType: string;
+            Dim?: number;
+        }>;
+        indexNames: string[];
+    } | null>;
+    /**
+     * Volcengine: 连通性自检（不写入数据）
+     */
+    volcengineSelfCheck(): Promise<{
+        ok: boolean;
+        provider: 'volcengine';
+        region?: string;
+        collectionName?: string;
+        indexName?: string;
+        latencyMs: number;
+        checks: {
+            collectionInfo: boolean;
+            indexInfo: boolean;
+            sampleSearch: boolean;
+        };
+        warnings: string[];
+        error?: string;
+    } | null>;
+    /**
+     * Volcengine: 校验远程 collection schema 是否匹配预期
+     * 返回缺失字段/类型不匹配的警告列表
+     */
+    validateVolcengineSchema(): Promise<{
+        ok: boolean;
+        collectionName: string;
+        warnings: string[];
+        errors: string[];
+        remoteFields?: Array<{
+            FieldName: string;
+            FieldType: string;
+            Dim?: number;
+        }>;
+        expectedFields: VolcengineFieldInfo[];
+    } | null>;
+    /**
+     * 健康检查
+     */
+    healthCheck(): Promise<{
+        healthy: boolean;
+        latency: number;
+        error?: string;
+    }>;
+    /**
+     * 向量搜索
+     */
+    vectorSearch(vector: number[], scope: ScopeMap, options?: {
+        topK?: number;
+        taskType?: string;
+    }): Promise<RetrievedCandidate[]>;
+    /**
+     * BM25 搜索
+     */
+    bm25Search(query: string, scope: ScopeMap, options?: {
+        topK?: number;
+        taskType?: string;
+    }): Promise<RetrievedCandidate[]>;
+    /**
+     * 混合搜索（向量 + BM25）
+     */
+    hybridSearch(vector: number[], query: string, scope: ScopeMap, options?: {
+        topK?: number;
+        candidateK?: number;
+        taskType?: string;
+        vectorWeight?: number;
+        bm25Weight?: number;
+    }): Promise<RetrievedCandidate[]>;
+    /**
+     * Upsert 文档
+     */
+    upsert(documents: MemoryDocument[]): Promise<void>;
+    /**
+     * 更新文档（部分更新，仅更新指定字段）
+     */
+    update(documents: Array<{
+        id: string;
+        [key: string]: unknown;
+    }>): Promise<void>;
+    /**
+     * 删除文档
+     */
+    delete(options: DeleteOptions): Promise<void>;
+    /**
+     * 按作用域删除所有文档
+     */
+    deleteByScope(scope: ScopeMap): Promise<void>;
+    private resolveIndexName;
+    private resolveCollectionName;
+    private resolveCollectionNameOrAlias;
+    private buildVolcFilter;
+    private normalizeVolcSearchResponse;
+    private parseMetadata;
+    private buildVolcUpsertFields;
+    private buildMetadata;
+    private request;
+    private buildUrl;
+    private buildHeaders;
+    private parseHeaders;
+    private normalizeCandidates;
+    /**
+     * Reciprocal Rank Fusion 算法
+     * 合并向量搜索和 BM25 搜索的结果
+     */
+    private reciprocalRankFusion;
+}
+//# sourceMappingURL=vikingdb-client.service.d.ts.map

package/dist/vikingdb-client.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"vikingdb-client.service.d.ts","sourceRoot":"","sources":["../src/vikingdb-client.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;AAI5C,OAAO,KAAK,EAEV,cAAc,EACd,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,mBAAmB,EACpB,MAAM,kBAAkB,CAAC;AAI1B,qBACa,qBAAsB,YAAW,YAAY;IAKrB,OAAO,CAAC,QAAQ,CAAC,MAAM;IACxD,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,WAAW;IAN9B,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,WAAW,CAAC,CAAoD;gBAGpB,MAAM,EAAE,MAAM,EAC/C,aAAa,EAAE,aAAa,EAC5B,WAAW,EAAE,WAAW;IAG3C,YAAY;IAuLZ;;OAEG;IACG,6BAA6B,IAAI,OAAO,CAAC;QAC7C,cAAc,EAAE,MAAM,CAAC;QACvB,MAAM,EAAE,KAAK,CAAC;YAAE,SAAS,EAAE,MAAM,CAAC;YAAC,SAAS,EAAE,MAAM,CAAC;YAAC,GAAG,CAAC,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;QACtE,UAAU,EAAE,MAAM,EAAE,CAAC;KACtB,GAAG,IAAI,CAAC;IA8BT;;OAEG;IACG,mBAAmB,IAAI,OAAO,CAAC;QACnC,EAAE,EAAE,OAAO,CAAC;QACZ,QAAQ,EAAE,YAAY,CAAC;QACvB,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,cAAc,CAAC,EAAE,MAAM,CAAC;QACxB,SAAS,CAAC,EAAE,MAAM,CAAC;QACnB,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE;YACN,cAAc,EAAE,OAAO,CAAC;YACxB,SAAS,EAAE,OAAO,CAAC;YACnB,YAAY,EAAE,OAAO,CAAC;SACvB,CAAC;QACF,QAAQ,EAAE,MAAM,EAAE,CAAC;QACnB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,GAAG,IAAI,CAAC;IAiFT;;;OAGG;IACG,wBAAwB,IAAI,OAAO,CAAC;QACxC,EAAE,EAAE,OAAO,CAAC;QACZ,cAAc,EAAE,MAAM,CAAC;QACvB,QAAQ,EAAE,MAAM,EAAE,CAAC;QACnB,MAAM,EAAE,MAAM,EAAE,CAAC;QACjB,YAAY,CAAC,EAAE,KAAK,CAAC;YACnB,SAAS,EAAE,MAAM,CAAC;YAClB,SAAS,EAAE,MAAM,CAAC;YAClB,GAAG,CAAC,EAAE,MAAM,CAAC;SACd,CAAC,CAAC;QACH,cAAc,EAAE,mBAAmB,EAAE,CAAC;KACvC,GAAG,IAAI,CAAC;IAyET;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC;QAC3B,OAAO,EAAE,OAAO,CAAC;QACjB,OAAO,EAAE,MAAM,CAAC;QAChB,KAAK,CAAC,EAAE,MAAM,CAAC;KAChB,CAAC;IAyCF;;OAEG;IACG,YAAY,CAChB,MAAM,EAAE,MAAM,EAAE,EAChB,KAAK,EAAE,QAAQ,EACf,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAO,GACjD,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAmDhC;;OAEG;IACG,UAAU,CACd,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,QAAQ,EACf,OAAO,GAAE;QAAE,IAAI,CAAC,EAAE,MAAM,CAAC;QAAC,QAAQ,CAAC,EAAE,MAAM,CAAA;KAAO,GACjD,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAkDhC;;OAEG;IACG,YAAY,CAChB,MAAM,EAAE,MAAM,EAAE,EAChB,KAAK,EAAE,MAAM,EACb,KAAK,EAAE,QAAQ,EACf,OAAO,GAAE;QACP,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,YAAY,CAAC,EAAE,MAAM,CAAC;QACtB,UAAU,CAAC,EAAE,MAAM,CAAC;KAChB,GACL,OAAO,CAAC,kBAAkB,EAAE,CAAC;IA2BhC;;OAEG;IACG,MAAM,CAAC,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAwCxD;;OAEG;IACG,MAAM,CACV,SAAS,EAAE,KAAK,CAAC;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAA;KAAE,CAAC,GACvD,OAAO,CAAC,IAAI,CAAC;IAsChB;;OAEG;IACG,MAAM,CAAC,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAkDnD;;OAEG;IACG,aAAa,CAAC,KAAK,EAAE,QAAQ,GAAG,OAAO,CAAC,IAAI,CAAC;IAQnD,OAAO,CAAC,gBAAgB;IASxB,OAAO,CAAC,qBAAqB;IAe7B,OAAO,CAAC,4BAA4B;IAcpC,OAAO,CAAC,eAAe;IAkDvB,OAAO,CAAC,2BAA2B;IA0DnC,OAAO,CAAC,aAAa;IAarB,OAAO,CAAC,qBAAqB;IA0E7B,OAAO,CAAC,aAAa;YAuBP,OAAO;IAiDrB,OAAO,CAAC,QAAQ;IAQhB,OAAO,CAAC,YAAY;IAapB,OAAO,CAAC,YAAY;IAgBpB,OAAO,CAAC,mBAAmB;IA6C3B;;;OAGG;IACH,OAAO,CAAC,oBAAoB;CAkD7B"}