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