triangle-utils 1.4.84 → 1.4.85

package/dist/src/UtilsBedrock.d.ts

@@ -1,10 +1,13 @@
+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;
 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: "user" | "assistant" | "system";
+        role: RoleID;
         text: string;
     }[], options?: {
         system_prompt?: string;
@@ -23,3 +26,4 @@ export declare class UtilsBedrock {
     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>;
 }
+export {};

package/dist/src/UtilsBedrock.js

@@ -1,5 +1,9 @@
 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);
+}
 export class UtilsBedrock {
     bedrock_runtime;
     anthropic_bedrock_mantle;

package/package.json

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

package/src/UtilsBedrock.ts

@@ -1,6 +1,14 @@
 import { BedrockRuntime } from "@aws-sdk/client-bedrock-runtime"
 import { AnthropicBedrockMantle } from "@anthropic-ai/bedrock-sdk"
 
+const role_ids = ["user", "assistant", "system"] as const
+
+export type RoleID = typeof role_ids[number]
+
+export function is_role_id(role_id : any) : role_id is RoleID {
+    return role_ids.includes(role_id)
+}
+
 export class UtilsBedrock {
 
     private readonly bedrock_runtime : BedrockRuntime
@@ -16,7 +24,7 @@ export class UtilsBedrock {
 
     async claude_converse(
         model_id : string,
-        messages : { role : "user" | "assistant" | "system", text : string }[],
+        messages : { role : RoleID, text : string }[],
         options : {
             system_prompt? : string,
             max_tokens? : number,