UNPKG

@anaplian/model-context-size

Version:

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

14 lines 1.19 kB
import { modelContextSizes } from './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. */ export const isModelSupported = (modelName) => new Set(Object.keys(modelContextSizes)).has(modelName); /** * Gets the context window size in tokens for the model. * * @param modelName - The supported model name as used in LangChain.js. */ export const getModelContextSize = (modelName) => modelContextSizes[modelName]; //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiZ2V0LW1vZGVsLWNvbnRleHQtc2l6ZS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIi4uLy4uL3NyYy9nZXQtbW9kZWwtY29udGV4dC1zaXplLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiJBQUFBLE9BQU8sRUFBRSxpQkFBaUIsRUFBRSxNQUFNLHVCQUF1QixDQUFDO0FBSTFEOzs7O0dBSUc7QUFDSCxNQUFNLENBQUMsTUFBTSxnQkFBZ0IsR0FBRyxDQUM5QixTQUFpQixFQUNnQixFQUFFLENBQ25DLElBQUksR0FBRyxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsaUJBQWlCLENBQUMsQ0FBQyxDQUFDLEdBQUcsQ0FBQyxTQUFTLENBQUMsQ0FBQztBQUV6RDs7OztHQUlHO0FBQ0gsTUFBTSxDQUFDLE1BQU0sbUJBQW1CLEdBQUcsQ0FBQyxTQUE2QixFQUFVLEVBQUUsQ0FDM0UsaUJBQWlCLENBQUMsU0FBUyxDQUFDLENBQUMifQ==