triangle-utils 1.4.173 → 1.4.175

package/dist/src/UtilsBedrock.js

@@ -1,5 +1,6 @@
 import { BedrockRuntime } from "@aws-sdk/client-bedrock-runtime";
 import { AnthropicBedrockMantle } from "@anthropic-ai/bedrock-sdk";
+import { TriangleUtils } from "./index.js";
 const message_roles = ["user", "assistant", "system"];
 export function is_message_role(message_role) {
     return message_roles.includes(message_role);
@@ -252,34 +253,36 @@ export class UtilsBedrock {
         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;
+        for (let i = 0; i < 3; i++) {
+            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.message);
+                await TriangleUtils.wait(5000);
             }
         }
-        catch (error) {
-            console.log(error.stack);
-            console.log("Failed to get from Bedrock.");
-        }
-        return undefined;
+        throw new Error("Failed to Cohere");
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "triangle-utils",
-    "version": "1.4.173",
+    "version": "1.4.175",
     "main": "dist/src/index.js",
     "types": "dist/src/index.d.ts",
     "directories": {
@@ -36,8 +36,7 @@
         "googleapis": "^170.0.0",
         "jsdom": "^29.0.1",
         "pg": "^8.23.0",
-        "scrapingbee": "^1.8.2",
-        "triangle-types": "^1.0.259"
+        "scrapingbee": "^1.8.2"
     },
     "devDependencies": {
         "@eslint/js": "^9.39.2",

package/src/UtilsBedrock.ts

@@ -1,6 +1,7 @@
 import { BedrockRuntime, InvokeModelCommandInput } from "@aws-sdk/client-bedrock-runtime"
 import { AnthropicBedrockMantle } from "@anthropic-ai/bedrock-sdk"
 import { ContentBlockParam, MessageCreateParams, MessageParam, OutputConfig } from "@anthropic-ai/sdk/resources"
+import { TriangleUtils } from "."
 
 const message_roles = ["user", "assistant", "system"] as const
 
@@ -284,34 +285,36 @@ export class UtilsBedrock {
     }
 
     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
+        for (let i = 0; i < 3; i++) {
+            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.message)
+                await TriangleUtils.wait(5000)
             }
-        } catch (error : any) {
-            console.log(error.stack)
-            console.log("Failed to get from Bedrock.")
         }
-        return undefined
+        throw new Error("Failed to Cohere")
     }
 
 }