Files
gemini-cli/packages/core/src/utils/modelUtils.test.ts

33 lines
892 B
TypeScript

/**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { describe, it, expect } from 'vitest';
import { normalizeModelId } from './modelUtils.js';
describe('modelUtils', () => {
describe('normalizeModelId', () => {
it('should strip "models/" prefix if present', () => {
expect(normalizeModelId('models/gemini-3.1-pro-preview')).toBe(
'gemini-3.1-pro-preview',
);
expect(normalizeModelId('models/gemini-1.5-flash')).toBe(
'gemini-1.5-flash',
);
});
it('should leave model ID untouched if prefix is not present', () => {
expect(normalizeModelId('gemini-3.1-pro-preview')).toBe(
'gemini-3.1-pro-preview',
);
expect(normalizeModelId('auto')).toBe('auto');
});
it('should handle empty string', () => {
expect(normalizeModelId('')).toBe('');
});
});
});