@mastra/opensearch 0.0.0-pass-headers-for-create-mastra-client-20250530010057 → 0.0.0-rag-chunk-extract-llm-option-20250926183645
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 +455 -2
- package/LICENSE.md +11 -42
- package/dist/index.cjs +127 -53
- package/dist/index.cjs.map +1 -0
- package/dist/index.d.ts +2 -1
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +119 -45
- package/dist/index.js.map +1 -0
- package/dist/vector/filter.d.ts +47 -0
- package/dist/vector/filter.d.ts.map +1 -0
- package/dist/vector/index.d.ts +101 -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 +28 -14
- package/.turbo/turbo-build.log +0 -23
- package/dist/_tsup-dts-rollup.d.cts +0 -159
- package/dist/_tsup-dts-rollup.d.ts +0 -159
- package/dist/index.d.cts +0 -1
- package/docker-compose.yaml +0 -23
- package/eslint.config.js +0 -6
- package/src/index.ts +0 -1
- package/src/vector/filter.test.ts +0 -660
- package/src/vector/filter.ts +0 -461
- package/src/vector/index.test.ts +0 -1559
- package/src/vector/index.ts +0 -357
- package/src/vector/prompt.ts +0 -82
- package/tsconfig.json +0 -5
- package/vitest.config.ts +0 -11
package/dist/index.cjs
CHANGED
|
@@ -1,5 +1,6 @@
|
|
|
1
1
|
'use strict';
|
|
2
2
|
|
|
3
|
+
var error = require('@mastra/core/error');
|
|
3
4
|
var vector = require('@mastra/core/vector');
|
|
4
5
|
var opensearch = require('@opensearch-project/opensearch');
|
|
5
6
|
var filter = require('@mastra/core/vector/filter');
|
|
@@ -11,7 +12,6 @@ var OpenSearchFilterTranslator = class extends filter.BaseFilterTranslator {
|
|
|
11
12
|
...filter.BaseFilterTranslator.DEFAULT_OPERATORS,
|
|
12
13
|
logical: ["$and", "$or", "$not"],
|
|
13
14
|
array: ["$in", "$nin", "$all"],
|
|
14
|
-
element: ["$exists"],
|
|
15
15
|
regex: ["$regex"],
|
|
16
16
|
custom: []
|
|
17
17
|
};
|
|
@@ -383,7 +383,13 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
383
383
|
*/
|
|
384
384
|
async createIndex({ indexName, dimension, metric = "cosine" }) {
|
|
385
385
|
if (!Number.isInteger(dimension) || dimension <= 0) {
|
|
386
|
-
throw new
|
|
386
|
+
throw new error.MastraError({
|
|
387
|
+
id: "STORAGE_OPENSEARCH_VECTOR_CREATE_INDEX_INVALID_ARGS",
|
|
388
|
+
domain: error.ErrorDomain.STORAGE,
|
|
389
|
+
category: error.ErrorCategory.USER,
|
|
390
|
+
text: "Dimension must be a positive integer",
|
|
391
|
+
details: { indexName, dimension }
|
|
392
|
+
});
|
|
387
393
|
}
|
|
388
394
|
try {
|
|
389
395
|
await this.client.indices.create({
|
|
@@ -408,14 +414,21 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
408
414
|
}
|
|
409
415
|
}
|
|
410
416
|
});
|
|
411
|
-
} catch (error) {
|
|
412
|
-
const message = error?.message || error?.toString();
|
|
417
|
+
} catch (error$1) {
|
|
418
|
+
const message = error$1?.message || error$1?.toString();
|
|
413
419
|
if (message && message.toLowerCase().includes("already exists")) {
|
|
414
420
|
await this.validateExistingIndex(indexName, dimension, metric);
|
|
415
421
|
return;
|
|
416
422
|
}
|
|
417
|
-
|
|
418
|
-
|
|
423
|
+
throw new error.MastraError(
|
|
424
|
+
{
|
|
425
|
+
id: "STORAGE_OPENSEARCH_VECTOR_CREATE_INDEX_FAILED",
|
|
426
|
+
domain: error.ErrorDomain.STORAGE,
|
|
427
|
+
category: error.ErrorCategory.THIRD_PARTY,
|
|
428
|
+
details: { indexName, dimension, metric }
|
|
429
|
+
},
|
|
430
|
+
error$1
|
|
431
|
+
);
|
|
419
432
|
}
|
|
420
433
|
}
|
|
421
434
|
/**
|
|
@@ -428,9 +441,15 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
428
441
|
const response = await this.client.cat.indices({ format: "json" });
|
|
429
442
|
const indexes = response.body.map((record) => record.index).filter((index) => index !== void 0);
|
|
430
443
|
return indexes;
|
|
431
|
-
} catch (error) {
|
|
432
|
-
|
|
433
|
-
|
|
444
|
+
} catch (error$1) {
|
|
445
|
+
throw new error.MastraError(
|
|
446
|
+
{
|
|
447
|
+
id: "STORAGE_OPENSEARCH_VECTOR_LIST_INDEXES_FAILED",
|
|
448
|
+
domain: error.ErrorDomain.STORAGE,
|
|
449
|
+
category: error.ErrorCategory.THIRD_PARTY
|
|
450
|
+
},
|
|
451
|
+
error$1
|
|
452
|
+
);
|
|
434
453
|
}
|
|
435
454
|
}
|
|
436
455
|
/**
|
|
@@ -460,8 +479,18 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
460
479
|
async deleteIndex({ indexName }) {
|
|
461
480
|
try {
|
|
462
481
|
await this.client.indices.delete({ index: indexName });
|
|
463
|
-
} catch (error) {
|
|
464
|
-
|
|
482
|
+
} catch (error$1) {
|
|
483
|
+
const mastraError = new error.MastraError(
|
|
484
|
+
{
|
|
485
|
+
id: "STORAGE_OPENSEARCH_VECTOR_DELETE_INDEX_FAILED",
|
|
486
|
+
domain: error.ErrorDomain.STORAGE,
|
|
487
|
+
category: error.ErrorCategory.THIRD_PARTY,
|
|
488
|
+
details: { indexName }
|
|
489
|
+
},
|
|
490
|
+
error$1
|
|
491
|
+
);
|
|
492
|
+
this.logger?.error(mastraError.toString());
|
|
493
|
+
this.logger?.trackException(mastraError);
|
|
465
494
|
}
|
|
466
495
|
}
|
|
467
496
|
/**
|
|
@@ -476,31 +505,38 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
476
505
|
async upsert({ indexName, vectors, metadata = [], ids }) {
|
|
477
506
|
const vectorIds = ids || vectors.map(() => crypto.randomUUID());
|
|
478
507
|
const operations = [];
|
|
479
|
-
const indexInfo = await this.describeIndex({ indexName });
|
|
480
|
-
this.validateVectorDimensions(vectors, indexInfo.dimension);
|
|
481
|
-
for (let i = 0; i < vectors.length; i++) {
|
|
482
|
-
const operation = {
|
|
483
|
-
index: {
|
|
484
|
-
_index: indexName,
|
|
485
|
-
_id: vectorIds[i]
|
|
486
|
-
}
|
|
487
|
-
};
|
|
488
|
-
const document = {
|
|
489
|
-
id: vectorIds[i],
|
|
490
|
-
embedding: vectors[i],
|
|
491
|
-
metadata: metadata[i] || {}
|
|
492
|
-
};
|
|
493
|
-
operations.push(operation);
|
|
494
|
-
operations.push(document);
|
|
495
|
-
}
|
|
496
508
|
try {
|
|
509
|
+
const indexInfo = await this.describeIndex({ indexName });
|
|
510
|
+
this.validateVectorDimensions(vectors, indexInfo.dimension);
|
|
511
|
+
for (let i = 0; i < vectors.length; i++) {
|
|
512
|
+
const operation = {
|
|
513
|
+
index: {
|
|
514
|
+
_index: indexName,
|
|
515
|
+
_id: vectorIds[i]
|
|
516
|
+
}
|
|
517
|
+
};
|
|
518
|
+
const document = {
|
|
519
|
+
id: vectorIds[i],
|
|
520
|
+
embedding: vectors[i],
|
|
521
|
+
metadata: metadata[i] || {}
|
|
522
|
+
};
|
|
523
|
+
operations.push(operation);
|
|
524
|
+
operations.push(document);
|
|
525
|
+
}
|
|
497
526
|
if (operations.length > 0) {
|
|
498
527
|
await this.client.bulk({ body: operations, refresh: true });
|
|
499
528
|
}
|
|
500
529
|
return vectorIds;
|
|
501
|
-
} catch (error) {
|
|
502
|
-
|
|
503
|
-
|
|
530
|
+
} catch (error$1) {
|
|
531
|
+
throw new error.MastraError(
|
|
532
|
+
{
|
|
533
|
+
id: "STORAGE_OPENSEARCH_VECTOR_UPSERT_FAILED",
|
|
534
|
+
domain: error.ErrorDomain.STORAGE,
|
|
535
|
+
category: error.ErrorCategory.THIRD_PARTY,
|
|
536
|
+
details: { indexName, vectorCount: vectors?.length || 0 }
|
|
537
|
+
},
|
|
538
|
+
error$1
|
|
539
|
+
);
|
|
504
540
|
}
|
|
505
541
|
}
|
|
506
542
|
/**
|
|
@@ -544,9 +580,16 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
544
580
|
};
|
|
545
581
|
});
|
|
546
582
|
return results;
|
|
547
|
-
} catch (error) {
|
|
548
|
-
|
|
549
|
-
|
|
583
|
+
} catch (error$1) {
|
|
584
|
+
throw new error.MastraError(
|
|
585
|
+
{
|
|
586
|
+
id: "STORAGE_OPENSEARCH_VECTOR_QUERY_FAILED",
|
|
587
|
+
domain: error.ErrorDomain.STORAGE,
|
|
588
|
+
category: error.ErrorCategory.THIRD_PARTY,
|
|
589
|
+
details: { indexName, topK }
|
|
590
|
+
},
|
|
591
|
+
error$1
|
|
592
|
+
);
|
|
550
593
|
}
|
|
551
594
|
}
|
|
552
595
|
/**
|
|
@@ -564,7 +607,7 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
564
607
|
/**
|
|
565
608
|
* Transforms the filter to the OpenSearch DSL.
|
|
566
609
|
*
|
|
567
|
-
* @param {
|
|
610
|
+
* @param {OpenSearchVectorFilter} filter - The filter to transform.
|
|
568
611
|
* @returns {Record<string, any>} The transformed filter.
|
|
569
612
|
*/
|
|
570
613
|
transformFilter(filter) {
|
|
@@ -582,34 +625,48 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
582
625
|
* @throws Will throw an error if no updates are provided or if the update operation fails.
|
|
583
626
|
*/
|
|
584
627
|
async updateVector({ indexName, id, update }) {
|
|
585
|
-
|
|
586
|
-
throw new Error("No updates provided");
|
|
587
|
-
}
|
|
628
|
+
let existingDoc;
|
|
588
629
|
try {
|
|
589
|
-
|
|
630
|
+
if (!update.vector && !update.metadata) {
|
|
631
|
+
throw new Error("No updates provided");
|
|
632
|
+
}
|
|
633
|
+
const { body } = await this.client.get({
|
|
590
634
|
index: indexName,
|
|
591
635
|
id
|
|
592
636
|
}).catch(() => {
|
|
593
637
|
throw new Error(`Document with ID ${id} not found in index ${indexName}`);
|
|
594
638
|
});
|
|
595
|
-
if (!
|
|
639
|
+
if (!body || !body._source) {
|
|
596
640
|
throw new Error(`Document with ID ${id} has no source data in index ${indexName}`);
|
|
597
641
|
}
|
|
598
|
-
|
|
599
|
-
|
|
600
|
-
|
|
601
|
-
|
|
642
|
+
existingDoc = body;
|
|
643
|
+
} catch (error$1) {
|
|
644
|
+
throw new error.MastraError(
|
|
645
|
+
{
|
|
646
|
+
id: "STORAGE_OPENSEARCH_VECTOR_UPDATE_VECTOR_FAILED",
|
|
647
|
+
domain: error.ErrorDomain.STORAGE,
|
|
648
|
+
category: error.ErrorCategory.USER,
|
|
649
|
+
details: { indexName, id }
|
|
650
|
+
},
|
|
651
|
+
error$1
|
|
652
|
+
);
|
|
653
|
+
}
|
|
654
|
+
const source = existingDoc._source;
|
|
655
|
+
const updatedDoc = {
|
|
656
|
+
id: source?.id || id
|
|
657
|
+
};
|
|
658
|
+
try {
|
|
602
659
|
if (update.vector) {
|
|
603
660
|
const indexInfo = await this.describeIndex({ indexName });
|
|
604
661
|
this.validateVectorDimensions([update.vector], indexInfo.dimension);
|
|
605
662
|
updatedDoc.embedding = update.vector;
|
|
606
|
-
} else if (source
|
|
663
|
+
} else if (source?.embedding) {
|
|
607
664
|
updatedDoc.embedding = source.embedding;
|
|
608
665
|
}
|
|
609
666
|
if (update.metadata) {
|
|
610
667
|
updatedDoc.metadata = update.metadata;
|
|
611
668
|
} else {
|
|
612
|
-
updatedDoc.metadata = source
|
|
669
|
+
updatedDoc.metadata = source?.metadata || {};
|
|
613
670
|
}
|
|
614
671
|
await this.client.index({
|
|
615
672
|
index: indexName,
|
|
@@ -617,9 +674,16 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
617
674
|
body: updatedDoc,
|
|
618
675
|
refresh: true
|
|
619
676
|
});
|
|
620
|
-
} catch (error) {
|
|
621
|
-
|
|
622
|
-
|
|
677
|
+
} catch (error$1) {
|
|
678
|
+
throw new error.MastraError(
|
|
679
|
+
{
|
|
680
|
+
id: "STORAGE_OPENSEARCH_VECTOR_UPDATE_VECTOR_FAILED",
|
|
681
|
+
domain: error.ErrorDomain.STORAGE,
|
|
682
|
+
category: error.ErrorCategory.THIRD_PARTY,
|
|
683
|
+
details: { indexName, id }
|
|
684
|
+
},
|
|
685
|
+
error$1
|
|
686
|
+
);
|
|
623
687
|
}
|
|
624
688
|
}
|
|
625
689
|
/**
|
|
@@ -636,13 +700,23 @@ var OpenSearchVector = class extends vector.MastraVector {
|
|
|
636
700
|
id,
|
|
637
701
|
refresh: true
|
|
638
702
|
});
|
|
639
|
-
} catch (error) {
|
|
640
|
-
|
|
641
|
-
|
|
642
|
-
throw error;
|
|
703
|
+
} catch (error$1) {
|
|
704
|
+
if (error$1 && typeof error$1 === "object" && "statusCode" in error$1 && error$1.statusCode === 404) {
|
|
705
|
+
return;
|
|
643
706
|
}
|
|
707
|
+
throw new error.MastraError(
|
|
708
|
+
{
|
|
709
|
+
id: "STORAGE_OPENSEARCH_VECTOR_DELETE_VECTOR_FAILED",
|
|
710
|
+
domain: error.ErrorDomain.STORAGE,
|
|
711
|
+
category: error.ErrorCategory.THIRD_PARTY,
|
|
712
|
+
details: { indexName, id }
|
|
713
|
+
},
|
|
714
|
+
error$1
|
|
715
|
+
);
|
|
644
716
|
}
|
|
645
717
|
}
|
|
646
718
|
};
|
|
647
719
|
|
|
648
720
|
exports.OpenSearchVector = OpenSearchVector;
|
|
721
|
+
//# sourceMappingURL=index.cjs.map
|
|
722
|
+
//# sourceMappingURL=index.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"sources":["../src/vector/filter.ts","../src/vector/index.ts"],"names":["BaseFilterTranslator","fieldWithKeyword","MastraVector","OpenSearchClient","MastraError","ErrorDomain","ErrorCategory","error"],"mappings":";;;;;;;;AA2BO,IAAM,0BAAA,GAAN,cAAyCA,2BAAA,CAA6C;AAAA,EACxE,qBAAA,GAAyC;AAC1D,IAAA,OAAO;AAAA,MACL,GAAGA,2BAAA,CAAqB,iBAAA;AAAA,MACxB,OAAA,EAAS,CAAC,MAAA,EAAQ,KAAA,EAAO,MAAM,CAAA;AAAA,MAC/B,KAAA,EAAO,CAAC,KAAA,EAAO,MAAA,EAAQ,MAAM,CAAA;AAAA,MAC7B,KAAA,EAAO,CAAC,QAAQ,CAAA;AAAA,MAChB,QAAQ;AAAC,KACX;AAAA,EACF;AAAA,EAEA,UAAU,MAAA,EAAyD;AACjE,IAAA,IAAI,IAAA,CAAK,OAAA,CAAQ,MAAM,CAAA,EAAG,OAAO,MAAA;AACjC,IAAA,IAAA,CAAK,eAAe,MAAM,CAAA;AAC1B,IAAA,OAAO,IAAA,CAAK,cAAc,MAAM,CAAA;AAAA,EAClC;AAAA,EAEQ,cAAc,IAAA,EAAmC;AAEvD,IAAA,IAAI,KAAK,WAAA,CAAY,IAAI,KAAK,KAAA,CAAM,OAAA,CAAQ,IAAI,CAAA,EAAG;AACjD,MAAA,OAAO,IAAA;AAAA,IACT;AAEA,IAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,IAA2B,CAAA;AAG1D,IAAA,MAAM,mBAAoC,EAAC;AAC3C,IAAA,MAAM,kBAAmC,EAAC;AAE1C,IAAA,OAAA,CAAQ,OAAA,CAAQ,CAAC,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AAChC,MAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,GAAG,CAAA,EAAG;AAC/B,QAAA,gBAAA,CAAiB,IAAA,CAAK,CAAC,GAAA,EAAK,KAAK,CAAC,CAAA;AAAA,MACpC,CAAA,MAAO;AACL,QAAA,eAAA,CAAgB,IAAA,CAAK,CAAC,GAAA,EAAK,KAAK,CAAC,CAAA;AAAA,MACnC;AAAA,IACF,CAAC,CAAA;AAGD,IAAA,IAAI,gBAAA,CAAiB,MAAA,KAAW,CAAA,IAAK,eAAA,CAAgB,WAAW,CAAA,EAAG;AACjE,MAAA,MAAM,CAAC,QAAA,EAAU,KAAK,CAAA,GAAI,iBAAiB,CAAC,CAAA;AAC5C,MAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,OAAO,UAAU,QAAA,EAAU;AACtD,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,oCAAA,EAAuC,QAAQ,CAAA,mCAAA,CAAqC,CAAA;AAAA,MACtG;AACA,MAAA,OAAO,IAAA,CAAK,wBAAA,CAAyB,QAAA,EAAU,KAAK,CAAA;AAAA,IACtD;AAGA,IAAA,MAAM,wBAAwB,eAAA,CAAgB,GAAA,CAAI,CAAC,CAAC,GAAA,EAAK,KAAK,CAAA,KAAM;AAElE,MAAA,IAAI,OAAO,UAAU,QAAA,IAAY,KAAA,KAAU,QAAQ,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AAExE,QAAA,MAAM,YAAA,GAAe,MAAA,CAAO,IAAA,CAAK,KAAK,CAAA,CAAE,KAAK,CAAA,CAAA,KAAK,IAAA,CAAK,UAAA,CAAW,CAAC,CAAC,CAAA;AAGpE,QAAA,MAAM,WAAA,GAAc,YAAY,GAAG,CAAA,CAAA;AACnC,QAAA,OAAO,YAAA,GACH,KAAK,wBAAA,CAAyB,WAAA,EAAa,KAAK,CAAA,GAChD,IAAA,CAAK,qBAAA,CAAsB,WAAA,EAAa,KAAK,CAAA;AAAA,MACnD;AAGA,MAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AACxB,QAAA,MAAMC,oBAAmB,IAAA,CAAK,kBAAA,CAAmB,CAAA,SAAA,EAAY,GAAG,IAAI,KAAK,CAAA;AACzE,QAAA,OAAO,EAAE,KAAA,EAAO,EAAE,CAACA,iBAAgB,GAAG,OAAM,EAAE;AAAA,MAChD;AAGA,MAAA,MAAM,mBAAmB,IAAA,CAAK,kBAAA,CAAmB,CAAA,SAAA,EAAY,GAAG,IAAI,KAAK,CAAA;AACzE,MAAA,OAAO,EAAE,IAAA,EAAM,EAAE,CAAC,gBAAgB,GAAG,OAAM,EAAE;AAAA,IAC/C,CAAC,CAAA;AAGD,IAAA,IAAI,gBAAA,CAAiB,SAAS,CAAA,EAAG;AAC/B,MAAA,MAAM,oBAAoB,gBAAA,CAAiB,GAAA;AAAA,QAAI,CAAC,CAAC,QAAA,EAAU,KAAK,MAC9D,IAAA,CAAK,iBAAA,CAAkB,UAA2B,KAAK;AAAA,OACzD;AAEA,MAAA,OAAO;AAAA,QACL,IAAA,EAAM;AAAA,UACJ,IAAA,EAAM,CAAC,GAAG,iBAAA,EAAmB,GAAG,qBAAqB;AAAA;AACvD,OACF;AAAA,IACF;AAGA,IAAA,IAAI,qBAAA,CAAsB,SAAS,CAAA,EAAG;AACpC,MAAA,OAAO;AAAA,QACL,IAAA,EAAM;AAAA,UACJ,IAAA,EAAM;AAAA;AACR,OACF;AAAA,IACF;AAGA,IAAA,IAAI,qBAAA,CAAsB,WAAW,CAAA,EAAG;AACtC,MAAA,OAAO,sBAAsB,CAAC,CAAA;AAAA,IAChC;AAGA,IAAA,OAAO,EAAE,SAAA,EAAW,EAAC,EAAE;AAAA,EACzB;AAAA;AAAA;AAAA;AAAA,EAKQ,qBAAA,CAAsB,OAAe,KAAA,EAAiC;AAC5E,IAAA,MAAM,UAAA,GAAa,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA,CAAE,IAAI,CAAC,CAAC,QAAA,EAAU,QAAQ,CAAA,KAAM;AACrE,MAAA,MAAM,SAAA,GAAY,CAAA,EAAG,KAAK,CAAA,CAAA,EAAI,QAAQ,CAAA,CAAA;AAGtC,MAAA,IAAI,IAAA,CAAK,UAAA,CAAW,QAAQ,CAAA,EAAG;AAC7B,QAAA,OAAO,IAAA,CAAK,iBAAA,CAAkB,QAAA,EAA2B,QAAA,EAAU,KAAK,CAAA;AAAA,MAC1E;AAEA,MAAA,IAAI,OAAO,aAAa,QAAA,IAAY,QAAA,KAAa,QAAQ,CAAC,KAAA,CAAM,OAAA,CAAQ,QAAQ,CAAA,EAAG;AAEjF,QAAA,MAAM,YAAA,GAAe,MAAA,CAAO,IAAA,CAAK,QAAQ,CAAA,CAAE,KAAK,CAAA,CAAA,KAAK,IAAA,CAAK,UAAA,CAAW,CAAC,CAAC,CAAA;AACvE,QAAA,IAAI,YAAA,EAAc;AAChB,UAAA,OAAO,IAAA,CAAK,wBAAA,CAAyB,SAAA,EAAW,QAAQ,CAAA;AAAA,QAC1D;AACA,QAAA,OAAO,IAAA,CAAK,qBAAA,CAAsB,SAAA,EAAW,QAAQ,CAAA;AAAA,MACvD;AACA,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,kBAAA,CAAmB,SAAA,EAAW,QAAQ,CAAA;AACpE,MAAA,OAAO,EAAE,IAAA,EAAM,EAAE,CAAC,gBAAgB,GAAG,UAAS,EAAE;AAAA,IAClD,CAAC,CAAA;AAED,IAAA,OAAO;AAAA,MACL,IAAA,EAAM;AAAA,QACJ,IAAA,EAAM;AAAA;AACR,KACF;AAAA,EACF;AAAA,EAEQ,wBAAA,CAAyB,UAAyB,KAAA,EAAiB;AACzE,IAAA,MAAM,aAAa,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,GAAI,MAAM,GAAA,CAAI,CAAA,IAAA,KAAQ,IAAA,CAAK,aAAA,CAAc,IAAI,CAAC,CAAA,GAAI,CAAC,IAAA,CAAK,aAAA,CAAc,KAAK,CAAC,CAAA;AAClH,IAAA,QAAQ,QAAA;AAAU,MAChB,KAAK,MAAA;AAEH,QAAA,IAAI,MAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,KAAA,CAAM,WAAW,CAAA,EAAG;AAC9C,UAAA,OAAO,EAAE,SAAA,EAAW,EAAC,EAAE;AAAA,QACzB;AACA,QAAA,OAAO;AAAA,UACL,IAAA,EAAM;AAAA,YACJ,IAAA,EAAM;AAAA;AACR,SACF;AAAA,MACF,KAAK,KAAA;AAEH,QAAA,IAAI,MAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,KAAA,CAAM,WAAW,CAAA,EAAG;AAC9C,UAAA,OAAO;AAAA,YACL,IAAA,EAAM;AAAA,cACJ,UAAU,CAAC,EAAE,SAAA,EAAW,IAAI;AAAA;AAC9B,WACF;AAAA,QACF;AACA,QAAA,OAAO;AAAA,UACL,IAAA,EAAM;AAAA,YACJ,MAAA,EAAQ;AAAA;AACV,SACF;AAAA,MACF,KAAK,MAAA;AACH,QAAA,OAAO;AAAA,UACL,IAAA,EAAM;AAAA,YACJ,QAAA,EAAU;AAAA;AACZ,SACF;AAAA,MACF;AACE,QAAA,OAAO,KAAA;AAAA;AACX,EACF;AAAA,EAEQ,sBAAA,CAAuB,KAAA,EAAe,QAAA,EAAyB,KAAA,EAAiB;AAEtF,IAAA,IAAI,IAAA,CAAK,eAAA,CAAgB,QAAQ,CAAA,EAAG;AAClC,MAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,wBAAA,CAAyB,KAAK,CAAA;AAC3D,MAAA,MAAMA,iBAAAA,GAAmB,IAAA,CAAK,kBAAA,CAAmB,KAAA,EAAO,KAAK,CAAA;AAC7D,MAAA,QAAQ,QAAA;AAAU,QAChB,KAAK,KAAA;AACH,UAAA,OAAO,EAAE,IAAA,EAAM,EAAE,CAACA,iBAAgB,GAAG,iBAAgB,EAAE;AAAA,QACzD,KAAK,KAAA;AACH,UAAA,OAAO;AAAA,YACL,IAAA,EAAM;AAAA,cACJ,QAAA,EAAU,CAAC,EAAE,IAAA,EAAM,EAAE,CAACA,iBAAgB,GAAG,eAAA,EAAgB,EAAG;AAAA;AAC9D,WACF;AAAA,QACF;AACE,UAAA,OAAO,EAAE,IAAA,EAAM,EAAE,CAACA,iBAAgB,GAAG,iBAAgB,EAAE;AAAA;AAC3D,IACF;AAGA,IAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,EAAG;AACpC,MAAA,MAAM,eAAA,GAAkB,IAAA,CAAK,wBAAA,CAAyB,KAAK,CAAA;AAC3D,MAAA,MAAM,OAAA,GAAU,QAAA,CAAS,OAAA,CAAQ,GAAA,EAAK,EAAE,CAAA;AACxC,MAAA,OAAO,EAAE,KAAA,EAAO,EAAE,CAAC,KAAK,GAAG,EAAE,CAAC,OAAO,GAAG,eAAA,EAAgB,EAAE,EAAE;AAAA,IAC9D;AAGA,IAAA,IAAI,IAAA,CAAK,eAAA,CAAgB,QAAQ,CAAA,EAAG;AAClC,MAAA,IAAI,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AACzB,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,8BAAA,EAAiC,QAAQ,CAAA,wBAAA,CAA0B,CAAA;AAAA,MACrF;AACA,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,oBAAA,CAAqB,KAAK,CAAA;AACxD,MAAA,MAAMA,iBAAAA,GAAmB,IAAA,CAAK,kBAAA,CAAmB,KAAA,EAAO,KAAK,CAAA;AAC7D,MAAA,QAAQ,QAAA;AAAU,QAChB,KAAK,KAAA;AACH,UAAA,OAAO,EAAE,KAAA,EAAO,EAAE,CAACA,iBAAgB,GAAG,kBAAiB,EAAE;AAAA,QAC3D,KAAK,MAAA;AAEH,UAAA,IAAI,gBAAA,CAAiB,WAAW,CAAA,EAAG;AACjC,YAAA,OAAO,EAAE,SAAA,EAAW,EAAC,EAAE;AAAA,UACzB;AACA,UAAA,OAAO;AAAA,YACL,IAAA,EAAM;AAAA,cACJ,QAAA,EAAU,CAAC,EAAE,KAAA,EAAO,EAAE,CAACA,iBAAgB,GAAG,gBAAA,EAAiB,EAAG;AAAA;AAChE,WACF;AAAA,QACF,KAAK,MAAA;AAEH,UAAA,IAAI,gBAAA,CAAiB,WAAW,CAAA,EAAG;AACjC,YAAA,OAAO;AAAA,cACL,IAAA,EAAM;AAAA,gBACJ,UAAU,CAAC,EAAE,SAAA,EAAW,IAAI;AAAA;AAC9B,aACF;AAAA,UACF;AACA,UAAA,OAAO;AAAA,YACL,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,gBAAA,CAAiB,GAAA,CAAI,CAAA,CAAA,MAAM,EAAE,IAAA,EAAM,EAAE,CAACA,iBAAgB,GAAG,CAAA,EAAE,EAAE,CAAE;AAAA;AACvE,WACF;AAAA,QACF;AACE,UAAA,OAAO,EAAE,KAAA,EAAO,EAAE,CAACA,iBAAgB,GAAG,kBAAiB,EAAE;AAAA;AAC7D,IACF;AAGA,IAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,EAAG;AACpC,MAAA,QAAQ,QAAA;AAAU,QAChB,KAAK,SAAA;AACH,UAAA,OAAO,QAAQ,EAAE,MAAA,EAAQ,EAAE,KAAA,EAAM,KAAM,EAAE,IAAA,EAAM,EAAE,QAAA,EAAU,CAAC,EAAE,MAAA,EAAQ,EAAE,OAAM,EAAG,GAAE,EAAE;AAAA,QACvF;AACE,UAAA,OAAO,EAAE,MAAA,EAAQ,EAAE,KAAA,EAAM,EAAE;AAAA;AAC/B,IACF;AAGA,IAAA,IAAI,IAAA,CAAK,eAAA,CAAgB,QAAQ,CAAA,EAAG;AAClC,MAAA,OAAO,IAAA,CAAK,sBAAA,CAAuB,KAAA,EAAO,KAAK,CAAA;AAAA,IACjD;AAEA,IAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,kBAAA,CAAmB,KAAA,EAAO,KAAK,CAAA;AAC7D,IAAA,OAAO,EAAE,IAAA,EAAM,EAAE,CAAC,gBAAgB,GAAG,OAAM,EAAE;AAAA,EAC/C;AAAA;AAAA;AAAA;AAAA,EAKQ,sBAAA,CAAuB,OAAe,KAAA,EAAiB;AAE7D,IAAA,MAAM,aAAa,OAAO,KAAA,KAAU,QAAA,GAAW,KAAA,GAAQ,MAAM,QAAA,EAAS;AAGtE,IAAA,IAAI,WAAW,QAAA,CAAS,IAAI,KAAK,UAAA,CAAW,QAAA,CAAS,IAAI,CAAA,EAAG;AAG1D,MAAA,OAAO,EAAE,KAAA,EAAO,EAAE,CAAC,KAAK,GAAG,OAAM,EAAE;AAAA,IACrC;AAGA,IAAA,IAAI,cAAA,GAAiB,UAAA;AACrB,IAAA,MAAM,cAAA,GAAiB,UAAA,CAAW,UAAA,CAAW,GAAG,CAAA;AAChD,IAAA,MAAM,YAAA,GAAe,UAAA,CAAW,QAAA,CAAS,GAAG,CAAA;AAG5C,IAAA,IAAI,kBAAkB,YAAA,EAAc;AAElC,MAAA,IAAI,cAAA,EAAgB;AAClB,QAAA,cAAA,GAAiB,cAAA,CAAe,UAAU,CAAC,CAAA;AAAA,MAC7C;AACA,MAAA,IAAI,YAAA,EAAc;AAChB,QAAA,cAAA,GAAiB,cAAA,CAAe,SAAA,CAAU,CAAA,EAAG,cAAA,CAAe,SAAS,CAAC,CAAA;AAAA,MACxE;AAGA,MAAA,IAAI,eAAA,GAAkB,cAAA;AACtB,MAAA,IAAI,CAAC,cAAA,EAAgB;AACnB,QAAA,eAAA,GAAkB,GAAA,GAAM,eAAA;AAAA,MAC1B;AACA,MAAA,IAAI,CAAC,YAAA,EAAc;AACjB,QAAA,eAAA,GAAkB,eAAA,GAAkB,GAAA;AAAA,MACtC;AAEA,MAAA,OAAO,EAAE,QAAA,EAAU,EAAE,CAAC,KAAK,GAAG,iBAAgB,EAAE;AAAA,IAClD;AAIA,IAAA,MAAM,YAAA,GAAe,UAAA,CAAW,OAAA,CAAQ,KAAA,EAAO,MAAM,CAAA;AACrD,IAAA,OAAO,EAAE,MAAA,EAAQ,EAAE,CAAC,KAAK,GAAG,cAAa,EAAE;AAAA,EAC7C;AAAA,EAEQ,kBAAA,CAAmB,OAAe,KAAA,EAAoB;AAE5D,IAAA,IAAI,OAAO,UAAU,QAAA,EAAU;AAC7B,MAAA,OAAO,GAAG,KAAK,CAAA,QAAA,CAAA;AAAA,IACjB;AAEA,IAAA,IAAI,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,IAAK,KAAA,CAAM,MAAM,CAAA,IAAA,KAAQ,OAAO,IAAA,KAAS,QAAQ,CAAA,EAAG;AACzE,MAAA,OAAO,GAAG,KAAK,CAAA,QAAA,CAAA;AAAA,IACjB;AACA,IAAA,OAAO,KAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA,EAKQ,6BAAA,CAA8B,OAAY,KAAA,EAA2B;AAE3E,IAAA,IAAI,UAAU,IAAA,EAAM;AAClB,MAAA,OAAO,EAAE,MAAA,EAAQ,EAAE,KAAA,EAAM,EAAE;AAAA,IAC7B;AAEA,IAAA,IAAI,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,EAAM;AAE/C,MAAA,IAAI,KAAA,IAAS,KAAA,IAAS,KAAA,CAAM,GAAA,KAAQ,IAAA,EAAM;AACxC,QAAA,OAAO,EAAE,MAAA,EAAQ,EAAE,KAAA,EAAM,EAAE;AAAA,MAC7B;AAGA,MAAA,IAAI,KAAA,IAAS,KAAA,IAAS,KAAA,CAAM,GAAA,KAAQ,IAAA,EAAM;AACxC,QAAA,OAAO;AAAA,UACL,IAAA,EAAM;AAAA,YACJ,UAAU,CAAC,EAAE,QAAQ,EAAE,KAAA,IAAS;AAAA;AAClC,SACF;AAAA,MACF;AAAA,IACF;AAEA,IAAA,OAAO,IAAA;AAAA,EACT;AAAA,EAEQ,iBAAA,CAAkB,QAAA,EAAyB,KAAA,EAAY,KAAA,EAAqB;AAElF,IAAA,IAAI,CAAC,IAAA,CAAK,UAAA,CAAW,QAAQ,CAAA,EAAG;AAC9B,MAAA,MAAM,IAAI,KAAA,CAAM,CAAA,sBAAA,EAAyB,QAAQ,CAAA,CAAE,CAAA;AAAA,IACrD;AAGA,IAAA,IAAI,QAAA,KAAa,UAAU,KAAA,EAAO;AAChC,MAAA,MAAM,iBAAA,GAAoB,IAAA,CAAK,6BAAA,CAA8B,KAAA,EAAO,KAAK,CAAA;AACzE,MAAA,IAAI,iBAAA,EAAmB;AACrB,QAAA,OAAO,iBAAA;AAAA,MACT;AAAA,IACF;AAGA,IAAA,IAAI,IAAA,CAAK,iBAAA,CAAkB,QAAQ,CAAA,EAAG;AAEpC,MAAA,IAAI,QAAA,KAAa,MAAA,IAAU,KAAA,IAAS,OAAO,KAAA,KAAU,QAAA,IAAY,KAAA,KAAU,IAAA,IAAQ,CAAC,KAAA,CAAM,OAAA,CAAQ,KAAK,CAAA,EAAG;AACxG,QAAA,MAAM,OAAA,GAAU,MAAA,CAAO,OAAA,CAAQ,KAAK,CAAA;AAGpC,QAAA,IAAI,OAAA,CAAQ,SAAS,CAAA,EAAG;AAEtB,UAAA,IAAI,OAAA,CAAQ,KAAA,CAAM,CAAC,CAAC,EAAE,MAAM,IAAA,CAAK,UAAA,CAAW,EAAE,CAAC,CAAA,EAAG;AAChD,YAAA,MAAM,mBAAA,GAAsB,IAAA,CAAK,wBAAA,CAAyB,KAAA,EAAO,KAAK,CAAA;AACtE,YAAA,OAAO;AAAA,cACL,IAAA,EAAM;AAAA,gBACJ,QAAA,EAAU,CAAC,mBAAmB;AAAA;AAChC,aACF;AAAA,UACF;AAGA,UAAA,IAAI,OAAA,CAAQ,MAAA,KAAW,CAAA,IAAK,OAAA,CAAQ,CAAC,CAAA,IAAK,IAAA,CAAK,UAAA,CAAW,OAAA,CAAQ,CAAC,CAAA,CAAE,CAAC,CAAC,CAAA,EAAG;AACxE,YAAA,MAAM,CAAC,QAAA,EAAU,SAAS,CAAA,GAAI,QAAQ,CAAC,CAAA;AACvC,YAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,sBAAA,CAAuB,KAAA,EAAO,UAAU,SAAS,CAAA;AAC/E,YAAA,OAAO;AAAA,cACL,IAAA,EAAM;AAAA,gBACJ,QAAA,EAAU,CAAC,gBAAgB;AAAA;AAC7B,aACF;AAAA,UACF;AAAA,QACF;AAAA,MACF;AACA,MAAA,OAAO,IAAA,CAAK,wBAAA,CAAyB,QAAA,EAAU,KAAK,CAAA;AAAA,IACtD;AAGA,IAAA,IAAI,KAAA,EAAO;AACT,MAAA,OAAO,IAAA,CAAK,sBAAA,CAAuB,KAAA,EAAO,QAAA,EAAU,KAAK,CAAA;AAAA,IAC3D;AAIA,IAAA,OAAO,KAAA;AAAA,EACT;AAAA;AAAA;AAAA;AAAA;AAAA,EAMQ,wBAAA,CAAyB,OAAe,UAAA,EAAsC;AAEpF,IAAA,IAAI,IAAA,CAAK,uBAAA,CAAwB,UAAU,CAAA,EAAG;AAC5C,MAAA,OAAO,IAAA,CAAK,gBAAA,CAAiB,KAAA,EAAO,UAAU,CAAA;AAAA,IAChD;AAGA,IAAA,MAAM,kBAAyB,EAAC;AAChC,IAAA,MAAA,CAAO,OAAA,CAAQ,UAAU,CAAA,CAAE,OAAA,CAAQ,CAAC,CAAC,QAAA,EAAU,KAAK,CAAA,KAAM;AACxD,MAAA,IAAI,IAAA,CAAK,UAAA,CAAW,QAAQ,CAAA,EAAG;AAC7B,QAAA,eAAA,CAAgB,KAAK,IAAA,CAAK,iBAAA,CAAkB,QAAA,EAA2B,KAAA,EAAO,KAAK,CAAC,CAAA;AAAA,MACtF,CAAA,MAAO;AAEL,QAAA,MAAM,gBAAA,GAAmB,KAAK,kBAAA,CAAmB,CAAA,EAAG,KAAK,CAAA,CAAA,EAAI,QAAQ,IAAI,KAAK,CAAA;AAC9E,QAAA,eAAA,CAAgB,IAAA,CAAK,EAAE,IAAA,EAAM,EAAE,CAAC,gBAAgB,GAAG,KAAA,EAAM,EAAG,CAAA;AAAA,MAC9D;AAAA,IACF,CAAC,CAAA;AAGD,IAAA,IAAI,eAAA,CAAgB,WAAW,CAAA,EAAG;AAChC,MAAA,OAAO,gBAAgB,CAAC,CAAA;AAAA,IAC1B;AAGA,IAAA,OAAO;AAAA,MACL,IAAA,EAAM;AAAA,QACJ,IAAA,EAAM;AAAA;AACR,KACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA,EAKQ,wBAAwB,UAAA,EAA0C;AACxE,IAAA,OAAO,MAAA,CAAO,IAAA,CAAK,UAAU,CAAA,CAAE,MAAM,CAAA,EAAA,KAAM,IAAA,CAAK,iBAAA,CAAkB,EAAE,CAAC,CAAA,IAAK,MAAA,CAAO,IAAA,CAAK,UAAU,EAAE,MAAA,GAAS,CAAA;AAAA,EAC7G;AAAA;AAAA;AAAA;AAAA,EAKQ,gBAAA,CAAiB,OAAe,UAAA,EAAsC;AAC5E,IAAA,MAAM,cAAc,MAAA,CAAO,WAAA;AAAA,MACzB,MAAA,CAAO,QAAQ,UAAU,CAAA,CAAE,IAAI,CAAC,CAAC,IAAI,GAAG,CAAA,KAAM,CAAC,EAAA,CAAG,OAAA,CAAQ,KAAK,EAAE,CAAA,EAAG,KAAK,wBAAA,CAAyB,GAAG,CAAC,CAAC;AAAA,KACzG;AAEA,IAAA,OAAO,EAAE,KAAA,EAAO,EAAE,CAAC,KAAK,GAAG,aAAY,EAAE;AAAA,EAC3C;AACF,CAAA;;;AC7cA,IAAM,cAAA,GAAiB;AAAA,EACrB,MAAA,EAAQ,aAAA;AAAA,EACR,SAAA,EAAW,IAAA;AAAA,EACX,UAAA,EAAY;AACd,CAAA;AAEA,IAAM,sBAAA,GAAyB;AAAA,EAC7B,WAAA,EAAa,QAAA;AAAA,EACb,EAAA,EAAI,WAAA;AAAA,EACJ,YAAA,EAAc;AAChB,CAAA;AAIO,IAAM,gBAAA,GAAN,cAA+BC,mBAAA,CAAqC;AAAA,EACjE,MAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOR,WAAA,CAAY,EAAE,GAAA,EAAI,EAAoB;AACpC,IAAA,KAAA,EAAM;AACN,IAAA,IAAA,CAAK,SAAS,IAAIC,iBAAA,CAAiB,EAAE,IAAA,EAAM,KAAK,CAAA;AAAA,EAClD;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAUA,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,IAAIC,iBAAA,CAAY;AAAA,QACpB,EAAA,EAAI,qDAAA;AAAA,QACJ,QAAQC,iBAAA,CAAY,OAAA;AAAA,QACpB,UAAUC,mBAAA,CAAc,IAAA;AAAA,QACxB,IAAA,EAAM,sCAAA;AAAA,QACN,OAAA,EAAS,EAAE,SAAA,EAAW,SAAA;AAAU,OACjC,CAAA;AAAA,IACH;AAEA,IAAA,IAAI;AACF,MAAA,MAAM,IAAA,CAAK,MAAA,CAAO,OAAA,CAAQ,MAAA,CAAO;AAAA,QAC/B,KAAA,EAAO,SAAA;AAAA,QACP,IAAA,EAAM;AAAA,UACJ,UAAU,EAAE,KAAA,EAAO,EAAE,GAAA,EAAK,MAAK,EAAE;AAAA,UACjC,QAAA,EAAU;AAAA,YACR,UAAA,EAAY;AAAA,cACV,QAAA,EAAU,EAAE,IAAA,EAAM,QAAA,EAAS;AAAA,cAC3B,EAAA,EAAI,EAAE,IAAA,EAAM,SAAA,EAAU;AAAA,cACtB,SAAA,EAAW;AAAA,gBACT,IAAA,EAAM,YAAA;AAAA,gBACN,SAAA;AAAA,gBACA,MAAA,EAAQ;AAAA,kBACN,IAAA,EAAM,MAAA;AAAA,kBACN,UAAA,EAAY,eAAe,MAAM,CAAA;AAAA,kBACjC,MAAA,EAAQ,OAAA;AAAA,kBACR,UAAA,EAAY,EAAE,eAAA,EAAiB,GAAA,EAAK,GAAG,EAAA;AAAG;AAC5C;AACF;AACF;AACF;AACF,OACD,CAAA;AAAA,IACH,SAASC,OAAA,EAAY;AACnB,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,+CAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,SAAA,EAAW,MAAA;AAAO,SAC1C;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAOA,MAAM,WAAA,GAAiC;AACrC,IAAA,IAAI;AACF,MAAA,MAAM,QAAA,GAAW,MAAM,IAAA,CAAK,MAAA,CAAO,IAAI,OAAA,CAAQ,EAAE,MAAA,EAAQ,MAAA,EAAQ,CAAA;AACjE,MAAA,MAAM,OAAA,GAAU,QAAA,CAAS,IAAA,CACtB,GAAA,CAAI,CAAC,MAAA,KAA+B,MAAA,CAAO,KAAK,CAAA,CAChD,MAAA,CAAO,CAAC,KAAA,KAA8B,UAAU,MAAS,CAAA;AAE5D,MAAA,OAAO,OAAA;AAAA,IACT,SAASA,OAAA,EAAO;AACd,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,+CAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;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,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,MAAM,IAAA,CAAK,MAAA,CAAO,OAAA,CAAQ,GAAA,CAAI,EAAE,KAAA,EAAO,SAAA,EAAW,CAAA;AAC9E,IAAA,MAAM,QAAA,GAAW,SAAA,CAAU,SAAS,CAAA,EAAG,QAAA;AACvC,IAAA,MAAM,SAAA,GAAiB,UAAU,UAAA,EAAY,SAAA;AAC7C,IAAA,MAAM,SAAA,GAAY,UAAU,MAAA,CAAO,UAAA;AAEnC,IAAA,MAAM,EAAE,IAAA,EAAM,SAAA,EAAU,GAAI,MAAM,IAAA,CAAK,MAAA,CAAO,KAAA,CAAM,EAAE,KAAA,EAAO,SAAA,EAAW,CAAA;AAExE,IAAA,OAAO;AAAA,MACL,SAAA,EAAW,MAAA,CAAO,SAAA,CAAU,SAAS,CAAA;AAAA,MACrC,KAAA,EAAO,MAAA,CAAO,SAAA,CAAU,KAAK,CAAA;AAAA,MAC7B,MAAA,EAAQ,uBAAuB,SAAS;AAAA,KAC1C;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQA,MAAM,WAAA,CAAY,EAAE,SAAA,EAAU,EAAqC;AACjE,IAAA,IAAI;AACF,MAAA,MAAM,KAAK,MAAA,CAAO,OAAA,CAAQ,OAAO,EAAE,KAAA,EAAO,WAAW,CAAA;AAAA,IACvD,SAASA,OAAA,EAAO;AACd,MAAA,MAAM,cAAc,IAAIH,iBAAA;AAAA,QACtB;AAAA,UACE,EAAA,EAAI,+CAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA;AAAU,SACvB;AAAA,QACAC;AAAA,OACF;AACA,MAAA,IAAA,CAAK,MAAA,EAAQ,KAAA,CAAM,WAAA,CAAY,QAAA,EAAU,CAAA;AACzC,MAAA,IAAA,CAAK,MAAA,EAAQ,eAAe,WAAW,CAAA;AAAA,IACzC;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAWA,MAAM,OAAO,EAAE,SAAA,EAAW,SAAS,QAAA,GAAW,EAAC,EAAG,GAAA,EAAI,EAA0C;AAC9F,IAAA,MAAM,YAAY,GAAA,IAAO,OAAA,CAAQ,IAAI,MAAM,MAAA,CAAO,YAAY,CAAA;AAC9D,IAAA,MAAM,aAAa,EAAC;AAEpB,IAAA,IAAI;AAEF,MAAA,MAAM,YAAY,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAGxD,MAAA,IAAA,CAAK,wBAAA,CAAyB,OAAA,EAAS,SAAA,CAAU,SAAS,CAAA;AAE1D,MAAA,KAAA,IAAS,CAAA,GAAI,CAAA,EAAG,CAAA,GAAI,OAAA,CAAQ,QAAQ,CAAA,EAAA,EAAK;AACvC,QAAA,MAAM,SAAA,GAAY;AAAA,UAChB,KAAA,EAAO;AAAA,YACL,MAAA,EAAQ,SAAA;AAAA,YACR,GAAA,EAAK,UAAU,CAAC;AAAA;AAClB,SACF;AAEA,QAAA,MAAM,QAAA,GAAW;AAAA,UACf,EAAA,EAAI,UAAU,CAAC,CAAA;AAAA,UACf,SAAA,EAAW,QAAQ,CAAC,CAAA;AAAA,UACpB,QAAA,EAAU,QAAA,CAAS,CAAC,CAAA,IAAK;AAAC,SAC5B;AAEA,QAAA,UAAA,CAAW,KAAK,SAAS,CAAA;AACzB,QAAA,UAAA,CAAW,KAAK,QAAQ,CAAA;AAAA,MAC1B;AAEA,MAAA,IAAI,UAAA,CAAW,SAAS,CAAA,EAAG;AACzB,QAAA,MAAM,IAAA,CAAK,OAAO,IAAA,CAAK,EAAE,MAAM,UAAA,EAAY,OAAA,EAAS,MAAM,CAAA;AAAA,MAC5D;AAEA,MAAA,OAAO,SAAA;AAAA,IACT,SAASA,OAAA,EAAO;AACd,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,yCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,SAAS,EAAE,SAAA,EAAW,WAAA,EAAa,OAAA,EAAS,UAAU,CAAA;AAAE,SAC1D;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAYA,MAAM,KAAA,CAAM;AAAA,IACV,SAAA;AAAA,IACA,WAAA;AAAA,IACA,MAAA;AAAA,IACA,IAAA,GAAO,EAAA;AAAA,IACP,aAAA,GAAgB;AAAA,GAClB,EAAmD;AACjD,IAAA,IAAI;AACF,MAAA,MAAM,gBAAA,GAAmB,IAAA,CAAK,eAAA,CAAgB,MAAM,CAAA;AAEpD,MAAA,MAAM,QAAA,GAAW,MAAM,IAAA,CAAK,MAAA,CAAO,MAAA,CAAO;AAAA,QACxC,KAAA,EAAO,SAAA;AAAA,QACP,IAAA,EAAM;AAAA,UACJ,KAAA,EAAO;AAAA,YACL,IAAA,EAAM;AAAA,cACJ,IAAA,EAAM,EAAE,GAAA,EAAK,EAAE,SAAA,EAAW,EAAE,MAAA,EAAQ,WAAA,EAAa,CAAA,EAAG,IAAA,EAAK,EAAE,EAAE;AAAA,cAC7D,MAAA,EAAQ,gBAAA,GAAmB,CAAC,gBAAgB,IAAI;AAAC;AACnD,WACF;AAAA,UACA,OAAA,EAAS,CAAC,IAAA,EAAM,UAAA,EAAY,WAAW;AAAA;AACzC,OACD,CAAA;AAED,MAAA,MAAM,UAAU,QAAA,CAAS,IAAA,CAAK,KAAK,IAAA,CAAK,GAAA,CAAI,CAAC,GAAA,KAAa;AACxD,QAAA,MAAM,MAAA,GAAS,GAAA,CAAI,OAAA,IAAW,EAAC;AAC/B,QAAA,OAAO;AAAA,UACL,EAAA,EAAI,MAAA,CAAO,MAAA,CAAO,EAAA,IAAM,EAAE,CAAA;AAAA,UAC1B,OAAO,OAAO,GAAA,CAAI,MAAA,KAAW,QAAA,GAAW,IAAI,MAAA,GAAS,CAAA;AAAA,UACrD,QAAA,EAAU,MAAA,CAAO,QAAA,IAAY,EAAC;AAAA,UAC9B,GAAI,aAAA,IAAiB,EAAE,MAAA,EAAQ,OAAO,SAAA;AAAsB,SAC9D;AAAA,MACF,CAAC,CAAA;AAED,MAAA,OAAO,OAAA;AAAA,IACT,SAASA,OAAA,EAAO;AACd,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,wCAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,WAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,IAAA;AAAK,SAC7B;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASQ,wBAAA,CAAyB,SAAqB,SAAA,EAAmB;AACvE,IAAA,IAAI,QAAQ,IAAA,CAAK,CAAA,MAAA,KAAU,MAAA,CAAO,MAAA,KAAW,SAAS,CAAA,EAAG;AACvD,MAAA,MAAM,IAAI,MAAM,iDAAiD,CAAA;AAAA,IACnE;AAAA,EACF;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EAQQ,gBAAgB,MAAA,EAAsC;AAC5D,IAAA,MAAM,UAAA,GAAa,IAAI,0BAAA,EAA2B;AAClD,IAAA,OAAO,UAAA,CAAW,UAAU,MAAM,CAAA;AAAA,EACpC;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,WAAA;AACJ,IAAA,IAAI;AACF,MAAA,IAAI,CAAC,MAAA,CAAO,MAAA,IAAU,CAAC,OAAO,QAAA,EAAU;AACtC,QAAA,MAAM,IAAI,MAAM,qBAAqB,CAAA;AAAA,MACvC;AAGA,MAAA,MAAM,EAAE,IAAA,EAAK,GAAI,MAAM,IAAA,CAAK,OAEzB,GAAA,CAAI;AAAA,QACH,KAAA,EAAO,SAAA;AAAA,QACP;AAAA,OACD,CAAA,CACA,KAAA,CAAM,MAAM;AACX,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,iBAAA,EAAoB,EAAE,CAAA,oBAAA,EAAuB,SAAS,CAAA,CAAE,CAAA;AAAA,MAC1E,CAAC,CAAA;AAEH,MAAA,IAAI,CAAC,IAAA,IAAQ,CAAC,IAAA,CAAK,OAAA,EAAS;AAC1B,QAAA,MAAM,IAAI,KAAA,CAAM,CAAA,iBAAA,EAAoB,EAAE,CAAA,6BAAA,EAAgC,SAAS,CAAA,CAAE,CAAA;AAAA,MACnF;AACA,MAAA,WAAA,GAAc,IAAA;AAAA,IAChB,SAASA,OAAA,EAAO;AACd,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,gDAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;AAAA,UACpB,UAAUC,mBAAA,CAAc,IAAA;AAAA,UACxB,OAAA,EAAS,EAAE,SAAA,EAAW,EAAA;AAAG,SAC3B;AAAA,QACAC;AAAA,OACF;AAAA,IACF;AAEA,IAAA,MAAM,SAAS,WAAA,CAAY,OAAA;AAC3B,IAAA,MAAM,UAAA,GAAkC;AAAA,MACtC,EAAA,EAAI,QAAQ,EAAA,IAAM;AAAA,KACpB;AAEA,IAAA,IAAI;AAEF,MAAA,IAAI,OAAO,MAAA,EAAQ;AAEjB,QAAA,MAAM,YAAY,MAAM,IAAA,CAAK,aAAA,CAAc,EAAE,WAAW,CAAA;AAGxD,QAAA,IAAA,CAAK,yBAAyB,CAAC,MAAA,CAAO,MAAM,CAAA,EAAG,UAAU,SAAS,CAAA;AAElE,QAAA,UAAA,CAAW,YAAY,MAAA,CAAO,MAAA;AAAA,MAChC,CAAA,MAAA,IAAW,QAAQ,SAAA,EAAW;AAC5B,QAAA,UAAA,CAAW,YAAY,MAAA,CAAO,SAAA;AAAA,MAChC;AAGA,MAAA,IAAI,OAAO,QAAA,EAAU;AACnB,QAAA,UAAA,CAAW,WAAW,MAAA,CAAO,QAAA;AAAA,MAC/B,CAAA,MAAO;AACL,QAAA,UAAA,CAAW,QAAA,GAAW,MAAA,EAAQ,QAAA,IAAY,EAAC;AAAA,MAC7C;AAGA,MAAA,MAAM,IAAA,CAAK,OAAO,KAAA,CAAM;AAAA,QACtB,KAAA,EAAO,SAAA;AAAA,QACP,EAAA;AAAA,QACA,IAAA,EAAM,UAAA;AAAA,QACN,OAAA,EAAS;AAAA,OACV,CAAA;AAAA,IACH,SAASA,OAAA,EAAO;AACd,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,gDAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;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;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,EASA,MAAM,YAAA,CAAa,EAAE,SAAA,EAAW,IAAG,EAAsC;AACvE,IAAA,IAAI;AACF,MAAA,MAAM,IAAA,CAAK,OAAO,MAAA,CAAO;AAAA,QACvB,KAAA,EAAO,SAAA;AAAA,QACP,EAAA;AAAA,QACA,OAAA,EAAS;AAAA,OACV,CAAA;AAAA,IACH,SAASA,OAAA,EAAgB;AAEvB,MAAA,IAAIA,OAAA,IAAS,OAAOA,OAAA,KAAU,QAAA,IAAY,gBAAgBA,OAAA,IAASA,OAAA,CAAM,eAAe,GAAA,EAAK;AAC3F,QAAA;AAAA,MACF;AACA,MAAA,MAAM,IAAIH,iBAAA;AAAA,QACR;AAAA,UACE,EAAA,EAAI,gDAAA;AAAA,UACJ,QAAQC,iBAAA,CAAY,OAAA;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","file":"index.cjs","sourcesContent":["import type {\n BlacklistedRootOperators,\n LogicalOperatorValueMap,\n OperatorSupport,\n OperatorValueMap,\n QueryOperator,\n VectorFilter,\n} from '@mastra/core/vector/filter';\nimport { BaseFilterTranslator } from '@mastra/core/vector/filter';\n\ntype OpenSearchOperatorValueMap = Omit<OperatorValueMap, '$options' | '$nor' | '$elemMatch'>;\n\ntype OpenSearchLogicalOperatorValueMap = Omit<LogicalOperatorValueMap, '$nor'>;\n\ntype OpenSearchBlacklisted = BlacklistedRootOperators | '$nor';\n\nexport type OpenSearchVectorFilter = VectorFilter<\n keyof OpenSearchOperatorValueMap,\n OpenSearchOperatorValueMap,\n OpenSearchLogicalOperatorValueMap,\n OpenSearchBlacklisted\n>;\n/**\n * Translator for OpenSearch filter queries.\n * Maintains OpenSearch-compatible syntax while ensuring proper validation\n * and normalization of values.\n */\nexport class OpenSearchFilterTranslator extends BaseFilterTranslator<OpenSearchVectorFilter> {\n protected override getSupportedOperators(): OperatorSupport {\n return {\n ...BaseFilterTranslator.DEFAULT_OPERATORS,\n logical: ['$and', '$or', '$not'],\n array: ['$in', '$nin', '$all'],\n regex: ['$regex'],\n custom: [],\n };\n }\n\n translate(filter?: OpenSearchVectorFilter): OpenSearchVectorFilter {\n if (this.isEmpty(filter)) return undefined;\n this.validateFilter(filter);\n return this.translateNode(filter);\n }\n\n private translateNode(node: OpenSearchVectorFilter): any {\n // Handle primitive values and arrays\n if (this.isPrimitive(node) || Array.isArray(node)) {\n return node;\n }\n\n const entries = Object.entries(node as Record<string, any>);\n\n // Extract logical operators and field conditions\n const logicalOperators: [string, any][] = [];\n const fieldConditions: [string, any][] = [];\n\n entries.forEach(([key, value]) => {\n if (this.isLogicalOperator(key)) {\n logicalOperators.push([key, value]);\n } else {\n fieldConditions.push([key, value]);\n }\n });\n\n // If we have a single logical operator\n if (logicalOperators.length === 1 && fieldConditions.length === 0) {\n const [operator, value] = logicalOperators[0] as [QueryOperator, any];\n if (!Array.isArray(value) && typeof value !== 'object') {\n throw new Error(`Invalid logical operator structure: ${operator} must have an array or object value`);\n }\n return this.translateLogicalOperator(operator, value);\n }\n\n // Process field conditions\n const fieldConditionQueries = fieldConditions.map(([key, value]) => {\n // Handle nested objects\n if (typeof value === 'object' && value !== null && !Array.isArray(value)) {\n // Check if the object contains operators\n const hasOperators = Object.keys(value).some(k => this.isOperator(k));\n\n // Use a more direct approach based on whether operators are present\n const nestedField = `metadata.${key}`;\n return hasOperators\n ? this.translateFieldConditions(nestedField, value)\n : this.translateNestedObject(nestedField, value);\n }\n\n // Handle arrays\n if (Array.isArray(value)) {\n const fieldWithKeyword = this.addKeywordIfNeeded(`metadata.${key}`, value);\n return { terms: { [fieldWithKeyword]: value } };\n }\n\n // Handle simple field equality\n const fieldWithKeyword = this.addKeywordIfNeeded(`metadata.${key}`, value);\n return { term: { [fieldWithKeyword]: value } };\n });\n\n // Handle case with both logical operators and field conditions or multiple logical operators\n if (logicalOperators.length > 0) {\n const logicalConditions = logicalOperators.map(([operator, value]) =>\n this.translateOperator(operator as QueryOperator, value),\n );\n\n return {\n bool: {\n must: [...logicalConditions, ...fieldConditionQueries],\n },\n };\n }\n\n // If we only have field conditions\n if (fieldConditionQueries.length > 1) {\n return {\n bool: {\n must: fieldConditionQueries,\n },\n };\n }\n\n // If we have only one field condition\n if (fieldConditionQueries.length === 1) {\n return fieldConditionQueries[0];\n }\n\n // If we have no conditions (e.g., only empty $and arrays)\n return { match_all: {} };\n }\n\n /**\n * Handles translation of nested objects with dot notation fields\n */\n private translateNestedObject(field: string, value: Record<string, any>): any {\n const conditions = Object.entries(value).map(([subField, subValue]) => {\n const fullField = `${field}.${subField}`;\n\n // Check if this is an operator in a nested field\n if (this.isOperator(subField)) {\n return this.translateOperator(subField as QueryOperator, subValue, field);\n }\n\n if (typeof subValue === 'object' && subValue !== null && !Array.isArray(subValue)) {\n // Check if the nested object contains operators\n const hasOperators = Object.keys(subValue).some(k => this.isOperator(k));\n if (hasOperators) {\n return this.translateFieldConditions(fullField, subValue);\n }\n return this.translateNestedObject(fullField, subValue);\n }\n const fieldWithKeyword = this.addKeywordIfNeeded(fullField, subValue);\n return { term: { [fieldWithKeyword]: subValue } };\n });\n\n return {\n bool: {\n must: conditions,\n },\n };\n }\n\n private translateLogicalOperator(operator: QueryOperator, value: any): any {\n const conditions = Array.isArray(value) ? value.map(item => this.translateNode(item)) : [this.translateNode(value)];\n switch (operator) {\n case '$and':\n // For empty $and, return a query that matches everything\n if (Array.isArray(value) && value.length === 0) {\n return { match_all: {} };\n }\n return {\n bool: {\n must: conditions,\n },\n };\n case '$or':\n // For empty $or, return a query that matches nothing\n if (Array.isArray(value) && value.length === 0) {\n return {\n bool: {\n must_not: [{ match_all: {} }],\n },\n };\n }\n return {\n bool: {\n should: conditions,\n },\n };\n case '$not':\n return {\n bool: {\n must_not: conditions,\n },\n };\n default:\n return value;\n }\n }\n\n private translateFieldOperator(field: string, operator: QueryOperator, value: any): any {\n // Handle basic comparison operators\n if (this.isBasicOperator(operator)) {\n const normalizedValue = this.normalizeComparisonValue(value);\n const fieldWithKeyword = this.addKeywordIfNeeded(field, value);\n switch (operator) {\n case '$eq':\n return { term: { [fieldWithKeyword]: normalizedValue } };\n case '$ne':\n return {\n bool: {\n must_not: [{ term: { [fieldWithKeyword]: normalizedValue } }],\n },\n };\n default:\n return { term: { [fieldWithKeyword]: normalizedValue } };\n }\n }\n\n // Handle numeric operators\n if (this.isNumericOperator(operator)) {\n const normalizedValue = this.normalizeComparisonValue(value);\n const rangeOp = operator.replace('$', '');\n return { range: { [field]: { [rangeOp]: normalizedValue } } };\n }\n\n // Handle array operators\n if (this.isArrayOperator(operator)) {\n if (!Array.isArray(value)) {\n throw new Error(`Invalid array operator value: ${operator} requires an array value`);\n }\n const normalizedValues = this.normalizeArrayValues(value);\n const fieldWithKeyword = this.addKeywordIfNeeded(field, value);\n switch (operator) {\n case '$in':\n return { terms: { [fieldWithKeyword]: normalizedValues } };\n case '$nin':\n // For empty arrays, return a query that matches everything\n if (normalizedValues.length === 0) {\n return { match_all: {} };\n }\n return {\n bool: {\n must_not: [{ terms: { [fieldWithKeyword]: normalizedValues } }],\n },\n };\n case '$all':\n // For empty arrays, return a query that will match nothing\n if (normalizedValues.length === 0) {\n return {\n bool: {\n must_not: [{ match_all: {} }],\n },\n };\n }\n return {\n bool: {\n must: normalizedValues.map(v => ({ term: { [fieldWithKeyword]: v } })),\n },\n };\n default:\n return { terms: { [fieldWithKeyword]: normalizedValues } };\n }\n }\n\n // Handle element operators\n if (this.isElementOperator(operator)) {\n switch (operator) {\n case '$exists':\n return value ? { exists: { field } } : { bool: { must_not: [{ exists: { field } }] } };\n default:\n return { exists: { field } };\n }\n }\n\n // Handle regex operators\n if (this.isRegexOperator(operator)) {\n return this.translateRegexOperator(field, value);\n }\n\n const fieldWithKeyword = this.addKeywordIfNeeded(field, value);\n return { term: { [fieldWithKeyword]: value } };\n }\n\n /**\n * Translates regex patterns to OpenSearch query syntax\n */\n private translateRegexOperator(field: string, value: any): any {\n // Convert value to string if it's not already\n const regexValue = typeof value === 'string' ? value : value.toString();\n\n // Check for problematic patterns (like newlines, etc.)\n if (regexValue.includes('\\n') || regexValue.includes('\\r')) {\n // For patterns with newlines, use a simpler approach\n // OpenSearch doesn't support dotall flag like JavaScript\n return { match: { [field]: value } };\n }\n\n // Process regex pattern to handle anchors properly\n let processedRegex = regexValue;\n const hasStartAnchor = regexValue.startsWith('^');\n const hasEndAnchor = regexValue.endsWith('$');\n\n // If we have anchors, use wildcard query for better handling\n if (hasStartAnchor || hasEndAnchor) {\n // Remove anchors\n if (hasStartAnchor) {\n processedRegex = processedRegex.substring(1);\n }\n if (hasEndAnchor) {\n processedRegex = processedRegex.substring(0, processedRegex.length - 1);\n }\n\n // Create wildcard pattern\n let wildcardPattern = processedRegex;\n if (!hasStartAnchor) {\n wildcardPattern = '*' + wildcardPattern;\n }\n if (!hasEndAnchor) {\n wildcardPattern = wildcardPattern + '*';\n }\n\n return { wildcard: { [field]: wildcardPattern } };\n }\n\n // Use regexp for other regex patterns\n // Escape any backslashes to prevent OpenSearch from misinterpreting them\n const escapedRegex = regexValue.replace(/\\\\/g, '\\\\\\\\');\n return { regexp: { [field]: escapedRegex } };\n }\n\n private addKeywordIfNeeded(field: string, value: any): string {\n // Add .keyword suffix for string fields\n if (typeof value === 'string') {\n return `${field}.keyword`;\n }\n // Add .keyword suffix for string array fields\n if (Array.isArray(value) && value.every(item => typeof item === 'string')) {\n return `${field}.keyword`;\n }\n return field;\n }\n\n /**\n * Helper method to handle special cases for the $not operator\n */\n private handleNotOperatorSpecialCases(value: any, field: string): any | null {\n // For \"not null\", we need to use exists query\n if (value === null) {\n return { exists: { field } };\n }\n\n if (typeof value === 'object' && value !== null) {\n // For \"not {$eq: null}\", we need to use exists query\n if ('$eq' in value && value.$eq === null) {\n return { exists: { field } };\n }\n\n // For \"not {$ne: null}\", we need to use must_not exists query\n if ('$ne' in value && value.$ne === null) {\n return {\n bool: {\n must_not: [{ exists: { field } }],\n },\n };\n }\n }\n\n return null; // No special case applies\n }\n\n private translateOperator(operator: QueryOperator, value: any, field?: string): any {\n // Check if this is a valid operator\n if (!this.isOperator(operator)) {\n throw new Error(`Unsupported operator: ${operator}`);\n }\n\n // Special case for $not with null or $eq: null\n if (operator === '$not' && field) {\n const specialCaseResult = this.handleNotOperatorSpecialCases(value, field);\n if (specialCaseResult) {\n return specialCaseResult;\n }\n }\n\n // Handle logical operators\n if (this.isLogicalOperator(operator)) {\n // For $not operator with field context and nested operators, handle specially\n if (operator === '$not' && field && typeof value === 'object' && value !== null && !Array.isArray(value)) {\n const entries = Object.entries(value);\n\n // Handle multiple operators in $not\n if (entries.length > 0) {\n // If all entries are operators, handle them as a single condition\n if (entries.every(([op]) => this.isOperator(op))) {\n const translatedCondition = this.translateFieldConditions(field, value);\n return {\n bool: {\n must_not: [translatedCondition],\n },\n };\n }\n\n // Handle single nested operator\n if (entries.length === 1 && entries[0] && this.isOperator(entries[0][0])) {\n const [nestedOp, nestedVal] = entries[0] as [QueryOperator, any];\n const translatedNested = this.translateFieldOperator(field, nestedOp, nestedVal);\n return {\n bool: {\n must_not: [translatedNested],\n },\n };\n }\n }\n }\n return this.translateLogicalOperator(operator, value);\n }\n\n // If a field is provided, use translateFieldOperator for more specific translation\n if (field) {\n return this.translateFieldOperator(field, operator, value);\n }\n\n // For non-logical operators without a field context, just return the value\n // The actual translation happens in translateFieldConditions where we have the field context\n return value;\n }\n\n /**\n * Translates field conditions to OpenSearch query syntax\n * Handles special cases like range queries and multiple operators\n */\n private translateFieldConditions(field: string, conditions: Record<string, any>): any {\n // Special case: Optimize multiple numeric operators into a single range query\n if (this.canOptimizeToRangeQuery(conditions)) {\n return this.createRangeQuery(field, conditions);\n }\n\n // Handle all other operators consistently\n const queryConditions: any[] = [];\n Object.entries(conditions).forEach(([operator, value]) => {\n if (this.isOperator(operator)) {\n queryConditions.push(this.translateOperator(operator as QueryOperator, value, field));\n } else {\n // Handle non-operator keys (should not happen in normal usage)\n const fieldWithKeyword = this.addKeywordIfNeeded(`${field}.${operator}`, value);\n queryConditions.push({ term: { [fieldWithKeyword]: value } });\n }\n });\n\n // Return single condition without wrapping\n if (queryConditions.length === 1) {\n return queryConditions[0];\n }\n\n // Combine multiple conditions with AND logic\n return {\n bool: {\n must: queryConditions,\n },\n };\n }\n\n /**\n * Checks if conditions can be optimized to a range query\n */\n private canOptimizeToRangeQuery(conditions: Record<string, any>): boolean {\n return Object.keys(conditions).every(op => this.isNumericOperator(op)) && Object.keys(conditions).length > 0;\n }\n\n /**\n * Creates a range query from numeric operators\n */\n private createRangeQuery(field: string, conditions: Record<string, any>): any {\n const rangeParams = Object.fromEntries(\n Object.entries(conditions).map(([op, val]) => [op.replace('$', ''), this.normalizeComparisonValue(val)]),\n );\n\n return { range: { [field]: rangeParams } };\n }\n}\n","import { MastraError, ErrorDomain, ErrorCategory } from '@mastra/core/error';\nimport type {\n CreateIndexParams,\n DeleteIndexParams,\n DeleteVectorParams,\n DescribeIndexParams,\n IndexStats,\n QueryResult,\n QueryVectorParams,\n UpdateVectorParams,\n UpsertVectorParams,\n} from '@mastra/core/vector';\nimport { MastraVector } from '@mastra/core/vector';\nimport { Client as OpenSearchClient } from '@opensearch-project/opensearch';\nimport { OpenSearchFilterTranslator } from './filter';\nimport type { OpenSearchVectorFilter } from './filter';\n\nconst METRIC_MAPPING = {\n cosine: 'cosinesimil',\n euclidean: 'l2',\n dotproduct: 'innerproduct',\n} as const;\n\nconst REVERSE_METRIC_MAPPING = {\n cosinesimil: 'cosine',\n l2: 'euclidean',\n innerproduct: 'dotproduct',\n} as const;\n\ntype OpenSearchVectorParams = QueryVectorParams<OpenSearchVectorFilter>;\n\nexport class OpenSearchVector extends MastraVector<OpenSearchVectorFilter> {\n private client: OpenSearchClient;\n\n /**\n * Creates a new OpenSearchVector client.\n *\n * @param {string} url - The url of the OpenSearch node.\n */\n constructor({ url }: { url: string }) {\n super();\n this.client = new OpenSearchClient({ node: url });\n }\n\n /**\n * Creates a new collection with the specified configuration.\n *\n * @param {string} indexName - The name of the collection to create.\n * @param {number} dimension - The dimension of the vectors to be stored in the collection.\n * @param {'cosine' | 'euclidean' | 'dotproduct'} [metric=cosine] - The metric to use to sort vectors in the collection.\n * @returns {Promise<void>} A promise that resolves when the collection is created.\n */\n async createIndex({ indexName, dimension, metric = 'cosine' }: CreateIndexParams): Promise<void> {\n if (!Number.isInteger(dimension) || dimension <= 0) {\n throw new MastraError({\n id: 'STORAGE_OPENSEARCH_VECTOR_CREATE_INDEX_INVALID_ARGS',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.USER,\n text: 'Dimension must be a positive integer',\n details: { indexName, dimension },\n });\n }\n\n try {\n await this.client.indices.create({\n index: indexName,\n body: {\n settings: { index: { knn: true } },\n mappings: {\n properties: {\n metadata: { type: 'object' },\n id: { type: 'keyword' },\n embedding: {\n type: 'knn_vector',\n dimension: dimension,\n method: {\n name: 'hnsw',\n space_type: METRIC_MAPPING[metric],\n engine: 'faiss',\n parameters: { ef_construction: 128, m: 16 },\n },\n },\n },\n },\n },\n });\n } catch (error: any) {\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: 'STORAGE_OPENSEARCH_VECTOR_CREATE_INDEX_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, dimension, metric },\n },\n error,\n );\n }\n }\n\n /**\n * Lists all indexes.\n *\n * @returns {Promise<string[]>} A promise that resolves to an array of indexes.\n */\n async listIndexes(): Promise<string[]> {\n try {\n const response = await this.client.cat.indices({ format: 'json' });\n const indexes = response.body\n .map((record: { index?: string }) => record.index)\n .filter((index: string | undefined) => index !== undefined);\n\n return indexes;\n } catch (error) {\n throw new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_LIST_INDEXES_FAILED',\n domain: ErrorDomain.STORAGE,\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 const { body: indexInfo } = await this.client.indices.get({ index: indexName });\n const mappings = indexInfo[indexName]?.mappings;\n const embedding: any = mappings?.properties?.embedding;\n const spaceType = embedding.method.space_type as keyof typeof REVERSE_METRIC_MAPPING;\n\n const { body: countInfo } = await this.client.count({ index: indexName });\n\n return {\n dimension: Number(embedding.dimension),\n count: Number(countInfo.count),\n metric: REVERSE_METRIC_MAPPING[spaceType],\n };\n }\n\n /**\n * Deletes the specified index.\n *\n * @param {string} indexName - The name of the index to delete.\n * @returns {Promise<void>} A promise that resolves when the index is deleted.\n */\n async deleteIndex({ indexName }: DeleteIndexParams): Promise<void> {\n try {\n await this.client.indices.delete({ index: indexName });\n } catch (error) {\n const mastraError = new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_DELETE_INDEX_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName },\n },\n error,\n );\n this.logger?.error(mastraError.toString());\n this.logger?.trackException(mastraError);\n }\n }\n\n /**\n * Inserts or updates vectors in the specified collection.\n *\n * @param {string} indexName - The name of the collection to upsert into.\n * @param {number[][]} vectors - An array of vectors to upsert.\n * @param {Record<string, any>[]} [metadata] - An optional array of metadata objects corresponding to each vector.\n * @param {string[]} [ids] - An optional array of IDs corresponding to each vector. If not provided, new IDs will be generated.\n * @returns {Promise<string[]>} A promise that resolves to an array of IDs of the upserted vectors.\n */\n async upsert({ indexName, vectors, metadata = [], ids }: UpsertVectorParams): Promise<string[]> {\n const vectorIds = ids || vectors.map(() => crypto.randomUUID());\n const operations = [];\n\n try {\n // Get index stats to check dimension\n const indexInfo = await this.describeIndex({ indexName });\n\n // Validate vector dimensions\n this.validateVectorDimensions(vectors, indexInfo.dimension);\n\n for (let i = 0; i < vectors.length; i++) {\n const operation = {\n index: {\n _index: indexName,\n _id: vectorIds[i],\n },\n };\n\n const document = {\n id: vectorIds[i],\n embedding: vectors[i],\n metadata: metadata[i] || {},\n };\n\n operations.push(operation);\n operations.push(document);\n }\n\n if (operations.length > 0) {\n await this.client.bulk({ body: operations, refresh: true });\n }\n\n return vectorIds;\n } catch (error) {\n throw new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_UPSERT_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, vectorCount: vectors?.length || 0 },\n },\n error,\n );\n }\n }\n\n /**\n * Queries the specified collection using a vector and optional filter.\n *\n * @param {string} indexName - The name of the collection to query.\n * @param {number[]} queryVector - The vector to query with.\n * @param {number} [topK] - The maximum number of results to return.\n * @param {Record<string, any>} [filter] - An optional filter to apply to the query. For more on filters in OpenSearch, see the filtering reference: https://opensearch.org/docs/latest/query-dsl/\n * @param {boolean} [includeVectors=false] - Whether to include the vectors in the response.\n * @returns {Promise<QueryResult[]>} A promise that resolves to an array of query results.\n */\n async query({\n indexName,\n queryVector,\n filter,\n topK = 10,\n includeVector = false,\n }: OpenSearchVectorParams): Promise<QueryResult[]> {\n try {\n const translatedFilter = this.transformFilter(filter);\n\n const response = await this.client.search({\n index: indexName,\n body: {\n query: {\n bool: {\n must: { knn: { embedding: { vector: queryVector, k: topK } } },\n filter: translatedFilter ? [translatedFilter] : [],\n },\n },\n _source: ['id', 'metadata', 'embedding'],\n },\n });\n\n const results = response.body.hits.hits.map((hit: any) => {\n const source = hit._source || {};\n return {\n id: String(source.id || ''),\n score: typeof hit._score === 'number' ? hit._score : 0,\n metadata: source.metadata || {},\n ...(includeVector && { vector: source.embedding as number[] }),\n };\n });\n\n return results;\n } catch (error) {\n throw new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_QUERY_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, topK },\n },\n error,\n );\n }\n }\n\n /**\n * Validates the dimensions of the vectors.\n *\n * @param {number[][]} vectors - The vectors to validate.\n * @param {number} dimension - The dimension of the vectors.\n * @returns {void}\n */\n private validateVectorDimensions(vectors: number[][], dimension: number) {\n if (vectors.some(vector => vector.length !== dimension)) {\n throw new Error('Vector dimension does not match index dimension');\n }\n }\n\n /**\n * Transforms the filter to the OpenSearch DSL.\n *\n * @param {OpenSearchVectorFilter} filter - The filter to transform.\n * @returns {Record<string, any>} The transformed filter.\n */\n private transformFilter(filter?: OpenSearchVectorFilter): any {\n const translator = new OpenSearchFilterTranslator();\n return translator.translate(filter);\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 let existingDoc;\n try {\n if (!update.vector && !update.metadata) {\n throw new Error('No updates provided');\n }\n\n // First get the current document to merge with updates\n const { body } = await this.client\n\n .get({\n index: indexName,\n id: id,\n })\n .catch(() => {\n throw new Error(`Document with ID ${id} not found in index ${indexName}`);\n });\n\n if (!body || !body._source) {\n throw new Error(`Document with ID ${id} has no source data in index ${indexName}`);\n }\n existingDoc = body;\n } catch (error) {\n throw new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_UPDATE_VECTOR_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.USER,\n details: { indexName, id },\n },\n error,\n );\n }\n\n const source = existingDoc._source;\n const updatedDoc: Record<string, any> = {\n id: source?.id || id,\n };\n\n try {\n // Update vector if provided\n if (update.vector) {\n // Get index stats to check dimension\n const indexInfo = await this.describeIndex({ indexName });\n\n // Validate vector dimensions\n this.validateVectorDimensions([update.vector], indexInfo.dimension);\n\n updatedDoc.embedding = update.vector;\n } else if (source?.embedding) {\n updatedDoc.embedding = source.embedding;\n }\n\n // Update metadata if provided\n if (update.metadata) {\n updatedDoc.metadata = update.metadata;\n } else {\n updatedDoc.metadata = source?.metadata || {};\n }\n\n // Update the document\n await this.client.index({\n index: indexName,\n id: id,\n body: updatedDoc,\n refresh: true,\n });\n } catch (error) {\n throw new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_UPDATE_VECTOR_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, id },\n },\n error,\n );\n }\n }\n\n /**\n * Deletes a vector by its ID.\n * @param indexName - The name of the index containing the vector.\n * @param id - The ID of the vector to delete.\n * @returns A promise that resolves when the deletion is complete.\n * @throws Will throw an error if the deletion operation fails.\n */\n async deleteVector({ indexName, id }: DeleteVectorParams): Promise<void> {\n try {\n await this.client.delete({\n index: indexName,\n id: id,\n refresh: true,\n });\n } catch (error: unknown) {\n // Don't throw error if document doesn't exist (404)\n if (error && typeof error === 'object' && 'statusCode' in error && error.statusCode === 404) {\n return;\n }\n throw new MastraError(\n {\n id: 'STORAGE_OPENSEARCH_VECTOR_DELETE_VECTOR_FAILED',\n domain: ErrorDomain.STORAGE,\n category: ErrorCategory.THIRD_PARTY,\n details: { indexName, id },\n },\n error,\n );\n }\n }\n}\n"]}
|
package/dist/index.d.ts
CHANGED
|
@@ -1 +1,2 @@
|
|
|
1
|
-
export
|
|
1
|
+
export * from './vector/index.js';
|
|
2
|
+
//# sourceMappingURL=index.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,UAAU,CAAC"}
|