@thenajs/qdrant-client 0.6.0 → 0.9.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/qdrant.store.d.ts +2 -2
- package/dist/qdrant.store.d.ts.map +1 -1
- package/dist/qdrant.store.js +10 -10
- package/dist/qdrant.store.js.map +1 -1
- package/package.json +5 -2
package/dist/qdrant.store.d.ts
CHANGED
|
@@ -31,7 +31,7 @@ export declare class QdrantStore extends VectorStore {
|
|
|
31
31
|
* apenas a forma abreviada é aceita, então caímos para ela: o índice sai
|
|
32
32
|
* igual, só sem a otimização de co-locação.
|
|
33
33
|
*/
|
|
34
|
-
private
|
|
34
|
+
private createPartitionIndex;
|
|
35
35
|
collectionExists(): Promise<boolean>;
|
|
36
36
|
dropCollection(): Promise<void>;
|
|
37
37
|
upsert(docs: VectorDocument[]): Promise<void>;
|
|
@@ -42,7 +42,7 @@ export declare class QdrantStore extends VectorStore {
|
|
|
42
42
|
* `rawFilter` vence quando presente — é o escape hatch para o que a
|
|
43
43
|
* igualdade não cobre (ranges, geo, aninhamento).
|
|
44
44
|
*/
|
|
45
|
-
private
|
|
45
|
+
private buildFilter;
|
|
46
46
|
/** Uma chamada à API, já com auth, retry e a mensagem de erro do Qdrant. */
|
|
47
47
|
private chamar;
|
|
48
48
|
}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"qdrant.store.d.ts","sourceRoot":"","sources":["../src/qdrant.store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,KAAK,
|
|
1
|
+
{"version":3,"file":"qdrant.store.d.ts","sourceRoot":"","sources":["../src/qdrant.store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,KAAK,EACV,iBAAiB,EAEjB,cAAc,EACd,WAAW,EACX,YAAY,EACZ,cAAc,EACd,sBAAsB,EACvB,MAAM,eAAe,CAAC;AAEvB,MAAM,MAAM,iBAAiB,GAAG,sBAAsB,CAAC;AAUvD;;;;;;;;;;;;GAYG;AACH,qBAAa,WAAY,SAAQ,WAAW;IAC1C,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;IACjC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAS;IACpC,iEAAiE;IACjE,SAAgB,YAAY,EAAE,MAAM,CAAC;gBAEzB,WAAW,EAAE,iBAAiB;IASpC,gBAAgB,CAAC,OAAO,EAAE,iBAAiB,GAAG,OAAO,CAAC,IAAI,CAAC;IAajE;;;;;;;;OAQG;YACW,oBAAoB;IAgB5B,gBAAgB,IAAI,OAAO,CAAC,OAAO,CAAC;IAQpC,cAAc,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,MAAM,CAAC,IAAI,EAAE,cAAc,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAY7C,MAAM,CAAC,MAAM,EAAE,YAAY,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;IAkBpD,MAAM,CAAC,QAAQ,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC;IAkBrD;;;;OAIG;IACH,OAAO,CAAC,WAAW;IAYnB,4EAA4E;YAC9D,MAAM;CAuBrB"}
|
package/dist/qdrant.store.js
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { VectorStore } from "@thenajs/core";
|
|
2
2
|
/** O shape neutro de distância → o nome que o Qdrant espera. */
|
|
3
|
-
const
|
|
3
|
+
const DISTANCES = {
|
|
4
4
|
cosine: "Cosine",
|
|
5
5
|
euclid: "Euclid",
|
|
6
6
|
dot: "Dot",
|
|
@@ -39,10 +39,10 @@ export class QdrantStore extends VectorStore {
|
|
|
39
39
|
await this.chamar(`/collections/${this.collection}`, "PUT", {
|
|
40
40
|
vectors: {
|
|
41
41
|
size: options.size,
|
|
42
|
-
distance:
|
|
42
|
+
distance: DISTANCES[options.distance ?? "cosine"],
|
|
43
43
|
},
|
|
44
44
|
});
|
|
45
|
-
await this.
|
|
45
|
+
await this.createPartitionIndex();
|
|
46
46
|
}
|
|
47
47
|
/**
|
|
48
48
|
* Índice no campo que particiona os datasets.
|
|
@@ -53,7 +53,7 @@ export class QdrantStore extends VectorStore {
|
|
|
53
53
|
* apenas a forma abreviada é aceita, então caímos para ela: o índice sai
|
|
54
54
|
* igual, só sem a otimização de co-locação.
|
|
55
55
|
*/
|
|
56
|
-
async
|
|
56
|
+
async createPartitionIndex() {
|
|
57
57
|
const rota = `/collections/${this.collection}/index?wait=true`;
|
|
58
58
|
try {
|
|
59
59
|
await this.chamar(rota, "PUT", {
|
|
@@ -91,7 +91,7 @@ export class QdrantStore extends VectorStore {
|
|
|
91
91
|
const data = await this.chamar(`/collections/${this.collection}/points/query`, "POST", {
|
|
92
92
|
query: params.vector,
|
|
93
93
|
limit: params.limit ?? 5,
|
|
94
|
-
filter: this.
|
|
94
|
+
filter: this.buildFilter(params.where, params.rawFilter),
|
|
95
95
|
with_payload: params.withPayload ?? true,
|
|
96
96
|
score_threshold: params.scoreThreshold,
|
|
97
97
|
});
|
|
@@ -102,23 +102,23 @@ export class QdrantStore extends VectorStore {
|
|
|
102
102
|
}));
|
|
103
103
|
}
|
|
104
104
|
async remove(selector) {
|
|
105
|
-
const filter = this.
|
|
106
|
-
const
|
|
105
|
+
const filter = this.buildFilter(selector.where);
|
|
106
|
+
const body = selector.ids?.length
|
|
107
107
|
? { points: selector.ids }
|
|
108
108
|
: filter
|
|
109
109
|
? { filter }
|
|
110
110
|
: undefined;
|
|
111
111
|
// Sem seletor nenhum, apagar tudo seria destrutivo demais para ser implícito.
|
|
112
|
-
if (!
|
|
112
|
+
if (!body)
|
|
113
113
|
return;
|
|
114
|
-
await this.chamar(`/collections/${this.collection}/points/delete?wait=true`, "POST",
|
|
114
|
+
await this.chamar(`/collections/${this.collection}/points/delete?wait=true`, "POST", body);
|
|
115
115
|
}
|
|
116
116
|
/**
|
|
117
117
|
* Traduz o `where` neutro (igualdade) para o formato do Qdrant. O
|
|
118
118
|
* `rawFilter` vence quando presente — é o escape hatch para o que a
|
|
119
119
|
* igualdade não cobre (ranges, geo, aninhamento).
|
|
120
120
|
*/
|
|
121
|
-
|
|
121
|
+
buildFilter(where, rawFilter) {
|
|
122
122
|
if (rawFilter !== undefined)
|
|
123
123
|
return rawFilter;
|
|
124
124
|
if (!where || !Object.keys(where).length)
|
package/dist/qdrant.store.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"qdrant.store.js","sourceRoot":"","sources":["../src/qdrant.store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;
|
|
1
|
+
{"version":3,"file":"qdrant.store.js","sourceRoot":"","sources":["../src/qdrant.store.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAa5C,gEAAgE;AAChE,MAAM,SAAS,GAAmC;IAChD,MAAM,EAAE,QAAQ;IAChB,MAAM,EAAE,QAAQ;IAChB,GAAG,EAAE,KAAK;IACV,SAAS,EAAE,WAAW;CACvB,CAAC;AAEF;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,WAAY,SAAQ,WAAW;IACzB,GAAG,CAAS;IACZ,MAAM,CAAU;IAChB,UAAU,CAAS;IACpC,iEAAiE;IACjD,YAAY,CAAS;IAErC,YAAY,WAA8B;QACxC,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QACrC,IAAI,CAAC,GAAG,GAAG,WAAW,CAAC,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,GAAG,WAAW,CAAC,MAAM,CAAC;QACjC,IAAI,CAAC,UAAU,GAAG,WAAW,CAAC,UAAU,IAAI,cAAc,CAAC;QAC3D,IAAI,CAAC,YAAY,GAAG,WAAW,CAAC,YAAY,IAAI,SAAS,CAAC;IAC5D,CAAC;IAED,KAAK,CAAC,gBAAgB,CAAC,OAA0B;QAC/C,IAAI,MAAM,IAAI,CAAC,gBAAgB,EAAE;YAAE,OAAO;QAE1C,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,UAAU,EAAE,EAAE,KAAK,EAAE;YAC1D,OAAO,EAAE;gBACP,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,QAAQ,EAAE,SAAS,CAAC,OAAO,CAAC,QAAQ,IAAI,QAAQ,CAAC;aAClD;SACF,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;IACpC,CAAC;IAED;;;;;;;;OAQG;IACK,KAAK,CAAC,oBAAoB;QAChC,MAAM,IAAI,GAAG,gBAAgB,IAAI,CAAC,UAAU,kBAAkB,CAAC;QAC/D,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE;gBAC7B,UAAU,EAAE,IAAI,CAAC,YAAY;gBAC7B,YAAY,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,SAAS,EAAE,IAAI,EAAE;aACnD,CAAC,CAAC;QACL,CAAC;QAAC,MAAM,CAAC;YACP,uEAAuE;YACvE,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,KAAK,EAAE;gBAC7B,UAAU,EAAE,IAAI,CAAC,YAAY;gBAC7B,YAAY,EAAE,SAAS;aACxB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,KAAK,CAAC,gBAAgB;QACpB,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAC5B,gBAAgB,IAAI,CAAC,UAAU,SAAS,EACxC,KAAK,CACN,CAAC;QACF,OAAO,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,cAAc;QAClB,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,UAAU,EAAE,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,IAAsB;QACjC,IAAI,CAAC,IAAI,CAAC,MAAM;YAAE,OAAO;QAEzB,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,UAAU,mBAAmB,EAAE,KAAK,EAAE;YAC3E,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;gBACvB,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,MAAM,EAAE,CAAC,CAAC,MAAM;gBAChB,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,EAAE;aACzB,CAAC,CAAC;SACJ,CAAC,CAAC;IACL,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,MAAoB;QAC/B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,MAAM,CAE3B,gBAAgB,IAAI,CAAC,UAAU,eAAe,EAAE,MAAM,EAAE;YACzD,KAAK,EAAE,MAAM,CAAC,MAAM;YACpB,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC;YACxB,MAAM,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC,SAAS,CAAC;YACxD,YAAY,EAAE,MAAM,CAAC,WAAW,IAAI,IAAI;YACxC,eAAe,EAAE,MAAM,CAAC,cAAc;SACvC,CAAC,CAAC;QAEH,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC9C,EAAE,EAAE,CAAC,CAAC,EAAE;YACR,KAAK,EAAE,CAAC,CAAC,KAAK;YACd,OAAO,EAAE,CAAC,CAAC,OAAO,IAAI,SAAS;SAChC,CAAC,CAAC,CAAC;IACN,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,QAAwB;QACnC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,IAAI,GAAG,QAAQ,CAAC,GAAG,EAAE,MAAM;YAC/B,CAAC,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,GAAG,EAAE;YAC1B,CAAC,CAAC,MAAM;gBACN,CAAC,CAAC,EAAE,MAAM,EAAE;gBACZ,CAAC,CAAC,SAAS,CAAC;QAEhB,8EAA8E;QAC9E,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,MAAM,IAAI,CAAC,MAAM,CACf,gBAAgB,IAAI,CAAC,UAAU,0BAA0B,EACzD,MAAM,EACN,IAAI,CACL,CAAC;IACJ,CAAC;IAED;;;;OAIG;IACK,WAAW,CAAC,KAA+B,EAAE,SAAmB;QACtE,IAAI,SAAS,KAAK,SAAS;YAAE,OAAO,SAAS,CAAC;QAC9C,IAAI,CAAC,KAAK,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,MAAM;YAAE,OAAO,SAAS,CAAC;QAE3D,OAAO;YACL,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjD,GAAG;gBACH,KAAK,EAAE,EAAE,KAAK,EAAE;aACjB,CAAC,CAAC;SACJ,CAAC;IACJ,CAAC;IAED,4EAA4E;IACpE,KAAK,CAAC,MAAM,CAClB,OAAe,EACf,MAAc,EACd,IAAc;QAEd,MAAM,EAAE,QAAQ,EAAE,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,GAAG,IAAI,CAAC,GAAG,GAAG,OAAO,EAAE,EAAE;YAC/D,MAAM;YACN,OAAO,EAAE;gBACP,cAAc,EAAE,kBAAkB;gBAClC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACnD;YACD,IAAI,EAAE,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC;SAC5D,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CACb,UAAU,MAAM,IAAI,OAAO,YAAY,QAAQ,CAAC,MAAM,MAAM,OAAO,EAAE,CACtE,CAAC;QACJ,CAAC;QAED,OAAO,CAAC,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAM,CAAC;IACtC,CAAC;CACF"}
|
package/package.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
{
|
|
2
2
|
"name": "@thenajs/qdrant-client",
|
|
3
|
-
"version": "0.
|
|
3
|
+
"version": "0.9.0",
|
|
4
4
|
"description": "Cliente Qdrant nativo para ThenaJS \u2014 implementa\u00e7\u00e3o de VectorStore sobre a API REST, sem SDK.",
|
|
5
5
|
"license": "MIT",
|
|
6
6
|
"author": "castroneto",
|
|
@@ -30,7 +30,10 @@
|
|
|
30
30
|
"publishConfig": {
|
|
31
31
|
"access": "public"
|
|
32
32
|
},
|
|
33
|
+
"engines": {
|
|
34
|
+
"node": ">=20.19"
|
|
35
|
+
},
|
|
33
36
|
"dependencies": {
|
|
34
|
-
"@thenajs/core": "^0.
|
|
37
|
+
"@thenajs/core": "^0.9.0"
|
|
35
38
|
}
|
|
36
39
|
}
|