genkitx-aws-bedrock
Version:
Firebase Genkit AI framework plugin for AWS Bedrock APIs.
91 lines • 10.3 kB
JavaScript
;
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.cohereEmbedMultilingualV3 = exports.cohereEmbedEnglishV3 = exports.amazonTitanEmbedTextG1V1 = exports.amazonTitanEmbedMultimodalV2 = exports.amazonTitanEmbedTextV2 = exports.anthropicClaudeInstantV1 = exports.anthropicClaude2V1 = exports.anthropicClaude21V1 = exports.anthropicClaude3SonnetV1 = exports.anthropicClaude3HaikuV1 = exports.anthropicClaude3OpusV1 = exports.anthropicClaude35SonnetV2 = exports.anthropicClaude37SonnetV1 = exports.anthropicClaude35HaikuV1 = exports.metaLlama38BInstruct = exports.metaLlama370BInstruct = exports.metaLlama318BInstruct = exports.metaLlama3170BInstruct = exports.metaLlama3290BInstruct = exports.metaLlama323BInstruct = exports.metaLlama321BInstruct = exports.metaLlama3211BInstruct = exports.metaLlama3370BInstruct = exports.ai21Jurassic2UltraV1 = exports.ai21Jurassic2MidV1 = exports.ai21JambaInstructV1 = exports.ai21Jamba15MiniV1 = exports.ai21Jamba15LargeV1 = exports.mistral8x7BInstructV01 = exports.mistral7BInstructV02 = exports.mistralSmall2402V1 = exports.mistralLarge2402V1 = exports.cohereCommandV14 = exports.cohereCommandLightV14 = exports.cohereCommandRPlusV1 = exports.cohereCommandRV1 = exports.amazonTitanTextG1LiteV1 = exports.amazonTitanTextG1ExpressV1 = exports.amazonTitanTextG1PremierV1 = exports.amazonNovaMicroV1 = exports.amazonNovaLiteV1 = exports.amazonNovaProV1 = void 0;
exports.awsBedrock = awsBedrock;
const plugin_1 = require("genkit/plugin");
const client_bedrock_runtime_1 = require("@aws-sdk/client-bedrock-runtime");
const aws_bedrock_llms_js_1 = require("./aws_bedrock_llms.js");
Object.defineProperty(exports, "amazonNovaProV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.amazonNovaProV1; } });
Object.defineProperty(exports, "amazonNovaLiteV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.amazonNovaLiteV1; } });
Object.defineProperty(exports, "amazonNovaMicroV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.amazonNovaMicroV1; } });
Object.defineProperty(exports, "amazonTitanTextG1PremierV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.amazonTitanTextG1PremierV1; } });
Object.defineProperty(exports, "amazonTitanTextG1ExpressV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.amazonTitanTextG1ExpressV1; } });
Object.defineProperty(exports, "amazonTitanTextG1LiteV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.amazonTitanTextG1LiteV1; } });
Object.defineProperty(exports, "cohereCommandRV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.cohereCommandRV1; } });
Object.defineProperty(exports, "cohereCommandRPlusV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.cohereCommandRPlusV1; } });
Object.defineProperty(exports, "cohereCommandLightV14", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.cohereCommandLightV14; } });
Object.defineProperty(exports, "cohereCommandV14", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.cohereCommandV14; } });
Object.defineProperty(exports, "mistralLarge2402V1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.mistralLarge2402V1; } });
Object.defineProperty(exports, "mistralSmall2402V1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.mistralSmall2402V1; } });
Object.defineProperty(exports, "mistral7BInstructV02", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.mistral7BInstructV02; } });
Object.defineProperty(exports, "mistral8x7BInstructV01", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.mistral8x7BInstructV01; } });
Object.defineProperty(exports, "anthropicClaude35HaikuV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude35HaikuV1; } });
Object.defineProperty(exports, "anthropicClaude37SonnetV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude37SonnetV1; } });
Object.defineProperty(exports, "anthropicClaude35SonnetV2", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude35SonnetV2; } });
Object.defineProperty(exports, "anthropicClaude3OpusV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude3OpusV1; } });
Object.defineProperty(exports, "anthropicClaude3HaikuV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude3HaikuV1; } });
Object.defineProperty(exports, "anthropicClaude3SonnetV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude3SonnetV1; } });
Object.defineProperty(exports, "anthropicClaude21V1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude21V1; } });
Object.defineProperty(exports, "anthropicClaude2V1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaude2V1; } });
Object.defineProperty(exports, "anthropicClaudeInstantV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.anthropicClaudeInstantV1; } });
Object.defineProperty(exports, "ai21Jamba15LargeV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.ai21Jamba15LargeV1; } });
Object.defineProperty(exports, "ai21Jamba15MiniV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.ai21Jamba15MiniV1; } });
Object.defineProperty(exports, "ai21JambaInstructV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.ai21JambaInstructV1; } });
Object.defineProperty(exports, "ai21Jurassic2MidV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.ai21Jurassic2MidV1; } });
Object.defineProperty(exports, "ai21Jurassic2UltraV1", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.ai21Jurassic2UltraV1; } });
Object.defineProperty(exports, "metaLlama3370BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama3370BInstruct; } });
Object.defineProperty(exports, "metaLlama3211BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama3211BInstruct; } });
Object.defineProperty(exports, "metaLlama321BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama321BInstruct; } });
Object.defineProperty(exports, "metaLlama323BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama323BInstruct; } });
Object.defineProperty(exports, "metaLlama3290BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama3290BInstruct; } });
Object.defineProperty(exports, "metaLlama3170BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama3170BInstruct; } });
Object.defineProperty(exports, "metaLlama318BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama318BInstruct; } });
Object.defineProperty(exports, "metaLlama370BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama370BInstruct; } });
Object.defineProperty(exports, "metaLlama38BInstruct", { enumerable: true, get: function () { return aws_bedrock_llms_js_1.metaLlama38BInstruct; } });
const aws_bedrock_embedders_js_1 = require("./aws_bedrock_embedders.js");
Object.defineProperty(exports, "amazonTitanEmbedTextV2", { enumerable: true, get: function () { return aws_bedrock_embedders_js_1.amazonTitanEmbedTextV2; } });
Object.defineProperty(exports, "amazonTitanEmbedMultimodalV2", { enumerable: true, get: function () { return aws_bedrock_embedders_js_1.amazonTitanEmbedMultimodalV2; } });
Object.defineProperty(exports, "amazonTitanEmbedTextG1V1", { enumerable: true, get: function () { return aws_bedrock_embedders_js_1.amazonTitanEmbedTextG1V1; } });
Object.defineProperty(exports, "cohereEmbedEnglishV3", { enumerable: true, get: function () { return aws_bedrock_embedders_js_1.cohereEmbedEnglishV3; } });
Object.defineProperty(exports, "cohereEmbedMultilingualV3", { enumerable: true, get: function () { return aws_bedrock_embedders_js_1.cohereEmbedMultilingualV3; } });
function awsBedrock(options) {
return (0, plugin_1.genkitPlugin)("aws-bedrock", (ai) => __awaiter(this, void 0, void 0, function* () {
const client = new client_bedrock_runtime_1.BedrockRuntimeClient(options || {});
const region = typeof (options === null || options === void 0 ? void 0 : options.region) === "string"
? options.region
: typeof (options === null || options === void 0 ? void 0 : options.region) === "function"
? yield options.region()
: undefined;
let inferenceRegion = '';
if (!region) {
inferenceRegion = "us"; // default to us
}
else if (region.includes('us-gov')) {
inferenceRegion = "us-gov";
}
else if (region.includes('us')) {
inferenceRegion = "us";
}
else if (region.includes('eu-')) {
inferenceRegion = "eu";
}
else if (region.includes('ap-')) {
inferenceRegion = "apac";
}
Object.keys((0, aws_bedrock_llms_js_1.SUPPORTED_AWS_BEDROCK_MODELS)(inferenceRegion)).forEach((name) => {
(0, aws_bedrock_llms_js_1.awsBedrockModel)(name, client, ai, inferenceRegion);
});
Object.keys(aws_bedrock_embedders_js_1.SUPPORTED_EMBEDDING_MODELS).forEach((name) => (0, aws_bedrock_embedders_js_1.awsBedrockEmbedder)(name, ai, client));
}));
}
exports.default = awsBedrock;
//# sourceMappingURL=index.js.map