triangle-utils 1.4.57 → 1.4.59

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. package/dist/src/UtilsAnthropic.d.ts +5 -0
  2. package/dist/src/UtilsAnthropic.js +36 -0
  3. package/dist/src/UtilsMisc.js +6 -5
  4. package/package.json +1 -1
  5. package/src/UtilsAnthropic.ts +46 -0
  6. package/src/UtilsMisc.ts +6 -5
package/dist/src/UtilsAnthropic.d.ts CHANGED
@@ -1,6 +1,11 @@
1
1
  export declare class UtilsAnthropic {
2
2
  private readonly anthropic;
3
3
  constructor(anthropic_api_key: string | undefined);
4
+ claude_multi_query(model_id: string, prompts: string[], options?: {
5
+ max_tokens?: number;
6
+ print_usage?: boolean;
7
+ json_format?: Record<string, any>;
8
+ }): Promise<string | undefined>;
4
9
  claude_query(model_id: string, prompt: string, attachment_pdf?: string, options?: {
5
10
  max_tokens?: number;
6
11
  print_usage?: boolean;
package/dist/src/UtilsAnthropic.js CHANGED
@@ -4,6 +4,42 @@ export class UtilsAnthropic {
4
4
  constructor(anthropic_api_key) {
5
5
  this.anthropic = new Anthropic({ apiKey: anthropic_api_key });
6
6
  }
7
+ async claude_multi_query(model_id, prompts, options = {}) {
8
+ try {
9
+ const response = await this.anthropic.messages.create({
10
+ model: model_id,
11
+ messages: prompts.map(prompt => ({
12
+ role: "user",
13
+ content: [
14
+ {
15
+ type: "text",
16
+ text: prompt
17
+ }
18
+ ]
19
+ })),
20
+ max_tokens: options.max_tokens || 1000,
21
+ output_config: options.json_format !== undefined ? {
22
+ format: {
23
+ type: "json_schema",
24
+ schema: options.json_format
25
+ }
26
+ } : undefined
27
+ });
28
+ const usage = response.usage;
29
+ if (options.print_usage !== undefined && options.print_usage) {
30
+ console.log(usage);
31
+ }
32
+ const text = response.content.filter(content => content.type === "text")[0].text;
33
+ return text;
34
+ }
35
+ catch (error) {
36
+ if (error instanceof Error) {
37
+ console.log(error.stack);
38
+ }
39
+ console.log("Failed to get from Anthropic.");
40
+ }
41
+ return undefined;
42
+ }
7
43
  async claude_query(model_id, prompt, attachment_pdf, options = {}) {
8
44
  try {
9
45
  const message_contents = [
package/dist/src/UtilsMisc.js CHANGED
@@ -25,16 +25,17 @@ export class UtilsMisc {
25
25
  const print_freq = typeof print_option === "boolean" ? (print_option === true ? 1 : 0) : print_option;
26
26
  let index = 0;
27
27
  let iterators = [];
28
- const outputs = []; //new Array(inputs.length).fill(undefined)
28
+ const outputs = new Array(inputs.length).fill(undefined);
29
29
  for (let iterator_id = 0; iterator_id < num_iterators; iterator_id++) {
30
30
  iterators.push((async () => {
31
31
  while (index < inputs.length) {
32
+ const local_index = index;
32
33
  index += 1;
33
- if (print_freq !== undefined && ((index - 1) % print_freq === 0)) {
34
- console.log(iterator_id + ":" + (index - 1) + "/" + inputs.length);
34
+ if (print_freq !== undefined && (local_index % print_freq === 0)) {
35
+ console.log(iterator_id + ":" + local_index + "/" + inputs.length);
35
36
  }
36
- const output = await this.safe_run(() => f(inputs[index - 1], iterator_id, index));
37
- outputs.push(output);
37
+ const output = await this.safe_run(() => f(inputs[local_index], iterator_id, local_index));
38
+ outputs[local_index] = output;
38
39
  }
39
40
  })());
40
41
  }
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "triangle-utils",
3
- "version": "1.4.57",
3
+ "version": "1.4.59",
4
4
  "main": "dist/src/index.js",
5
5
  "types": "dist/src/index.d.ts",
6
6
  "directories": {
package/src/UtilsAnthropic.ts CHANGED
@@ -8,6 +8,52 @@ export class UtilsAnthropic {
8
8
  this.anthropic = new Anthropic({ apiKey : anthropic_api_key })
9
9
  }
10
10
 
11
+ async claude_multi_query(
12
+ model_id : string,
13
+ prompts : string[],
14
+ options : {
15
+ max_tokens? : number,
16
+ print_usage? : boolean,
17
+ json_format? : Record<string, any>
18
+ } = {}
19
+ ) : Promise<string | undefined> {
20
+ try {
21
+
22
+ const response = await this.anthropic.messages.create({
23
+ model : model_id,
24
+ messages : prompts.map(prompt => ({
25
+ role : "user",
26
+ content : [
27
+ {
28
+ type : "text",
29
+ text : prompt
30
+ }
31
+ ]
32
+ })),
33
+ max_tokens : options.max_tokens || 1000,
34
+ output_config : options.json_format !== undefined ? {
35
+ format : {
36
+ type : "json_schema",
37
+ schema : options.json_format
38
+ }
39
+ } : undefined
40
+ })
41
+
42
+ const usage = response.usage
43
+ if (options.print_usage !== undefined && options.print_usage) {
44
+ console.log(usage)
45
+ }
46
+ const text = response.content.filter(content => content.type === "text")[0].text
47
+ return text
48
+ } catch (error) {
49
+ if (error instanceof Error) {
50
+ console.log(error.stack)
51
+ }
52
+ console.log("Failed to get from Anthropic.")
53
+ }
54
+ return undefined
55
+ }
56
+
11
57
  async claude_query(
12
58
  model_id : string,
13
59
  prompt : string,
package/src/UtilsMisc.ts CHANGED
@@ -35,16 +35,17 @@ export class UtilsMisc {
35
35
  const print_freq : number | undefined = typeof print_option === "boolean" ? (print_option === true ? 1 : 0) : print_option
36
36
  let index = 0
37
37
  let iterators = []
38
- const outputs : (U | undefined)[] = []//new Array(inputs.length).fill(undefined)
38
+ const outputs : (U | undefined)[] = new Array(inputs.length).fill(undefined)
39
39
  for (let iterator_id = 0; iterator_id < num_iterators; iterator_id++) {
40
40
  iterators.push((async () => {
41
41
  while (index < inputs.length) {
42
+ const local_index = index
42
43
  index += 1
43
- if (print_freq !== undefined && ((index - 1) % print_freq === 0)) {
44
- console.log(iterator_id + ":" + (index - 1) + "/" + inputs.length)
44
+ if (print_freq !== undefined && (local_index % print_freq === 0)) {
45
+ console.log(iterator_id + ":" + local_index + "/" + inputs.length)
45
46
  }
46
- const output = await this.safe_run(() => f(inputs[index - 1], iterator_id, index))
47
- outputs.push(output)
47
+ const output = await this.safe_run(() => f(inputs[local_index], iterator_id, local_index))
48
+ outputs[local_index] = output
48
49
  }
49
50
  })())
50
51
  }