chromadb 1.1.0 → 1.1.2
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/main/index.d.ts +4 -2
- package/dist/main/index.d.ts.map +1 -1
- package/dist/main/index.js +12 -6
- package/dist/main/index.js.map +1 -1
- package/dist/module/index.d.ts +4 -2
- package/dist/module/index.d.ts.map +1 -1
- package/dist/module/index.js +11 -4
- package/dist/module/index.js.map +1 -1
- package/package.json +1 -1
- package/src/index.ts +22 -31
package/dist/main/index.d.ts
CHANGED
|
@@ -21,6 +21,8 @@ export declare class ChromaClient {
|
|
|
21
21
|
getCollection(name: string): Promise<Collection>;
|
|
22
22
|
deleteCollection(name: string): Promise<import("axios").AxiosResponse<any, any>>;
|
|
23
23
|
}
|
|
24
|
-
|
|
25
|
-
|
|
24
|
+
declare const _default: {
|
|
25
|
+
ChromaClient: typeof ChromaClient;
|
|
26
|
+
};
|
|
27
|
+
export default _default;
|
|
26
28
|
//# sourceMappingURL=index.d.ts.map
|
package/dist/main/index.d.ts.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAW7C,qBAAa,UAAU;IACrB,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,GAAG,CAAa;gBAEZ,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU;IAK5B,GAAG,CACd,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,EAAE,EAAE,0BAA0B;IAChD,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,EAC/B,eAAe,GAAE,OAAc;IAcpB,KAAK;IAKL,GAAG,CACd,GAAG,CAAC,EAAE,MAAM,EAAE,EACd,KAAK,CAAC,EAAE,MAAM,EACd,KAAK,CAAC,EAAE,MAAM,EACd,MAAM,CAAC,EAAE,MAAM;IAaJ,KAAK,CAChB,gBAAgB,EAAE,MAAM,EAAE,EAC1B,SAAS,GAAE,MAAW,EACtB,KAAK,CAAC,EAAE,MAAM;IAaH,IAAI,CAAC,KAAK,GAAE,MAAW;IAOvB,WAAW;IAIX,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM;CAOnD;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,GAAG,CAAa;gBAEZ,QAAQ,EAAE,MAAM;IAOf,KAAK;IAIL,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM;IAOhD,eAAe;IAKf,aAAa,CAAC,IAAI,EAAE,MAAM;IAI1B,gBAAgB,CAAC,IAAI,EAAE,MAAM;CAI3C;;;;AAMD,wBAAgC"}
|
package/dist/main/index.js
CHANGED
|
@@ -1,8 +1,15 @@
|
|
|
1
1
|
"use strict";
|
|
2
2
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
-
exports.
|
|
3
|
+
exports.ChromaClient = exports.Collection = void 0;
|
|
4
4
|
const api_1 = require("./generated/api");
|
|
5
5
|
const configuration_1 = require("./generated/configuration");
|
|
6
|
+
// export class ChromaClientT {
|
|
7
|
+
// reset: () => Promise<any>;
|
|
8
|
+
// createCollection: (name: string, metadata?: object) => Promise<any>;
|
|
9
|
+
// listCollections: () => Promise<any>;
|
|
10
|
+
// getCollection: (name: string) => Promise<any>;
|
|
11
|
+
// deleteCollection: (name: string) => Promise<any>;
|
|
12
|
+
// };
|
|
6
13
|
class Collection {
|
|
7
14
|
constructor(name, api) {
|
|
8
15
|
this.name = name;
|
|
@@ -92,9 +99,8 @@ class ChromaClient {
|
|
|
92
99
|
}
|
|
93
100
|
}
|
|
94
101
|
exports.ChromaClient = ChromaClient;
|
|
95
|
-
const chromaClient = (basePath) => {
|
|
96
|
-
|
|
97
|
-
}
|
|
98
|
-
exports.
|
|
99
|
-
exports.default = exports.chromaClient;
|
|
102
|
+
// export const chromaClient = (basePath: string): ChromaClient => {
|
|
103
|
+
// return new ChromaClient(basePath);
|
|
104
|
+
// }
|
|
105
|
+
exports.default = { ChromaClient };
|
|
100
106
|
//# sourceMappingURL=index.js.map
|
package/dist/main/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAAA,yCAA6C;AAC7C,6DAA0D;AAE1D,MAAa,UAAU;IAIrB,YAAY,IAAY,EAAE,GAAe;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAW,EACX,UAAoB,EAAE,0BAA0B;IAChD,SAAkB,EAClB,SAA+B,EAC/B,kBAA2B,IAAI;QAE/B,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,UAAU;gBACV,SAAS;gBACT,SAAS;gBACT,eAAe,EAAE,eAAe;aACjC;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACrE,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAc,EACd,KAAc,EACd,KAAc,EACd,MAAe;QAEf,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,KAAK;gBACL,KAAK;gBACL,MAAM;aACP;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK,CAChB,gBAA0B,EAC1B,YAAoB,EAAE,EACtB,KAAc;QAEd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC;YAClD,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,cAAc,EAAE;gBACd,gBAAgB;gBAChB,KAAK;gBACL,SAAS;aACV;SACF,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE;QAClC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,WAAW;QACtB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,GAAc,EAAE,KAAc;QAChD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;YAC5B,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,eAAe,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":";;;AAAA,yCAA6C;AAC7C,6DAA0D;AAE1D,+BAA+B;AAC/B,+BAA+B;AAC/B,yEAAyE;AACzE,yCAAyC;AACzC,mDAAmD;AACnD,sDAAsD;AACtD,KAAK;AAEL,MAAa,UAAU;IAIrB,YAAY,IAAY,EAAE,GAAe;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAW,EACX,UAAoB,EAAE,0BAA0B;IAChD,SAAkB,EAClB,SAA+B,EAC/B,kBAA2B,IAAI;QAE/B,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,UAAU;gBACV,SAAS;gBACT,SAAS;gBACT,eAAe,EAAE,eAAe;aACjC;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACrE,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAc,EACd,KAAc,EACd,KAAc,EACd,MAAe;QAEf,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,KAAK;gBACL,KAAK;gBACL,MAAM;aACP;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK,CAChB,gBAA0B,EAC1B,YAAoB,EAAE,EACtB,KAAc;QAEd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC;YAClD,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,cAAc,EAAE;gBACd,gBAAgB;gBAChB,KAAK;gBACL,SAAS;aACV;SACF,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE;QAClC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,WAAW;QACtB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,GAAc,EAAE,KAAc;QAChD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;YAC5B,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,eAAe,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;CAEF;AApFD,gCAoFC;AAED,MAAa,YAAY;IAGvB,YAAY,QAAgB;QAC1B,MAAM,SAAS,GAAkB,IAAI,6BAAa,CAAC;YACjD,QAAQ;SACT,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,GAAG,IAAI,gBAAU,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAY,EAAE,QAAiB;QAC3D,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC;YACpD,gBAAgB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SACrC,CAAC,CAAC;QACH,OAAO,IAAI,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;QAClD,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,IAAY;QACrC,OAAO,IAAI,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACxC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;CAEF;AAlCD,oCAkCC;AAED,oEAAoE;AACpE,uCAAuC;AACvC,IAAI;AAEJ,kBAAe,EAAE,YAAY,EAAE,CAAC"}
|
package/dist/module/index.d.ts
CHANGED
|
@@ -21,6 +21,8 @@ export declare class ChromaClient {
|
|
|
21
21
|
getCollection(name: string): Promise<Collection>;
|
|
22
22
|
deleteCollection(name: string): Promise<import("axios").AxiosResponse<any, any>>;
|
|
23
23
|
}
|
|
24
|
-
|
|
25
|
-
|
|
24
|
+
declare const _default: {
|
|
25
|
+
ChromaClient: typeof ChromaClient;
|
|
26
|
+
};
|
|
27
|
+
export default _default;
|
|
26
28
|
//# sourceMappingURL=index.d.ts.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAW7C,qBAAa,UAAU;IACrB,OAAO,CAAC,IAAI,CAAS;IACrB,OAAO,CAAC,GAAG,CAAa;gBAEZ,IAAI,EAAE,MAAM,EAAE,GAAG,EAAE,UAAU;IAK5B,GAAG,CACd,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,MAAM,EAAE,EAAE,0BAA0B;IAChD,SAAS,CAAC,EAAE,MAAM,EAClB,SAAS,CAAC,EAAE,MAAM,GAAG,KAAK,CAAC,GAAG,CAAC,EAC/B,eAAe,GAAE,OAAc;IAcpB,KAAK;IAKL,GAAG,CACd,GAAG,CAAC,EAAE,MAAM,EAAE,EACd,KAAK,CAAC,EAAE,MAAM,EACd,KAAK,CAAC,EAAE,MAAM,EACd,MAAM,CAAC,EAAE,MAAM;IAaJ,KAAK,CAChB,gBAAgB,EAAE,MAAM,EAAE,EAC1B,SAAS,GAAE,MAAW,EACtB,KAAK,CAAC,EAAE,MAAM;IAaH,IAAI,CAAC,KAAK,GAAE,MAAW;IAOvB,WAAW;IAIX,MAAM,CAAC,GAAG,CAAC,EAAE,MAAM,EAAE,EAAE,KAAK,CAAC,EAAE,MAAM;CAOnD;AAED,qBAAa,YAAY;IACvB,OAAO,CAAC,GAAG,CAAa;gBAEZ,QAAQ,EAAE,MAAM;IAOf,KAAK;IAIL,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM;IAOhD,eAAe;IAKf,aAAa,CAAC,IAAI,EAAE,MAAM;IAI1B,gBAAgB,CAAC,IAAI,EAAE,MAAM;CAI3C;;;;AAMD,wBAAgC"}
|
package/dist/module/index.js
CHANGED
|
@@ -1,5 +1,12 @@
|
|
|
1
1
|
import { DefaultApi } from "./generated/api";
|
|
2
2
|
import { Configuration } from "./generated/configuration";
|
|
3
|
+
// export class ChromaClientT {
|
|
4
|
+
// reset: () => Promise<any>;
|
|
5
|
+
// createCollection: (name: string, metadata?: object) => Promise<any>;
|
|
6
|
+
// listCollections: () => Promise<any>;
|
|
7
|
+
// getCollection: (name: string) => Promise<any>;
|
|
8
|
+
// deleteCollection: (name: string) => Promise<any>;
|
|
9
|
+
// };
|
|
3
10
|
export class Collection {
|
|
4
11
|
constructor(name, api) {
|
|
5
12
|
this.name = name;
|
|
@@ -87,8 +94,8 @@ export class ChromaClient {
|
|
|
87
94
|
return await this.api.deleteCollection({ collectionName: name });
|
|
88
95
|
}
|
|
89
96
|
}
|
|
90
|
-
export const chromaClient = (basePath) => {
|
|
91
|
-
|
|
92
|
-
}
|
|
93
|
-
export default
|
|
97
|
+
// export const chromaClient = (basePath: string): ChromaClient => {
|
|
98
|
+
// return new ChromaClient(basePath);
|
|
99
|
+
// }
|
|
100
|
+
export default { ChromaClient };
|
|
94
101
|
//# sourceMappingURL=index.js.map
|
package/dist/module/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,MAAM,OAAO,UAAU;IAIrB,YAAY,IAAY,EAAE,GAAe;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAW,EACX,UAAoB,EAAE,0BAA0B;IAChD,SAAkB,EAClB,SAA+B,EAC/B,kBAA2B,IAAI;QAE/B,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,UAAU;gBACV,SAAS;gBACT,SAAS;gBACT,eAAe,EAAE,eAAe;aACjC;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACrE,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAc,EACd,KAAc,EACd,KAAc,EACd,MAAe;QAEf,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,KAAK;gBACL,KAAK;gBACL,MAAM;aACP;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK,CAChB,gBAA0B,EAC1B,YAAoB,EAAE,EACtB,KAAc;QAEd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC;YAClD,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,cAAc,EAAE;gBACd,gBAAgB;gBAChB,KAAK;gBACL,SAAS;aACV;SACF,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE;QAClC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,WAAW;QACtB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,GAAc,EAAE,KAAc;QAChD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;YAC5B,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,eAAe,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAE1D,+BAA+B;AAC/B,+BAA+B;AAC/B,yEAAyE;AACzE,yCAAyC;AACzC,mDAAmD;AACnD,sDAAsD;AACtD,KAAK;AAEL,MAAM,OAAO,UAAU;IAIrB,YAAY,IAAY,EAAE,GAAe;QACvC,IAAI,CAAC,IAAI,GAAG,IAAI,CAAC;QACjB,IAAI,CAAC,GAAG,GAAG,GAAG,CAAC;IACjB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAW,EACX,UAAoB,EAAE,0BAA0B;IAChD,SAAkB,EAClB,SAA+B,EAC/B,kBAA2B,IAAI;QAE/B,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,UAAU;gBACV,SAAS;gBACT,SAAS;gBACT,eAAe,EAAE,eAAe;aACjC;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QACrE,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,GAAG,CACd,GAAc,EACd,KAAc,EACd,KAAc,EACd,MAAe;QAEf,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE;gBACZ,GAAG;gBACH,KAAK;gBACL,KAAK;gBACL,MAAM;aACP;SACF,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,KAAK,CAChB,gBAA0B,EAC1B,YAAoB,EAAE,EACtB,KAAc;QAEd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,mBAAmB,CAAC;YAClD,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,cAAc,EAAE;gBACd,gBAAgB;gBAChB,KAAK;gBACL,SAAS;aACV;SACF,CAAC,CAAC;QACH,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,IAAI,CAAC,QAAgB,EAAE;QAClC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;YACxB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,YAAY,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;SAC/B,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,WAAW;QACtB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;IAEM,KAAK,CAAC,MAAM,CAAC,GAAc,EAAE,KAAc;QAChD,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;YAC5B,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,eAAe,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,KAAK,EAAE,KAAK,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;CAEF;AAED,MAAM,OAAO,YAAY;IAGvB,YAAY,QAAgB;QAC1B,MAAM,SAAS,GAAkB,IAAI,aAAa,CAAC;YACjD,QAAQ;SACT,CAAC,CAAC;QACH,IAAI,CAAC,GAAG,GAAG,IAAI,UAAU,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAEM,KAAK,CAAC,KAAK;QAChB,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;IAChC,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAY,EAAE,QAAiB;QAC3D,MAAM,aAAa,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC;YACpD,gBAAgB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;SACrC,CAAC,CAAC;QACH,OAAO,IAAI,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,eAAe;QAC1B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,GAAG,CAAC,eAAe,EAAE,CAAC;QAClD,OAAO,QAAQ,CAAC,IAAI,CAAC;IACvB,CAAC;IAEM,KAAK,CAAC,aAAa,CAAC,IAAY;QACrC,OAAO,IAAI,UAAU,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAY;QACxC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;IACnE,CAAC;CAEF;AAED,oEAAoE;AACpE,uCAAuC;AACvC,IAAI;AAEJ,eAAe,EAAE,YAAY,EAAE,CAAC"}
|
package/package.json
CHANGED
package/src/index.ts
CHANGED
|
@@ -1,6 +1,14 @@
|
|
|
1
1
|
import { DefaultApi } from "./generated/api";
|
|
2
2
|
import { Configuration } from "./generated/configuration";
|
|
3
3
|
|
|
4
|
+
// export class ChromaClientT {
|
|
5
|
+
// reset: () => Promise<any>;
|
|
6
|
+
// createCollection: (name: string, metadata?: object) => Promise<any>;
|
|
7
|
+
// listCollections: () => Promise<any>;
|
|
8
|
+
// getCollection: (name: string) => Promise<any>;
|
|
9
|
+
// deleteCollection: (name: string) => Promise<any>;
|
|
10
|
+
// };
|
|
11
|
+
|
|
4
12
|
export class Collection {
|
|
5
13
|
private name: string;
|
|
6
14
|
private api: DefaultApi;
|
|
@@ -11,12 +19,12 @@ export class Collection {
|
|
|
11
19
|
}
|
|
12
20
|
|
|
13
21
|
public async add(
|
|
14
|
-
ids: string,
|
|
22
|
+
ids: string,
|
|
15
23
|
embeddings: number[], // this should be optional
|
|
16
24
|
metadatas?: object,
|
|
17
25
|
documents?: string | Array<any>,
|
|
18
26
|
increment_index: boolean = true,
|
|
19
|
-
|
|
27
|
+
) {
|
|
20
28
|
return await this.api.add({
|
|
21
29
|
collectionName: this.name,
|
|
22
30
|
addEmbedding: {
|
|
@@ -28,7 +36,7 @@ export class Collection {
|
|
|
28
36
|
},
|
|
29
37
|
});
|
|
30
38
|
}
|
|
31
|
-
|
|
39
|
+
|
|
32
40
|
public async count() {
|
|
33
41
|
const response = await this.api.count({ collectionName: this.name });
|
|
34
42
|
return response.data;
|
|
@@ -39,11 +47,11 @@ export class Collection {
|
|
|
39
47
|
where?: object,
|
|
40
48
|
limit?: number,
|
|
41
49
|
offset?: number,
|
|
42
|
-
|
|
50
|
+
) {
|
|
43
51
|
return await this.api.get({
|
|
44
52
|
collectionName: this.name,
|
|
45
|
-
getEmbedding: {
|
|
46
|
-
ids,
|
|
53
|
+
getEmbedding: {
|
|
54
|
+
ids,
|
|
47
55
|
where,
|
|
48
56
|
limit,
|
|
49
57
|
offset,
|
|
@@ -52,13 +60,13 @@ export class Collection {
|
|
|
52
60
|
}
|
|
53
61
|
|
|
54
62
|
public async query(
|
|
55
|
-
query_embeddings: number[],
|
|
63
|
+
query_embeddings: number[],
|
|
56
64
|
n_results: number = 10,
|
|
57
65
|
where?: object,
|
|
58
|
-
|
|
66
|
+
) {
|
|
59
67
|
const response = await this.api.getNearestNeighbors({
|
|
60
68
|
collectionName: this.name,
|
|
61
|
-
queryEmbedding: {
|
|
69
|
+
queryEmbedding: {
|
|
62
70
|
query_embeddings,
|
|
63
71
|
where,
|
|
64
72
|
n_results,
|
|
@@ -79,28 +87,12 @@ export class Collection {
|
|
|
79
87
|
}
|
|
80
88
|
|
|
81
89
|
public async delete(ids?: string[], where?: object) {
|
|
82
|
-
return await this.api._delete({
|
|
90
|
+
return await this.api._delete({
|
|
83
91
|
collectionName: this.name,
|
|
84
92
|
deleteEmbedding: { ids: ids, where: where },
|
|
85
93
|
});
|
|
86
94
|
}
|
|
87
95
|
|
|
88
|
-
// public async modify(new_name: string) {
|
|
89
|
-
// const collection = await this.api.updateCollection({
|
|
90
|
-
// collectionName: this.name,
|
|
91
|
-
// updateCollection: { new_name },
|
|
92
|
-
// });
|
|
93
|
-
// this.name = new_name;
|
|
94
|
-
// return this
|
|
95
|
-
// }
|
|
96
|
-
|
|
97
|
-
// public async deleteEmbeddings(ids: string[]) {
|
|
98
|
-
// return await this.api.deleteEmbeddings({
|
|
99
|
-
// collectionName: this.name,
|
|
100
|
-
// deleteEmbedding: { ids },
|
|
101
|
-
// });
|
|
102
|
-
// }
|
|
103
|
-
|
|
104
96
|
}
|
|
105
97
|
|
|
106
98
|
export class ChromaClient {
|
|
@@ -137,11 +129,10 @@ export class ChromaClient {
|
|
|
137
129
|
return await this.api.deleteCollection({ collectionName: name });
|
|
138
130
|
}
|
|
139
131
|
|
|
140
|
-
|
|
141
132
|
}
|
|
142
133
|
|
|
143
|
-
export const chromaClient = (basePath: string): ChromaClient => {
|
|
144
|
-
|
|
145
|
-
}
|
|
134
|
+
// export const chromaClient = (basePath: string): ChromaClient => {
|
|
135
|
+
// return new ChromaClient(basePath);
|
|
136
|
+
// }
|
|
146
137
|
|
|
147
|
-
export default
|
|
138
|
+
export default { ChromaClient };
|