@bedrockio/ai 0.1.0 → 0.2.0

package/CHANGELOG.md

@@ -1,3 +1,7 @@
+## 0.2.0
+
+- Added Gemini
+
 ## 0.1.0
 
 - Initial commit

package/README.md

@@ -6,6 +6,10 @@ usage.
 
 - [Install](#install)
 - [Usage](#usage)
+- [Streaming](#stream)
+- [Templates](#templates)
+- [Platforms](#platforms)
+- [Models](#models)
 
 ## Install
 
@@ -42,26 +46,28 @@ const response = await client.prompt({
   text: 'a long yellow fruit',
   fruit: 'banana, apple, pear',
 });
+```
+
+## Streaming
 
+Responses may be streamed:
+
+```js
 // Stream the results
 const stream = await client.stream({
   file: 'classify-fruits',
-  // ...
 });
 
 // Will return an AsyncIterator
 for await (const chunk of stream) {
   console.info(chunk.text);
 }
-
-// List available models
-const models = await client.models();
 ```
 
 ## Templates
 
-Template files must have be markdown (`.md`) and live in your templates
-directory. They may be a simple text description or delineated roles:
+Template files must be markdown (`.md`) and live in your templates directory.
+They may be a simple text prompt or delineated roles:
 
 ````
 --- SYSTEM ---
@@ -82,3 +88,19 @@ Please provide your response as a JSON object containing:
 {{text}}
 ```
 ````
+
+## Platforms
+
+Currently supported platforms:
+
+- OpenAI (ChatGPT)
+- Anthropic (Claude)
+- Google (Gemini).
+
+## Models
+
+Available models can be listed with:
+
+```js
+const models = await client.models();
+```

package/__mocks__/@anthropic-ai/sdk.js

@@ -1,6 +1,6 @@
 let mock;
 
-function Anthropic() {
+function MockAnthropicClient() {
   return {
     messages: {
       create(options) {
@@ -38,6 +38,6 @@ function wrapChunk(str, type) {
   };
 }
 
-Anthropic.setResponse = setResponse;
+MockAnthropicClient.setResponse = setResponse;
 
-module.exports = Anthropic;
+module.exports = MockAnthropicClient;

package/__mocks__/@google/generative-ai.js

@@ -0,0 +1,59 @@
+let mock;
+
+class MockGoogleClient {
+  constructor() {
+    return {
+      getGenerativeModel() {
+        return {
+          generateContent() {
+            return mock;
+          },
+          generateContentStream() {
+            return {
+              stream: streamMock(),
+            };
+          },
+        };
+      },
+    };
+  }
+}
+
+function setResponse(data) {
+  mock = data;
+}
+
+async function* streamMock() {
+  const content = mock.response.candidates[0].content.parts[0].text;
+  const size = Math.floor(content.length / 3);
+  const one = content.slice(0, size);
+  const two = content.slice(size, 2 * size);
+  const three = content.slice(2 * size);
+  yield wrapChunk(one);
+  yield wrapChunk(two);
+  yield wrapChunk(three, true);
+}
+
+function wrapChunk(str, finish) {
+  return {
+    candidates: [
+      {
+        ...(finish && {
+          finishReason: 'STOP',
+        }),
+        content: {
+          parts: [
+            {
+              text: str,
+            },
+          ],
+        },
+      },
+    ],
+  };
+}
+
+module.exports = {
+  setResponse,
+  GoogleGenerativeAI: MockGoogleClient,
+};

package/__mocks__/openai.js

@@ -1,6 +1,6 @@
 let mock;
 
-function OpenAI() {
+function MockOpenAiClient() {
   return {
     chat: {
       completions: {
@@ -43,6 +43,6 @@ function setResponse(data) {
   mock = data;
 }
 
-OpenAI.setResponse = setResponse;
+MockOpenAiClient.setResponse = setResponse;
 
-module.exports = OpenAI;
+module.exports = MockOpenAiClient;

package/dist/cjs/BaseClient.js

@@ -0,0 +1,130 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.default = void 0;
+var _mustache = _interopRequireDefault(require("mustache"));
+var _util = require("./util.js");
+function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+const MESSAGES_REG = /(?:^|\n)-{3,}\s*(\w+)\s*-{3,}(.*?)(?=\n-{3,}|$)/gs;
+class BaseClient {
+  constructor(options) {
+    this.options = options;
+    this.templates = null;
+  }
+
+  /**
+   * Interpolates vars into the provided template and
+   * runs the chat completion. The "output" option may
+   * be omitted and will default to `"text"`.
+   * {@link https://github.com/bedrockio/ai?tab=readme-ov-file#bedrockioai Documentation}
+   *
+   * @param {object} options
+   * @param {string} options.model - The model to use.
+   * @param {"raw" | "text" | "json" | "messages"} [options.output] - The output to use.
+   * @param {Object.<string, any>} [options.other] - Additional props
+   *             will be interpolated in the template.
+   */
+  async prompt(options) {
+    options = {
+      ...this.options,
+      ...options
+    };
+    const messages = await this.getMessages(options);
+    return await this.getCompletion({
+      ...options,
+      messages
+    });
+  }
+
+  /**
+   * Streams the prompt response.
+   * @returns {AsyncIterator}
+   */
+  async *stream(options) {
+    const stream = await this.getStream(options);
+    let started = false;
+
+    // @ts-ignore
+    for await (const chunk of stream) {
+      const resolved = this.getStreamedChunk(chunk, started);
+      started = true;
+
+      // @ts-ignore
+      if (resolved) {
+        yield resolved;
+      }
+    }
+  }
+  async getMessages(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({
+        role: 'user',
+        content: raw.trim()
+      });
+    }
+    return messages;
+  }
+  async loadTemplates() {
+    const {
+      templates
+    } = this.options;
+    this.templates ||= await (0, _util.loadTemplates)(templates);
+  }
+  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.');
+    }
+    return template;
+  }
+  async getStream(options) {
+    return await this.prompt({
+      ...options,
+      output: 'raw',
+      stream: true
+    });
+  }
+  getCompletion(options) {
+    void options;
+    new Error('Method not implemented.');
+  }
+  getStreamedChunk(chunk, started) {
+    void chunk;
+    void started;
+    new Error('Method not implemented.');
+  }
+}
+exports.default = BaseClient;
+function transformParams(params) {
+  const result = {};
+  for (let [key, value] of Object.entries(params)) {
+    if (Array.isArray(value)) {
+      value = value.map(el => {
+        return `- ${el}`;
+      }).join('\n');
+    } else if (typeof value === 'object') {
+      value = JSON.stringify(value, null, 2);
+    }
+    result[key] = value;
+  }
+  return result;
+}

package/dist/cjs/anthropic.js

@@ -1,19 +1,101 @@
 "use strict";
 
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.AnthropicClient = void 0;
 var _sdk = _interopRequireDefault(require("@anthropic-ai/sdk"));
+var _BaseClient = _interopRequireDefault(require("./BaseClient.js"));
+var _util = require("./util.js");
 function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
-const client = new _sdk.default({
-  apiKey: process.env['ANTHROPIC_API_KEY'] // This is the default and can be omitted
-});
-async function main() {
-  const message = await client.messages.create({
-    max_tokens: 1024,
-    messages: [{
-      role: 'user',
-      content: 'Hello, Claude'
-    }],
-    model: 'claude-3-5-sonnet-latest'
-  });
-  console.log(message.content);
+const MODELS_URL = 'https://docs.anthropic.com/en/docs/about-claude/models';
+const DEFAULT_MODEL = 'claude-3-5-sonnet-latest';
+class AnthropicClient extends _BaseClient.default {
+  constructor(options) {
+    super(options);
+    this.client = new _sdk.default({
+      ...options
+    });
+  }
+
+  /**
+   * Lists available models.
+   * {@link https://docs.anthropic.com/en/docs/about-claude/models Documentation}
+   */
+  async models() {
+    const {
+      data
+    } = await this.client.models.list();
+    return data.map(o => o.id);
+  }
+  async getCompletion(options) {
+    const {
+      model = DEFAULT_MODEL,
+      max_tokens = 2048,
+      output = 'text',
+      stream = false,
+      messages
+    } = options;
+    const {
+      client
+    } = this;
+    const {
+      system,
+      user
+    } = splitMessages(messages);
+    if (!model) {
+      throw new Error(`No model specified. Available models are here: ${MODELS_URL}.`);
+    }
+    const response = await client.messages.create({
+      max_tokens,
+      messages: user,
+      system,
+      model,
+      stream
+    });
+    if (output === 'raw') {
+      return response;
+    }
+
+    // @ts-ignore
+    const message = response.content[0];
+    return (0, _util.transformResponse)({
+      ...options,
+      messages,
+      message
+    });
+  }
+  getStreamedChunk(chunk) {
+    // @ts-ignore
+    let type;
+    if (chunk.type === 'content_block_start') {
+      type = 'start';
+    } else if (chunk.type === 'content_block_delta') {
+      type = 'chunk';
+    } else if (chunk.type === 'message_stop') {
+      type = 'stop';
+    }
+    if (type) {
+      return {
+        type,
+        text: chunk.delta?.text || ''
+      };
+    }
+  }
 }
-main();
+exports.AnthropicClient = AnthropicClient;
+function splitMessages(messages) {
+  const system = [];
+  const user = [];
+  for (let message of messages) {
+    if (message.role === 'system') {
+      system.push(message);
+    } else {
+      user.push(message);
+    }
+  }
+  return {
+    system: system.join('\n'),
+    user
+  };
+}

package/dist/cjs/google.js

@@ -0,0 +1,94 @@
+"use strict";
+
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.GoogleClient = void 0;
+var _generativeAi = require("@google/generative-ai");
+var _BaseClient = _interopRequireDefault(require("./BaseClient.js"));
+var _util = require("./util.js");
+function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
+const DEFAULT_MODEL = 'models/gemini-2.0-flash-exp';
+class GoogleClient extends _BaseClient.default {
+  constructor(options) {
+    super(options);
+    const {
+      apiKey
+    } = options;
+    this.client = new _generativeAi.GoogleGenerativeAI(apiKey);
+  }
+
+  /**
+   * Lists available models.
+   * {@link https://ai.google.dev/gemini-api/docs/models/gemini#gemini-2.0-flashl Documentation}
+   */
+  async models() {
+    return ['gemini-2.0-flash-exp', 'gemini-1.5-flash', 'gemini-1.5-flash-8b', 'gemini-1.5-pro'];
+  }
+  async getCompletion(options) {
+    const {
+      model = DEFAULT_MODEL,
+      output = 'text',
+      stream = false
+    } = options;
+    const {
+      client
+    } = this;
+    const generator = client.getGenerativeModel({
+      model
+    });
+    const messages = await this.getMessages(options);
+    const prompts = messages.map(message => {
+      return message.content;
+    });
+    let response;
+    if (stream) {
+      response = await generator.generateContentStream(prompts);
+    } else {
+      response = await generator.generateContent(prompts);
+    }
+    // const response = await client.chat.completions.create({
+    //   model,
+    //   messages,
+    //   stream,
+    // });
+
+    if (output === 'raw') {
+      return response;
+    }
+
+    // @ts-ignore
+    const parts = response.response.candidates.flatMap(candidate => {
+      return candidate.content.parts;
+    });
+    const [message] = parts;
+    return (0, _util.transformResponse)({
+      ...options,
+      messages,
+      message
+    });
+  }
+  async getStream(options) {
+    const response = await super.getStream(options);
+    // @ts-ignore
+    return response.stream;
+  }
+  getStreamedChunk(chunk, started) {
+    const [candidate] = chunk.candidates;
+    let type;
+    if (!started) {
+      type = 'start';
+    } else if (candidate.finishReason === 'STOP') {
+      type = 'stop';
+    } else {
+      type = 'chunk';
+    }
+    if (type) {
+      return {
+        type,
+        text: candidate.content.parts[0].text || ''
+      };
+    }
+  }
+}
+exports.GoogleClient = GoogleClient;

package/dist/cjs/index.js

@@ -1,3 +1,29 @@
 "use strict";
 
-console.info('ohai');
+Object.defineProperty(exports, "__esModule", {
+  value: true
+});
+exports.Client = void 0;
+var _openai = require("./openai.js");
+var _google = require("./google.js");
+var _anthropic = require("./anthropic.js");
+class Client {
+  constructor(options) {
+    const {
+      platform,
+      ...rest
+    } = 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.');
+    }
+  }
+}
+exports.Client = Client;

package/dist/cjs/openai.js

@@ -5,45 +5,71 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.OpenAiClient = void 0;
 var _openai = _interopRequireDefault(require("openai"));
-var _Wrapper = _interopRequireDefault(require("./Wrapper.js"));
+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';
-class OpenAiClient extends _Wrapper.default {
+class OpenAiClient extends _BaseClient.default {
   constructor(options) {
     super(options);
     this.client = new _openai.default({
       ...options
     });
   }
-  async prompt(options) {
-    const messages = await this.getMessages(options);
-    return await runCompletion(this.client, messages, options);
+
+  /**
+   * Lists available models.
+   * {@link https://platform.openai.com/docs/models Documentation}
+   */
+  async models() {
+    const {
+      data
+    } = await this.client.models.list();
+    return data.map(o => o.id);
   }
-}
-exports.OpenAiClient = OpenAiClient;
-async function runCompletion(client, messages, options) {
-  const {
-    output = 'text',
-    model = DEFAULT_MODEL
-  } = options;
-  const response = await client.chat.completions.create({
-    model,
-    messages
-  });
-  let content = response.choices[0].message.content;
-  if (output === 'raw') {
-    return response;
-  } else if (output === 'text') {
-    return content;
-  } else if (output === 'messages') {
+  async getCompletion(options) {
+    const {
+      model = DEFAULT_MODEL,
+      output = 'text',
+      stream = false
+    } = options;
+    const {
+      client
+    } = this;
+    const messages = await this.getMessages(options);
+    const response = await client.chat.completions.create({
+      model,
+      messages,
+      stream
+    });
+    if (output === 'raw') {
+      return response;
+    }
     const {
       message
     } = response.choices[0];
-    return [...messages, message];
-  } else if (output === 'json') {
-    return (0, _util.parse)(content);
-  } else {
-    throw new Error(`Unknown output type "${output}".`);
+    return (0, _util.transformResponse)({
+      ...options,
+      messages,
+      message
+    });
+  }
+  getStreamedChunk(chunk, started) {
+    const [choice] = chunk.choices;
+    let type;
+    if (!started) {
+      type = 'start';
+    } else if (choice.finish_reason === 'stop') {
+      type = 'stop';
+    } else {
+      type = 'chunk';
+    }
+    if (type) {
+      return {
+        type,
+        text: choice.delta.content || ''
+      };
+    }
   }
-}
+}
+exports.OpenAiClient = OpenAiClient;

package/dist/cjs/util.js

@@ -5,6 +5,7 @@ Object.defineProperty(exports, "__esModule", {
 });
 exports.loadTemplates = loadTemplates;
 exports.parse = parse;
+exports.transformResponse = transformResponse;
 var _promises = _interopRequireDefault(require("fs/promises"));
 var _path = _interopRequireDefault(require("path"));
 var _glob = require("glob");
@@ -26,4 +27,21 @@ function parse(content) {
   } catch (error) {
     throw new Error('Unable to derive JSON object in response.');
   }
+}
+function transformResponse(options) {
+  const {
+    output = 'text',
+    messages,
+    message
+  } = options;
+  const content = message.content || message.text;
+  if (output === 'text') {
+    return content;
+  } else if (output === 'messages') {
+    return [...messages, message];
+  } else if (output === 'json') {
+    return parse(content);
+  } else {
+    throw new Error(`Unknown output type "${output}".`);
+  }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@bedrockio/ai",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "description": "Bedrock wrapper for common AI chatbots.",
   "type": "module",
   "scripts": {
@@ -26,6 +26,7 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.33.1",
+    "@google/generative-ai": "^0.21.0",
     "glob": "^11.0.1",
     "mustache": "^4.2.0",
     "openai": "^4.79.1"

package/src/BaseClient.js

@@ -40,11 +40,7 @@ export default class BaseClient {
    * @returns {AsyncIterator}
    */
   async *stream(options) {
-    const stream = await this.prompt({
-      ...options,
-      output: 'raw',
-      stream: true,
-    });
+    const stream = await this.getStream(options);
 
     let started = false;
 
@@ -104,6 +100,14 @@ export default class BaseClient {
     return template;
   }
 
+  async getStream(options) {
+    return await this.prompt({
+      ...options,
+      output: 'raw',
+      stream: true,
+    });
+  }
+
   getCompletion(options) {
     void options;
     new Error('Method not implemented.');

package/src/anthropic.js

@@ -16,6 +16,7 @@ export class AnthropicClient extends BaseClient {
 
   /**
    * Lists available models.
+   * {@link https://docs.anthropic.com/en/docs/about-claude/models Documentation}
    */
   async models() {
     const { data } = await this.client.models.list();

package/src/google.js

@@ -0,0 +1,94 @@
+import { GoogleGenerativeAI } from '@google/generative-ai';
+
+import BaseClient from './BaseClient.js';
+import { transformResponse } from './util.js';
+
+const DEFAULT_MODEL = 'models/gemini-2.0-flash-exp';
+
+export class GoogleClient extends BaseClient {
+  constructor(options) {
+    super(options);
+    const { apiKey } = options;
+    this.client = new GoogleGenerativeAI(apiKey);
+  }
+
+  /**
+   * Lists available models.
+   * {@link https://ai.google.dev/gemini-api/docs/models/gemini#gemini-2.0-flashl Documentation}
+   */
+  async models() {
+    return [
+      'gemini-2.0-flash-exp',
+      'gemini-1.5-flash',
+      'gemini-1.5-flash-8b',
+      'gemini-1.5-pro',
+    ];
+  }
+
+  async getCompletion(options) {
+    const { model = DEFAULT_MODEL, output = 'text', stream = false } = options;
+    const { client } = this;
+
+    const generator = client.getGenerativeModel({ model });
+
+    const messages = await this.getMessages(options);
+
+    const prompts = messages.map((message) => {
+      return message.content;
+    });
+
+    let response;
+
+    if (stream) {
+      response = await generator.generateContentStream(prompts);
+    } else {
+      response = await generator.generateContent(prompts);
+    }
+    // const response = await client.chat.completions.create({
+    //   model,
+    //   messages,
+    //   stream,
+    // });
+
+    if (output === 'raw') {
+      return response;
+    }
+
+    // @ts-ignore
+    const parts = response.response.candidates.flatMap((candidate) => {
+      return candidate.content.parts;
+    });
+    const [message] = parts;
+
+    return transformResponse({
+      ...options,
+      messages,
+      message,
+    });
+  }
+  async getStream(options) {
+    const response = await super.getStream(options);
+    // @ts-ignore
+    return response.stream;
+  }
+
+  getStreamedChunk(chunk, started) {
+    const [candidate] = chunk.candidates;
+
+    let type;
+    if (!started) {
+      type = 'start';
+    } else if (candidate.finishReason === 'STOP') {
+      type = 'stop';
+    } else {
+      type = 'chunk';
+    }
+
+    if (type) {
+      return {
+        type,
+        text: candidate.content.parts[0].text || '',
+      };
+    }
+  }
+}

package/src/index.js

@@ -1,4 +1,5 @@
 import { OpenAiClient } from './openai.js';
+import { GoogleClient } from './google.js';
 import { AnthropicClient } from './anthropic.js';
 
 export class Client {
@@ -6,6 +7,8 @@ export class Client {
     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) {

package/src/openai.js

@@ -15,6 +15,7 @@ export class OpenAiClient extends BaseClient {
 
   /**
    * Lists available models.
+   * {@link https://platform.openai.com/docs/models Documentation}
    */
   async models() {
     const { data } = await this.client.models.list();

package/types/BaseClient.d.ts

@@ -0,0 +1,39 @@
+export default class BaseClient {
+    constructor(options: any);
+    options: any;
+    templates: any;
+    /**
+     * Interpolates vars into the provided template and
+     * runs the chat completion. The "output" option may
+     * be omitted and will default to `"text"`.
+     * {@link https://github.com/bedrockio/ai?tab=readme-ov-file#bedrockioai Documentation}
+     *
+     * @param {object} options
+     * @param {string} options.model - The model to use.
+     * @param {"raw" | "text" | "json" | "messages"} [options.output] - The output to use.
+     * @param {Object.<string, any>} [options.other] - Additional props
+     *             will be interpolated in the template.
+     */
+    prompt(options: {
+        model: string;
+        output?: "raw" | "text" | "json" | "messages";
+        other?: {
+            [x: string]: any;
+        };
+    }): Promise<void>;
+    /**
+     * Streams the prompt response.
+     * @returns {AsyncIterator}
+     */
+    stream(options: any): AsyncIterator<any, any, any>;
+    getMessages(options: any): Promise<{
+        role: any;
+        content: any;
+    }[]>;
+    loadTemplates(): Promise<void>;
+    resolveTemplate(options: any): Promise<any>;
+    getStream(options: any): Promise<void>;
+    getCompletion(options: any): void;
+    getStreamedChunk(chunk: any, started: any): void;
+}
+//# sourceMappingURL=BaseClient.d.ts.map

package/types/BaseClient.d.ts.map

@@ -0,0 +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"}

package/types/anthropic.d.ts

@@ -0,0 +1,16 @@
+export class AnthropicClient extends BaseClient {
+    client: Anthropic;
+    /**
+     * Lists available models.
+     * {@link https://docs.anthropic.com/en/docs/about-claude/models Documentation}
+     */
+    models(): Promise<string[]>;
+    getCompletion(options: any): Promise<any>;
+    getStreamedChunk(chunk: any): {
+        type: string;
+        text: any;
+    };
+}
+import BaseClient from './BaseClient.js';
+import Anthropic from '@anthropic-ai/sdk';
+//# sourceMappingURL=anthropic.d.ts.map

package/types/anthropic.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.d.ts","sourceRoot":"","sources":["../src/anthropic.js"],"names":[],"mappings":"AAQA;IAGI,kBAEE;IAGJ;;;OAGG;IACH,4BAGC;IAED,0CAsCC;IAED;;;MAiBC;CACF;uBAjFsB,iBAAiB;sBAFlB,mBAAmB"}

package/types/google.d.ts

@@ -0,0 +1,17 @@
+export class GoogleClient extends BaseClient {
+    client: GoogleGenerativeAI;
+    /**
+     * Lists available models.
+     * {@link https://ai.google.dev/gemini-api/docs/models/gemini#gemini-2.0-flashl Documentation}
+     */
+    models(): Promise<string[]>;
+    getCompletion(options: any): Promise<any>;
+    getStream(options: any): Promise<any>;
+    getStreamedChunk(chunk: any, started: any): {
+        type: string;
+        text: any;
+    };
+}
+import BaseClient from './BaseClient.js';
+import { GoogleGenerativeAI } from '@google/generative-ai';
+//# sourceMappingURL=google.d.ts.map

package/types/google.d.ts.map

@@ -0,0 +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"}

package/types/index.d.ts

@@ -1 +1,4 @@
+export class Client {
+    constructor(options: 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":""}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.js"],"names":[],"mappings":"AAIA;IACE,0BAaC;CACF"}

package/types/openai.d.ts

@@ -0,0 +1,16 @@
+export class OpenAiClient extends BaseClient {
+    client: OpenAI;
+    /**
+     * Lists available models.
+     * {@link https://platform.openai.com/docs/models Documentation}
+     */
+    models(): Promise<string[]>;
+    getCompletion(options: any): Promise<any>;
+    getStreamedChunk(chunk: any, started: any): {
+        type: string;
+        text: any;
+    };
+}
+import BaseClient from './BaseClient.js';
+import OpenAI from 'openai';
+//# sourceMappingURL=openai.d.ts.map

package/types/openai.d.ts.map

@@ -0,0 +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"}

package/types/util.d.ts

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

package/types/util.d.ts.map

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

package/dist/cjs/Wrapper.js

@@ -1,63 +0,0 @@
-"use strict";
-
-Object.defineProperty(exports, "__esModule", {
-  value: true
-});
-exports.default = void 0;
-var _mustache = _interopRequireDefault(require("mustache"));
-var _util = require("./util.js");
-function _interopRequireDefault(e) { return e && e.__esModule ? e : { default: e }; }
-const MESSAGES_REG = /(?:^|\n)-{3,}\s*(\w+)\s*-{3,}(.*?)(?=\n-{3,}|$)/gs;
-class Wrapper {
-  constructor(options) {
-    this.options = options;
-  }
-  async getMessages(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()
-      });
-    }
-    return messages;
-  }
-  async loadTemplates() {
-    const {
-      templates
-    } = this.options;
-    this.templates ||= await (0, _util.loadTemplates)(templates);
-  }
-  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.');
-    }
-    return template;
-  }
-}
-exports.default = Wrapper;
-function transformParams(params) {
-  const result = {};
-  for (let [key, value] of Object.entries(params)) {
-    if (Array.isArray(value)) {
-      value = value.map(el => {
-        return `- ${el}`;
-      }).join('\n');
-    } else if (typeof value === 'object') {
-      value = JSON.stringify(value, null, 2);
-    }
-    result[key] = value;
-  }
-  return result;
-}