33 lines
996 B
TypeScript
33 lines
996 B
TypeScript
|
import process from "node:process"
|
||
|
|
|
||
|
|
const DEFAULT_ANTHROPIC_ACTUAL_LIMIT = 200_000
|
||
|
|
|
||
|
|
export type ContextLimitModelCacheState = {
|
||
|
|
anthropicContext1MEnabled: boolean
|
||
|
|
modelContextLimitsCache?: Map<string, number>
|
||
|
|
}
|
||
|
|
|
||
|
|
function isAnthropicProvider(providerID: string): boolean {
|
||
|
|
return providerID.toLowerCase().includes("anthropic")
|
||
|
|
}
|
||
|
|
|
||
|
|
function getAnthropicActualLimit(modelCacheState?: ContextLimitModelCacheState): number {
|
||
|
|
return (modelCacheState?.anthropicContext1MEnabled ?? false) ||
|
||
|
|
process.env.ANTHROPIC_1M_CONTEXT === "true" ||
|
||
|
|
process.env.VERTEX_ANTHROPIC_1M_CONTEXT === "true"
|
||
|
|
? 1_000_000
|
||
|
|
: DEFAULT_ANTHROPIC_ACTUAL_LIMIT
|
||
|
|
}
|
||
|
|
|
||
|
|
export function resolveActualContextLimit(
|
||
|
|
providerID: string,
|
||
|
|
modelID: string,
|
||
|
|
modelCacheState?: ContextLimitModelCacheState,
|
||
|
|
): number | null {
|
||
|
|
if (isAnthropicProvider(providerID)) {
|
||
|
|
return getAnthropicActualLimit(modelCacheState)
|
||
|
|
}
|
||
|
|
|
||
|
|
return modelCacheState?.modelContextLimitsCache?.get(`${providerID}/${modelID}`) ?? null
|
||
|
|
}
|