triangle-utils 1.4.174 → 1.4.176

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,7 +253,7 @@ export class UtilsBedrock {
         return undefined;
     }
     async cohere_invoke(model_id, text, options = {}) {
-        for (let i = 0; i < 3; i++) {
+        for (let i = 0; i < 5; i++) {
             try {
                 const output = await this.bedrock_runtime.invokeModel({
                     modelId: model_id,
@@ -278,10 +279,10 @@ export class UtilsBedrock {
                 }
             }
             catch (error) {
-                console.log(error.stack);
-                console.log("Failed to get from Bedrock.");
+                console.log(error.message);
+                await TriangleUtils.wait(5000 * (i + 1));
             }
         }
-        throw new Error();
+        throw new Error("Failed to Cohere");
     }
 }

package/package.json

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

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,7 +285,7 @@ export class UtilsBedrock {
     }
 
     async cohere_invoke(model_id : string, text : string, options : { input_type_id? : string } = {}) : Promise<number[] | undefined> {
-        for (let i = 0; i < 3; i++) {
+        for (let i = 0; i < 5; i++) {
             try {
                 const output = await this.bedrock_runtime.invokeModel({
                     modelId : model_id,
@@ -309,11 +310,11 @@ export class UtilsBedrock {
                     return coordinates
                 }
             } catch (error : any) {
-                console.log(error.stack)
-                console.log("Failed to get from Bedrock.")
+                console.log(error.message)
+                await TriangleUtils.wait(5000 * (i + 1))
             }
         }
-        throw new Error()
+        throw new Error("Failed to Cohere")
     }
 
 }