@mastra/chroma 0.0.0-vnext-inngest-20250508122351 → 0.0.0-vnext-20251104230439
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/CHANGELOG.md +715 -4
- package/LICENSE.md +12 -4
- package/README.md +47 -21
- package/dist/index.cjs +301 -105
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.ts +3 -2
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +299 -103
- package/dist/index.js.map +1 -0
- package/dist/vector/filter.d.ts +19 -0
- package/dist/vector/filter.d.ts.map +1 -0
- package/dist/vector/index.d.ts +70 -0
- package/dist/vector/index.d.ts.map +1 -0
- package/dist/vector/prompt.d.ts +6 -0
- package/dist/vector/prompt.d.ts.map +1 -0
- package/package.json +32 -14
- package/dist/_tsup-dts-rollup.d.cts +0 -75
- package/dist/_tsup-dts-rollup.d.ts +0 -75
- package/dist/index.d.cts +0 -2
- package/docker-compose.yaml +0 -7
- package/eslint.config.js +0 -6
- package/src/index.ts +0 -2
- package/src/vector/filter.test.ts +0 -413
- package/src/vector/filter.ts +0 -115
- package/src/vector/index.test.ts +0 -1657
- package/src/vector/index.ts +0 -231
- package/src/vector/prompt.ts +0 -72
- package/tsconfig.json +0 -5
- package/vitest.config.ts +0 -8
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/vector/filter.ts","../src/vector/index.ts","../src/vector/prompt.ts"],"names":["BaseFilterTranslator","MastraVector","CloudClient","ChromaClient","MastraError","ErrorDomain","ErrorCategory","error"],"mappings":";;;;;;;;AA4BO,IAAM,sBAAA,GAAN,cAAqCA,2BAAA,CAAyC;AAAA,EAChE,qBAAA,GAAyC;AAC1D,IAAA,OAAO;AAAA,MACL,GAAGA,2BAAA,CAAqB,iBAAA;AAAA,MACxB,OAAA,EAAS,CAAC,MAAA,EAAQ,KAAK,CAAA;AAAA,MACvB,KAAA,EAAO,CAAC,KAAA,EAAO,MAAM,CAAA;AAAA,MACrB,SAAS,EAAC;AAAA,MACV,OAAO,EAAC;AAAA,MACR,QAAQ;AAAC,KACX;AAAA,EACF;AAAA,EAEA,UAAU,MAAA,EAAiD;AACzD,IAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,MAAM,CAAA,EAAG,OAAO,MAAA;AACjC,IAAA,IAAA,CAAK,eAAe,MAAM,CAAA;AAE1B,IAAA,OAAO,IAAA,CAAK,cAAc,MAAM,CAAA;AAAA,EAClC;AAAA,EAEQ,aAAA,CAAc,IAAA,EAA0B,WAAA,GAAsB,EAAA,EAAS;AAE7E,IAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,IAAI,CAAA,EAAG;AACtB,MAAA,MAAM,IAAI,MAAM,gEAAgE,CAAA;AAAA,IAClF;AACA,IAAA,IAAI,KAAK,WAAA,CAAY,IAAI,GAAG,OAAO,IAAA,CAAK,yBAAyB,IAAI,CAAA;AACrE,IAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,EAAG,OAAO,EAAE,GAAA,EAAK,IAAA,CAAK,oBAAA,CAAqB,IAAI,CAAA,EAAE;AAEvE,IAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,IAA2B,CAAA;AAC1D,IAAA,MAAM,UAAA,GAAa,QAAQ,CAAC,CAAA;AAE5B,IAAA,IAAI,OAAA,CAAQ,WAAW,CAAA,IAAK,UAAA,IAAc,KAAK,UAAA,CAAW,UAAA,CAAW,CAAC,CAAC,CAAA,EAAG;AACxE,MAAA,MAAM,CAAC,QAAA,EAAU,KAAK,CAAA,GAAI,UAAA;AAC1B,MAAA,MAAM,UAAA,GAAa,IAAA,CAAK,iBAAA,CAAkB,QAAA,EAAU,KAAK,CAAA;AACzD,MAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,IAAK,KAAA,CAAM,QAAQ,UAAU,CAAA,IAAK,UAAA,CAAW,MAAA,KAAW,CAAA,EAAG;AAC5F,QAAA,OAAO,WAAW,CAAC,CAAA;AAAA,MACrB;AACA,MAAA,OAAO,IAAA,CAAK,kBAAkB,QAAQ,CAAA,GAAI,EAAE,CAAC,QAAQ,GAAG,UAAA,EAAW,GAAI,UAAA;AAAA,IACzE;AAGA,IAAA,MAAM,SAA8B,EAAC;AACrC,IAAA,MAAM,0BAAiC,EAAC;AAExC,IAAA,KAAA,MAAW,CAAC,GAAA,EAAK,KAAK,CAAA,IAAK,OAAA,EAAS;AAClC,MAAA,MAAM,UAAU,WAAA,GAAc,CAAA,EAAG,WAAW,CAAA,CAAA,EAAI,GAAG,CAAA,CAAA,GAAK,GAAA;AAExD,MAAA,IAAI,IAAA,CAAK,UAAA,CAAW,GAAG,CAAA,EAAG;AACxB,QAAA,MAAA,CAAO,GAAG,CAAA,GAAI,IAAA,CAAK,iBAAA,CAAkB,KAAK,KAAK,CAAA;AAC/C,QAAA;AAAA,MACF;AAEA,MAAA,IAAI,OAAO,UAAU,QAAA,IAAY,KAAA,KAAU,QAAQ,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AAExE,QAAA,MAAM,YAAA,GAAe,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA;AACzC,QAAA,IAAI,YAAA,CAAa,KAAA,CAAM,CAAC,CAAC,EAAE,CAAA,KAAM,IAAA,CAAK,UAAA,CAAW,EAAE,CAAC,CAAA,IAAK,YAAA,CAAa,SAAS,CAAA,EAAG;AAChF,UAAA,YAAA,CAAa,OAAA,CAAQ,CAAC,CAAC,EAAA,EAAI,OAAO,CAAA,KAAM;AACtC,YAAA,uBAAA,CAAwB,IAAA,CAAK;AAAA,cAC3B,CAAC,OAAO,GAAG,EAAE,CAAC,EAAE,GAAG,IAAA,CAAK,wBAAA,CAAyB,OAAO,CAAA;AAAE,aAC3D,CAAA;AAAA,UACH,CAAC,CAAA;AACD,UAAA;AAAA,QACF;AAGA,QAAA,IAAI,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA,CAAE,WAAW,CAAA,EAAG;AACnC,UAAA,MAAA,CAAO,OAAO,CAAA,GAAI,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAAA,QAC5C,CAAA,MAAO;AACL,UAAA,MAAM,YAAA,GAAe,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA,CAAE,KAAK,CAAA,CAAA,KAAK,IAAA,CAAK,UAAA,CAAW,CAAC,CAAC,CAAA;AACpE,UAAA,IAAI,YAAA,EAAc;AAEhB,YAAA,MAAM,kBAAuC,EAAC;AAC9C,YAAA,KAAA,MAAW,CAAC,EAAA,EAAI,OAAO,KAAK,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,EAAG;AACjD,cAAA,eAAA,CAAgB,EAAE,CAAA,GAAI,IAAA,CAAK,UAAA,CAAW,EAAE,IAAI,IAAA,CAAK,iBAAA,CAAkB,EAAA,EAAI,OAAO,CAAA,GAAI,OAAA;AAAA,YACpF;AACA,YAAA,MAAA,CAAO,OAAO,CAAA,GAAI,eAAA;AAAA,UACpB,CAAA,MAAO;AAEL,YAAA,MAAA,CAAO,OAAO,MAAA,EAAQ,IAAA,CAAK,aAAA,CAAc,KAAA,EAAO,OAAO,CAAC,CAAA;AAAA,UAC1D;AAAA,QACF;AAAA,MACF,CAAA,MAAO;AACL,QAAA,MAAA,CAAO,OAAO,CAAA,GAAI,IAAA,CAAK,aAAA,CAAc,KAAK,CAAA;AAAA,MAC5C;AAAA,IACF;AAGA,IAAA,IAAI,uBAAA,CAAwB,SAAS,CAAA,EAAG;AACtC,MAAA,OAAO,EAAE,MAAM,uBAAA,EAAwB;AAAA,IACzC;AAGA,IAAA,IAAI,OAAO,IAAA,CAAK,MAAM,EAAE,MAAA,GAAS,CAAA,IAAK,CAAC,WAAA,EAAa;AAClD,MAAA,OAAO;AAAA,QACL,MAAM,MAAA,CAAO,OAAA,CAAQ,MAAM,CAAA,CAAE,IAAI,CAAC,CAAC,GAAA,EAAK,KAAK,OAAO,EAAE,CAAC,GAAG,GAAG,OAAM,CAAE;AAAA,OACvE;AAAA,IACF;AAEA,IAAA,OAAO,MAAA;AAAA,EACT;AAAA,EAEQ,iBAAA,CAAkB,UAAyB,KAAA,EAAiB;AAElE,IAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,EAAG;AACpC,MAAA,OAAO,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,GAAI,MAAM,GAAA,CAAI,CAAA,IAAA,KAAQ,IAAA,CAAK,aAAA,CAAc,IAAI,CAAC,CAAA,GAAI,IAAA,CAAK,cAAc,KAAK,CAAA;AAAA,IACtG;AAGA,IAAA,OAAO,IAAA,CAAK,yBAAyB,KAAK,CAAA;AAAA,EAC5C;AACF,CAAA;;;AC/FA,IAAM,aAAA,GAAgB;AAAA,EACpB,MAAA,EAAQ,QAAA;AAAA,EACR,SAAA,EAAW,IAAA;AAAA,EACX,UAAA,EAAY,IAAA;AAAA,EACZ,EAAA,EAAI,WAAA;AAAA,EACJ,EAAA,EAAI;AACN,CAAA;AAEO,IAAM,YAAA,GAAN,cAA2BC,mBAAA,CAAiC;AAAA,EACzD,MAAA;AAAA,EACA,WAAA;AAAA,EAER,YAAY,gBAAA,EAAqC;AAC/C,IAAA,KAAA,EAAM;AACN,IAAA,IAAI,kBAAkB,MAAA,EAAQ;AAC5B,MAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,oBAAA,CAAY;AAAA,QAC5B,QAAQ,gBAAA,CAAiB,MAAA;AAAA,QACzB,QAAQ,gBAAA,CAAiB,MAAA;AAAA,QACzB,UAAU,gBAAA,CAAiB;AAAA,OAC5B,CAAA;AAAA,IACH,CAAA,MAAO;AACL,MAAA,IAAA,CAAK,MAAA,GAAS,IAAIC,qBAAA,CAAa,gBAAgB,CAAA;AAAA,IACjD;AACA,IAAA,IAAA,CAAK,WAAA,uBAAkB,GAAA,EAAI;AAAA,EAC7B;AAAA,EAEA,MAAM,aAAA,CAAc,EAAE,SAAA,EAAW,WAAA,GAAc,OAAM,EAAiD;AACpG,IAAA,IAAI,UAAA,GAAa,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAS,CAAA;AAC/C,IAAA,IAAI,WAAA,IAAe,CAAC,UAAA,EAAY;AAC9B,MAAA,IAAI;AACF,QAAA,UAAA,GAAa,MAAM,IAAA,CAAK,MAAA,CAAO,cAAc,EAAE,IAAA,EAAM,WAAW,CAAA;AAChE,QAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAA,EAAW,UAAU,CAAA;AAC1C,QAAA,OAAO,UAAA;AAAA,MACT,CAAA,CAAA,MAAQ;AACN,QAAA,MAAM,IAAIC,iBAAA,CAAY;AAAA,UACpB,EAAA,EAAI,8BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACtB,CAAA;AAAA,MACH;AAAA,IACF;AACA,IAAA,OAAO,UAAA;AAAA,EACT;AAAA,EAEQ,wBAAA,CAAyB,SAAqB,SAAA,EAAyB;AAC7E,IAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,OAAA,CAAQ,QAAQ,CAAA,EAAA,EAAK;AACvC,MAAA,IAAI,OAAA,GAAU,CAAC,CAAA,EAAG,MAAA,KAAW,SAAA,EAAW;AACtC,QAAA,MAAM,IAAI,KAAA;AAAA,UACR,CAAA,gBAAA,EAAmB,CAAC,CAAA,uBAAA,EAA0B,OAAA,GAAU,CAAC,CAAA,EAAG,MAAM,cAAc,SAAS,CAAA,YAAA;AAAA,SAC3F;AAAA,MACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,OAAO,EAAE,SAAA,EAAW,SAAS,QAAA,EAAU,GAAA,EAAK,WAAU,EAAgD;AAC1G,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAEzD,MAAA,MAAM,QAAQ,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACpD,MAAA,IAAA,CAAK,wBAAA,CAAyB,OAAA,EAAS,KAAA,CAAM,SAAS,CAAA;AACtD,MAAA,MAAM,eAAe,GAAA,IAAO,OAAA,CAAQ,IAAI,MAAM,MAAA,CAAO,YAAY,CAAA;AAEjE,MAAA,MAAM,WAAW,MAAA,CAAO;AAAA,QACtB,GAAA,EAAK,YAAA;AAAA,QACL,UAAA,EAAY,OAAA;AAAA,QACZ,SAAA,EAAW,QAAA;AAAA,QACX;AAAA,OACD,CAAA;AAED,MAAA,OAAO,YAAA;AAAA,IACT,SAASC,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,CAAY,EAAE,WAAW,SAAA,EAAW,MAAA,GAAS,UAAS,EAAqC;AAC/F,IAAA,IAAI,CAAC,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA,IAAK,aAAa,CAAA,EAAG;AAClD,MAAA,MAAM,IAAIH,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,8CAAA;AAAA,QACJ,IAAA,EAAM,sCAAA;AAAA,QACN,QAAQC,iBAAA,CAAY,aAAA;AAAA,QACpB,UAAUC,mBAAA,CAAc,IAAA;AAAA,QACxB,OAAA,EAAS,EAAE,SAAA;AAAU,OACtB,CAAA;AAAA,IACH;AAEA,IAAA,MAAM,SAAA,GAAY,cAAc,MAAM,CAAA;AAEtC,IAAA,IAAI,CAAC,SAAA,IAAa,CAAC,CAAC,QAAA,EAAU,MAAM,IAAI,CAAA,CAAE,QAAA,CAAS,SAAS,CAAA,EAAG;AAC7D,MAAA,MAAM,IAAIF,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,2CAAA;AAAA,QACJ,IAAA,EAAM,oBAAoB,MAAM,CAAA,gDAAA,CAAA;AAAA,QAChC,QAAQC,iBAAA,CAAY,aAAA;AAAA,QACpB,UAAUC,mBAAA,CAAc,IAAA;AAAA,QACxB,OAAA,EAAS,EAAE,MAAA;AAAO,OACnB,CAAA;AAAA,IACH;AAEA,IAAA,IAAI;AACF,MAAA,MAAM,UAAA,GAAa,MAAM,IAAA,CAAK,MAAA,CAAO,gBAAA,CAAiB;AAAA,QACpD,IAAA,EAAM,SAAA;AAAA,QACN,QAAA,EAAU,EAAE,SAAA,EAAU;AAAA,QACtB,eAAe,EAAE,IAAA,EAAM,EAAE,KAAA,EAAO,WAAU,EAAE;AAAA,QAC5C,iBAAA,EAAmB;AAAA,OACpB,CAAA;AACD,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,SAAA,EAAW,UAAU,CAAA;AAAA,IAC5C,SAASC,OAAA,EAAY;AAEnB,MAAA,MAAM,OAAA,GAAUA,OAAA,EAAO,OAAA,IAAWA,OAAA,EAAO,QAAA,EAAS;AAClD,MAAA,IAAI,WAAW,OAAA,CAAQ,WAAA,EAAY,CAAE,QAAA,CAAS,gBAAgB,CAAA,EAAG;AAE/D,QAAA,MAAM,IAAA,CAAK,qBAAA,CAAsB,SAAA,EAAW,SAAA,EAAW,MAAM,CAAA;AAC7D,QAAA;AAAA,MACF;AACA,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,gBAAgB,MAAA,EAA6B;AAC3C,IAAA,MAAM,UAAA,GAAa,IAAI,sBAAA,EAAuB;AAC9C,IAAA,MAAM,gBAAA,GAAmB,UAAA,CAAW,SAAA,CAAU,MAAM,CAAA;AACpD,IAAA,OAAO,mBAAoB,gBAAA,GAA6B,MAAA;AAAA,EAC1D;AAAA,EAEA,MAAM,KAAA,CAAqC;AAAA,IACzC,SAAA;AAAA,IACA,WAAA;AAAA,IACA,IAAA,GAAO,EAAA;AAAA,IACP,MAAA;AAAA,IACA,aAAA,GAAgB,KAAA;AAAA,IAChB;AAAA,GACF,EAAoD;AAClD,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAEzD,MAAA,MAAM,cAAA,GAA0D,CAAC,WAAA,EAAa,WAAA,EAAa,WAAW,CAAA;AAEtG,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AACpD,MAAA,MAAM,OAAA,GAAU,MAAM,UAAA,CAAW,KAAA,CAAS;AAAA,QACxC,eAAA,EAAiB,CAAC,WAAW,CAAA;AAAA,QAC7B,QAAA,EAAU,IAAA;AAAA,QACV,OAAO,gBAAA,IAAoB,MAAA;AAAA,QAC3B,eAAe,cAAA,IAAkB,MAAA;AAAA,QACjC,SAAS,aAAA,GAAgB,CAAC,GAAG,cAAA,EAAgB,YAAY,CAAA,GAAI;AAAA,OAC9D,CAAA;AAED,MAAA,OAAA,CAAQ,OAAA,CAAQ,IAAI,CAAC,CAAA,IAAK,EAAC,EAAG,GAAA,CAAI,CAAC,EAAA,EAAY,KAAA,MAAmB;AAAA,QAChE,EAAA;AAAA,QACA,OAAO,OAAA,CAAQ,SAAA,GAAY,CAAC,CAAA,GAAI,KAAK,CAAA,IAAK,CAAA;AAAA,QAC1C,UAAU,OAAA,CAAQ,SAAA,GAAY,CAAC,CAAA,GAAI,KAAK,KAAK,EAAC;AAAA,QAC9C,UAAU,OAAA,CAAQ,SAAA,GAAY,CAAC,CAAA,GAAI,KAAK,CAAA,IAAK,MAAA;AAAA,QAC7C,GAAI,aAAA,IAAiB,EAAE,MAAA,EAAQ,OAAA,CAAQ,UAAA,GAAa,CAAC,CAAA,GAAI,KAAK,CAAA,IAAK,EAAC;AAAE,OACxE,CAAE,CAAA;AAAA,IACJ,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,4BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,GAAA,CAAmC;AAAA,IACvC,SAAA;AAAA,IACA,GAAA;AAAA,IACA,MAAA;AAAA,IACA,aAAA,GAAgB,KAAA;AAAA,IAChB,cAAA;AAAA,IACA,MAAA;AAAA,IACA;AAAA,GACF,EAA2B;AACzB,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAEzD,MAAA,MAAM,cAAA,GAA6C,CAAC,WAAA,EAAa,WAAW,CAAA;AAC5E,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AAEpD,MAAA,MAAM,MAAA,GAAS,MAAM,UAAA,CAAW,GAAA,CAAO;AAAA,QACrC,GAAA;AAAA,QACA,OAAO,gBAAA,IAAoB,MAAA;AAAA,QAC3B,eAAe,cAAA,IAAkB,MAAA;AAAA,QACjC,MAAA;AAAA,QACA,KAAA;AAAA,QACA,SAAS,aAAA,GAAgB,CAAC,GAAG,cAAA,EAAgB,YAAY,CAAA,GAAI;AAAA,OAC9D,CAAA;AACD,MAAA,OAAO,OAAO,IAAA,EAAK;AAAA,IACrB,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,GAAiC;AACrC,IAAA,IAAI;AACF,MAAA,MAAM,WAAA,GAAc,MAAM,IAAA,CAAK,MAAA,CAAO,eAAA,EAAgB;AACtD,MAAA,OAAO,WAAA,CAAY,GAAA,CAAI,CAAA,UAAA,KAAc,UAAA,CAAW,IAAI,CAAA;AAAA,IACtD,SAASA,OAAA,EAAY;AACnB,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc;AAAA,SAC1B;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,aAAA,CAAc,EAAE,SAAA,EAAU,EAA6C;AAC3E,IAAA,IAAI;AACF,MAAA,MAAM,aAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACzD,MAAA,MAAM,KAAA,GAAQ,MAAM,UAAA,CAAW,KAAA,EAAM;AACrC,MAAA,MAAM,WAAW,UAAA,CAAW,QAAA;AAC5B,MAAA,MAAM,KAAA,GAAQ,WAAW,aAAA,CAAc,IAAA,EAAM,SAAS,UAAA,CAAW,aAAA,CAAc,OAAO,KAAA,IAAS,MAAA;AAE/F,MAAA,OAAO;AAAA,QACL,SAAA,EAAW,UAAU,SAAA,IAAa,CAAA;AAAA,QAClC,KAAA;AAAA,QACA,MAAA,EAAQ,KAAA,GAAS,aAAA,CAAc,KAAK,CAAA,GAA8C;AAAA,OACpF;AAAA,IACF,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,qCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,WAAA,CAAY,EAAE,SAAA,EAAU,EAAqC;AACjE,IAAA,IAAI;AACF,MAAA,MAAM,KAAK,MAAA,CAAO,gBAAA,CAAiB,EAAE,IAAA,EAAM,WAAW,CAAA;AACtD,MAAA,IAAA,CAAK,WAAA,CAAY,OAAO,SAAS,CAAA;AAAA,IACnC,SAASA,OAAA,EAAY;AACnB,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,mCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,SAAA,CAAU,EAAE,SAAA,EAAW,cAAa,EAA+D;AACvG,IAAA,IAAI;AACF,MAAA,MAAM,UAAA,GAAa,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,SAAA,EAAW,WAAA,EAAa,MAAM,CAAA;AAC5E,MAAA,MAAM,mBAAmB,MAAM,UAAA,CAAW,KAAK,EAAE,IAAA,EAAM,cAAc,CAAA;AACrE,MAAA,IAAA,CAAK,WAAA,CAAY,GAAA,CAAI,YAAA,EAAc,gBAAgB,CAAA;AAAA,IACrD,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,0BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,YAAA,CAAa,EAAE,SAAA,EAAW,EAAA,EAAI,QAAO,EAAsC;AAC/E,IAAA,IAAI,CAAC,MAAA,CAAO,MAAA,IAAU,CAAC,OAAO,QAAA,EAAU;AACtC,MAAA,MAAM,IAAIH,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,iCAAA;AAAA,QACJ,IAAA,EAAM,gCAAA;AAAA,QACN,QAAQC,iBAAA,CAAY,aAAA;AAAA,QACpB,UAAUC,mBAAA,CAAc,IAAA;AAAA,QACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,OAC1B,CAAA;AAAA,IACH;AAEA,IAAA,IAAI;AACF,MAAA,MAAM,aAAyB,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAErE,MAAA,MAAM,eAAA,GAA6B,EAAE,GAAA,EAAK,CAAC,EAAE,CAAA,EAAE;AAE/C,MAAA,IAAI,QAAQ,MAAA,EAAQ;AAClB,QAAA,MAAM,QAAQ,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACpD,QAAA,IAAA,CAAK,yBAAyB,CAAC,MAAA,CAAO,MAAM,CAAA,EAAG,MAAM,SAAS,CAAA;AAC9D,QAAA,eAAA,CAAgB,UAAA,GAAa,CAAC,MAAA,CAAO,MAAM,CAAA;AAAA,MAC7C;AAEA,MAAA,IAAI,QAAQ,QAAA,EAAU;AACpB,QAAA,eAAA,CAAgB,SAAA,GAAY,CAAC,MAAA,CAAO,QAAQ,CAAA;AAAA,MAC9C;AAEA,MAAA,OAAO,MAAM,UAAA,CAAW,MAAA,CAAO,eAAe,CAAA;AAAA,IAChD,SAASC,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA,EAEA,MAAM,YAAA,CAAa,EAAE,SAAA,EAAW,IAAG,EAAsC;AACvE,IAAA,IAAI;AACF,MAAA,MAAM,aAAyB,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AACrE,MAAA,MAAM,WAAW,MAAA,CAAO,EAAE,KAAK,CAAC,EAAE,GAAG,CAAA;AAAA,IACvC,SAASA,OAAA,EAAY;AACnB,MAAA,IAAIA,OAAA,YAAiBH,mBAAa,MAAMG,OAAA;AACxC,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,6BAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,aAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AACF;;;AC7ZO,IAAM,aAAA,GAAgB,CAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,CAAA","file":"index.cjs","sourcesContent":["import { BaseFilterTranslator } from '@mastra/core/vector/filter';\nimport type {\n VectorFilter,\n OperatorSupport,\n QueryOperator,\n OperatorValueMap,\n LogicalOperatorValueMap,\n BlacklistedRootOperators,\n} from '@mastra/core/vector/filter';\n\ntype ChromaOperatorValueMap = Omit<OperatorValueMap, '$exists' | '$elemMatch' | '$regex' | '$options'>;\n\ntype ChromaLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor' | '$not'>;\n\ntype ChromaBlacklisted = BlacklistedRootOperators | '$nor' | '$not';\n\nexport type ChromaVectorFilter = VectorFilter<\n keyof ChromaOperatorValueMap,\n ChromaOperatorValueMap,\n ChromaLogicalOperatorValueMap,\n ChromaBlacklisted\n>;\n\n/**\n * Translator for Chroma filter queries.\n * Maintains MongoDB-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class ChromaFilterTranslator extends BaseFilterTranslator<ChromaVectorFilter> {\n protected override getSupportedOperators(): OperatorSupport {\n return {\n ...BaseFilterTranslator.DEFAULT_OPERATORS,\n logical: ['$and', '$or'],\n array: ['$in', '$nin'],\n element: [],\n regex: [],\n custom: [],\n };\n }\n\n translate(filter?: ChromaVectorFilter): ChromaVectorFilter {\n if (this.isEmpty(filter)) return filter;\n this.validateFilter(filter);\n\n return this.translateNode(filter);\n }\n\n private translateNode(node: ChromaVectorFilter, currentPath: string = ''): any {\n // Handle primitive values and arrays\n if (this.isRegex(node)) {\n throw new Error('Regex is supported in Chroma via the `documentFilter` argument');\n }\n if (this.isPrimitive(node)) return this.normalizeComparisonValue(node);\n if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };\n\n const entries = Object.entries(node as Record<string, any>);\n const firstEntry = entries[0];\n // Handle single operator case\n if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {\n const [operator, value] = firstEntry;\n const translated = this.translateOperator(operator, value);\n if (this.isLogicalOperator(operator) && Array.isArray(translated) && translated.length === 1) {\n return translated[0];\n }\n return this.isLogicalOperator(operator) ? { [operator]: translated } : translated;\n }\n\n // Process each entry\n const result: Record<string, any> = {};\n const multiOperatorConditions: any[] = [];\n\n for (const [key, value] of entries) {\n const newPath = currentPath ? `${currentPath}.${key}` : key;\n\n if (this.isOperator(key)) {\n result[key] = this.translateOperator(key, value);\n continue;\n }\n\n if (typeof value === 'object' && value !== null && !Array.isArray(value)) {\n // Check for multiple operators on same field\n const valueEntries = Object.entries(value);\n if (valueEntries.every(([op]) => this.isOperator(op)) && valueEntries.length > 1) {\n valueEntries.forEach(([op, opValue]) => {\n multiOperatorConditions.push({\n [newPath]: { [op]: this.normalizeComparisonValue(opValue) },\n });\n });\n continue;\n }\n\n // Check if the nested object contains operators\n if (Object.keys(value).length === 0) {\n result[newPath] = this.translateNode(value);\n } else {\n const hasOperators = Object.keys(value).some(k => this.isOperator(k));\n if (hasOperators) {\n // For objects with operators, normalize each operator value\n const normalizedValue: Record<string, any> = {};\n for (const [op, opValue] of Object.entries(value)) {\n normalizedValue[op] = this.isOperator(op) ? this.translateOperator(op, opValue) : opValue;\n }\n result[newPath] = normalizedValue;\n } else {\n // For objects without operators, flatten them\n Object.assign(result, this.translateNode(value, newPath));\n }\n }\n } else {\n result[newPath] = this.translateNode(value);\n }\n }\n\n // If we have multiple operators, return them combined with $and\n if (multiOperatorConditions.length > 0) {\n return { $and: multiOperatorConditions };\n }\n\n // Wrap in $and if there are multiple top-level fields\n if (Object.keys(result).length > 1 && !currentPath) {\n return {\n $and: Object.entries(result).map(([key, value]) => ({ [key]: value })),\n };\n }\n\n return result;\n }\n\n private translateOperator(operator: QueryOperator, value: any): any {\n // Handle logical operators\n if (this.isLogicalOperator(operator)) {\n return Array.isArray(value) ? value.map(item => this.translateNode(item)) : this.translateNode(value);\n }\n\n // Handle comparison and element operators\n return this.normalizeComparisonValue(value);\n }\n}\n","import { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport { MastraVector } from '@mastra/core/vector';\nimport type {\n QueryResult,\n IndexStats,\n CreateIndexParams,\n UpsertVectorParams,\n QueryVectorParams,\n DescribeIndexParams,\n DeleteIndexParams,\n DeleteVectorParams,\n UpdateVectorParams,\n} from '@mastra/core/vector';\nimport { ChromaClient, CloudClient } from 'chromadb';\nimport type { ChromaClientArgs, RecordSet, Where, WhereDocument, Collection, Metadata } from 'chromadb';\nimport type { ChromaVectorFilter } from './filter';\nimport { ChromaFilterTranslator } from './filter';\n\ninterface ChromaUpsertVectorParams extends UpsertVectorParams {\n documents?: string[];\n}\n\ninterface ChromaQueryVectorParams extends QueryVectorParams<ChromaVectorFilter> {\n documentFilter?: WhereDocument | null;\n}\n\ninterface ChromaGetRecordsParams {\n indexName: string;\n ids?: string[];\n filter?: ChromaVectorFilter;\n documentFilter?: WhereDocument | null;\n includeVector?: boolean;\n limit?: number;\n offset?: number;\n}\n\ntype MastraMetadata = {\n dimension?: number;\n};\n\ntype ChromaVectorArgs = ChromaClientArgs & { apiKey?: string };\n\nconst spaceMappings = {\n cosine: 'cosine',\n euclidean: 'l2',\n dotproduct: 'ip',\n l2: 'euclidean',\n ip: 'dotproduct',\n};\n\nexport class ChromaVector extends MastraVector<ChromaVectorFilter> {\n private client: ChromaClient;\n private collections: Map<string, Collection>;\n\n constructor(chromaClientArgs?: ChromaVectorArgs) {\n super();\n if (chromaClientArgs?.apiKey) {\n this.client = new CloudClient({\n apiKey: chromaClientArgs.apiKey,\n tenant: chromaClientArgs.tenant,\n database: chromaClientArgs.database,\n });\n } else {\n this.client = new ChromaClient(chromaClientArgs);\n }\n this.collections = new Map();\n }\n\n async getCollection({ indexName, forceUpdate = false }: { indexName: string; forceUpdate?: boolean }) {\n let collection = this.collections.get(indexName);\n if (forceUpdate || !collection) {\n try {\n collection = await this.client.getCollection({ name: indexName });\n this.collections.set(indexName, collection);\n return collection;\n } catch {\n throw new MastraError({\n id: 'CHROMA_COLLECTION_GET_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n });\n }\n }\n return collection;\n }\n\n private validateVectorDimensions(vectors: number[][], dimension: number): void {\n for (let i = 0; i < vectors.length; i++) {\n if (vectors?.[i]?.length !== dimension) {\n throw new Error(\n `Vector at index ${i} has invalid dimension ${vectors?.[i]?.length}. Expected ${dimension} dimensions.`,\n );\n }\n }\n }\n\n async upsert({ indexName, vectors, metadata, ids, documents }: ChromaUpsertVectorParams): Promise<string[]> {\n try {\n const collection = await this.getCollection({ indexName });\n\n const stats = await this.describeIndex({ indexName });\n this.validateVectorDimensions(vectors, stats.dimension);\n const generatedIds = ids || vectors.map(() => crypto.randomUUID());\n\n await collection.upsert({\n ids: generatedIds,\n embeddings: vectors,\n metadatas: metadata,\n documents: documents,\n });\n\n return generatedIds;\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_UPSERT_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n if (!Number.isInteger(dimension) || dimension <= 0) {\n throw new MastraError({\n id: 'CHROMA_VECTOR_CREATE_INDEX_INVALID_DIMENSION',\n text: 'Dimension must be a positive integer',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.USER,\n details: { dimension },\n });\n }\n\n const hnswSpace = spaceMappings[metric] as 'cosine' | 'l2' | 'ip' | undefined;\n\n if (!hnswSpace || !['cosine', 'l2', 'ip'].includes(hnswSpace)) {\n throw new MastraError({\n id: 'CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC',\n text: `Invalid metric: \"${metric}\". Must be one of: cosine, euclidean, dotproduct`,\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.USER,\n details: { metric },\n });\n }\n\n try {\n const collection = await this.client.createCollection({\n name: indexName,\n metadata: { dimension },\n configuration: { hnsw: { space: hnswSpace } },\n embeddingFunction: null,\n });\n this.collections.set(indexName, collection);\n } catch (error: any) {\n // Check for 'already exists' error\n const message = error?.message || error?.toString();\n if (message && message.toLowerCase().includes('already exists')) {\n // Fetch collection info and check dimension\n await this.validateExistingIndex(indexName, dimension, metric);\n return;\n }\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_CREATE_INDEX_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n transformFilter(filter?: ChromaVectorFilter) {\n const translator = new ChromaFilterTranslator();\n const translatedFilter = translator.translate(filter);\n return translatedFilter ? (translatedFilter as Where) : undefined;\n }\n\n async query<T extends Metadata = Metadata>({\n indexName,\n queryVector,\n topK = 10,\n filter,\n includeVector = false,\n documentFilter,\n }: ChromaQueryVectorParams): Promise<QueryResult[]> {\n try {\n const collection = await this.getCollection({ indexName });\n\n const defaultInclude: ['documents', 'metadatas', 'distances'] = ['documents', 'metadatas', 'distances'];\n\n const translatedFilter = this.transformFilter(filter);\n const results = await collection.query<T>({\n queryEmbeddings: [queryVector],\n nResults: topK,\n where: translatedFilter ?? undefined,\n whereDocument: documentFilter ?? undefined,\n include: includeVector ? [...defaultInclude, 'embeddings'] : defaultInclude,\n });\n\n return (results.ids[0] || []).map((id: string, index: number) => ({\n id,\n score: results.distances?.[0]?.[index] || 0,\n metadata: results.metadatas?.[0]?.[index] || {},\n document: results.documents?.[0]?.[index] ?? undefined,\n ...(includeVector && { vector: results.embeddings?.[0]?.[index] || [] }),\n }));\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_QUERY_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async get<T extends Metadata = Metadata>({\n indexName,\n ids,\n filter,\n includeVector = false,\n documentFilter,\n offset,\n limit,\n }: ChromaGetRecordsParams) {\n try {\n const collection = await this.getCollection({ indexName });\n\n const defaultInclude: ['documents', 'metadatas'] = ['documents', 'metadatas'];\n const translatedFilter = this.transformFilter(filter);\n\n const result = await collection.get<T>({\n ids,\n where: translatedFilter ?? undefined,\n whereDocument: documentFilter ?? undefined,\n offset,\n limit,\n include: includeVector ? [...defaultInclude, 'embeddings'] : defaultInclude,\n });\n return result.rows();\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_GET_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async listIndexes(): Promise<string[]> {\n try {\n const collections = await this.client.listCollections();\n return collections.map(collection => collection.name);\n } catch (error: any) {\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_LIST_INDEXES_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n },\n error,\n );\n }\n }\n\n /**\n * Retrieves statistics about a vector index.\n *\n * @param {string} indexName - The name of the index to describe\n * @returns A promise that resolves to the index statistics including dimension, count and metric\n */\n async describeIndex({ indexName }: DescribeIndexParams): Promise<IndexStats> {\n try {\n const collection = await this.getCollection({ indexName });\n const count = await collection.count();\n const metadata = collection.metadata as MastraMetadata | undefined;\n const space = collection.configuration.hnsw?.space || collection.configuration.spann?.space || undefined;\n\n return {\n dimension: metadata?.dimension || 0,\n count,\n metric: space ? (spaceMappings[space] as 'cosine' | 'euclidean' | 'dotproduct') : undefined,\n };\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_DESCRIBE_INDEX_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n try {\n await this.client.deleteCollection({ name: indexName });\n this.collections.delete(indexName);\n } catch (error: any) {\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_DELETE_INDEX_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n async forkIndex({ indexName, newIndexName }: { indexName: string; newIndexName: string }): Promise<void> {\n try {\n const collection = await this.getCollection({ indexName, forceUpdate: true });\n const forkedCollection = await collection.fork({ name: newIndexName });\n this.collections.set(newIndexName, forkedCollection);\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_INDEX_FORK_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n }\n }\n\n /**\n * Updates a vector by its ID with the provided vector and/or metadata.\n * @param indexName - The name of the index containing the vector.\n * @param id - The ID of the vector to update.\n * @param update - An object containing the vector and/or metadata to update.\n * @param update.vector - An optional array of numbers representing the new vector.\n * @param update.metadata - An optional record containing the new metadata.\n * @returns A promise that resolves when the update is complete.\n * @throws Will throw an error if no updates are provided or if the update operation fails.\n */\n async updateVector({ indexName, id, update }: UpdateVectorParams): Promise<void> {\n if (!update.vector && !update.metadata) {\n throw new MastraError({\n id: 'CHROMA_VECTOR_UPDATE_NO_PAYLOAD',\n text: 'No updates provided for vector',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.USER,\n details: { indexName, id },\n });\n }\n\n try {\n const collection: Collection = await this.getCollection({ indexName });\n\n const updateRecordSet: RecordSet = { ids: [id] };\n\n if (update?.vector) {\n const stats = await this.describeIndex({ indexName });\n this.validateVectorDimensions([update.vector], stats.dimension);\n updateRecordSet.embeddings = [update.vector];\n }\n\n if (update?.metadata) {\n updateRecordSet.metadatas = [update.metadata];\n }\n\n return await collection.update(updateRecordSet);\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_UPDATE_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, id },\n },\n error,\n );\n }\n }\n\n async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n try {\n const collection: Collection = await this.getCollection({ indexName });\n await collection.delete({ ids: [id] });\n } catch (error: any) {\n if (error instanceof MastraError) throw error;\n throw new MastraError(\n {\n id: 'CHROMA_VECTOR_DELETE_FAILED',\n domain: ErrorDomain.MASTRA_VECTOR,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, id },\n },\n error,\n );\n }\n }\n}\n","/**\n * Vector store specific prompt that details supported operators and examples.\n * This prompt helps users construct valid filters for Chroma Vector.\n */\nexport const CHROMA_PROMPT = `When querying Chroma, you can ONLY use the operators listed below. Any other operators will be rejected.\nImportant: Don't explain how to construct the filter - use the specified operators and fields to search the content and return relevant results.\nIf a user tries to give an explicit operator that is not supported, reject the filter entirely and let them know that the operator is not supported.\n\nBasic Comparison Operators:\n- $eq: Exact match (default when using field: value)\n Example: { \"category\": \"electronics\" }\n- $ne: Not equal\n Example: { \"category\": { \"$ne\": \"electronics\" } }\n- $gt: Greater than\n Example: { \"price\": { \"$gt\": 100 } }\n- $gte: Greater than or equal\n Example: { \"price\": { \"$gte\": 100 } }\n- $lt: Less than\n Example: { \"price\": { \"$lt\": 100 } }\n- $lte: Less than or equal\n Example: { \"price\": { \"$lte\": 100 } }\n\nArray Operators:\n- $in: Match any value in array\n Example: { \"category\": { \"$in\": [\"electronics\", \"books\"] } }\n- $nin: Does not match any value in array\n Example: { \"category\": { \"$nin\": [\"electronics\", \"books\"] } }\n\nLogical Operators:\n- $and: Logical AND\n Example: { \"$and\": [{ \"price\": { \"$gt\": 100 } }, { \"category\": \"electronics\" }] }\n- $or: Logical OR\n Example: { \"$or\": [{ \"price\": { \"$lt\": 50 } }, { \"category\": \"books\" }] }\n\nRestrictions:\n- Regex patterns are not supported\n- Element operators are not supported\n- Only $and and $or logical operators are supported\n- Nested fields are supported using dot notation\n- Multiple conditions on the same field are supported with both implicit and explicit $and\n- Empty arrays in $in/$nin will return no results\n- If multiple top-level fields exist, they're wrapped in $and\n- Only logical operators ($and, $or) can be used at the top level\n- All other operators must be used within a field condition\n Valid: { \"field\": { \"$gt\": 100 } }\n Valid: { \"$and\": [...] }\n Invalid: { \"$gt\": 100 }\n Invalid: { \"$in\": [...] }\n- Logical operators must contain field conditions, not direct operators\n Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n Invalid: { \"$and\": [{ \"$gt\": 100 }] }\n- Logical operators ($and, $or):\n - Can only be used at top level or nested within other logical operators\n - Can not be used on a field level, or be nested inside a field\n - Can not be used inside an operator\n - Valid: { \"$and\": [{ \"field\": { \"$gt\": 100 } }] }\n - Valid: { \"$or\": [{ \"$and\": [{ \"field\": { \"$gt\": 100 } }] }] }\n - Invalid: { \"field\": { \"$and\": [{ \"$gt\": 100 }] } }\n - Invalid: { \"field\": { \"$or\": [{ \"$gt\": 100 }] } }\n - Invalid: { \"field\": { \"$gt\": { \"$and\": [{...}] } } }\n\nExample Complex Query:\n{\n \"$and\": [\n { \"category\": { \"$in\": [\"electronics\", \"computers\"] } },\n { \"price\": { \"$gte\": 100, \"$lte\": 1000 } },\n { \"$or\": [\n { \"inStock\": true },\n { \"preorder\": true }\n ]}\n ]\n}`;\n"]}
|
package/dist/index.d.ts
CHANGED
|
@@ -1,2 +1,3 @@
|
|
|
1
|
-
export
|
|
2
|
-
export {
|
|
1
|
+
export * from './vector/index.js';
|
|
2
|
+
export { CHROMA_PROMPT } from './vector/prompt.js';
|
|
3
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,gBAAgB,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC"}
|
package/dist/index.js
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
|
+
import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
|
|
1
2
|
import { MastraVector } from '@mastra/core/vector';
|
|
2
|
-
import { ChromaClient } from 'chromadb';
|
|
3
|
+
import { CloudClient, ChromaClient } from 'chromadb';
|
|
3
4
|
import { BaseFilterTranslator } from '@mastra/core/vector/filter';
|
|
4
5
|
|
|
5
6
|
// src/vector/index.ts
|
|
@@ -21,7 +22,7 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
|
|
|
21
22
|
}
|
|
22
23
|
translateNode(node, currentPath = "") {
|
|
23
24
|
if (this.isRegex(node)) {
|
|
24
|
-
throw new Error("Regex is
|
|
25
|
+
throw new Error("Regex is supported in Chroma via the `documentFilter` argument");
|
|
25
26
|
}
|
|
26
27
|
if (this.isPrimitive(node)) return this.normalizeComparisonValue(node);
|
|
27
28
|
if (Array.isArray(node)) return { $in: this.normalizeArrayValues(node) };
|
|
@@ -30,6 +31,9 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
|
|
|
30
31
|
if (entries.length === 1 && firstEntry && this.isOperator(firstEntry[0])) {
|
|
31
32
|
const [operator, value] = firstEntry;
|
|
32
33
|
const translated = this.translateOperator(operator, value);
|
|
34
|
+
if (this.isLogicalOperator(operator) && Array.isArray(translated) && translated.length === 1) {
|
|
35
|
+
return translated[0];
|
|
36
|
+
}
|
|
33
37
|
return this.isLogicalOperator(operator) ? { [operator]: translated } : translated;
|
|
34
38
|
}
|
|
35
39
|
const result = {};
|
|
@@ -87,31 +91,46 @@ var ChromaFilterTranslator = class extends BaseFilterTranslator {
|
|
|
87
91
|
};
|
|
88
92
|
|
|
89
93
|
// src/vector/index.ts
|
|
94
|
+
var spaceMappings = {
|
|
95
|
+
cosine: "cosine",
|
|
96
|
+
euclidean: "l2",
|
|
97
|
+
dotproduct: "ip",
|
|
98
|
+
l2: "euclidean",
|
|
99
|
+
ip: "dotproduct"
|
|
100
|
+
};
|
|
90
101
|
var ChromaVector = class extends MastraVector {
|
|
91
102
|
client;
|
|
92
103
|
collections;
|
|
93
|
-
constructor({
|
|
94
|
-
path,
|
|
95
|
-
auth
|
|
96
|
-
}) {
|
|
104
|
+
constructor(chromaClientArgs) {
|
|
97
105
|
super();
|
|
98
|
-
|
|
99
|
-
|
|
100
|
-
|
|
101
|
-
|
|
106
|
+
if (chromaClientArgs?.apiKey) {
|
|
107
|
+
this.client = new CloudClient({
|
|
108
|
+
apiKey: chromaClientArgs.apiKey,
|
|
109
|
+
tenant: chromaClientArgs.tenant,
|
|
110
|
+
database: chromaClientArgs.database
|
|
111
|
+
});
|
|
112
|
+
} else {
|
|
113
|
+
this.client = new ChromaClient(chromaClientArgs);
|
|
114
|
+
}
|
|
102
115
|
this.collections = /* @__PURE__ */ new Map();
|
|
103
116
|
}
|
|
104
|
-
async getCollection(indexName,
|
|
105
|
-
|
|
106
|
-
|
|
107
|
-
|
|
108
|
-
|
|
109
|
-
|
|
110
|
-
|
|
117
|
+
async getCollection({ indexName, forceUpdate = false }) {
|
|
118
|
+
let collection = this.collections.get(indexName);
|
|
119
|
+
if (forceUpdate || !collection) {
|
|
120
|
+
try {
|
|
121
|
+
collection = await this.client.getCollection({ name: indexName });
|
|
122
|
+
this.collections.set(indexName, collection);
|
|
123
|
+
return collection;
|
|
124
|
+
} catch {
|
|
125
|
+
throw new MastraError({
|
|
126
|
+
id: "CHROMA_COLLECTION_GET_FAILED",
|
|
127
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
128
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
129
|
+
details: { indexName }
|
|
130
|
+
});
|
|
111
131
|
}
|
|
112
|
-
return null;
|
|
113
132
|
}
|
|
114
|
-
return
|
|
133
|
+
return collection;
|
|
115
134
|
}
|
|
116
135
|
validateVectorDimensions(vectors, dimension) {
|
|
117
136
|
for (let i = 0; i < vectors.length; i++) {
|
|
@@ -122,120 +141,295 @@ var ChromaVector = class extends MastraVector {
|
|
|
122
141
|
}
|
|
123
142
|
}
|
|
124
143
|
}
|
|
125
|
-
async upsert(
|
|
126
|
-
|
|
127
|
-
|
|
128
|
-
|
|
129
|
-
|
|
130
|
-
|
|
131
|
-
|
|
132
|
-
|
|
133
|
-
|
|
134
|
-
|
|
135
|
-
|
|
136
|
-
|
|
137
|
-
|
|
138
|
-
})
|
|
139
|
-
|
|
144
|
+
async upsert({ indexName, vectors, metadata, ids, documents }) {
|
|
145
|
+
try {
|
|
146
|
+
const collection = await this.getCollection({ indexName });
|
|
147
|
+
const stats = await this.describeIndex({ indexName });
|
|
148
|
+
this.validateVectorDimensions(vectors, stats.dimension);
|
|
149
|
+
const generatedIds = ids || vectors.map(() => crypto.randomUUID());
|
|
150
|
+
await collection.upsert({
|
|
151
|
+
ids: generatedIds,
|
|
152
|
+
embeddings: vectors,
|
|
153
|
+
metadatas: metadata,
|
|
154
|
+
documents
|
|
155
|
+
});
|
|
156
|
+
return generatedIds;
|
|
157
|
+
} catch (error) {
|
|
158
|
+
if (error instanceof MastraError) throw error;
|
|
159
|
+
throw new MastraError(
|
|
160
|
+
{
|
|
161
|
+
id: "CHROMA_VECTOR_UPSERT_FAILED",
|
|
162
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
163
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
164
|
+
details: { indexName }
|
|
165
|
+
},
|
|
166
|
+
error
|
|
167
|
+
);
|
|
168
|
+
}
|
|
140
169
|
}
|
|
141
|
-
|
|
142
|
-
cosine: "cosine",
|
|
143
|
-
euclidean: "l2",
|
|
144
|
-
dotproduct: "ip",
|
|
145
|
-
l2: "euclidean",
|
|
146
|
-
ip: "dotproduct"
|
|
147
|
-
};
|
|
148
|
-
async createIndex(...args) {
|
|
149
|
-
const params = this.normalizeArgs("createIndex", args);
|
|
150
|
-
const { indexName, dimension, metric = "cosine" } = params;
|
|
170
|
+
async createIndex({ indexName, dimension, metric = "cosine" }) {
|
|
151
171
|
if (!Number.isInteger(dimension) || dimension <= 0) {
|
|
152
|
-
throw new
|
|
172
|
+
throw new MastraError({
|
|
173
|
+
id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_DIMENSION",
|
|
174
|
+
text: "Dimension must be a positive integer",
|
|
175
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
176
|
+
category: ErrorCategory.USER,
|
|
177
|
+
details: { dimension }
|
|
178
|
+
});
|
|
153
179
|
}
|
|
154
|
-
const hnswSpace =
|
|
155
|
-
if (!["cosine", "l2", "ip"].includes(hnswSpace)) {
|
|
156
|
-
throw new
|
|
180
|
+
const hnswSpace = spaceMappings[metric];
|
|
181
|
+
if (!hnswSpace || !["cosine", "l2", "ip"].includes(hnswSpace)) {
|
|
182
|
+
throw new MastraError({
|
|
183
|
+
id: "CHROMA_VECTOR_CREATE_INDEX_INVALID_METRIC",
|
|
184
|
+
text: `Invalid metric: "${metric}". Must be one of: cosine, euclidean, dotproduct`,
|
|
185
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
186
|
+
category: ErrorCategory.USER,
|
|
187
|
+
details: { metric }
|
|
188
|
+
});
|
|
157
189
|
}
|
|
158
190
|
try {
|
|
159
|
-
await this.client.createCollection({
|
|
191
|
+
const collection = await this.client.createCollection({
|
|
160
192
|
name: indexName,
|
|
161
|
-
metadata: {
|
|
162
|
-
|
|
163
|
-
|
|
164
|
-
}
|
|
193
|
+
metadata: { dimension },
|
|
194
|
+
configuration: { hnsw: { space: hnswSpace } },
|
|
195
|
+
embeddingFunction: null
|
|
165
196
|
});
|
|
197
|
+
this.collections.set(indexName, collection);
|
|
166
198
|
} catch (error) {
|
|
167
199
|
const message = error?.message || error?.toString();
|
|
168
200
|
if (message && message.toLowerCase().includes("already exists")) {
|
|
169
201
|
await this.validateExistingIndex(indexName, dimension, metric);
|
|
170
202
|
return;
|
|
171
203
|
}
|
|
172
|
-
throw
|
|
204
|
+
throw new MastraError(
|
|
205
|
+
{
|
|
206
|
+
id: "CHROMA_VECTOR_CREATE_INDEX_FAILED",
|
|
207
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
208
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
209
|
+
details: { indexName }
|
|
210
|
+
},
|
|
211
|
+
error
|
|
212
|
+
);
|
|
173
213
|
}
|
|
174
214
|
}
|
|
175
215
|
transformFilter(filter) {
|
|
176
216
|
const translator = new ChromaFilterTranslator();
|
|
177
|
-
|
|
217
|
+
const translatedFilter = translator.translate(filter);
|
|
218
|
+
return translatedFilter ? translatedFilter : void 0;
|
|
178
219
|
}
|
|
179
|
-
async query(
|
|
180
|
-
|
|
181
|
-
|
|
182
|
-
|
|
183
|
-
|
|
184
|
-
|
|
185
|
-
|
|
186
|
-
|
|
187
|
-
|
|
188
|
-
|
|
189
|
-
|
|
190
|
-
|
|
191
|
-
|
|
192
|
-
|
|
193
|
-
|
|
194
|
-
|
|
195
|
-
|
|
196
|
-
|
|
197
|
-
|
|
198
|
-
|
|
220
|
+
async query({
|
|
221
|
+
indexName,
|
|
222
|
+
queryVector,
|
|
223
|
+
topK = 10,
|
|
224
|
+
filter,
|
|
225
|
+
includeVector = false,
|
|
226
|
+
documentFilter
|
|
227
|
+
}) {
|
|
228
|
+
try {
|
|
229
|
+
const collection = await this.getCollection({ indexName });
|
|
230
|
+
const defaultInclude = ["documents", "metadatas", "distances"];
|
|
231
|
+
const translatedFilter = this.transformFilter(filter);
|
|
232
|
+
const results = await collection.query({
|
|
233
|
+
queryEmbeddings: [queryVector],
|
|
234
|
+
nResults: topK,
|
|
235
|
+
where: translatedFilter ?? void 0,
|
|
236
|
+
whereDocument: documentFilter ?? void 0,
|
|
237
|
+
include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
|
|
238
|
+
});
|
|
239
|
+
return (results.ids[0] || []).map((id, index) => ({
|
|
240
|
+
id,
|
|
241
|
+
score: results.distances?.[0]?.[index] || 0,
|
|
242
|
+
metadata: results.metadatas?.[0]?.[index] || {},
|
|
243
|
+
document: results.documents?.[0]?.[index] ?? void 0,
|
|
244
|
+
...includeVector && { vector: results.embeddings?.[0]?.[index] || [] }
|
|
245
|
+
}));
|
|
246
|
+
} catch (error) {
|
|
247
|
+
if (error instanceof MastraError) throw error;
|
|
248
|
+
throw new MastraError(
|
|
249
|
+
{
|
|
250
|
+
id: "CHROMA_VECTOR_QUERY_FAILED",
|
|
251
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
252
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
253
|
+
details: { indexName }
|
|
254
|
+
},
|
|
255
|
+
error
|
|
256
|
+
);
|
|
257
|
+
}
|
|
258
|
+
}
|
|
259
|
+
async get({
|
|
260
|
+
indexName,
|
|
261
|
+
ids,
|
|
262
|
+
filter,
|
|
263
|
+
includeVector = false,
|
|
264
|
+
documentFilter,
|
|
265
|
+
offset,
|
|
266
|
+
limit
|
|
267
|
+
}) {
|
|
268
|
+
try {
|
|
269
|
+
const collection = await this.getCollection({ indexName });
|
|
270
|
+
const defaultInclude = ["documents", "metadatas"];
|
|
271
|
+
const translatedFilter = this.transformFilter(filter);
|
|
272
|
+
const result = await collection.get({
|
|
273
|
+
ids,
|
|
274
|
+
where: translatedFilter ?? void 0,
|
|
275
|
+
whereDocument: documentFilter ?? void 0,
|
|
276
|
+
offset,
|
|
277
|
+
limit,
|
|
278
|
+
include: includeVector ? [...defaultInclude, "embeddings"] : defaultInclude
|
|
279
|
+
});
|
|
280
|
+
return result.rows();
|
|
281
|
+
} catch (error) {
|
|
282
|
+
if (error instanceof MastraError) throw error;
|
|
283
|
+
throw new MastraError(
|
|
284
|
+
{
|
|
285
|
+
id: "CHROMA_VECTOR_GET_FAILED",
|
|
286
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
287
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
288
|
+
details: { indexName }
|
|
289
|
+
},
|
|
290
|
+
error
|
|
291
|
+
);
|
|
292
|
+
}
|
|
199
293
|
}
|
|
200
294
|
async listIndexes() {
|
|
201
|
-
|
|
202
|
-
|
|
295
|
+
try {
|
|
296
|
+
const collections = await this.client.listCollections();
|
|
297
|
+
return collections.map((collection) => collection.name);
|
|
298
|
+
} catch (error) {
|
|
299
|
+
throw new MastraError(
|
|
300
|
+
{
|
|
301
|
+
id: "CHROMA_VECTOR_LIST_INDEXES_FAILED",
|
|
302
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
303
|
+
category: ErrorCategory.THIRD_PARTY
|
|
304
|
+
},
|
|
305
|
+
error
|
|
306
|
+
);
|
|
307
|
+
}
|
|
203
308
|
}
|
|
204
|
-
|
|
205
|
-
|
|
206
|
-
|
|
207
|
-
|
|
208
|
-
|
|
209
|
-
|
|
210
|
-
|
|
211
|
-
|
|
212
|
-
|
|
213
|
-
|
|
309
|
+
/**
|
|
310
|
+
* Retrieves statistics about a vector index.
|
|
311
|
+
*
|
|
312
|
+
* @param {string} indexName - The name of the index to describe
|
|
313
|
+
* @returns A promise that resolves to the index statistics including dimension, count and metric
|
|
314
|
+
*/
|
|
315
|
+
async describeIndex({ indexName }) {
|
|
316
|
+
try {
|
|
317
|
+
const collection = await this.getCollection({ indexName });
|
|
318
|
+
const count = await collection.count();
|
|
319
|
+
const metadata = collection.metadata;
|
|
320
|
+
const space = collection.configuration.hnsw?.space || collection.configuration.spann?.space || void 0;
|
|
321
|
+
return {
|
|
322
|
+
dimension: metadata?.dimension || 0,
|
|
323
|
+
count,
|
|
324
|
+
metric: space ? spaceMappings[space] : void 0
|
|
325
|
+
};
|
|
326
|
+
} catch (error) {
|
|
327
|
+
if (error instanceof MastraError) throw error;
|
|
328
|
+
throw new MastraError(
|
|
329
|
+
{
|
|
330
|
+
id: "CHROMA_VECTOR_DESCRIBE_INDEX_FAILED",
|
|
331
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
332
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
333
|
+
details: { indexName }
|
|
334
|
+
},
|
|
335
|
+
error
|
|
336
|
+
);
|
|
337
|
+
}
|
|
338
|
+
}
|
|
339
|
+
async deleteIndex({ indexName }) {
|
|
340
|
+
try {
|
|
341
|
+
await this.client.deleteCollection({ name: indexName });
|
|
342
|
+
this.collections.delete(indexName);
|
|
343
|
+
} catch (error) {
|
|
344
|
+
throw new MastraError(
|
|
345
|
+
{
|
|
346
|
+
id: "CHROMA_VECTOR_DELETE_INDEX_FAILED",
|
|
347
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
348
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
349
|
+
details: { indexName }
|
|
350
|
+
},
|
|
351
|
+
error
|
|
352
|
+
);
|
|
353
|
+
}
|
|
214
354
|
}
|
|
215
|
-
async
|
|
216
|
-
|
|
217
|
-
|
|
355
|
+
async forkIndex({ indexName, newIndexName }) {
|
|
356
|
+
try {
|
|
357
|
+
const collection = await this.getCollection({ indexName, forceUpdate: true });
|
|
358
|
+
const forkedCollection = await collection.fork({ name: newIndexName });
|
|
359
|
+
this.collections.set(newIndexName, forkedCollection);
|
|
360
|
+
} catch (error) {
|
|
361
|
+
if (error instanceof MastraError) throw error;
|
|
362
|
+
throw new MastraError(
|
|
363
|
+
{
|
|
364
|
+
id: "CHROMA_INDEX_FORK_FAILED",
|
|
365
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
366
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
367
|
+
details: { indexName }
|
|
368
|
+
},
|
|
369
|
+
error
|
|
370
|
+
);
|
|
371
|
+
}
|
|
218
372
|
}
|
|
219
|
-
|
|
373
|
+
/**
|
|
374
|
+
* Updates a vector by its ID with the provided vector and/or metadata.
|
|
375
|
+
* @param indexName - The name of the index containing the vector.
|
|
376
|
+
* @param id - The ID of the vector to update.
|
|
377
|
+
* @param update - An object containing the vector and/or metadata to update.
|
|
378
|
+
* @param update.vector - An optional array of numbers representing the new vector.
|
|
379
|
+
* @param update.metadata - An optional record containing the new metadata.
|
|
380
|
+
* @returns A promise that resolves when the update is complete.
|
|
381
|
+
* @throws Will throw an error if no updates are provided or if the update operation fails.
|
|
382
|
+
*/
|
|
383
|
+
async updateVector({ indexName, id, update }) {
|
|
220
384
|
if (!update.vector && !update.metadata) {
|
|
221
|
-
throw new
|
|
222
|
-
|
|
223
|
-
|
|
224
|
-
|
|
225
|
-
|
|
226
|
-
|
|
385
|
+
throw new MastraError({
|
|
386
|
+
id: "CHROMA_VECTOR_UPDATE_NO_PAYLOAD",
|
|
387
|
+
text: "No updates provided for vector",
|
|
388
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
389
|
+
category: ErrorCategory.USER,
|
|
390
|
+
details: { indexName, id }
|
|
391
|
+
});
|
|
227
392
|
}
|
|
228
|
-
|
|
229
|
-
|
|
393
|
+
try {
|
|
394
|
+
const collection = await this.getCollection({ indexName });
|
|
395
|
+
const updateRecordSet = { ids: [id] };
|
|
396
|
+
if (update?.vector) {
|
|
397
|
+
const stats = await this.describeIndex({ indexName });
|
|
398
|
+
this.validateVectorDimensions([update.vector], stats.dimension);
|
|
399
|
+
updateRecordSet.embeddings = [update.vector];
|
|
400
|
+
}
|
|
401
|
+
if (update?.metadata) {
|
|
402
|
+
updateRecordSet.metadatas = [update.metadata];
|
|
403
|
+
}
|
|
404
|
+
return await collection.update(updateRecordSet);
|
|
405
|
+
} catch (error) {
|
|
406
|
+
if (error instanceof MastraError) throw error;
|
|
407
|
+
throw new MastraError(
|
|
408
|
+
{
|
|
409
|
+
id: "CHROMA_VECTOR_UPDATE_FAILED",
|
|
410
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
411
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
412
|
+
details: { indexName, id }
|
|
413
|
+
},
|
|
414
|
+
error
|
|
415
|
+
);
|
|
230
416
|
}
|
|
231
|
-
return await collection.update(updateOptions);
|
|
232
417
|
}
|
|
233
|
-
async
|
|
418
|
+
async deleteVector({ indexName, id }) {
|
|
234
419
|
try {
|
|
235
|
-
const collection = await this.getCollection(indexName
|
|
420
|
+
const collection = await this.getCollection({ indexName });
|
|
236
421
|
await collection.delete({ ids: [id] });
|
|
237
422
|
} catch (error) {
|
|
238
|
-
|
|
423
|
+
if (error instanceof MastraError) throw error;
|
|
424
|
+
throw new MastraError(
|
|
425
|
+
{
|
|
426
|
+
id: "CHROMA_VECTOR_DELETE_FAILED",
|
|
427
|
+
domain: ErrorDomain.MASTRA_VECTOR,
|
|
428
|
+
category: ErrorCategory.THIRD_PARTY,
|
|
429
|
+
details: { indexName, id }
|
|
430
|
+
},
|
|
431
|
+
error
|
|
432
|
+
);
|
|
239
433
|
}
|
|
240
434
|
}
|
|
241
435
|
};
|
|
@@ -311,3 +505,5 @@ Example Complex Query:
|
|
|
311
505
|
}`;
|
|
312
506
|
|
|
313
507
|
export { CHROMA_PROMPT, ChromaVector };
|
|
508
|
+
//# sourceMappingURL=index.js.map
|
|
509
|
+
//# sourceMappingURL=index.js.map
|