@bedrockio/ai 0.6.1 → 0.7.0

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.
package/CHANGELOG.md CHANGED
@@ -1,3 +1,11 @@
1
+ ## 0.7.0
2
+
3
+ - Allow getting the template source.
4
+
5
+ ## 0.6.2
6
+
7
+ - Pass tool_choice param.
8
+
1
9
  ## 0.6.0
2
10
 
3
11
  - Moved template rendering out to external package.
@@ -105,6 +105,14 @@ class BaseClient {
105
105
  };
106
106
  }
107
107
  }
108
+ /**
109
+ * Gets the source for a given template.
110
+ *
111
+ * @param {string} name
112
+ */
113
+ getTemplateSource(name) {
114
+ return this.renderer.getTemplateSource(name);
115
+ }
108
116
  // Protected
109
117
  runPrompt(options) {
110
118
  void options;
@@ -21,12 +21,13 @@ class OpenAiClient extends BaseClient_js_1.default {
21
21
  return data.map((o) => o.id);
22
22
  }
23
23
  async runPrompt(options) {
24
- const { model, tools, verbosity, temperature, prevResponseId, messages: input, system: instructions, stream = false, } = options;
24
+ const { model, tools, verbosity, temperature, prevResponseId, messages: input, system: instructions, tool_choice = 'auto', stream = false, } = options;
25
25
  const params = {
26
26
  model,
27
27
  tools,
28
28
  input,
29
29
  stream,
30
+ tool_choice,
30
31
  temperature,
31
32
  instructions,
32
33
  previous_response_id: prevResponseId,
@@ -103,6 +103,14 @@ export default class BaseClient {
103
103
  };
104
104
  }
105
105
  }
106
+ /**
107
+ * Gets the source for a given template.
108
+ *
109
+ * @param {string} name
110
+ */
111
+ getTemplateSource(name) {
112
+ return this.renderer.getTemplateSource(name);
113
+ }
106
114
  // Protected
107
115
  runPrompt(options) {
108
116
  void options;
@@ -15,12 +15,13 @@ export class OpenAiClient extends BaseClient {
15
15
  return data.map((o) => o.id);
16
16
  }
17
17
  async runPrompt(options) {
18
- const { model, tools, verbosity, temperature, prevResponseId, messages: input, system: instructions, stream = false, } = options;
18
+ const { model, tools, verbosity, temperature, prevResponseId, messages: input, system: instructions, tool_choice = 'auto', stream = false, } = options;
19
19
  const params = {
20
20
  model,
21
21
  tools,
22
22
  input,
23
23
  stream,
24
+ tool_choice,
24
25
  temperature,
25
26
  instructions,
26
27
  previous_response_id: prevResponseId,
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@bedrockio/ai",
3
- "version": "0.6.1",
3
+ "version": "0.7.0",
4
4
  "description": "Bedrock wrapper for common AI chatbots.",
5
5
  "type": "module",
6
6
  "scripts": {
@@ -36,7 +36,7 @@
36
36
  },
37
37
  "dependencies": {
38
38
  "@anthropic-ai/sdk": "^0.66.0",
39
- "@bedrockio/templates": "^0.1.1",
39
+ "@bedrockio/templates": "^0.3.0",
40
40
  "@google/generative-ai": "^0.21.0",
41
41
  "openai": "^6.3.0",
42
42
  "partial-json": "^0.1.7"
@@ -16,6 +16,12 @@ export default class BaseClient {
16
16
  * @returns {AsyncIterator}
17
17
  */
18
18
  stream(options: PromptOptions & StreamOptions): AsyncIterator<any, any, any>;
19
+ /**
20
+ * Gets the source for a given template.
21
+ *
22
+ * @param {string} name
23
+ */
24
+ getTemplateSource(name: string): any;
19
25
  runPrompt(options: any): void;
20
26
  runStream(options: any): void;
21
27
  getTextResponse(response: any): void;
@@ -1 +1 @@
1
- {"version":3,"file":"BaseClient.d.ts","sourceRoot":"","sources":["../src/BaseClient.js"],"names":[],"mappings":"AAKA;IACE,0BASC;IARC,aAIC;IACD,2BAEE;IAKJ;;;;;OAKG;IACH,gBAFW,aAAa,gBAuCvB;IAED;;;;;OAKG;IACH,gBAHW,aAAa,GAAG,aAAa,gCAsDvC;IAID,8BAGC;IAED,8BAGC;IAED,qCAGC;IAED;;OAEG;IACH,0CAGC;IAED;;OAEG;IACH,oDAGC;IAED;;OAEG;IACH,sCAGC;IAID;;OAEG;IACH,oCAOC;IAED;;;MAsCC;IAED,kCAeC;IAED;;;MA4BC;IAED,uDAWC;IAED,oCAMC;CACF;;;;;WAIa,MAAM,GAAC,aAAa,EAAE;;;;YACtB,MAAM;;;;YACN,OAAO;;;;;;;;aAEP,KAAK,GAAG,MAAM,GAAG,MAAM,GAAG,UAAU;;;;;;;;;;;sBAOpC,MAAM;;;UAKN,QAAQ,GAAG,MAAM,GAAG,WAAW;aAC/B,MAAM;;iCAlTa,sBAAsB"}
1
+ {"version":3,"file":"BaseClient.d.ts","sourceRoot":"","sources":["../src/BaseClient.js"],"names":[],"mappings":"AAKA;IACE,0BASC;IARC,aAIC;IACD,2BAEE;IAKJ;;;;;OAKG;IACH,gBAFW,aAAa,gBAuCvB;IAED;;;;;OAKG;IACH,gBAHW,aAAa,GAAG,aAAa,gCAsDvC;IAED;;;;OAIG;IACH,wBAFW,MAAM,OAIhB;IAID,8BAGC;IAED,8BAGC;IAED,qCAGC;IAED;;OAEG;IACH,0CAGC;IAED;;OAEG;IACH,oDAGC;IAED;;OAEG;IACH,sCAGC;IAID;;OAEG;IACH,oCAOC;IAED;;;MAsCC;IAED,kCAeC;IAED;;;MA4BC;IAED,uDAWC;IAED,oCAMC;CACF;;;;;WAIa,MAAM,GAAC,aAAa,EAAE;;;;YACtB,MAAM;;;;YACN,OAAO;;;;;;;;aAEP,KAAK,GAAG,MAAM,GAAG,MAAM,GAAG,UAAU;;;;;;;;;;;sBAOpC,MAAM;;;UAKN,QAAQ,GAAG,MAAM,GAAG,WAAW;aAC/B,MAAM;;iCA3Ta,sBAAsB"}
@@ -1 +1 @@
1
- {"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.js"],"names":[],"mappings":"AAIA;IACE,6BAAoC;IAIlC,eAAiC;IAGnC;;;OAGG;IACH,4BAGC;IAED;;yFA+BC;IAED;;yFAKC;IAED,oCAEC;IAwBD;;;MAcC;IAID;;;;;;;;;;MAmBC;IAED;;;;;;;;;;;;;;;;;;;;;;;;MAyBC;CACF;uBAtJsB,iBAAiB;mBAFrB,QAAQ"}
1
+ {"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.js"],"names":[],"mappings":"AAIA;IACE,6BAAoC;IAIlC,eAAiC;IAGnC;;;OAGG;IACH,4BAGC;IAED;;yFAiCC;IAED;;yFAKC;IAED,oCAEC;IAwBD;;;MAcC;IAID;;;;;;;;;;MAmBC;IAED;;;;;;;;;;;;;;;;;;;;;;;;MAyBC;CACF;uBAxJsB,iBAAiB;mBAFrB,QAAQ"}