@anaplian/model-context-size
Version:
Anaplian support library for fetching the context window size for various models
19 lines • 1.46 kB
JavaScript
;
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==