@bedrockio/ai 0.3.0 → 0.4.4

package/src/openai.js

@@ -1,71 +0,0 @@
-import OpenAI from 'openai';
-
-import BaseClient from './BaseClient.js';
-import { transformResponse } from './util.js';
-
-const DEFAULT_MODEL = 'gpt-4o-mini';
-
-export class OpenAiClient extends BaseClient {
-  constructor(options) {
-    super(options);
-    this.client = new OpenAI({
-      ...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);
-  }
-
-  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,
-      response_format: {
-        type: output === 'json' ? 'json_object' : 'text',
-      },
-    });
-
-    if (output === 'raw') {
-      return response;
-    }
-
-    const { message } = response.choices[0];
-
-    return 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 || '',
-      };
-    }
-  }
-}

package/src/util.js

@@ -1,60 +0,0 @@
-import fs from 'fs/promises';
-
-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 loadTemplate(file);
-  }
-
-  return result;
-}
-
-export async function loadTemplate(file) {
-  return await fs.readFile(file, 'utf-8');
-}
-
-export 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 parseJson(content);
-  } else if (output === 'code') {
-    return parseCode(content);
-  } else {
-    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

@@ -1,19 +0,0 @@
-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,
-    });
-  }
-}