triangle-utils 1.4.90 → 1.4.92

package/dist/src/UtilsS3Vectors.d.ts

@@ -18,5 +18,6 @@ export declare class UtilsS3Vectors {
     query(s3_vector_id_prefix: string, coordinates: number[], num_vectors: number, filters?: Record<string, any> | undefined): Promise<{
         s3_vector_id: string;
         distance: number;
+        headers: Record<string, any>;
     }[]>;
 }

package/dist/src/UtilsS3Vectors.js

@@ -89,7 +89,7 @@ export class UtilsS3Vectors {
         const s3vectors_inputs = s3_vector_ids.map(get_s3vectors_input);
         const { s3_vector_bucket_name, s3_vector_index_name } = s3vectors_inputs[0];
         if (!s3vectors_inputs.every(s3vectors_input => s3vectors_input.s3_vector_bucket_name === s3_vector_bucket_name && s3vectors_input.s3_vector_index_name === s3_vector_index_name)) {
-            return [];
+            throw Error("Inconsistent prefixes.");
         }
         const s3_vectors = [];
         for (let i = 0; i < s3vectors_inputs.length; i += 100) {
@@ -143,7 +143,7 @@ export class UtilsS3Vectors {
         const s3vectors_inputs = s3_vectors.map(s3_vector => get_s3vectors_input(s3_vector.s3_vector_id));
         const { s3_vector_bucket_name, s3_vector_index_name } = s3vectors_inputs[0];
         if (!s3vectors_inputs.every(s3vectors_input => s3vectors_input.s3_vector_bucket_name === s3_vector_bucket_name && s3vectors_input.s3_vector_index_name === s3_vector_index_name)) {
-            return undefined;
+            throw Error("Inconsistent prefixes.");
         }
         for (let i = 0; i < s3_vectors.length; i += 500) {
             await this.s3vectors.putVectors({
@@ -171,7 +171,7 @@ export class UtilsS3Vectors {
         const s3vectors_inputs = s3_vector_ids.map(get_s3vectors_input);
         const { s3_vector_bucket_name, s3_vector_index_name } = s3vectors_inputs[0];
         if (!s3vectors_inputs.every(s3vectors_input => s3vectors_input.s3_vector_bucket_name === s3_vector_bucket_name && s3vectors_input.s3_vector_index_name === s3_vector_index_name)) {
-            return undefined;
+            throw Error("Inconsistent prefixes.");
         }
         for (let i = 0; i < s3vectors_inputs.length; i += 500) {
             await this.s3vectors.deleteVectors({
@@ -211,9 +211,15 @@ export class UtilsS3Vectors {
                 console.log("Bad distance:", distance);
                 continue;
             }
+            const headers = new_vector.metadata;
+            if (!(typeof headers === "object" && Object.keys(headers).every((header_id) => typeof header_id === "string"))) {
+                console.log("Bad headers:", headers);
+                continue;
+            }
             results.push({
                 s3_vector_id: s3_vector_id,
-                distance: distance
+                distance: distance,
+                headers: headers
             });
         }
         return results;

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "triangle-utils",
-    "version": "1.4.90",
+    "version": "1.4.92",
     "main": "dist/src/index.js",
     "types": "dist/src/index.d.ts",
     "directories": {

package/src/UtilsS3Vectors.ts

@@ -109,7 +109,7 @@ export class UtilsS3Vectors {
         const s3vectors_inputs : S3VectorsInput[] = s3_vector_ids.map(get_s3vectors_input)
         const { s3_vector_bucket_name, s3_vector_index_name } = s3vectors_inputs[0]
         if (!s3vectors_inputs.every(s3vectors_input => s3vectors_input.s3_vector_bucket_name === s3_vector_bucket_name && s3vectors_input.s3_vector_index_name === s3_vector_index_name)) {
-            return []
+            throw Error("Inconsistent prefixes.")
         }
         const s3_vectors : (S3Vector | undefined)[] = []
         for (let i = 0; i < s3vectors_inputs.length; i += 100) {
@@ -165,7 +165,7 @@ export class UtilsS3Vectors {
         const s3vectors_inputs : S3VectorsInput[] = s3_vectors.map(s3_vector => get_s3vectors_input(s3_vector.s3_vector_id))
         const { s3_vector_bucket_name, s3_vector_index_name } = s3vectors_inputs[0]
         if (!s3vectors_inputs.every(s3vectors_input => s3vectors_input.s3_vector_bucket_name === s3_vector_bucket_name && s3vectors_input.s3_vector_index_name === s3_vector_index_name)) {
-            return undefined
+            throw Error("Inconsistent prefixes.")
         }
         for (let i = 0; i < s3_vectors.length; i += 500) {
             await this.s3vectors.putVectors({
@@ -196,7 +196,7 @@ export class UtilsS3Vectors {
         const s3vectors_inputs : S3VectorsInput[] = s3_vector_ids.map(get_s3vectors_input)
         const { s3_vector_bucket_name, s3_vector_index_name } = s3vectors_inputs[0]
         if (!s3vectors_inputs.every(s3vectors_input => s3vectors_input.s3_vector_bucket_name === s3_vector_bucket_name && s3vectors_input.s3_vector_index_name === s3_vector_index_name)) {
-            return undefined
+            throw Error("Inconsistent prefixes.")
         }
         for (let i = 0; i < s3vectors_inputs.length; i += 500) {
             await this.s3vectors.deleteVectors({
@@ -224,7 +224,7 @@ export class UtilsS3Vectors {
         if (new_vectors === undefined) {
             return []
         }
-        const results : { s3_vector_id : string, distance : number }[] = []
+        const results : { s3_vector_id : string, distance : number, headers : Record<string, any> }[] = []
         for (const new_vector of new_vectors) {
             const s3_vector_key = new_vector.key
             if (s3_vector_key === undefined) {
@@ -237,9 +237,15 @@ export class UtilsS3Vectors {
                 console.log("Bad distance:", distance)
                 continue
             }
+            const headers : any = new_vector.metadata
+            if (!(typeof headers === "object" && Object.keys(headers).every((header_id : any) => typeof header_id === "string"))) {
+                console.log("Bad headers:", headers)
+                continue
+            }
             results.push({
                 s3_vector_id : s3_vector_id,
-                distance : distance
+                distance : distance,
+                headers : headers
             })
         }
         return results