triangle-utils 1.4.58 → 1.4.60

package/dist/src/UtilsAnthropic.d.ts

@@ -1,6 +1,11 @@
 export declare class UtilsAnthropic {
     private readonly anthropic;
     constructor(anthropic_api_key: string | undefined);
+    claude_multi_query(model_id: string, prompts: string[], options?: {
+        max_tokens?: number;
+        print_usage?: boolean;
+        json_format?: Record<string, any>;
+    }): Promise<string | undefined>;
     claude_query(model_id: string, prompt: string, attachment_pdf?: string, options?: {
         max_tokens?: number;
         print_usage?: boolean;

package/dist/src/UtilsAnthropic.js

@@ -4,6 +4,42 @@ export class UtilsAnthropic {
     constructor(anthropic_api_key) {
         this.anthropic = new Anthropic({ apiKey: anthropic_api_key });
     }
+    async claude_multi_query(model_id, prompts, options = {}) {
+        try {
+            const response = await this.anthropic.messages.create({
+                model: model_id,
+                messages: prompts.map(prompt => ({
+                    role: "user",
+                    content: [
+                        {
+                            type: "text",
+                            text: prompt
+                        }
+                    ]
+                })),
+                max_tokens: options.max_tokens || 1000,
+                output_config: options.json_format !== undefined ? {
+                    format: {
+                        type: "json_schema",
+                        schema: options.json_format
+                    }
+                } : undefined
+            });
+            const usage = response.usage;
+            if (options.print_usage !== undefined && options.print_usage) {
+                console.log(usage);
+            }
+            const text = response.content.filter(content => content.type === "text")[0].text;
+            return text;
+        }
+        catch (error) {
+            if (error instanceof Error) {
+                console.log(error.stack);
+            }
+            console.log("Failed to get from Anthropic.");
+        }
+        return undefined;
+    }
     async claude_query(model_id, prompt, attachment_pdf, options = {}) {
         try {
             const message_contents = [

package/dist/src/UtilsS3.d.ts

@@ -10,7 +10,7 @@ export declare class UtilsS3 {
     get_file(s3_filename: string, encoding?: string): Promise<string | undefined>;
     get_file_buffer(s3_filename: string): Promise<Buffer | undefined>;
     get_file_stream(s3_filename: string): Promise<Readable | undefined>;
-    create_file(s3_filename: string, content: string | Buffer, options?: {
+    create_file(s3_filename: string, content: string | Buffer | Readable, options?: {
         content_type?: string;
     }): Promise<import("@aws-sdk/client-s3").PutObjectCommandOutput | undefined>;
     delete_file(s3_filename: string): Promise<void>;

package/package.json

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

package/src/UtilsAnthropic.ts

@@ -8,6 +8,52 @@ export class UtilsAnthropic {
         this.anthropic = new Anthropic({ apiKey : anthropic_api_key })
     }
 
+    async claude_multi_query(
+        model_id : string,
+        prompts : string[],
+        options : {
+            max_tokens? : number,
+            print_usage? : boolean,
+            json_format? : Record<string, any>
+        } = {}
+    ) : Promise<string | undefined> {
+        try {
+            
+            const response = await this.anthropic.messages.create({
+                model : model_id,
+                messages : prompts.map(prompt => ({
+                    role : "user",
+                    content : [
+                        {
+                            type : "text",
+                            text : prompt
+                        }
+                    ]
+                })),
+                max_tokens : options.max_tokens || 1000,
+                output_config : options.json_format !== undefined ? {
+                    format : {
+                        type : "json_schema",
+                        schema : options.json_format
+                    }
+                } : undefined
+            })
+
+            const usage = response.usage
+            if (options.print_usage !== undefined && options.print_usage) {
+                console.log(usage)
+            }
+            const text = response.content.filter(content => content.type === "text")[0].text
+            return text
+        } catch (error) {
+            if (error instanceof Error) {
+                console.log(error.stack)
+            }
+            console.log("Failed to get from Anthropic.")
+        }
+        return undefined
+    }
+
     async claude_query(
         model_id : string,
         prompt : string,

package/src/UtilsS3.ts

@@ -147,7 +147,7 @@ export class UtilsS3 {
         }
     }
 
-    async create_file(s3_filename : string, content : string | Buffer, options : {
+    async create_file(s3_filename : string, content : string | Buffer | Readable, options : {
         content_type? : string
     } = {}) {
         const content_type = options.content_type !== undefined ? options.content_type : undefined