triangle-utils 1.4.102 → 1.4.103

package/dist/src/UtilsBedrock.d.ts

@@ -35,5 +35,8 @@ export declare class UtilsBedrock {
     llama_invoke(model_id: string, prompt: string, temperature?: number, max_gen_len?: number, top_p?: number): Promise<string | undefined>;
     gpt_converse(model_id: string, prompt: string, temperature?: number, max_gen_len?: number, top_p?: number): Promise<string | undefined>;
     titan_invoke(text: string): Promise<number[] | undefined>;
+    cohere_invoke(model_id: string, text: string, options?: {
+        input_type_id?: string;
+    }): Promise<number[] | undefined>;
 }
 export {};

package/dist/src/UtilsBedrock.js

@@ -272,4 +272,35 @@ export class UtilsBedrock {
         console.log("Failed to get from Bedrock, quitting.");
         return undefined;
     }
+    async cohere_invoke(model_id, text, options = {}) {
+        try {
+            const output = await this.bedrock_runtime.invokeModel({
+                modelId: model_id,
+                body: JSON.stringify({
+                    input_type: options.input_type_id || "search_document",
+                    inputs: [
+                        {
+                            content: [
+                                { type: "text", text: text }
+                            ]
+                        }
+                    ],
+                    output_dimension: 1024,
+                    truncate: "RIGHT",
+                    max_tokens: 128000
+                }),
+                contentType: "application/json"
+            });
+            const response = JSON.parse(this.text_decoder.decode(output.body));
+            const coordinates = response.embeddings.float[0];
+            if (coordinates !== undefined && Array.isArray(coordinates) && coordinates.every((i) => typeof i === "number")) {
+                return coordinates;
+            }
+        }
+        catch (error) {
+            console.log(error.stack);
+            console.log("Failed to get from Bedrock.");
+        }
+        return undefined;
+    }
 }

package/dist/src/f.js

@@ -15,9 +15,9 @@ const config = {
 };
 console.log(config);
 const utils = new TriangleUtils(config);
-const raw_s3_id_prefix = config.s3_scout + "/raw_scout_documents/57523929653b604e/";
-const raw_s3_ids = await utils.s3.query_prefix(raw_s3_id_prefix, { compile: true });
-console.log(raw_s3_ids);
+// const raw_s3_id_prefix = config.s3_scout + "/raw_scout_documents/57523929653b604e/"
+// const raw_s3_ids = await utils.s3.query_prefix(raw_s3_id_prefix, { compile : true })
+// console.log(raw_s3_ids)
 // const text = await utils.xai.grok_simple_query("grok-4.3", "Find the X username corresponding to the candidate of the committee \"Darializa for Congress\"", {
 //     print_usage : true,
 //     json_format : {
@@ -232,3 +232,5 @@ console.log(raw_s3_ids);
 //     }
 // )
 // console.log(response)
+const coordinates = await utils.bedrock.cohere_invoke("global.cohere.embed-v4:0", "yerrp");
+console.log(coordinates);

package/package.json

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

package/src/UtilsBedrock.ts

@@ -308,4 +308,35 @@ export class UtilsBedrock {
         return undefined
     }
 
+    async cohere_invoke(model_id : string, text : string, options : { input_type_id? : string } = {}) : Promise<number[] | undefined> {
+        try {
+            const output = await this.bedrock_runtime.invokeModel({
+                modelId : model_id,
+                body : JSON.stringify({
+                    input_type: options.input_type_id || "search_document",
+                    inputs: [
+                        {
+                            content: [
+                                { type : "text", text : text }
+                            ]
+                        }
+                    ],
+                    output_dimension: 1024,
+                    truncate: "RIGHT",
+                    max_tokens: 128000
+                }),
+                contentType : "application/json"
+            })
+            const response = JSON.parse(this.text_decoder.decode(output.body))
+            const coordinates = response.embeddings.float[0]
+            if (coordinates !== undefined && Array.isArray(coordinates) && coordinates.every((i : any) => typeof i === "number")) {
+                return coordinates
+            }
+        } catch (error : any) {
+            console.log(error.stack)
+            console.log("Failed to get from Bedrock.")
+        }
+        return undefined
+    }
+
 }

package/src/f.ts

@@ -24,10 +24,10 @@ console.log(config)
 
 const utils = new TriangleUtils(config)
 
-const raw_s3_id_prefix = config.s3_scout + "/raw_scout_documents/57523929653b604e/"
-const raw_s3_ids = await utils.s3.query_prefix(raw_s3_id_prefix, { compile : true })
+// const raw_s3_id_prefix = config.s3_scout + "/raw_scout_documents/57523929653b604e/"
+// const raw_s3_ids = await utils.s3.query_prefix(raw_s3_id_prefix, { compile : true })
 
-console.log(raw_s3_ids)
+// console.log(raw_s3_ids)
 
 // const text = await utils.xai.grok_simple_query("grok-4.3", "Find the X username corresponding to the candidate of the committee \"Darializa for Congress\"", {
 //     print_usage : true,
@@ -268,4 +268,8 @@ console.log(raw_s3_ids)
 //         print_usage : true
 //     }
 // )
-// console.log(response)
+// console.log(response)
+
+const coordinates = await utils.bedrock.cohere_invoke("global.cohere.embed-v4:0", "yerrp")
+
+console.log(coordinates)