@bedrockio/ai 0.2.1 → 0.3.0

package/CHANGELOG.md

@@ -1,3 +1,16 @@
+## 0.3.0
+
+- Added MultiClient.
+- Allow partial template interpolation.
+- Changed default openai model to `gpt-4o-mini`.
+- Allow passing in specific `params` object.
+- Allow passing in complex arrays.
+
+## 0.2.1
+
+- Better error handling for entry.
+- Allow parsing of unknown code.
+
 ## 0.2.0
 
 - Added Gemini

package/README.md

@@ -96,6 +96,7 @@ Currently supported platforms:
 - OpenAI (ChatGPT)
 - Anthropic (Claude)
 - Google (Gemini).
+- xAi (Grok).
 
 ## Models
 

package/dist/cjs/BaseClient.js

@@ -58,23 +58,39 @@ class BaseClient {
     }
   }
   async getMessages(options) {
+    const {
+      text
+    } = options;
     const template = await this.resolveTemplate(options);
-    const raw = _mustache.default.render(template, transformParams(options));
-    const messages = [];
-    for (let match of raw.matchAll(MESSAGES_REG)) {
-      const [, role, content] = match;
-      messages.push({
-        role: role.toLowerCase(),
-        content: content.trim()
-      });
-    }
-    if (!messages.length) {
-      messages.push({
+    if (template) {
+      const raw = render(template, options);
+      const messages = [];
+      for (let match of raw.matchAll(MESSAGES_REG)) {
+        const [, role, content] = match;
+        messages.push({
+          role: role.toLowerCase(),
+          content: content.trim()
+        });
+      }
+      if (!messages.length) {
+        messages.push({
+          role: 'user',
+          content: raw.trim()
+        });
+      }
+      return messages;
+    } else if (text) {
+      return [{
         role: 'user',
-        content: raw.trim()
-      });
+        content: text
+      }];
+    } else {
+      throw new Error('No input provided.');
     }
-    return messages;
+  }
+  async buildTemplate(options) {
+    const template = await this.resolveTemplate(options);
+    return render(template, options);
   }
   async loadTemplates() {
     const {
@@ -83,18 +99,18 @@ class BaseClient {
     this.templates ||= await (0, _util.loadTemplates)(templates);
   }
   async resolveTemplate(options) {
-    await this.loadTemplates();
-    let {
-      file,
-      template
+    const {
+      template,
+      file
     } = options;
-    if (!template && file) {
-      template = this.templates[file];
-    }
-    if (!template) {
-      throw new Error('No template provided.');
+    if (template) {
+      return template;
+    } else if (file?.endsWith('.md')) {
+      return await (0, _util.loadTemplate)(file);
+    } else if (file) {
+      await this.loadTemplates();
+      return this.templates[file];
     }
-    return template;
   }
   async getStream(options) {
     return await this.prompt({
@@ -114,17 +130,57 @@ class BaseClient {
   }
 }
 exports.default = BaseClient;
-function transformParams(params) {
+function render(template, options) {
+  let params = {
+    ...options,
+    ...options.params
+  };
+  params = mapObjects(params);
+  params = wrapProxy(params);
+  return _mustache.default.render(template, params);
+}
+
+// Transform arrays and object to versions
+// that are more understandable in the context
+// of a template that may have meaningful whitespace.
+function mapObjects(params) {
   const result = {};
   for (let [key, value] of Object.entries(params)) {
     if (Array.isArray(value)) {
-      value = value.map(el => {
-        return `- ${el}`;
-      }).join('\n');
+      value = mapArray(value);
     } else if (typeof value === 'object') {
       value = JSON.stringify(value, null, 2);
     }
     result[key] = value;
   }
   return result;
+}
+function mapArray(arr) {
+  // Only map simple arrays of primitives.
+  if (typeof arr[0] === 'string') {
+    arr = arr.map(el => {
+      return `- ${el}`;
+    }).join('\n');
+  }
+  return arr;
+}
+
+// Wrap params with a proxy object that reports
+// as having all properties. If one is accessed
+// that does not exist then return the original
+// token. This way templates can be partially
+// interpolated and re-interpolated later.
+function wrapProxy(params) {
+  return new Proxy(params, {
+    has() {
+      return true;
+    },
+    get(target, prop) {
+      if (prop in target) {
+        return target[prop];
+      } else {
+        return `{{{${prop.toString()}}}}`;
+      }
+    }
+  });
 }

package/dist/cjs/google.js

@@ -47,12 +47,6 @@ class GoogleClient extends _BaseClient.default {
     } else {
       response = await generator.generateContent(prompts);
     }
-    // const response = await client.chat.completions.create({
-    //   model,
-    //   messages,
-    //   stream,
-    // });
-
     if (output === 'raw') {
       return response;
     }

package/dist/cjs/index.js

@@ -3,27 +3,78 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
-exports.Client = void 0;
+exports.MultiClient = exports.Client = void 0;
 var _openai = require("./openai.js");
 var _google = require("./google.js");
 var _anthropic = require("./anthropic.js");
+var _xai = require("./xai.js");
 class Client {
+  constructor(options = {}) {
+    if (!options.platform) {
+      throw new Error('No platform specified.');
+    } else if (!options.templates) {
+      throw new Error('No templates directory specified.');
+    } else if (!options.apiKey) {
+      throw new Error('No API key specified.');
+    }
+    return getClientForPlatform(options);
+  }
+}
+exports.Client = Client;
+class MultiClient {
   constructor(options) {
     const {
-      platform,
-      ...rest
+      platforms
     } = options;
-    if (platform === 'openai' || platform === 'gpt') {
-      return new _openai.OpenAiClient(rest);
-    } else if (platform === 'google' || platform === 'gemini') {
-      return new _google.GoogleClient(rest);
-    } else if (platform === 'anthropic' || platform === 'claude') {
-      return new _anthropic.AnthropicClient(rest);
-    } else if (platform) {
-      throw new Error(`Unknown platform "${platform}".`);
-    } else {
-      throw new Error('Platform required.');
+    this.clients = {};
+    for (let platform of platforms) {
+      const {
+        name,
+        apiKey
+      } = platform;
+      const client = getClientForPlatform({
+        ...options,
+        platform: name,
+        apiKey
+      });
+      this.clients[name] = client;
+      this.clients[undefined] ||= client;
     }
   }
+  prompt(options) {
+    return this.getClient(options).prompt(options);
+  }
+  stream(options) {
+    return this.getClient(options).stream(options);
+  }
+  buildTemplate(options) {
+    return this.getClient(options).buildTemplate(options);
+  }
+  getClient(options) {
+    const {
+      platform
+    } = options;
+    const client = this.clients[platform];
+    if (!client) {
+      throw new Error(`Platform "${platform}" not found.`);
+    }
+    return client;
+  }
 }
-exports.Client = Client;
+exports.MultiClient = MultiClient;
+function getClientForPlatform(options) {
+  const {
+    platform
+  } = options;
+  if (platform === 'openai' || platform === 'gpt') {
+    return new _openai.OpenAiClient(options);
+  } else if (platform === 'google' || platform === 'gemini') {
+    return new _google.GoogleClient(options);
+  } else if (platform === 'anthropic' || platform === 'claude') {
+    return new _anthropic.AnthropicClient(options);
+  } else if (platform === 'xai' || platform === 'grok') {
+    return new _xai.XAiClient(options);
+  } else if (platform) {
+    throw new Error(`Unknown platform "${platform}".`);
+  }
+}

package/dist/cjs/openai.js

@@ -8,7 +8,7 @@ var _openai = _interopRequireDefault(require("openai"));
 var _BaseClient = _interopRequireDefault(require("./BaseClient.js"));
 var _util = require("./util.js");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
-const DEFAULT_MODEL = 'gpt-4o';
+const DEFAULT_MODEL = 'gpt-4o-mini';
 class OpenAiClient extends _BaseClient.default {
   constructor(options) {
     super(options);
@@ -40,7 +40,10 @@ class OpenAiClient extends _BaseClient.default {
     const response = await client.chat.completions.create({
       model,
       messages,
-      stream
+      stream,
+      response_format: {
+        type: output === 'json' ? 'json_object' : 'text'
+      }
     });
     if (output === 'raw') {
       return response;

package/dist/cjs/util.js

@@ -3,30 +3,29 @@
 Object.defineProperty(exports, "__esModule", {
   value: true
 });
+exports.loadTemplate = loadTemplate;
 exports.loadTemplates = loadTemplates;
-exports.parse = parse;
 exports.transformResponse = transformResponse;
 var _promises = _interopRequireDefault(require("fs/promises"));
 var _path = _interopRequireDefault(require("path"));
 var _glob = require("glob");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+const CODE_REG = /^```\w*$(.+)```/ms;
 const JSON_REG = /([{[].+[}\]])/s;
 async function loadTemplates(dir) {
   const result = {};
   const files = await (0, _glob.glob)(_path.default.join(dir, '*.md'));
+  if (!files.length) {
+    throw new Error(`No templates found in: ${dir}.`);
+  }
   for (let file of files) {
     const base = _path.default.basename(file, '.md');
-    result[base] = await _promises.default.readFile(file, 'utf-8');
+    result[base] = await loadTemplate(file);
   }
   return result;
 }
-function parse(content) {
-  try {
-    const match = content.match(JSON_REG);
-    return JSON.parse(match[1]);
-  } catch (error) {
-    throw new Error('Unable to derive JSON object in response.');
-  }
+async function loadTemplate(file) {
+  return await _promises.default.readFile(file, 'utf-8');
 }
 function transformResponse(options) {
   const {
@@ -40,8 +39,24 @@ function transformResponse(options) {
   } else if (output === 'messages') {
     return [...messages, message];
   } else if (output === 'json') {
-    return parse(content);
+    return parseJson(content);
+  } else if (output === 'code') {
+    return parseCode(content);
   } else {
-    throw new Error(`Unknown output type "${output}".`);
+    throw new Error(`No output type provided.`);
+  }
+}
+function parseJson(content) {
+  try {
+    return JSON.parse(content.match(JSON_REG)[0]);
+  } catch (error) {
+    throw new Error(`Unable to derive JSON from response:\n\n${content}`);
+  }
+}
+function parseCode(content) {
+  try {
+    return content.match(CODE_REG)[1].trim();
+  } catch (error) {
+    throw new Error(`Unable to derive code from response:\n\n${content}`);
   }
 }

package/dist/cjs/xai.js

@@ -0,0 +1,23 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.XAiClient = void 0;
+var _openai = require("./openai.js");
+const DEFAULT_MODEL = 'grok-2-1212';
+class XAiClient extends _openai.OpenAiClient {
+  constructor(options) {
+    super({
+      ...options,
+      baseURL: 'https://api.x.ai/v1'
+    });
+  }
+  async getCompletion(options) {
+    return super.getCompletion({
+      model: DEFAULT_MODEL,
+      ...options
+    });
+  }
+}
+exports.XAiClient = XAiClient;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bedrockio/ai",
-  "version": "0.2.1",
+  "version": "0.3.0",
   "description": "Bedrock wrapper for common AI chatbots.",
   "type": "module",
   "scripts": {
@@ -31,11 +31,11 @@
     "url": "https://github.com/bedrockio/router"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.33.1",
+    "@anthropic-ai/sdk": "^0.36.3",
     "@google/generative-ai": "^0.21.0",
     "glob": "^11.0.1",
     "mustache": "^4.2.0",
-    "openai": "^4.79.1"
+    "openai": "^4.83.0"
   },
   "devDependencies": {
     "@babel/cli": "^7.26.4",

package/src/BaseClient.js

@@ -1,6 +1,6 @@
 import Mustache from 'mustache';
 
-import { loadTemplates } from './util.js';
+import { loadTemplates, loadTemplate } from './util.js';
 
 const MESSAGES_REG = /(?:^|\n)-{3,}\s*(\w+)\s*-{3,}(.*?)(?=\n-{3,}|$)/gs;
 
@@ -57,26 +57,44 @@ export default class BaseClient {
   }
 
   async getMessages(options) {
+    const { text } = options;
     const template = await this.resolveTemplate(options);
-    const raw = Mustache.render(template, transformParams(options));
-
-    const messages = [];
-    for (let match of raw.matchAll(MESSAGES_REG)) {
-      const [, role, content] = match;
-      messages.push({
-        role: role.toLowerCase(),
-        content: content.trim(),
-      });
-    }
 
-    if (!messages.length) {
-      messages.push({
-        role: 'user',
-        content: raw.trim(),
-      });
+    if (template) {
+      const raw = render(template, options);
+
+      const messages = [];
+      for (let match of raw.matchAll(MESSAGES_REG)) {
+        const [, role, content] = match;
+        messages.push({
+          role: role.toLowerCase(),
+          content: content.trim(),
+        });
+      }
+
+      if (!messages.length) {
+        messages.push({
+          role: 'user',
+          content: raw.trim(),
+        });
+      }
+
+      return messages;
+    } else if (text) {
+      return [
+        {
+          role: 'user',
+          content: text,
+        },
+      ];
+    } else {
+      throw new Error('No input provided.');
     }
+  }
 
-    return messages;
+  async buildTemplate(options) {
+    const template = await this.resolveTemplate(options);
+    return render(template, options);
   }
 
   async loadTemplates() {
@@ -85,19 +103,15 @@ export default class BaseClient {
   }
 
   async resolveTemplate(options) {
-    await this.loadTemplates();
-
-    let { file, template } = options;
-
-    if (!template && file) {
-      template = this.templates[file];
-    }
-
-    if (!template) {
-      throw new Error('No template provided.');
+    const { template, file } = options;
+    if (template) {
+      return template;
+    } else if (file?.endsWith('.md')) {
+      return await loadTemplate(file);
+    } else if (file) {
+      await this.loadTemplates();
+      return this.templates[file];
     }
-
-    return template;
   }
 
   async getStream(options) {
@@ -120,15 +134,25 @@ export default class BaseClient {
   }
 }
 
-function transformParams(params) {
+function render(template, options) {
+  let params = {
+    ...options,
+    ...options.params,
+  };
+
+  params = mapObjects(params);
+  params = wrapProxy(params);
+  return Mustache.render(template, params);
+}
+
+// Transform arrays and object to versions
+// that are more understandable in the context
+// of a template that may have meaningful whitespace.
+function mapObjects(params) {
   const result = {};
   for (let [key, value] of Object.entries(params)) {
     if (Array.isArray(value)) {
-      value = value
-        .map((el) => {
-          return `- ${el}`;
-        })
-        .join('\n');
+      value = mapArray(value);
     } else if (typeof value === 'object') {
       value = JSON.stringify(value, null, 2);
     }
@@ -136,3 +160,36 @@ function transformParams(params) {
   }
   return result;
 }
+
+function mapArray(arr) {
+  // Only map simple arrays of primitives.
+  if (typeof arr[0] === 'string') {
+    arr = arr
+      .map((el) => {
+        return `- ${el}`;
+      })
+      .join('\n');
+  }
+  return arr;
+}
+
+// Wrap params with a proxy object that reports
+// as having all properties. If one is accessed
+// that does not exist then return the original
+// token. This way templates can be partially
+// interpolated and re-interpolated later.
+function wrapProxy(params) {
+  return new Proxy(params, {
+    has() {
+      return true;
+    },
+
+    get(target, prop) {
+      if (prop in target) {
+        return target[prop];
+      } else {
+        return `{{{${prop.toString()}}}}`;
+      }
+    },
+  });
+}

package/src/google.js

@@ -29,7 +29,9 @@ export class GoogleClient extends BaseClient {
     const { model = DEFAULT_MODEL, output = 'text', stream = false } = options;
     const { client } = this;
 
-    const generator = client.getGenerativeModel({ model });
+    const generator = client.getGenerativeModel({
+      model,
+    });
 
     const messages = await this.getMessages(options);
 
@@ -44,11 +46,6 @@ export class GoogleClient extends BaseClient {
     } else {
       response = await generator.generateContent(prompts);
     }
-    // const response = await client.chat.completions.create({
-    //   model,
-    //   messages,
-    //   stream,
-    // });
 
     if (output === 'raw') {
       return response;

package/src/index.js

@@ -1,20 +1,72 @@
 import { OpenAiClient } from './openai.js';
 import { GoogleClient } from './google.js';
 import { AnthropicClient } from './anthropic.js';
+import { XAiClient } from './xai.js';
 
 export class Client {
+  constructor(options = {}) {
+    if (!options.platform) {
+      throw new Error('No platform specified.');
+    } else if (!options.templates) {
+      throw new Error('No templates directory specified.');
+    } else if (!options.apiKey) {
+      throw new Error('No API key specified.');
+    }
+    return getClientForPlatform(options);
+  }
+}
+
+export class MultiClient {
   constructor(options) {
-    const { platform, ...rest } = options;
-    if (platform === 'openai' || platform === 'gpt') {
-      return new OpenAiClient(rest);
-    } else if (platform === 'google' || platform === 'gemini') {
-      return new GoogleClient(rest);
-    } else if (platform === 'anthropic' || platform === 'claude') {
-      return new AnthropicClient(rest);
-    } else if (platform) {
-      throw new Error(`Unknown platform "${platform}".`);
-    } else {
-      throw new Error('Platform required.');
+    const { platforms } = options;
+
+    this.clients = {};
+
+    for (let platform of platforms) {
+      const { name, apiKey } = platform;
+      const client = getClientForPlatform({
+        ...options,
+        platform: name,
+        apiKey,
+      });
+      this.clients[name] = client;
+      this.clients[undefined] ||= client;
+    }
+  }
+
+  prompt(options) {
+    return this.getClient(options).prompt(options);
+  }
+
+  stream(options) {
+    return this.getClient(options).stream(options);
+  }
+
+  buildTemplate(options) {
+    return this.getClient(options).buildTemplate(options);
+  }
+
+  getClient(options) {
+    const { platform } = options;
+    const client = this.clients[platform];
+    if (!client) {
+      throw new Error(`Platform "${platform}" not found.`);
     }
+    return client;
+  }
+}
+
+function getClientForPlatform(options) {
+  const { platform } = options;
+  if (platform === 'openai' || platform === 'gpt') {
+    return new OpenAiClient(options);
+  } else if (platform === 'google' || platform === 'gemini') {
+    return new GoogleClient(options);
+  } else if (platform === 'anthropic' || platform === 'claude') {
+    return new AnthropicClient(options);
+  } else if (platform === 'xai' || platform === 'grok') {
+    return new XAiClient(options);
+  } else if (platform) {
+    throw new Error(`Unknown platform "${platform}".`);
   }
 }

package/src/openai.js

@@ -3,7 +3,7 @@ import OpenAI from 'openai';
 import BaseClient from './BaseClient.js';
 import { transformResponse } from './util.js';
 
-const DEFAULT_MODEL = 'gpt-4o';
+const DEFAULT_MODEL = 'gpt-4o-mini';
 
 export class OpenAiClient extends BaseClient {
   constructor(options) {
@@ -31,6 +31,9 @@ export class OpenAiClient extends BaseClient {
       model,
       messages,
       stream,
+      response_format: {
+        type: output === 'json' ? 'json_object' : 'text',
+      },
     });
 
     if (output === 'raw') {

package/src/util.js

@@ -4,27 +4,27 @@ import path from 'path';
 
 import { glob } from 'glob';
 
+const CODE_REG = /^```\w*$(.+)```/ms;
 const JSON_REG = /([{[].+[}\]])/s;
 
 export async function loadTemplates(dir) {
   const result = {};
   const files = await glob(path.join(dir, '*.md'));
 
+  if (!files.length) {
+    throw new Error(`No templates found in: ${dir}.`);
+  }
+
   for (let file of files) {
     const base = path.basename(file, '.md');
-    result[base] = await fs.readFile(file, 'utf-8');
+    result[base] = await loadTemplate(file);
   }
 
   return result;
 }
 
-export function parse(content) {
-  try {
-    const match = content.match(JSON_REG);
-    return JSON.parse(match[1]);
-  } catch (error) {
-    throw new Error('Unable to derive JSON object in response.');
-  }
+export async function loadTemplate(file) {
+  return await fs.readFile(file, 'utf-8');
 }
 
 export function transformResponse(options) {
@@ -35,8 +35,26 @@ export function transformResponse(options) {
   } else if (output === 'messages') {
     return [...messages, message];
   } else if (output === 'json') {
-    return parse(content);
+    return parseJson(content);
+  } else if (output === 'code') {
+    return parseCode(content);
   } else {
-    throw new Error(`Unknown output type "${output}".`);
+    throw new Error(`No output type provided.`);
+  }
+}
+
+function parseJson(content) {
+  try {
+    return JSON.parse(content.match(JSON_REG)[0]);
+  } catch (error) {
+    throw new Error(`Unable to derive JSON from response:\n\n${content}`);
+  }
+}
+
+function parseCode(content) {
+  try {
+    return content.match(CODE_REG)[1].trim();
+  } catch (error) {
+    throw new Error(`Unable to derive code from response:\n\n${content}`);
   }
 }

package/src/xai.js

@@ -0,0 +1,19 @@
+import { OpenAiClient } from './openai.js';
+
+const DEFAULT_MODEL = 'grok-2-1212';
+
+export class XAiClient extends OpenAiClient {
+  constructor(options) {
+    super({
+      ...options,
+      baseURL: 'https://api.x.ai/v1',
+    });
+  }
+
+  async getCompletion(options) {
+    return super.getCompletion({
+      model: DEFAULT_MODEL,
+      ...options,
+    });
+  }
+}

package/types/BaseClient.d.ts

@@ -30,6 +30,7 @@ export default class BaseClient {
         role: any;
         content: any;
     }[]>;
+    buildTemplate(options: any): Promise<any>;
     loadTemplates(): Promise<void>;
     resolveTemplate(options: any): Promise<any>;
     getStream(options: any): Promise<void>;

package/types/BaseClient.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"BaseClient.d.ts","sourceRoot":"","sources":["../src/BaseClient.js"],"names":[],"mappings":"AAMA;IACE,0BAGC;IAFC,aAAsB;IACtB,eAAqB;IAGvB;;;;;;;;;;;OAWG;IACH,gBALG;QAAwB,KAAK,EAArB,MAAM;QACyC,MAAM,GAArD,KAAK,GAAG,MAAM,GAAG,MAAM,GAAG,UAAU;QACL,KAAK,GAApC;gBAAQ,MAAM,GAAE,GAAG;SAAC;KAE9B,iBAYA;IAED;;;OAGG;IACH,mDAeC;IAED;;;SAqBC;IAED,+BAGC;IAED,4CAcC;IAED,uCAMC;IAED,kCAGC;IAED,iDAIC;CACF"}
+{"version":3,"file":"BaseClient.d.ts","sourceRoot":"","sources":["../src/BaseClient.js"],"names":[],"mappings":"AAMA;IACE,0BAGC;IAFC,aAAsB;IACtB,eAAqB;IAGvB;;;;;;;;;;;OAWG;IACH,gBALG;QAAwB,KAAK,EAArB,MAAM;QACyC,MAAM,GAArD,KAAK,GAAG,MAAM,GAAG,MAAM,GAAG,UAAU;QACL,KAAK,GAApC;gBAAQ,MAAM,GAAE,GAAG;SAAC;KAE9B,iBAYA;IAED;;;OAGG;IACH,mDAeC;IAED;;;SAkCC;IAED,0CAGC;IAED,+BAGC;IAED,4CAUC;IAED,uCAMC;IAED,kCAGC;IAED,iDAIC;CACF"}

package/types/google.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../src/google.js"],"names":[],"mappings":"AAOA;IAII,2BAA4C;IAG9C;;;OAGG;IACH,4BAOC;IAED,0CAwCC;IACD,sCAIC;IAED;;;MAkBC;CACF;uBA3FsB,iBAAiB;mCAFL,uBAAuB"}
+{"version":3,"file":"google.d.ts","sourceRoot":"","sources":["../src/google.js"],"names":[],"mappings":"AAOA;IAII,2BAA4C;IAG9C;;;OAGG;IACH,4BAOC;IAED,0CAqCC;IACD,sCAIC;IAED;;;MAkBC;CACF;uBAxFsB,iBAAiB;mCAFL,uBAAuB"}

package/types/index.d.ts

@@ -1,4 +1,12 @@
 export class Client {
+    constructor(options?: {});
+}
+export class MultiClient {
     constructor(options: any);
+    clients: {};
+    prompt(options: any): any;
+    stream(options: any): any;
+    buildTemplate(options: any): any;
+    getClient(options: any): any;
 }
 //# sourceMappingURL=index.d.ts.map

package/types/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.js"],"names":[],"mappings":"AAIA;IACE,0BAaC;CACF"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.js"],"names":[],"mappings":"AAKA;IACE,0BASC;CACF;AAED;IACE,0BAeC;IAZC,YAAiB;IAcnB,0BAEC;IAED,0BAEC;IAED,iCAEC;IAED,6BAOC;CACF"}

package/types/openai.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.js"],"names":[],"mappings":"AAOA;IAGI,eAEE;IAGJ;;;OAGG;IACH,4BAGC;IAED,0CAsBC;IAED;;;MAkBC;CACF;uBAjEsB,iBAAiB;mBAFrB,QAAQ"}
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../src/openai.js"],"names":[],"mappings":"AAOA;IAGI,eAEE;IAGJ;;;OAGG;IACH,4BAGC;IAED,0CAyBC;IAED;;;MAkBC;CACF;uBApEsB,iBAAiB;mBAFrB,QAAQ"}

package/types/util.d.ts

@@ -1,4 +1,4 @@
 export function loadTemplates(dir: any): Promise<{}>;
-export function parse(content: any): any;
+export function loadTemplate(file: any): Promise<string>;
 export function transformResponse(options: any): any;
 //# sourceMappingURL=util.d.ts.map

package/types/util.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../src/util.js"],"names":[],"mappings":"AAQA,qDAUC;AAED,yCAOC;AAED,qDAYC"}
+{"version":3,"file":"util.d.ts","sourceRoot":"","sources":["../src/util.js"],"names":[],"mappings":"AASA,qDAcC;AAED,yDAEC;AAED,qDAcC"}

package/types/xai.d.ts

@@ -0,0 +1,4 @@
+export class XAiClient extends OpenAiClient {
+}
+import { OpenAiClient } from './openai.js';
+//# sourceMappingURL=xai.d.ts.map

package/types/xai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"xai.d.ts","sourceRoot":"","sources":["../src/xai.js"],"names":[],"mappings":"AAIA;CAcC;6BAlB4B,aAAa"}