@zilliz/claude-context-core 0.0.13 → 0.1.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/context.d.ts +50 -8
- package/dist/context.d.ts.map +1 -1
- package/dist/context.js +267 -79
- package/dist/context.js.map +1 -1
- package/dist/embedding/base-embedding.d.ts +6 -0
- package/dist/embedding/base-embedding.d.ts.map +1 -1
- package/dist/embedding/base-embedding.js.map +1 -1
- package/dist/embedding/gemini-embedding.d.ts +1 -0
- package/dist/embedding/gemini-embedding.d.ts.map +1 -1
- package/dist/embedding/gemini-embedding.js +4 -0
- package/dist/embedding/gemini-embedding.js.map +1 -1
- package/dist/embedding/ollama-embedding.d.ts +1 -10
- package/dist/embedding/ollama-embedding.d.ts.map +1 -1
- package/dist/embedding/ollama-embedding.js +42 -52
- package/dist/embedding/ollama-embedding.js.map +1 -1
- package/dist/embedding/openai-embedding.d.ts +2 -2
- package/dist/embedding/openai-embedding.d.ts.map +1 -1
- package/dist/embedding/openai-embedding.js +78 -32
- package/dist/embedding/openai-embedding.js.map +1 -1
- package/dist/embedding/voyageai-embedding.d.ts +1 -0
- package/dist/embedding/voyageai-embedding.d.ts.map +1 -1
- package/dist/embedding/voyageai-embedding.js +4 -0
- package/dist/embedding/voyageai-embedding.js.map +1 -1
- package/dist/vectordb/index.d.ts +1 -1
- package/dist/vectordb/index.d.ts.map +1 -1
- package/dist/vectordb/index.js.map +1 -1
- package/dist/vectordb/milvus-restful-vectordb.d.ts +5 -1
- package/dist/vectordb/milvus-restful-vectordb.d.ts.map +1 -1
- package/dist/vectordb/milvus-restful-vectordb.js +246 -0
- package/dist/vectordb/milvus-restful-vectordb.js.map +1 -1
- package/dist/vectordb/milvus-vectordb.d.ts +4 -1
- package/dist/vectordb/milvus-vectordb.d.ts.map +1 -1
- package/dist/vectordb/milvus-vectordb.js +201 -0
- package/dist/vectordb/milvus-vectordb.js.map +1 -1
- package/dist/vectordb/types.d.ts +41 -3
- package/dist/vectordb/types.d.ts.map +1 -1
- package/dist/vectordb/types.js.map +1 -1
- package/package.json +1 -1
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"voyageai-embedding.d.ts","sourceRoot":"","sources":["../../src/embedding/voyageai-embedding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAE9D,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,iBAAkB,SAAQ,SAAS;IAC5C,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,SAAS,CAAoC;IACrD,SAAS,CAAC,SAAS,EAAE,MAAM,CAAS;gBAExB,MAAM,EAAE,uBAAuB;IAW3C,OAAO,CAAC,mBAAmB;IAoB3B,OAAO,CAAC,uBAAuB;IAiBzB,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAoB7C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAyB7D,YAAY,IAAI,MAAM;IAItB,WAAW,IAAI,MAAM;IAIrB;;;OAGG;IACH,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAK7B;;;OAGG;IACH,YAAY,CAAC,SAAS,EAAE,UAAU,GAAG,OAAO,GAAG,IAAI;IAInD;;OAEG;IACH,SAAS,IAAI,cAAc;IAI3B;;OAEG;IACH,MAAM,CAAC,kBAAkB,IAAI,MAAM,CAAC,MAAM,EAAE;QAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC;CAkG1H"}
|
|
1
|
+
{"version":3,"file":"voyageai-embedding.d.ts","sourceRoot":"","sources":["../../src/embedding/voyageai-embedding.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAC1C,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,MAAM,kBAAkB,CAAC;AAE9D,MAAM,WAAW,uBAAuB;IACpC,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAClB;AAED,qBAAa,iBAAkB,SAAQ,SAAS;IAC5C,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,MAAM,CAA0B;IACxC,OAAO,CAAC,SAAS,CAAgB;IACjC,OAAO,CAAC,SAAS,CAAoC;IACrD,SAAS,CAAC,SAAS,EAAE,MAAM,CAAS;gBAExB,MAAM,EAAE,uBAAuB;IAW3C,OAAO,CAAC,mBAAmB;IAoB3B,OAAO,CAAC,uBAAuB;IAiBzB,eAAe,IAAI,OAAO,CAAC,MAAM,CAAC;IAKlC,KAAK,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,eAAe,CAAC;IAoB7C,UAAU,CAAC,KAAK,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAyB7D,YAAY,IAAI,MAAM;IAItB,WAAW,IAAI,MAAM;IAIrB;;;OAGG;IACH,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,IAAI;IAK7B;;;OAGG;IACH,YAAY,CAAC,SAAS,EAAE,UAAU,GAAG,OAAO,GAAG,IAAI;IAInD;;OAEG;IACH,SAAS,IAAI,cAAc;IAI3B;;OAEG;IACH,MAAM,CAAC,kBAAkB,IAAI,MAAM,CAAC,MAAM,EAAE;QAAE,SAAS,EAAE,MAAM,GAAG,MAAM,CAAC;QAAC,aAAa,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,MAAM,CAAA;KAAE,CAAC;CAkG1H"}
|
|
@@ -53,6 +53,10 @@ class VoyageAIEmbedding extends base_embedding_1.Embedding {
|
|
|
53
53
|
this.dimension = 1024;
|
|
54
54
|
}
|
|
55
55
|
}
|
|
56
|
+
async detectDimension() {
|
|
57
|
+
// VoyageAI doesn't need dynamic detection, return configured dimension
|
|
58
|
+
return this.dimension;
|
|
59
|
+
}
|
|
56
60
|
async embed(text) {
|
|
57
61
|
const processedText = this.preprocessText(text);
|
|
58
62
|
const model = this.config.model || 'voyage-code-3';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"voyageai-embedding.js","sourceRoot":"","sources":["../../src/embedding/voyageai-embedding.ts"],"names":[],"mappings":";;;AAAA,uCAA0C;AAC1C,qDAA8D;AAO9D,MAAa,iBAAkB,SAAQ,0BAAS;IAO5C,YAAY,MAA+B;QACvC,KAAK,EAAE,CAAC;QALJ,cAAS,GAAW,IAAI,CAAC,CAAC,sCAAsC;QAChE,cAAS,GAAyB,UAAU,CAAC;QAC3C,cAAS,GAAW,KAAK,CAAC,CAAC,qBAAqB;QAItD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,yBAAc,CAAC;YAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;SACxB,CAAC,CAAC;QAEH,6DAA6D;QAC7D,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,CAAC,CAAC;IAC9D,CAAC;IAEO,mBAAmB,CAAC,KAAa;QACrC,MAAM,eAAe,GAAG,iBAAiB,CAAC,kBAAkB,EAAE,CAAC;QAC/D,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAEzC,IAAI,SAAS,EAAE,CAAC;YACZ,mFAAmF;YACnF,IAAI,OAAO,SAAS,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC1C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,oBAAoB;YAC/C,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;YACzC,CAAC;YACD,iDAAiD;YACjD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,aAAa,CAAC;QAC7C,CAAC;aAAM,CAAC;YACJ,8DAA8D;YAC9D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QAC3B,CAAC;IACL,CAAC;IAEO,uBAAuB,CAAC,KAAa;QACzC,MAAM,eAAe,GAAG,iBAAiB,CAAC,kBAAkB,EAAE,CAAC;QAC/D,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAEzC,IAAI,SAAS,EAAE,CAAC;YACZ,mFAAmF;YACnF,IAAI,OAAO,SAAS,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC1C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,oBAAoB;YAC/C,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;YACzC,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,2CAA2C;YAC3C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAC1B,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACpB,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACrC,KAAK,EAAE,aAAa;YACpB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACrE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO;YACH,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YAClC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC;IACN,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACrC,KAAK,EAAE,cAAc;YACrB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC9B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YACpE,CAAC;YACD,OAAO;gBACH,MAAM,EAAE,IAAI,CAAC,SAAS;gBACtB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC5B,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAED,YAAY;QACR,OAAO,IAAI,CAAC,SAAS,CAAC;IAC1B,CAAC;IAED,WAAW;QACP,OAAO,UAAU,CAAC;IACtB,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,KAAa;QAClB,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,YAAY,CAAC,SAA+B;QACxC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,SAAS;QACL,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,kBAAkB;QACrB,OAAO;YACH,4BAA4B;YAC5B,gBAAgB,EAAE;gBACd,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,6DAA6D;aAC7E;YACD,YAAY,EAAE;gBACV,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,kEAAkE;aAClF;YACD,iBAAiB,EAAE;gBACf,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,eAAe,EAAE;gBACb,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,qDAAqD;aACrE;YACD,6BAA6B;YAC7B,kBAAkB,EAAE;gBAChB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,yCAAyC;aACzD;YACD,cAAc,EAAE;gBACZ,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,uCAAuC;aACvD;YACD,uBAAuB,EAAE;gBACrB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,+CAA+C;aAC/D;YACD,yBAAyB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,gBAAgB;YAChB,gBAAgB,EAAE;gBACd,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,eAAe,EAAE;gBACb,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,wCAAwC;aACxD;YACD,UAAU,EAAE;gBACR,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,eAAe,EAAE;gBACb,SAAS,EAAE,GAAG;gBACd,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,qCAAqC;aACrD;YACD,UAAU,EAAE;gBACR,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,qCAAqC;aACrD;YACD,sBAAsB;YACtB,WAAW,EAAE;gBACT,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,WAAW,EAAE;gBACT,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,gBAAgB,EAAE;gBACd,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,yBAAyB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,yBAAyB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;SACJ,CAAC;IACN,CAAC;CACJ;
|
|
1
|
+
{"version":3,"file":"voyageai-embedding.js","sourceRoot":"","sources":["../../src/embedding/voyageai-embedding.ts"],"names":[],"mappings":";;;AAAA,uCAA0C;AAC1C,qDAA8D;AAO9D,MAAa,iBAAkB,SAAQ,0BAAS;IAO5C,YAAY,MAA+B;QACvC,KAAK,EAAE,CAAC;QALJ,cAAS,GAAW,IAAI,CAAC,CAAC,sCAAsC;QAChE,cAAS,GAAyB,UAAU,CAAC;QAC3C,cAAS,GAAW,KAAK,CAAC,CAAC,qBAAqB;QAItD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,IAAI,yBAAc,CAAC;YAC7B,MAAM,EAAE,MAAM,CAAC,MAAM;SACxB,CAAC,CAAC;QAEH,6DAA6D;QAC7D,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,CAAC,CAAC;IAC9D,CAAC;IAEO,mBAAmB,CAAC,KAAa;QACrC,MAAM,eAAe,GAAG,iBAAiB,CAAC,kBAAkB,EAAE,CAAC;QAC/D,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAEzC,IAAI,SAAS,EAAE,CAAC;YACZ,mFAAmF;YACnF,IAAI,OAAO,SAAS,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC1C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,oBAAoB;YAC/C,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;YACzC,CAAC;YACD,iDAAiD;YACjD,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,aAAa,CAAC;QAC7C,CAAC;aAAM,CAAC;YACJ,8DAA8D;YAC9D,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QAC3B,CAAC;IACL,CAAC;IAEO,uBAAuB,CAAC,KAAa;QACzC,MAAM,eAAe,GAAG,iBAAiB,CAAC,kBAAkB,EAAE,CAAC;QAC/D,MAAM,SAAS,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;QAEzC,IAAI,SAAS,EAAE,CAAC;YACZ,mFAAmF;YACnF,IAAI,OAAO,SAAS,CAAC,SAAS,KAAK,QAAQ,EAAE,CAAC;gBAC1C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC,CAAC,oBAAoB;YAC/C,CAAC;iBAAM,CAAC;gBACJ,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC,SAAS,CAAC;YACzC,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,2CAA2C;YAC3C,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAC1B,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,uEAAuE;QACvE,OAAO,IAAI,CAAC,SAAS,CAAC;IAC1B,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,IAAY;QACpB,MAAM,aAAa,GAAG,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC;QAChD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACrC,KAAK,EAAE,aAAa;YACpB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;YACrE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO;YACH,MAAM,EAAE,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS;YAClC,SAAS,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC;IACN,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,KAAe;QAC5B,MAAM,cAAc,GAAG,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,eAAe,CAAC;QAEnD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACrC,KAAK,EAAE,cAAc;YACrB,KAAK,EAAE,KAAK;YACZ,SAAS,EAAE,IAAI,CAAC,SAAS;SAC5B,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC9D,CAAC;QAED,OAAO,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YAC9B,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;gBAClB,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;YACpE,CAAC;YACD,OAAO;gBACH,MAAM,EAAE,IAAI,CAAC,SAAS;gBACtB,SAAS,EAAE,IAAI,CAAC,SAAS;aAC5B,CAAC;QACN,CAAC,CAAC,CAAC;IACP,CAAC;IAED,YAAY;QACR,OAAO,IAAI,CAAC,SAAS,CAAC;IAC1B,CAAC;IAED,WAAW;QACP,OAAO,UAAU,CAAC;IACtB,CAAC;IAED;;;OAGG;IACH,QAAQ,CAAC,KAAa;QAClB,IAAI,CAAC,MAAM,CAAC,KAAK,GAAG,KAAK,CAAC;QAC1B,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAED;;;OAGG;IACH,YAAY,CAAC,SAA+B;QACxC,IAAI,CAAC,SAAS,GAAG,SAAS,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,SAAS;QACL,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAED;;OAEG;IACH,MAAM,CAAC,kBAAkB;QACrB,OAAO;YACH,4BAA4B;YAC5B,gBAAgB,EAAE;gBACd,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,6DAA6D;aAC7E;YACD,YAAY,EAAE;gBACV,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,kEAAkE;aAClF;YACD,iBAAiB,EAAE;gBACf,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,eAAe,EAAE;gBACb,SAAS,EAAE,gCAAgC;gBAC3C,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,qDAAqD;aACrE;YACD,6BAA6B;YAC7B,kBAAkB,EAAE;gBAChB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,yCAAyC;aACzD;YACD,cAAc,EAAE;gBACZ,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,uCAAuC;aACvD;YACD,uBAAuB,EAAE;gBACrB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,+CAA+C;aAC/D;YACD,yBAAyB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,gBAAgB;YAChB,gBAAgB,EAAE;gBACd,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,eAAe,EAAE;gBACb,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,wCAAwC;aACxD;YACD,UAAU,EAAE;gBACR,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,gCAAgC;aAChD;YACD,eAAe,EAAE;gBACb,SAAS,EAAE,GAAG;gBACd,aAAa,EAAE,KAAK;gBACpB,WAAW,EAAE,qCAAqC;aACrD;YACD,UAAU,EAAE;gBACR,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,qCAAqC;aACrD;YACD,sBAAsB;YACtB,WAAW,EAAE;gBACT,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,WAAW,EAAE;gBACT,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,gBAAgB,EAAE;gBACd,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,yBAAyB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;YACD,yBAAyB,EAAE;gBACvB,SAAS,EAAE,IAAI;gBACf,aAAa,EAAE,IAAI;gBACnB,WAAW,EAAE,cAAc;aAC9B;SACJ,CAAC;IACN,CAAC;CACJ;AA9OD,8CA8OC"}
|
package/dist/vectordb/index.d.ts
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
export { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase, COLLECTION_LIMIT_MESSAGE } from './types';
|
|
1
|
+
export { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase, HybridSearchRequest, HybridSearchOptions, HybridSearchResult, RerankStrategy, COLLECTION_LIMIT_MESSAGE } from './types';
|
|
2
2
|
export { MilvusRestfulVectorDatabase, MilvusRestfulConfig } from './milvus-restful-vectordb';
|
|
3
3
|
export { MilvusVectorDatabase, MilvusConfig } from './milvus-vectordb';
|
|
4
4
|
export { ClusterManager, ZillizConfig, Project, Cluster, CreateFreeClusterRequest, CreateFreeClusterResponse, CreateFreeClusterWithDetailsResponse, DescribeClusterResponse } from './zilliz-utils';
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":"AACA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,EACd,wBAAwB,EAC3B,MAAM,SAAS,CAAC;AAGjB,OAAO,EAAE,2BAA2B,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EACH,cAAc,EACd,YAAY,EACZ,OAAO,EACP,OAAO,EACP,wBAAwB,EACxB,yBAAyB,EACzB,oCAAoC,EACpC,uBAAuB,EAC1B,MAAM,gBAAgB,CAAC"}
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":"AACA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,EACd,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,EAClB,cAAc,EACd,wBAAwB,EAC3B,MAAM,SAAS,CAAC;AAGjB,OAAO,EAAE,2BAA2B,EAAE,mBAAmB,EAAE,MAAM,2BAA2B,CAAC;AAC7F,OAAO,EAAE,oBAAoB,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACvE,OAAO,EACH,cAAc,EACd,YAAY,EACZ,OAAO,EACP,OAAO,EACP,wBAAwB,EACxB,yBAAyB,EACzB,oCAAoC,EACpC,uBAAuB,EAC1B,MAAM,gBAAgB,CAAC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":";;;AAAA,iCAAiC;AACjC,
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/vectordb/index.ts"],"names":[],"mappings":";;;AAAA,iCAAiC;AACjC,iCAUiB;AADb,iHAAA,wBAAwB,OAAA;AAG5B,+BAA+B;AAC/B,qEAA6F;AAApF,sIAAA,2BAA2B,OAAA;AACpC,qDAAuE;AAA9D,uHAAA,oBAAoB,OAAA;AAC7B,+CASwB;AARpB,8GAAA,cAAc,OAAA"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase } from './types';
|
|
1
|
+
import { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase, HybridSearchRequest, HybridSearchOptions, HybridSearchResult } from './types';
|
|
2
2
|
export interface MilvusRestfulConfig {
|
|
3
3
|
address?: string;
|
|
4
4
|
token?: string;
|
|
@@ -47,5 +47,9 @@ export declare class MilvusRestfulVectorDatabase implements VectorDatabase {
|
|
|
47
47
|
search(collectionName: string, queryVector: number[], options?: SearchOptions): Promise<VectorSearchResult[]>;
|
|
48
48
|
delete(collectionName: string, ids: string[]): Promise<void>;
|
|
49
49
|
query(collectionName: string, filter: string, outputFields: string[], limit?: number): Promise<Record<string, any>[]>;
|
|
50
|
+
createHybridCollection(collectionName: string, dimension: number, description?: string): Promise<void>;
|
|
51
|
+
private createHybridIndexes;
|
|
52
|
+
insertHybrid(collectionName: string, documents: VectorDocument[]): Promise<void>;
|
|
53
|
+
hybridSearch(collectionName: string, searchRequests: HybridSearchRequest[], options?: HybridSearchOptions): Promise<HybridSearchResult[]>;
|
|
50
54
|
}
|
|
51
55
|
//# sourceMappingURL=milvus-restful-vectordb.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"milvus-restful-vectordb.d.ts","sourceRoot":"","sources":["../../src/vectordb/milvus-restful-vectordb.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,
|
|
1
|
+
{"version":3,"file":"milvus-restful-vectordb.d.ts","sourceRoot":"","sources":["../../src/vectordb/milvus-restful-vectordb.ts"],"names":[],"mappings":"AAAA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,EACd,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,EAErB,MAAM,SAAS,CAAC;AAGjB,MAAM,WAAW,mBAAmB;IAChC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAwBD;;;;GAIG;AACH,qBAAa,2BAA4B,YAAW,cAAc;IAC9D,SAAS,CAAC,MAAM,EAAE,mBAAmB,CAAC;IACtC,OAAO,CAAC,OAAO,CAAuB;IACtC,SAAS,CAAC,qBAAqB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBAEnC,MAAM,EAAE,mBAAmB;YAOzB,UAAU;YAKV,gBAAgB;IAY9B;;;OAGG;cACa,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAejD;;OAEG;cACa,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAOlD;;OAEG;YACW,WAAW;IA4CnB,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoFtG;;OAEG;YACW,WAAW;IAuBzB;;OAEG;YACW,cAAc;IAatB,cAAc,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAerD,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAkBvD,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IAgBpC,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA+B1E,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IAgE7G,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAuB5D,KAAK,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;IA4BrH,sBAAsB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;YA+F9F,mBAAmB;IAwC3B,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAmChF,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;CAsGlJ"}
|
|
@@ -401,6 +401,252 @@ class MilvusRestfulVectorDatabase {
|
|
|
401
401
|
throw error;
|
|
402
402
|
}
|
|
403
403
|
}
|
|
404
|
+
async createHybridCollection(collectionName, dimension, description) {
|
|
405
|
+
try {
|
|
406
|
+
const restfulConfig = this.config;
|
|
407
|
+
const collectionSchema = {
|
|
408
|
+
collectionName,
|
|
409
|
+
dbName: restfulConfig.database,
|
|
410
|
+
schema: {
|
|
411
|
+
enableDynamicField: false,
|
|
412
|
+
functions: [
|
|
413
|
+
{
|
|
414
|
+
name: "content_bm25_emb",
|
|
415
|
+
description: "content bm25 function",
|
|
416
|
+
type: "BM25",
|
|
417
|
+
inputFieldNames: ["content"],
|
|
418
|
+
outputFieldNames: ["sparse_vector"],
|
|
419
|
+
params: {},
|
|
420
|
+
},
|
|
421
|
+
],
|
|
422
|
+
fields: [
|
|
423
|
+
{
|
|
424
|
+
fieldName: "id",
|
|
425
|
+
dataType: "VarChar",
|
|
426
|
+
isPrimary: true,
|
|
427
|
+
elementTypeParams: {
|
|
428
|
+
max_length: 512
|
|
429
|
+
}
|
|
430
|
+
},
|
|
431
|
+
{
|
|
432
|
+
fieldName: "content",
|
|
433
|
+
dataType: "VarChar",
|
|
434
|
+
elementTypeParams: {
|
|
435
|
+
max_length: 65535,
|
|
436
|
+
enable_analyzer: true
|
|
437
|
+
}
|
|
438
|
+
},
|
|
439
|
+
{
|
|
440
|
+
fieldName: "vector",
|
|
441
|
+
dataType: "FloatVector",
|
|
442
|
+
elementTypeParams: {
|
|
443
|
+
dim: dimension
|
|
444
|
+
}
|
|
445
|
+
},
|
|
446
|
+
{
|
|
447
|
+
fieldName: "sparse_vector",
|
|
448
|
+
dataType: "SparseFloatVector"
|
|
449
|
+
},
|
|
450
|
+
{
|
|
451
|
+
fieldName: "relativePath",
|
|
452
|
+
dataType: "VarChar",
|
|
453
|
+
elementTypeParams: {
|
|
454
|
+
max_length: 1024
|
|
455
|
+
}
|
|
456
|
+
},
|
|
457
|
+
{
|
|
458
|
+
fieldName: "startLine",
|
|
459
|
+
dataType: "Int64"
|
|
460
|
+
},
|
|
461
|
+
{
|
|
462
|
+
fieldName: "endLine",
|
|
463
|
+
dataType: "Int64"
|
|
464
|
+
},
|
|
465
|
+
{
|
|
466
|
+
fieldName: "fileExtension",
|
|
467
|
+
dataType: "VarChar",
|
|
468
|
+
elementTypeParams: {
|
|
469
|
+
max_length: 32
|
|
470
|
+
}
|
|
471
|
+
},
|
|
472
|
+
{
|
|
473
|
+
fieldName: "metadata",
|
|
474
|
+
dataType: "VarChar",
|
|
475
|
+
elementTypeParams: {
|
|
476
|
+
max_length: 65535
|
|
477
|
+
}
|
|
478
|
+
}
|
|
479
|
+
]
|
|
480
|
+
}
|
|
481
|
+
};
|
|
482
|
+
// Step 1: Create collection with schema and functions
|
|
483
|
+
await createCollectionWithLimitCheck(this.makeRequest.bind(this), collectionSchema);
|
|
484
|
+
// Step 2: Create indexes for both vector fields
|
|
485
|
+
await this.createHybridIndexes(collectionName);
|
|
486
|
+
// Step 3: Load collection to memory for searching
|
|
487
|
+
await this.loadCollection(collectionName);
|
|
488
|
+
}
|
|
489
|
+
catch (error) {
|
|
490
|
+
console.error(`❌ Failed to create hybrid collection '${collectionName}':`, error);
|
|
491
|
+
throw error;
|
|
492
|
+
}
|
|
493
|
+
}
|
|
494
|
+
async createHybridIndexes(collectionName) {
|
|
495
|
+
try {
|
|
496
|
+
const restfulConfig = this.config;
|
|
497
|
+
// Create index for dense vector
|
|
498
|
+
const denseIndexParams = {
|
|
499
|
+
collectionName,
|
|
500
|
+
dbName: restfulConfig.database,
|
|
501
|
+
indexParams: [
|
|
502
|
+
{
|
|
503
|
+
fieldName: "vector",
|
|
504
|
+
indexName: "vector_index",
|
|
505
|
+
metricType: "COSINE",
|
|
506
|
+
index_type: "AUTOINDEX"
|
|
507
|
+
}
|
|
508
|
+
]
|
|
509
|
+
};
|
|
510
|
+
await this.makeRequest('/indexes/create', 'POST', denseIndexParams);
|
|
511
|
+
// Create index for sparse vector
|
|
512
|
+
const sparseIndexParams = {
|
|
513
|
+
collectionName,
|
|
514
|
+
dbName: restfulConfig.database,
|
|
515
|
+
indexParams: [
|
|
516
|
+
{
|
|
517
|
+
fieldName: "sparse_vector",
|
|
518
|
+
indexName: "sparse_vector_index",
|
|
519
|
+
metricType: "BM25",
|
|
520
|
+
index_type: "SPARSE_INVERTED_INDEX"
|
|
521
|
+
}
|
|
522
|
+
]
|
|
523
|
+
};
|
|
524
|
+
await this.makeRequest('/indexes/create', 'POST', sparseIndexParams);
|
|
525
|
+
}
|
|
526
|
+
catch (error) {
|
|
527
|
+
console.error(`❌ Failed to create hybrid indexes for collection '${collectionName}':`, error);
|
|
528
|
+
throw error;
|
|
529
|
+
}
|
|
530
|
+
}
|
|
531
|
+
async insertHybrid(collectionName, documents) {
|
|
532
|
+
await this.ensureInitialized();
|
|
533
|
+
try {
|
|
534
|
+
const restfulConfig = this.config;
|
|
535
|
+
const data = documents.map(doc => ({
|
|
536
|
+
id: doc.id,
|
|
537
|
+
content: doc.content,
|
|
538
|
+
vector: doc.vector,
|
|
539
|
+
relativePath: doc.relativePath,
|
|
540
|
+
startLine: doc.startLine,
|
|
541
|
+
endLine: doc.endLine,
|
|
542
|
+
fileExtension: doc.fileExtension,
|
|
543
|
+
metadata: JSON.stringify(doc.metadata),
|
|
544
|
+
}));
|
|
545
|
+
const insertRequest = {
|
|
546
|
+
collectionName,
|
|
547
|
+
dbName: restfulConfig.database,
|
|
548
|
+
data: data
|
|
549
|
+
};
|
|
550
|
+
const response = await this.makeRequest('/entities/insert', 'POST', insertRequest);
|
|
551
|
+
if (response.code !== 0) {
|
|
552
|
+
throw new Error(`Insert failed: ${response.message || 'Unknown error'}`);
|
|
553
|
+
}
|
|
554
|
+
}
|
|
555
|
+
catch (error) {
|
|
556
|
+
console.error(`❌ Failed to insert hybrid documents to collection '${collectionName}':`, error);
|
|
557
|
+
throw error;
|
|
558
|
+
}
|
|
559
|
+
}
|
|
560
|
+
async hybridSearch(collectionName, searchRequests, options) {
|
|
561
|
+
await this.ensureInitialized();
|
|
562
|
+
try {
|
|
563
|
+
const restfulConfig = this.config;
|
|
564
|
+
console.log(`🔍 Preparing hybrid search for collection: ${collectionName}`);
|
|
565
|
+
// Prepare search requests according to Milvus REST API hybrid search specification
|
|
566
|
+
// For dense vector search - data must be array of vectors: [[0.1, 0.2, 0.3, ...]]
|
|
567
|
+
const search_param_1 = {
|
|
568
|
+
data: Array.isArray(searchRequests[0].data) ? [searchRequests[0].data] : [[searchRequests[0].data]],
|
|
569
|
+
annsField: searchRequests[0].anns_field, // "vector"
|
|
570
|
+
limit: searchRequests[0].limit,
|
|
571
|
+
outputFields: ["*"],
|
|
572
|
+
searchParams: {
|
|
573
|
+
metricType: "COSINE",
|
|
574
|
+
params: searchRequests[0].param || { "nprobe": 10 }
|
|
575
|
+
}
|
|
576
|
+
};
|
|
577
|
+
// For sparse vector search - data must be array of queries: ["query text"]
|
|
578
|
+
const search_param_2 = {
|
|
579
|
+
data: Array.isArray(searchRequests[1].data) ? searchRequests[1].data : [searchRequests[1].data],
|
|
580
|
+
annsField: searchRequests[1].anns_field, // "sparse_vector"
|
|
581
|
+
limit: searchRequests[1].limit,
|
|
582
|
+
outputFields: ["*"],
|
|
583
|
+
searchParams: {
|
|
584
|
+
metricType: "BM25",
|
|
585
|
+
params: searchRequests[1].param || { "drop_ratio_search": 0.2 }
|
|
586
|
+
}
|
|
587
|
+
};
|
|
588
|
+
const rerank_strategy = {
|
|
589
|
+
strategy: "rrf",
|
|
590
|
+
params: {
|
|
591
|
+
k: 100
|
|
592
|
+
}
|
|
593
|
+
};
|
|
594
|
+
console.log(`🔍 Dense search params:`, JSON.stringify({
|
|
595
|
+
annsField: search_param_1.annsField,
|
|
596
|
+
limit: search_param_1.limit,
|
|
597
|
+
data_length: Array.isArray(search_param_1.data[0]) ? search_param_1.data[0].length : 'N/A',
|
|
598
|
+
searchParams: search_param_1.searchParams
|
|
599
|
+
}, null, 2));
|
|
600
|
+
console.log(`🔍 Sparse search params:`, JSON.stringify({
|
|
601
|
+
annsField: search_param_2.annsField,
|
|
602
|
+
limit: search_param_2.limit,
|
|
603
|
+
query_text: typeof search_param_2.data[0] === 'string' ? search_param_2.data[0].substring(0, 50) + '...' : 'N/A',
|
|
604
|
+
searchParams: search_param_2.searchParams
|
|
605
|
+
}, null, 2));
|
|
606
|
+
const hybridSearchRequest = {
|
|
607
|
+
collectionName,
|
|
608
|
+
dbName: restfulConfig.database,
|
|
609
|
+
search: [search_param_1, search_param_2],
|
|
610
|
+
rerank: rerank_strategy,
|
|
611
|
+
limit: options?.limit || searchRequests[0]?.limit || 10,
|
|
612
|
+
outputFields: ['id', 'content', 'relativePath', 'startLine', 'endLine', 'fileExtension', 'metadata'],
|
|
613
|
+
};
|
|
614
|
+
console.log(`🔍 Complete REST API request:`, JSON.stringify({
|
|
615
|
+
collectionName: hybridSearchRequest.collectionName,
|
|
616
|
+
dbName: hybridSearchRequest.dbName,
|
|
617
|
+
search_count: hybridSearchRequest.search.length,
|
|
618
|
+
rerank: hybridSearchRequest.rerank,
|
|
619
|
+
limit: hybridSearchRequest.limit,
|
|
620
|
+
outputFields: hybridSearchRequest.outputFields
|
|
621
|
+
}, null, 2));
|
|
622
|
+
console.log(`🔍 Executing REST API hybrid search...`);
|
|
623
|
+
const response = await this.makeRequest('/entities/hybrid_search', 'POST', hybridSearchRequest);
|
|
624
|
+
if (response.code !== 0) {
|
|
625
|
+
throw new Error(`Hybrid search failed: ${response.message || 'Unknown error'}`);
|
|
626
|
+
}
|
|
627
|
+
const results = response.data || [];
|
|
628
|
+
console.log(`✅ Found ${results.length} results from hybrid search`);
|
|
629
|
+
// Transform response to HybridSearchResult format
|
|
630
|
+
return results.map((result) => ({
|
|
631
|
+
document: {
|
|
632
|
+
id: result.id,
|
|
633
|
+
content: result.content,
|
|
634
|
+
vector: [], // Vector not returned in search results
|
|
635
|
+
sparse_vector: [], // Vector not returned in search results
|
|
636
|
+
relativePath: result.relativePath,
|
|
637
|
+
startLine: result.startLine,
|
|
638
|
+
endLine: result.endLine,
|
|
639
|
+
fileExtension: result.fileExtension,
|
|
640
|
+
metadata: JSON.parse(result.metadata || '{}'),
|
|
641
|
+
},
|
|
642
|
+
score: result.score || result.distance || 0,
|
|
643
|
+
}));
|
|
644
|
+
}
|
|
645
|
+
catch (error) {
|
|
646
|
+
console.error(`❌ Failed to perform hybrid search on collection '${collectionName}':`, error);
|
|
647
|
+
throw error;
|
|
648
|
+
}
|
|
649
|
+
}
|
|
404
650
|
}
|
|
405
651
|
exports.MilvusRestfulVectorDatabase = MilvusRestfulVectorDatabase;
|
|
406
652
|
//# sourceMappingURL=milvus-restful-vectordb.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"milvus-restful-vectordb.js","sourceRoot":"","sources":["../../src/vectordb/milvus-restful-vectordb.ts"],"names":[],"mappings":";;;AAAA,mCAMiB;AACjB,iDAAgD;AAUhD;;;GAGG;AACH,KAAK,UAAU,8BAA8B,CACzC,aAAqF,EACrF,gBAAqB;IAErB,IAAI,CAAC;QACD,MAAM,aAAa,CAAC,qBAAqB,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACzE,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,4EAA4E;QAC5E,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;QAC7D,IAAI,2CAA2C,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACjE,sDAAsD;YACtD,MAAM,gCAAwB,CAAC;QACnC,CAAC;QACD,8BAA8B;QAC9B,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAa,2BAA2B;IAKpC,YAAY,MAA2B;QAH/B,YAAO,GAAkB,IAAI,CAAC;QAIlC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,sDAAsD;QACtD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,UAAU;QACpB,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACpD,MAAM,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAe;QAC1C,qCAAqC;QACrC,IAAI,gBAAgB,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACtF,gBAAgB,GAAG,UAAU,gBAAgB,EAAE,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,cAAc,CAAC;QAEpE,OAAO,CAAC,GAAG,CAAC,wCAAwC,gBAAgB,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED;;;OAGG;IACO,KAAK,CAAC,cAAc;QAC1B,IAAI,WAAW,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAErC,iDAAiD;QACjD,IAAI,CAAC,WAAW,CAAC,OAAO,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;YAC5C,WAAW,CAAC,OAAO,GAAG,MAAM,6BAAc,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACtF,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAChF,CAAC;QAED,OAAO,WAAW,CAAC,OAAO,CAAC;IAC/B,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,iBAAiB;QAC7B,MAAM,IAAI,CAAC,qBAAqB,CAAC;QACjC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAChD,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,SAAyB,MAAM,EAAE,IAAU;QACnF,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,GAAG,QAAQ,EAAE,CAAC;QAEzC,MAAM,OAAO,GAA2B;YACpC,cAAc,EAAE,kBAAkB;YAClC,QAAQ,EAAE,kBAAkB;SAC/B,CAAC;QAEF,wBAAwB;QACxB,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACpB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC7D,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACtD,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxF,CAAC;QAED,MAAM,cAAc,GAAgB;YAChC,MAAM;YACN,OAAO;SACV,CAAC;QAEF,IAAI,IAAI,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YAC5B,cAAc,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YAElD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,MAAM,MAAM,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAE1C,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;YAC9E,CAAC;YAED,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,cAAsB,EAAE,SAAiB,EAAE,WAAoB;QAClF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,kFAAkF;YAClF,8EAA8E;YAC9E,wEAAwE;YACxE,MAAM,gBAAgB,GAAG;gBACrB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,MAAM,EAAE;oBACJ,kBAAkB,EAAE,KAAK;oBACzB,MAAM,EAAE;wBACJ;4BACI,SAAS,EAAE,IAAI;4BACf,QAAQ,EAAE,SAAS;4BACnB,SAAS,EAAE,IAAI;4BACf,iBAAiB,EAAE;gCACf,UAAU,EAAE,GAAG;6BAClB;yBACJ;wBACD;4BACI,SAAS,EAAE,QAAQ;4BACnB,QAAQ,EAAE,aAAa;4BACvB,iBAAiB,EAAE;gCACf,GAAG,EAAE,SAAS;6BACjB;yBACJ;wBACD;4BACI,SAAS,EAAE,SAAS;4BACpB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,KAAK;6BACpB;yBACJ;wBACD;4BACI,SAAS,EAAE,cAAc;4BACzB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,IAAI;6BACnB;yBACJ;wBACD;4BACI,SAAS,EAAE,WAAW;4BACtB,QAAQ,EAAE,OAAO;yBACpB;wBACD;4BACI,SAAS,EAAE,SAAS;4BACpB,QAAQ,EAAE,OAAO;yBACpB;wBACD;4BACI,SAAS,EAAE,eAAe;4BAC1B,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,EAAE;6BACjB;yBACJ;wBACD;4BACI,SAAS,EAAE,UAAU;4BACrB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,KAAK;6BACpB;yBACJ;qBACJ;iBACJ;aACJ,CAAC;YAEF,wCAAwC;YACxC,MAAM,8BAA8B,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,gBAAgB,CAAC,CAAC;YAEpF,4DAA4D;YAC5D,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;YAEvC,kDAAkD;YAClD,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QAE9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,kCAAkC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC3E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,cAAsB;QAC5C,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,WAAW,GAAG;gBAChB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,WAAW,EAAE;oBACT;wBACI,SAAS,EAAE,QAAQ;wBACnB,SAAS,EAAE,cAAc;wBACzB,UAAU,EAAE,QAAQ;wBACpB,UAAU,EAAE,WAAW;qBAC1B;iBACJ;aACJ,CAAC;YAEF,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,4CAA4C,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YACrF,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAAC,cAAsB;QAC/C,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,MAAM,EAAE;gBAChD,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YACzE,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,cAAsB;QACvC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,MAAM,EAAE;gBAChD,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YACzE,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,cAAsB;QACtC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE;gBAChE,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,GAAG,IAAI,KAAK,CAAC;YAC3C,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,cAAc,cAAc,EAAE,KAAK,CAAC,CAAC;YACpF,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,MAAM,EAAE;gBACjE,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,SAA2B;QAC5D,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,yDAAyD;YACzD,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,aAAa,EAAE,GAAG,CAAC,aAAa;gBAChC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,yCAAyC;aACnF,CAAC,CAAC,CAAC;YAEJ,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,IAAI;gBACJ,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC;YAEF,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QAEtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,WAAqB,EAAE,OAAuB;QAC/E,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC;QAEjC,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,kEAAkE;YAClE,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,IAAI,EAAE,CAAC,WAAW,CAAC,EAAE,yBAAyB;gBAC9C,SAAS,EAAE,QAAQ,EAAE,oBAAoB;gBACzC,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE;oBACV,SAAS;oBACT,cAAc;oBACd,WAAW;oBACX,SAAS;oBACT,eAAe;oBACf,UAAU;iBACb;gBACD,YAAY,EAAE;oBACV,UAAU,EAAE,QAAQ,EAAE,8BAA8B;oBACpD,MAAM,EAAE,EAAE;iBACb;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;YAEnF,kDAAkD;YAClD,MAAM,OAAO,GAAyB,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;gBAC1E,kCAAkC;gBAClC,IAAI,QAAQ,GAAG,EAAE,CAAC;gBAClB,IAAI,CAAC;oBACD,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,CAAC;gBACjD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,IAAI,CAAC,qCAAqC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;oBACrE,QAAQ,GAAG,EAAE,CAAC;gBAClB,CAAC;gBAED,OAAO;oBACH,QAAQ,EAAE;wBACN,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;wBAC7B,MAAM,EAAE,WAAW,EAAE,wCAAwC;wBAC7D,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,EAAE;wBACrC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,CAAC;wBAC9B,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,CAAC;wBAC1B,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,EAAE;wBACvC,QAAQ,EAAE,QAAQ;qBACrB;oBACD,KAAK,EAAE,IAAI,CAAC,QAAQ,IAAI,CAAC;iBAC5B,CAAC;YACN,CAAC,CAAC,CAAC;YAEH,OAAO,OAAO,CAAC;QAEnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,qCAAqC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC9E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,GAAa;QAC9C,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,8CAA8C;YAC9C,sCAAsC;YACtC,MAAM,MAAM,GAAG,UAAU,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;YAEhE,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,MAAM;gBACN,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC;YAEF,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QAEtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,cAAsB,EAAE,MAAc,EAAE,YAAsB,EAAE,KAAc;QACtF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,YAAY,GAAG;gBACjB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,MAAM;gBACN,YAAY;gBACZ,KAAK,EAAE,KAAK,IAAI,KAAK,EAAE,gCAAgC;gBACvD,MAAM,EAAE,CAAC;aACZ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;YAEjF,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;YACtF,CAAC;YAED,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QAE/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;CACJ;AAzaD,kEAyaC"}
|
|
1
|
+
{"version":3,"file":"milvus-restful-vectordb.js","sourceRoot":"","sources":["../../src/vectordb/milvus-restful-vectordb.ts"],"names":[],"mappings":";;;AAAA,mCASiB;AACjB,iDAAgD;AAUhD;;;GAGG;AACH,KAAK,UAAU,8BAA8B,CACzC,aAAqF,EACrF,gBAAqB;IAErB,IAAI,CAAC;QACD,MAAM,aAAa,CAAC,qBAAqB,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;IACzE,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,4EAA4E;QAC5E,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,IAAI,KAAK,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC;QAC7D,IAAI,2CAA2C,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,CAAC;YACjE,sDAAsD;YACtD,MAAM,gCAAwB,CAAC;QACnC,CAAC;QACD,8BAA8B;QAC9B,MAAM,KAAK,CAAC;IAChB,CAAC;AACL,CAAC;AAED;;;;GAIG;AACH,MAAa,2BAA2B;IAKpC,YAAY,MAA2B;QAH/B,YAAO,GAAkB,IAAI,CAAC;QAIlC,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QAErB,sDAAsD;QACtD,IAAI,CAAC,qBAAqB,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;IACnD,CAAC;IAEO,KAAK,CAAC,UAAU;QACpB,MAAM,eAAe,GAAG,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QACpD,MAAM,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAe;QAC1C,qCAAqC;QACrC,IAAI,gBAAgB,GAAG,OAAO,CAAC;QAC/B,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,SAAS,CAAC,IAAI,CAAC,gBAAgB,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACtF,gBAAgB,GAAG,UAAU,gBAAgB,EAAE,CAAC;QACpD,CAAC;QAED,IAAI,CAAC,OAAO,GAAG,gBAAgB,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,GAAG,cAAc,CAAC;QAEpE,OAAO,CAAC,GAAG,CAAC,wCAAwC,gBAAgB,EAAE,CAAC,CAAC;IAC5E,CAAC;IAED;;;OAGG;IACO,KAAK,CAAC,cAAc;QAC1B,IAAI,WAAW,GAAG,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;QAErC,iDAAiD;QACjD,IAAI,CAAC,WAAW,CAAC,OAAO,IAAI,WAAW,CAAC,KAAK,EAAE,CAAC;YAC5C,WAAW,CAAC,OAAO,GAAG,MAAM,6BAAc,CAAC,mBAAmB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACtF,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAChF,CAAC;QAED,OAAO,WAAW,CAAC,OAAO,CAAC;IAC/B,CAAC;IAED;;OAEG;IACO,KAAK,CAAC,iBAAiB;QAC7B,MAAM,IAAI,CAAC,qBAAqB,CAAC;QACjC,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAChD,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,QAAgB,EAAE,SAAyB,MAAM,EAAE,IAAU;QACnF,MAAM,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,GAAG,QAAQ,EAAE,CAAC;QAEzC,MAAM,OAAO,GAA2B;YACpC,cAAc,EAAE,kBAAkB;YAClC,QAAQ,EAAE,kBAAkB;SAC/B,CAAC;QAEF,wBAAwB;QACxB,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;YACpB,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;QAC7D,CAAC;aAAM,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;YACtD,OAAO,CAAC,eAAe,CAAC,GAAG,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACxF,CAAC;QAED,MAAM,cAAc,GAAgB;YAChC,MAAM;YACN,OAAO;SACV,CAAC;QAEF,IAAI,IAAI,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YAC5B,cAAc,CAAC,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QAC/C,CAAC;QAED,IAAI,CAAC;YACD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,EAAE,cAAc,CAAC,CAAC;YAElD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,CAAC,MAAM,KAAK,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;YACvE,CAAC;YAED,MAAM,MAAM,GAAQ,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YAE1C,IAAI,MAAM,CAAC,IAAI,KAAK,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,GAAG,EAAE,CAAC;gBAC3C,MAAM,IAAI,KAAK,CAAC,qBAAqB,MAAM,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;YAC9E,CAAC;YAED,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,KAAK,CAAC,CAAC;YACxD,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,cAAsB,EAAE,SAAiB,EAAE,WAAoB;QAClF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,kFAAkF;YAClF,8EAA8E;YAC9E,wEAAwE;YACxE,MAAM,gBAAgB,GAAG;gBACrB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,MAAM,EAAE;oBACJ,kBAAkB,EAAE,KAAK;oBACzB,MAAM,EAAE;wBACJ;4BACI,SAAS,EAAE,IAAI;4BACf,QAAQ,EAAE,SAAS;4BACnB,SAAS,EAAE,IAAI;4BACf,iBAAiB,EAAE;gCACf,UAAU,EAAE,GAAG;6BAClB;yBACJ;wBACD;4BACI,SAAS,EAAE,QAAQ;4BACnB,QAAQ,EAAE,aAAa;4BACvB,iBAAiB,EAAE;gCACf,GAAG,EAAE,SAAS;6BACjB;yBACJ;wBACD;4BACI,SAAS,EAAE,SAAS;4BACpB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,KAAK;6BACpB;yBACJ;wBACD;4BACI,SAAS,EAAE,cAAc;4BACzB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,IAAI;6BACnB;yBACJ;wBACD;4BACI,SAAS,EAAE,WAAW;4BACtB,QAAQ,EAAE,OAAO;yBACpB;wBACD;4BACI,SAAS,EAAE,SAAS;4BACpB,QAAQ,EAAE,OAAO;yBACpB;wBACD;4BACI,SAAS,EAAE,eAAe;4BAC1B,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,EAAE;6BACjB;yBACJ;wBACD;4BACI,SAAS,EAAE,UAAU;4BACrB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,KAAK;6BACpB;yBACJ;qBACJ;iBACJ;aACJ,CAAC;YAEF,wCAAwC;YACxC,MAAM,8BAA8B,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,gBAAgB,CAAC,CAAC;YAEpF,4DAA4D;YAC5D,MAAM,IAAI,CAAC,WAAW,CAAC,cAAc,CAAC,CAAC;YAEvC,kDAAkD;YAClD,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QAE9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,kCAAkC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC3E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,cAAsB;QAC5C,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,WAAW,GAAG;gBAChB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,WAAW,EAAE;oBACT;wBACI,SAAS,EAAE,QAAQ;wBACnB,SAAS,EAAE,cAAc;wBACzB,UAAU,EAAE,QAAQ;wBACpB,UAAU,EAAE,WAAW;qBAC1B;iBACJ;aACJ,CAAC;YAEF,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,WAAW,CAAC,CAAC;QACnE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,4CAA4C,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YACrF,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,cAAc,CAAC,cAAsB;QAC/C,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,MAAM,EAAE;gBAChD,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YACzE,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,cAAc,CAAC,cAAsB;QACvC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,MAAM,EAAE;gBAChD,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;QACP,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YACzE,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,aAAa,CAAC,cAAsB;QACtC,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE;gBAChE,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,QAAQ,CAAC,IAAI,EAAE,GAAG,IAAI,KAAK,CAAC;YAC3C,OAAO,MAAM,CAAC;QAClB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,cAAc,cAAc,EAAE,KAAK,CAAC,CAAC;YACpF,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,eAAe;QACjB,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,mBAAmB,EAAE,MAAM,EAAE;gBACjE,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC,CAAC;YAEH,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,+BAA+B,EAAE,KAAK,CAAC,CAAC;YACtD,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,SAA2B;QAC5D,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,yDAAyD;YACzD,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,aAAa,EAAE,GAAG,CAAC,aAAa;gBAChC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,yCAAyC;aACnF,CAAC,CAAC,CAAC;YAEJ,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,IAAI;gBACJ,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC;YAEF,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QAEtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,WAAqB,EAAE,OAAuB;QAC/E,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,MAAM,IAAI,GAAG,OAAO,EAAE,IAAI,IAAI,EAAE,CAAC;QAEjC,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,kEAAkE;YAClE,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,IAAI,EAAE,CAAC,WAAW,CAAC,EAAE,yBAAyB;gBAC9C,SAAS,EAAE,QAAQ,EAAE,oBAAoB;gBACzC,KAAK,EAAE,IAAI;gBACX,YAAY,EAAE;oBACV,SAAS;oBACT,cAAc;oBACd,WAAW;oBACX,SAAS;oBACT,eAAe;oBACf,UAAU;iBACb;gBACD,YAAY,EAAE;oBACV,UAAU,EAAE,QAAQ,EAAE,8BAA8B;oBACpD,MAAM,EAAE,EAAE;iBACb;aACJ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;YAEnF,kDAAkD;YAClD,MAAM,OAAO,GAAyB,CAAC,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,IAAS,EAAE,EAAE;gBAC1E,kCAAkC;gBAClC,IAAI,QAAQ,GAAG,EAAE,CAAC;gBAClB,IAAI,CAAC;oBACD,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,QAAQ,IAAI,IAAI,CAAC,CAAC;gBACjD,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACb,OAAO,CAAC,IAAI,CAAC,qCAAqC,IAAI,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,CAAC;oBACrE,QAAQ,GAAG,EAAE,CAAC;gBAClB,CAAC;gBAED,OAAO;oBACH,QAAQ,EAAE;wBACN,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,QAAQ,EAAE,IAAI,EAAE;wBAC7B,MAAM,EAAE,WAAW,EAAE,wCAAwC;wBAC7D,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,EAAE;wBAC3B,YAAY,EAAE,IAAI,CAAC,YAAY,IAAI,EAAE;wBACrC,SAAS,EAAE,IAAI,CAAC,SAAS,IAAI,CAAC;wBAC9B,OAAO,EAAE,IAAI,CAAC,OAAO,IAAI,CAAC;wBAC1B,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,EAAE;wBACvC,QAAQ,EAAE,QAAQ;qBACrB;oBACD,KAAK,EAAE,IAAI,CAAC,QAAQ,IAAI,CAAC;iBAC5B,CAAC;YACN,CAAC,CAAC,CAAC;YAEH,OAAO,OAAO,CAAC;QAEnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,qCAAqC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC9E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,cAAsB,EAAE,GAAa;QAC9C,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,8CAA8C;YAC9C,sCAAsC;YACtC,MAAM,MAAM,GAAG,UAAU,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC;YAEhE,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,MAAM;gBACN,MAAM,EAAE,aAAa,CAAC,QAAQ;aACjC,CAAC;YAEF,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;QAEtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iDAAiD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,cAAsB,EAAE,MAAc,EAAE,YAAsB,EAAE,KAAc;QACtF,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YACzD,MAAM,YAAY,GAAG;gBACjB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,MAAM;gBACN,YAAY;gBACZ,KAAK,EAAE,KAAK,IAAI,KAAK,EAAE,gCAAgC;gBACvD,MAAM,EAAE,CAAC;aACZ,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,YAAY,CAAC,CAAC;YAEjF,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,2BAA2B,QAAQ,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;YACtF,CAAC;YAED,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QAE/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,iCAAiC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC1E,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,sBAAsB,CAAC,cAAsB,EAAE,SAAiB,EAAE,WAAoB;QACxF,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YAEzD,MAAM,gBAAgB,GAAG;gBACrB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,MAAM,EAAE;oBACJ,kBAAkB,EAAE,KAAK;oBACzB,SAAS,EAAE;wBACP;4BACI,IAAI,EAAE,kBAAkB;4BACxB,WAAW,EAAE,uBAAuB;4BACpC,IAAI,EAAE,MAAM;4BACZ,eAAe,EAAE,CAAC,SAAS,CAAC;4BAC5B,gBAAgB,EAAE,CAAC,eAAe,CAAC;4BACnC,MAAM,EAAE,EAAE;yBACb;qBACJ;oBACD,MAAM,EAAE;wBACJ;4BACI,SAAS,EAAE,IAAI;4BACf,QAAQ,EAAE,SAAS;4BACnB,SAAS,EAAE,IAAI;4BACf,iBAAiB,EAAE;gCACf,UAAU,EAAE,GAAG;6BAClB;yBACJ;wBACD;4BACI,SAAS,EAAE,SAAS;4BACpB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,KAAK;gCACjB,eAAe,EAAE,IAAI;6BACxB;yBACJ;wBACD;4BACI,SAAS,EAAE,QAAQ;4BACnB,QAAQ,EAAE,aAAa;4BACvB,iBAAiB,EAAE;gCACf,GAAG,EAAE,SAAS;6BACjB;yBACJ;wBACD;4BACI,SAAS,EAAE,eAAe;4BAC1B,QAAQ,EAAE,mBAAmB;yBAChC;wBACD;4BACI,SAAS,EAAE,cAAc;4BACzB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,IAAI;6BACnB;yBACJ;wBACD;4BACI,SAAS,EAAE,WAAW;4BACtB,QAAQ,EAAE,OAAO;yBACpB;wBACD;4BACI,SAAS,EAAE,SAAS;4BACpB,QAAQ,EAAE,OAAO;yBACpB;wBACD;4BACI,SAAS,EAAE,eAAe;4BAC1B,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,EAAE;6BACjB;yBACJ;wBACD;4BACI,SAAS,EAAE,UAAU;4BACrB,QAAQ,EAAE,SAAS;4BACnB,iBAAiB,EAAE;gCACf,UAAU,EAAE,KAAK;6BACpB;yBACJ;qBACJ;iBACJ;aACJ,CAAC;YAEF,sDAAsD;YACtD,MAAM,8BAA8B,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,gBAAgB,CAAC,CAAC;YAEpF,gDAAgD;YAChD,MAAM,IAAI,CAAC,mBAAmB,CAAC,cAAc,CAAC,CAAC;YAE/C,kDAAkD;YAClD,MAAM,IAAI,CAAC,cAAc,CAAC,cAAc,CAAC,CAAC;QAE9C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,yCAAyC,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAClF,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,mBAAmB,CAAC,cAAsB;QACpD,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YAEzD,gCAAgC;YAChC,MAAM,gBAAgB,GAAG;gBACrB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,WAAW,EAAE;oBACT;wBACI,SAAS,EAAE,QAAQ;wBACnB,SAAS,EAAE,cAAc;wBACzB,UAAU,EAAE,QAAQ;wBACpB,UAAU,EAAE,WAAW;qBAC1B;iBACJ;aACJ,CAAC;YACF,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,gBAAgB,CAAC,CAAC;YAEpE,iCAAiC;YACjC,MAAM,iBAAiB,GAAG;gBACtB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,WAAW,EAAE;oBACT;wBACI,SAAS,EAAE,eAAe;wBAC1B,SAAS,EAAE,qBAAqB;wBAChC,UAAU,EAAE,MAAM;wBAClB,UAAU,EAAE,uBAAuB;qBACtC;iBACJ;aACJ,CAAC;YACF,MAAM,IAAI,CAAC,WAAW,CAAC,iBAAiB,EAAE,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAEzE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,qDAAqD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC9F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,cAAsB,EAAE,SAA2B;QAClE,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YAEzD,MAAM,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC/B,EAAE,EAAE,GAAG,CAAC,EAAE;gBACV,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,YAAY,EAAE,GAAG,CAAC,YAAY;gBAC9B,SAAS,EAAE,GAAG,CAAC,SAAS;gBACxB,OAAO,EAAE,GAAG,CAAC,OAAO;gBACpB,aAAa,EAAE,GAAG,CAAC,aAAa;gBAChC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,QAAQ,CAAC;aACzC,CAAC,CAAC,CAAC;YAEJ,MAAM,aAAa,GAAG;gBAClB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,IAAI,EAAE,IAAI;aACb,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,kBAAkB,EAAE,MAAM,EAAE,aAAa,CAAC,CAAC;YAEnF,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;YAC7E,CAAC;QAEL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,sDAAsD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC/F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;IAED,KAAK,CAAC,YAAY,CAAC,cAAsB,EAAE,cAAqC,EAAE,OAA6B;QAC3G,MAAM,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAE/B,IAAI,CAAC;YACD,MAAM,aAAa,GAAG,IAAI,CAAC,MAA6B,CAAC;YAEzD,OAAO,CAAC,GAAG,CAAC,8CAA8C,cAAc,EAAE,CAAC,CAAC;YAE5E,mFAAmF;YACnF,kFAAkF;YAClF,MAAM,cAAc,GAAG;gBACnB,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;gBACnG,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,WAAW;gBACpD,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,KAAK;gBAC9B,YAAY,EAAE,CAAC,GAAG,CAAC;gBACnB,YAAY,EAAE;oBACV,UAAU,EAAE,QAAQ;oBACpB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE;iBACtD;aACJ,CAAC;YAEF,2EAA2E;YAC3E,MAAM,cAAc,GAAG;gBACnB,IAAI,EAAE,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;gBAC/F,SAAS,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,UAAU,EAAE,kBAAkB;gBAC3D,KAAK,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,KAAK;gBAC9B,YAAY,EAAE,CAAC,GAAG,CAAC;gBACnB,YAAY,EAAE;oBACV,UAAU,EAAE,MAAM;oBAClB,MAAM,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,mBAAmB,EAAE,GAAG,EAAE;iBAClE;aACJ,CAAC;YAEF,MAAM,eAAe,GAAG;gBACpB,QAAQ,EAAE,KAAK;gBACf,MAAM,EAAE;oBACJ,CAAC,EAAE,GAAG;iBACT;aACJ,CAAC;YAEF,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE,IAAI,CAAC,SAAS,CAAC;gBAClD,SAAS,EAAE,cAAc,CAAC,SAAS;gBACnC,KAAK,EAAE,cAAc,CAAC,KAAK;gBAC3B,WAAW,EAAE,KAAK,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK;gBAC1F,YAAY,EAAE,cAAc,CAAC,YAAY;aAC5C,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YACb,OAAO,CAAC,GAAG,CAAC,0BAA0B,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnD,SAAS,EAAE,cAAc,CAAC,SAAS;gBACnC,KAAK,EAAE,cAAc,CAAC,KAAK;gBAC3B,UAAU,EAAE,OAAO,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,CAAC,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK;gBAChH,YAAY,EAAE,cAAc,CAAC,YAAY;aAC5C,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAEb,MAAM,mBAAmB,GAAG;gBACxB,cAAc;gBACd,MAAM,EAAE,aAAa,CAAC,QAAQ;gBAC9B,MAAM,EAAE,CAAC,cAAc,EAAE,cAAc,CAAC;gBACxC,MAAM,EAAE,eAAe;gBACvB,KAAK,EAAE,OAAO,EAAE,KAAK,IAAI,cAAc,CAAC,CAAC,CAAC,EAAE,KAAK,IAAI,EAAE;gBACvD,YAAY,EAAE,CAAC,IAAI,EAAE,SAAS,EAAE,cAAc,EAAE,WAAW,EAAE,SAAS,EAAE,eAAe,EAAE,UAAU,CAAC;aACvG,CAAC;YAEF,OAAO,CAAC,GAAG,CAAC,+BAA+B,EAAE,IAAI,CAAC,SAAS,CAAC;gBACxD,cAAc,EAAE,mBAAmB,CAAC,cAAc;gBAClD,MAAM,EAAE,mBAAmB,CAAC,MAAM;gBAClC,YAAY,EAAE,mBAAmB,CAAC,MAAM,CAAC,MAAM;gBAC/C,MAAM,EAAE,mBAAmB,CAAC,MAAM;gBAClC,KAAK,EAAE,mBAAmB,CAAC,KAAK;gBAChC,YAAY,EAAE,mBAAmB,CAAC,YAAY;aACjD,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;YAEb,OAAO,CAAC,GAAG,CAAC,wCAAwC,CAAC,CAAC;YACtD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,yBAAyB,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC;YAEhG,IAAI,QAAQ,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,yBAAyB,QAAQ,CAAC,OAAO,IAAI,eAAe,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,OAAO,GAAG,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;YACpC,OAAO,CAAC,GAAG,CAAC,WAAW,OAAO,CAAC,MAAM,6BAA6B,CAAC,CAAC;YAEpE,kDAAkD;YAClD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,MAAW,EAAE,EAAE,CAAC,CAAC;gBACjC,QAAQ,EAAE;oBACN,EAAE,EAAE,MAAM,CAAC,EAAE;oBACb,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,MAAM,EAAE,EAAE,EAAE,wCAAwC;oBACpD,aAAa,EAAE,EAAE,EAAE,wCAAwC;oBAC3D,YAAY,EAAE,MAAM,CAAC,YAAY;oBACjC,SAAS,EAAE,MAAM,CAAC,SAAS;oBAC3B,OAAO,EAAE,MAAM,CAAC,OAAO;oBACvB,aAAa,EAAE,MAAM,CAAC,aAAa;oBACnC,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,QAAQ,IAAI,IAAI,CAAC;iBAChD;gBACD,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,QAAQ,IAAI,CAAC;aAC9C,CAAC,CAAC,CAAC;QAER,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACb,OAAO,CAAC,KAAK,CAAC,oDAAoD,cAAc,IAAI,EAAE,KAAK,CAAC,CAAC;YAC7F,MAAM,KAAK,CAAC;QAChB,CAAC;IACL,CAAC;CACJ;AA1rBD,kEA0rBC"}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase } from './types';
|
|
1
|
+
import { VectorDocument, SearchOptions, VectorSearchResult, VectorDatabase, HybridSearchRequest, HybridSearchOptions, HybridSearchResult } from './types';
|
|
2
2
|
export interface MilvusConfig {
|
|
3
3
|
address?: string;
|
|
4
4
|
token?: string;
|
|
@@ -30,5 +30,8 @@ export declare class MilvusVectorDatabase implements VectorDatabase {
|
|
|
30
30
|
search(collectionName: string, queryVector: number[], options?: SearchOptions): Promise<VectorSearchResult[]>;
|
|
31
31
|
delete(collectionName: string, ids: string[]): Promise<void>;
|
|
32
32
|
query(collectionName: string, filter: string, outputFields: string[], limit?: number): Promise<Record<string, any>[]>;
|
|
33
|
+
createHybridCollection(collectionName: string, dimension: number, description?: string): Promise<void>;
|
|
34
|
+
insertHybrid(collectionName: string, documents: VectorDocument[]): Promise<void>;
|
|
35
|
+
hybridSearch(collectionName: string, searchRequests: HybridSearchRequest[], options?: HybridSearchOptions): Promise<HybridSearchResult[]>;
|
|
33
36
|
}
|
|
34
37
|
//# sourceMappingURL=milvus-vectordb.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"milvus-vectordb.d.ts","sourceRoot":"","sources":["../../src/vectordb/milvus-vectordb.ts"],"names":[],"mappings":"AACA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,
|
|
1
|
+
{"version":3,"file":"milvus-vectordb.d.ts","sourceRoot":"","sources":["../../src/vectordb/milvus-vectordb.ts"],"names":[],"mappings":"AACA,OAAO,EACH,cAAc,EACd,aAAa,EACb,kBAAkB,EAClB,cAAc,EACd,mBAAmB,EACnB,mBAAmB,EACnB,kBAAkB,EAErB,MAAM,SAAS,CAAC;AAGjB,MAAM,WAAW,YAAY;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,GAAG,CAAC,EAAE,OAAO,CAAC;CACjB;AAwBD,qBAAa,oBAAqB,YAAW,cAAc;IACvD,SAAS,CAAC,MAAM,EAAE,YAAY,CAAC;IAC/B,OAAO,CAAC,MAAM,CAA6B;IAC3C,SAAS,CAAC,qBAAqB,EAAE,OAAO,CAAC,IAAI,CAAC,CAAC;gBAEnC,MAAM,EAAE,YAAY;YAOlB,UAAU;YAKV,gBAAgB;IAY9B;;;OAGG;cACa,cAAc,IAAI,OAAO,CAAC,MAAM,CAAC;IAejD;;OAEG;cACa,iBAAiB,IAAI,OAAO,CAAC,IAAI,CAAC;IAO5C,gBAAgB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAoFhG,cAAc,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAQrD,aAAa,CAAC,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;IAUvD,eAAe,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;IASpC,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAqB1E,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;IA+B7G,MAAM,CAAC,cAAc,EAAE,MAAM,EAAE,GAAG,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAS5D,KAAK,CAAC,cAAc,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAAY,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,EAAE,CAAC;IA+BrH,sBAAsB,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAiHtG,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAoBhF,YAAY,CAAC,cAAc,EAAE,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,EAAE,OAAO,CAAC,EAAE,mBAAmB,GAAG,OAAO,CAAC,kBAAkB,EAAE,CAAC;CA6FlJ"}
|
|
@@ -243,6 +243,207 @@ class MilvusVectorDatabase {
|
|
|
243
243
|
throw error;
|
|
244
244
|
}
|
|
245
245
|
}
|
|
246
|
+
async createHybridCollection(collectionName, dimension, description) {
|
|
247
|
+
await this.ensureInitialized();
|
|
248
|
+
console.log('Beginning hybrid collection creation:', collectionName);
|
|
249
|
+
console.log('Collection dimension:', dimension);
|
|
250
|
+
const schema = [
|
|
251
|
+
{
|
|
252
|
+
name: 'id',
|
|
253
|
+
description: 'Document ID',
|
|
254
|
+
data_type: milvus2_sdk_node_1.DataType.VarChar,
|
|
255
|
+
max_length: 512,
|
|
256
|
+
is_primary_key: true,
|
|
257
|
+
},
|
|
258
|
+
{
|
|
259
|
+
name: 'content',
|
|
260
|
+
description: 'Full text content for BM25 and storage',
|
|
261
|
+
data_type: milvus2_sdk_node_1.DataType.VarChar,
|
|
262
|
+
max_length: 65535,
|
|
263
|
+
enable_analyzer: true,
|
|
264
|
+
},
|
|
265
|
+
{
|
|
266
|
+
name: 'vector',
|
|
267
|
+
description: 'Dense vector embedding',
|
|
268
|
+
data_type: milvus2_sdk_node_1.DataType.FloatVector,
|
|
269
|
+
dim: dimension,
|
|
270
|
+
},
|
|
271
|
+
{
|
|
272
|
+
name: 'sparse_vector',
|
|
273
|
+
description: 'Sparse vector embedding from BM25',
|
|
274
|
+
data_type: milvus2_sdk_node_1.DataType.SparseFloatVector,
|
|
275
|
+
},
|
|
276
|
+
{
|
|
277
|
+
name: 'relativePath',
|
|
278
|
+
description: 'Relative path to the codebase',
|
|
279
|
+
data_type: milvus2_sdk_node_1.DataType.VarChar,
|
|
280
|
+
max_length: 1024,
|
|
281
|
+
},
|
|
282
|
+
{
|
|
283
|
+
name: 'startLine',
|
|
284
|
+
description: 'Start line number of the chunk',
|
|
285
|
+
data_type: milvus2_sdk_node_1.DataType.Int64,
|
|
286
|
+
},
|
|
287
|
+
{
|
|
288
|
+
name: 'endLine',
|
|
289
|
+
description: 'End line number of the chunk',
|
|
290
|
+
data_type: milvus2_sdk_node_1.DataType.Int64,
|
|
291
|
+
},
|
|
292
|
+
{
|
|
293
|
+
name: 'fileExtension',
|
|
294
|
+
description: 'File extension',
|
|
295
|
+
data_type: milvus2_sdk_node_1.DataType.VarChar,
|
|
296
|
+
max_length: 32,
|
|
297
|
+
},
|
|
298
|
+
{
|
|
299
|
+
name: 'metadata',
|
|
300
|
+
description: 'Additional document metadata as JSON string',
|
|
301
|
+
data_type: milvus2_sdk_node_1.DataType.VarChar,
|
|
302
|
+
max_length: 65535,
|
|
303
|
+
},
|
|
304
|
+
];
|
|
305
|
+
// Add BM25 function
|
|
306
|
+
const functions = [
|
|
307
|
+
{
|
|
308
|
+
name: "content_bm25_emb",
|
|
309
|
+
description: "content bm25 function",
|
|
310
|
+
type: milvus2_sdk_node_1.FunctionType.BM25,
|
|
311
|
+
input_field_names: ["content"],
|
|
312
|
+
output_field_names: ["sparse_vector"],
|
|
313
|
+
params: {},
|
|
314
|
+
},
|
|
315
|
+
];
|
|
316
|
+
const createCollectionParams = {
|
|
317
|
+
collection_name: collectionName,
|
|
318
|
+
description: description || `Hybrid code context collection: ${collectionName}`,
|
|
319
|
+
fields: schema,
|
|
320
|
+
functions: functions,
|
|
321
|
+
};
|
|
322
|
+
await createCollectionWithLimitCheck(this.client, createCollectionParams);
|
|
323
|
+
// Create indexes for both vector fields
|
|
324
|
+
// Index for dense vector
|
|
325
|
+
const denseIndexParams = {
|
|
326
|
+
collection_name: collectionName,
|
|
327
|
+
field_name: 'vector',
|
|
328
|
+
index_type: 'AUTOINDEX',
|
|
329
|
+
metric_type: milvus2_sdk_node_1.MetricType.COSINE,
|
|
330
|
+
};
|
|
331
|
+
await this.client.createIndex(denseIndexParams);
|
|
332
|
+
// Index for sparse vector
|
|
333
|
+
const sparseIndexParams = {
|
|
334
|
+
collection_name: collectionName,
|
|
335
|
+
field_name: 'sparse_vector',
|
|
336
|
+
index_type: 'SPARSE_INVERTED_INDEX',
|
|
337
|
+
metric_type: milvus2_sdk_node_1.MetricType.BM25,
|
|
338
|
+
};
|
|
339
|
+
await this.client.createIndex(sparseIndexParams);
|
|
340
|
+
// Load collection to memory
|
|
341
|
+
await this.client.loadCollection({
|
|
342
|
+
collection_name: collectionName,
|
|
343
|
+
});
|
|
344
|
+
// Verify collection is created correctly
|
|
345
|
+
await this.client.describeCollection({
|
|
346
|
+
collection_name: collectionName,
|
|
347
|
+
});
|
|
348
|
+
}
|
|
349
|
+
async insertHybrid(collectionName, documents) {
|
|
350
|
+
await this.ensureInitialized();
|
|
351
|
+
const data = documents.map(doc => ({
|
|
352
|
+
id: doc.id,
|
|
353
|
+
content: doc.content,
|
|
354
|
+
vector: doc.vector,
|
|
355
|
+
relativePath: doc.relativePath,
|
|
356
|
+
startLine: doc.startLine,
|
|
357
|
+
endLine: doc.endLine,
|
|
358
|
+
fileExtension: doc.fileExtension,
|
|
359
|
+
metadata: JSON.stringify(doc.metadata),
|
|
360
|
+
}));
|
|
361
|
+
await this.client.insert({
|
|
362
|
+
collection_name: collectionName,
|
|
363
|
+
data: data,
|
|
364
|
+
});
|
|
365
|
+
}
|
|
366
|
+
async hybridSearch(collectionName, searchRequests, options) {
|
|
367
|
+
await this.ensureInitialized();
|
|
368
|
+
try {
|
|
369
|
+
// Generate OpenAI embedding for the first search request (dense)
|
|
370
|
+
console.log(`🔍 Preparing hybrid search for collection: ${collectionName}`);
|
|
371
|
+
// Prepare search requests in the correct Milvus format
|
|
372
|
+
const search_param_1 = {
|
|
373
|
+
data: Array.isArray(searchRequests[0].data) ? searchRequests[0].data : [searchRequests[0].data],
|
|
374
|
+
anns_field: searchRequests[0].anns_field, // "vector"
|
|
375
|
+
param: searchRequests[0].param, // {"nprobe": 10}
|
|
376
|
+
limit: searchRequests[0].limit
|
|
377
|
+
};
|
|
378
|
+
const search_param_2 = {
|
|
379
|
+
data: searchRequests[1].data, // query text for sparse search
|
|
380
|
+
anns_field: searchRequests[1].anns_field, // "sparse_vector"
|
|
381
|
+
param: searchRequests[1].param, // {"drop_ratio_search": 0.2}
|
|
382
|
+
limit: searchRequests[1].limit
|
|
383
|
+
};
|
|
384
|
+
// Set rerank strategy to RRF (100) by default
|
|
385
|
+
const rerank_strategy = {
|
|
386
|
+
strategy: "rrf",
|
|
387
|
+
params: {
|
|
388
|
+
k: 100
|
|
389
|
+
}
|
|
390
|
+
};
|
|
391
|
+
console.log(`🔍 Dense search params:`, JSON.stringify({
|
|
392
|
+
anns_field: search_param_1.anns_field,
|
|
393
|
+
param: search_param_1.param,
|
|
394
|
+
limit: search_param_1.limit,
|
|
395
|
+
data_length: Array.isArray(search_param_1.data[0]) ? search_param_1.data[0].length : 'N/A'
|
|
396
|
+
}, null, 2));
|
|
397
|
+
console.log(`🔍 Sparse search params:`, JSON.stringify({
|
|
398
|
+
anns_field: search_param_2.anns_field,
|
|
399
|
+
param: search_param_2.param,
|
|
400
|
+
limit: search_param_2.limit,
|
|
401
|
+
query_text: typeof search_param_2.data === 'string' ? search_param_2.data.substring(0, 50) + '...' : 'N/A'
|
|
402
|
+
}, null, 2));
|
|
403
|
+
console.log(`🔍 Rerank strategy:`, JSON.stringify(rerank_strategy, null, 2));
|
|
404
|
+
// Execute hybrid search using the correct client.search format
|
|
405
|
+
const searchParams = {
|
|
406
|
+
collection_name: collectionName,
|
|
407
|
+
data: [search_param_1, search_param_2],
|
|
408
|
+
limit: options?.limit || searchRequests[0]?.limit || 10,
|
|
409
|
+
rerank: rerank_strategy,
|
|
410
|
+
output_fields: ['id', 'content', 'relativePath', 'startLine', 'endLine', 'fileExtension', 'metadata'],
|
|
411
|
+
};
|
|
412
|
+
console.log(`🔍 Complete search request:`, JSON.stringify({
|
|
413
|
+
collection_name: searchParams.collection_name,
|
|
414
|
+
data_count: searchParams.data.length,
|
|
415
|
+
limit: searchParams.limit,
|
|
416
|
+
rerank: searchParams.rerank,
|
|
417
|
+
output_fields: searchParams.output_fields
|
|
418
|
+
}, null, 2));
|
|
419
|
+
const searchResult = await this.client.search(searchParams);
|
|
420
|
+
console.log(`🔍 Search executed, processing results...`);
|
|
421
|
+
if (!searchResult.results || searchResult.results.length === 0) {
|
|
422
|
+
console.log(`⚠️ No results returned from Milvus search`);
|
|
423
|
+
return [];
|
|
424
|
+
}
|
|
425
|
+
console.log(`✅ Found ${searchResult.results.length} results from hybrid search`);
|
|
426
|
+
// Transform results to HybridSearchResult format
|
|
427
|
+
return searchResult.results.map((result) => ({
|
|
428
|
+
document: {
|
|
429
|
+
id: result.id,
|
|
430
|
+
content: result.content,
|
|
431
|
+
vector: [],
|
|
432
|
+
sparse_vector: [],
|
|
433
|
+
relativePath: result.relativePath,
|
|
434
|
+
startLine: result.startLine,
|
|
435
|
+
endLine: result.endLine,
|
|
436
|
+
fileExtension: result.fileExtension,
|
|
437
|
+
metadata: JSON.parse(result.metadata || '{}'),
|
|
438
|
+
},
|
|
439
|
+
score: result.score,
|
|
440
|
+
}));
|
|
441
|
+
}
|
|
442
|
+
catch (error) {
|
|
443
|
+
console.error(`❌ Failed to perform hybrid search on collection '${collectionName}':`, error);
|
|
444
|
+
throw error;
|
|
445
|
+
}
|
|
446
|
+
}
|
|
246
447
|
}
|
|
247
448
|
exports.MilvusVectorDatabase = MilvusVectorDatabase;
|
|
248
449
|
//# sourceMappingURL=milvus-vectordb.js.map
|