mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-04-24 12:04:56 -07:00
Jacob Richman
30 lines
652 B
TypeScript
30 lines
652 B
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
import { tokenLimit } from '@google/gemini-cli-core';
|
|
|
|
export function getContextUsagePercentage(
|
|
promptTokenCount: number,
|
|
model: string | undefined,
|
|
): number {
|
|
if (!model || typeof model !== 'string' || model.length === 0) {
|
|
return 0;
|
|
}
|
|
const limit = tokenLimit(model);
|
|
if (limit <= 0) {
|
|
return 0;
|
|
}
|
|
return promptTokenCount / limit;
|
|
}
|
|
|
|
export function isContextUsageHigh(
|
|
promptTokenCount: number,
|
|
model: string | undefined,
|
|
threshold = 0.6,
|
|
): boolean {
|
|
return getContextUsagePercentage(promptTokenCount, model) > threshold;
|
|
}
|