triangle-utils 1.4.85 → 1.4.87

package/dist/src/UtilsBedrock.d.ts

@@ -1,18 +1,19 @@
-declare const role_ids: readonly ["user", "assistant", "system"];
-export type RoleID = typeof role_ids[number];
-export declare function is_role_id(role_id: any): role_id is RoleID;
+declare const roles: readonly ["user", "assistant", "system"];
+export type Role = typeof roles[number];
+export declare function is_role(role: any): role is Role;
 export declare class UtilsBedrock {
     private readonly bedrock_runtime;
     private readonly anthropic_bedrock_mantle;
     private readonly text_decoder;
     constructor(region: string);
     claude_converse(model_id: string, messages: {
-        role: RoleID;
+        role: Role;
         text: string;
     }[], options?: {
         system_prompt?: string;
         max_tokens?: number;
         print_usage?: boolean;
+        print_content?: boolean;
         json_format?: Record<string, any>;
         web?: boolean;
         cache_period?: "5m" | "1h";

package/dist/src/UtilsBedrock.js

@@ -1,8 +1,8 @@
 import { BedrockRuntime } from "@aws-sdk/client-bedrock-runtime";
 import { AnthropicBedrockMantle } from "@anthropic-ai/bedrock-sdk";
-const role_ids = ["user", "assistant", "system"];
-export function is_role_id(role_id) {
-    return role_ids.includes(role_id);
+const roles = ["user", "assistant", "system"];
+export function is_role(role) {
+    return roles.includes(role);
 }
 export class UtilsBedrock {
     bedrock_runtime;
@@ -44,6 +44,9 @@ export class UtilsBedrock {
             if (options.print_usage !== undefined && options.print_usage) {
                 console.log(usage);
             }
+            if (options.print_content !== undefined && options.print_content) {
+                console.log(response.content);
+            }
             const text = response.content.filter(content => content.type === "text")[0].text;
             return text;
         }

package/package.json

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

package/src/UtilsBedrock.ts

@@ -1,12 +1,12 @@
 import { BedrockRuntime } from "@aws-sdk/client-bedrock-runtime"
 import { AnthropicBedrockMantle } from "@anthropic-ai/bedrock-sdk"
 
-const role_ids = ["user", "assistant", "system"] as const
+const roles = ["user", "assistant", "system"] as const
 
-export type RoleID = typeof role_ids[number]
+export type Role = typeof roles[number]
 
-export function is_role_id(role_id : any) : role_id is RoleID {
-    return role_ids.includes(role_id)
+export function is_role(role : any) : role is Role {
+    return roles.includes(role)
 }
 
 export class UtilsBedrock {
@@ -24,11 +24,12 @@ export class UtilsBedrock {
 
     async claude_converse(
         model_id : string,
-        messages : { role : RoleID, text : string }[],
+        messages : { role : Role, text : string }[],
         options : {
             system_prompt? : string,
             max_tokens? : number,
             print_usage? : boolean,
+            print_content? : boolean,
             json_format? : Record<string, any>,
             web? : boolean,
             cache_period? : "5m" | "1h"
@@ -65,6 +66,9 @@ export class UtilsBedrock {
             if (options.print_usage !== undefined && options.print_usage) {
                 console.log(usage)
             }
+            if (options.print_content !== undefined && options.print_content) {
+                console.log(response.content)
+            }
             const text = response.content.filter(content => content.type === "text")[0].text
             return text
         } catch (error) {