UNPKG

@anaplian/model-context-size

Version:

Anaplian support library for fetching the context window size for various models

19 lines 1.46 kB
"use strict"; Object.defineProperty(exports, "__esModule", { value: true }); exports.getModelContextSize = exports.isModelSupported = void 0; const model_context_sizes_1 = require("./model-context-sizes"); /** * A type guard that checks if a particular model's context size is available from this package. * * @param modelName - A model name as used in LangChain.js. */ const isModelSupported = (modelName) => new Set(Object.keys(model_context_sizes_1.modelContextSizes)).has(modelName); exports.isModelSupported = isModelSupported; /** * Gets the context window size in tokens for the model. * * @param modelName - The supported model name as used in LangChain.js. */ const getModelContextSize = (modelName) => model_context_sizes_1.modelContextSizes[modelName]; exports.getModelContextSize = getModelContextSize; //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZ2V0LW1vZGVsLWNvbnRleHQtc2l6ZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9nZXQtbW9kZWwtY29udGV4dC1zaXplLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7OztBQUFBLCtEQUEwRDtBQUkxRDs7OztHQUlHO0FBQ0ksTUFBTSxnQkFBZ0IsR0FBRyxDQUM5QixTQUFpQixFQUNnQixFQUFFLENBQ25DLElBQUksR0FBRyxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsdUNBQWlCLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQyxTQUFTLENBQUMsQ0FBQztBQUg1QyxRQUFBLGdCQUFnQixvQkFHNEI7QUFFekQ7Ozs7R0FJRztBQUNJLE1BQU0sbUJBQW1CLEdBQUcsQ0FBQyxTQUE2QixFQUFVLEVBQUUsQ0FDM0UsdUNBQWlCLENBQUMsU0FBUyxDQUFDLENBQUM7QUFEbEIsUUFBQSxtQkFBbUIsdUJBQ0QifQ==