chuan/Cli-Proxy-API-Management-Center
1
0
Fork 0
mirror of https://github.com/router-for-me/Cli-Proxy-API-Management-Center.git synced 2026-02-19 11:10:49 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(quota): add normalization for Gemini CLI model IDs and update quota groups

Browse Source
This commit is contained in:
LTbinglingfeng
2026-02-06 19:11:57 +08:00
parent 3661530f5f
commit cade2647d6
4 changed files with 63 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
src/utils/quota/parsers.ts
View File

@@ -4,6 +4,8 @@
import type { CodexUsagePayload, GeminiCliQuotaPayload } from '@/types';
const GEMINI_CLI_MODEL_SUFFIX = '_vertex';
export function normalizeAuthIndexValue(value: unknown): string | null {
if (typeof value === 'number' && Number.isFinite(value)) {
return value.toString();
@@ -26,6 +28,15 @@ export function normalizeStringValue(value: unknown): string | null {
return null;
}
export function normalizeGeminiCliModelId(value: unknown): string | null {
const modelId = normalizeStringValue(value);
if (!modelId) return null;
if (modelId.endsWith(GEMINI_CLI_MODEL_SUFFIX)) {
return modelId.slice(0, -GEMINI_CLI_MODEL_SUFFIX.length);
}
return modelId;
}
export function normalizeNumberValue(value: unknown): number | null {
if (typeof value === 'number' && Number.isFinite(value)) return value;
if (typeof value === 'string') {