triangle-utils 1.4.83 → 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;
@@ -31,10 +35,10 @@ export class UtilsBedrock {
                         schema: options.json_format
                     }
                 } : undefined,
-                cache_control: {
-                    type: "ephemeral",
-                    ttl: options.cache_period
-                }
+                // cache_control : {
+                //     type : "ephemeral",
+                //     ttl : options.cache_period
+                // }
             });
             const usage = response.usage;
             if (options.print_usage !== undefined && options.print_usage) {

package/dist/src/f.js

@@ -3,10 +3,10 @@ import { TriangleUtils } from "./index.js";
 console.log("Starting triangle-utils!");
 const secret_name = "elections_config";
 const secrets_manager = new SecretsManager({ region: "us-east-1" });
-const response = await secrets_manager.getSecretValue({
+const secret = await secrets_manager.getSecretValue({
     SecretId: secret_name
 });
-const api_keys = JSON.parse(response?.SecretString || "");
+const api_keys = JSON.parse(secret?.SecretString || "");
 const config = {
     google_email: "louishou@triangleanalytics.com",
     alerts_email: "alerts@triangleanalytics.com",
@@ -216,5 +216,12 @@ const utils = new TriangleUtils(config);
 //     min_twitter_tweet_id : "2061610563679985813"
 // })
 // console.log(results)
-const f = await utils.dynamodb.query_range("federal_election_donations:donor_address_state_id.federal_election_committee_id-receipt_date-federal_election_donation_id", { donor_address_state_id: "NH", federal_election_committee_id: "C00694323" }, { receipt_date: ["2021", "2023"] }, { compile: false });
-console.log(f);
+// const f = await utils.dynamodb.query_range("federal_election_donations:donor_address_state_id.federal_election_committee_id-receipt_date-federal_election_donation_id", { donor_address_state_id : "NH", federal_election_committee_id : "C00694323" }, { receipt_date : ["2021", "2023"] }, { compile : false })
+// console.log(f)
+console.log("YERP");
+const response = await utils.bedrock.claude_converse("anthropic.claude-opus-4-8", [
+    { role: "user", text: "Hi" }
+], {
+    print_usage: true
+});
+console.log(response);

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "triangle-utils",
-    "version": "1.4.83",
+    "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,
@@ -27,7 +35,6 @@ export class UtilsBedrock {
         } = {}
     ) {
         try {
-            
             const response = await this.anthropic_bedrock_mantle.messages.create({
                 model : model_id,
                 messages : messages.map(message => ({
@@ -48,10 +55,10 @@ export class UtilsBedrock {
                         schema : options.json_format
                     }
                 } : undefined,
-                cache_control : {
-                    type : "ephemeral",
-                    ttl : options.cache_period
-                }
+                // cache_control : {
+                //     type : "ephemeral",
+                //     ttl : options.cache_period
+                // }
             })
 
             const usage = response.usage

package/src/f.ts

@@ -7,11 +7,11 @@ const secret_name = "elections_config"
 
 const secrets_manager = new SecretsManager({ region: "us-east-1" })
 
-const response = await secrets_manager.getSecretValue({
+const secret = await secrets_manager.getSecretValue({
     SecretId: secret_name
 })
 
-const api_keys = JSON.parse(response?.SecretString || "")
+const api_keys = JSON.parse(secret?.SecretString || "")
 
 const config = {
     google_email : "louishou@triangleanalytics.com",
@@ -250,6 +250,17 @@ const utils = new TriangleUtils(config)
 
 // console.log(results)
 
-const f = await utils.dynamodb.query_range("federal_election_donations:donor_address_state_id.federal_election_committee_id-receipt_date-federal_election_donation_id", { donor_address_state_id : "NH", federal_election_committee_id : "C00694323" }, { receipt_date : ["2021", "2023"] }, { compile : false })
-
-console.log(f)
+// const f = await utils.dynamodb.query_range("federal_election_donations:donor_address_state_id.federal_election_committee_id-receipt_date-federal_election_donation_id", { donor_address_state_id : "NH", federal_election_committee_id : "C00694323" }, { receipt_date : ["2021", "2023"] }, { compile : false })
+
+// console.log(f)
+console.log("YERP")
+const response = await utils.bedrock.claude_converse(
+    "anthropic.claude-opus-4-8",
+    [
+        { role : "user", text : "Hi" }
+    ],
+    {
+        print_usage : true
+    }
+)
+console.log(response)