mirror of
https://github.com/google-gemini/gemini-cli.git
synced 2026-04-21 18:44:30 -07:00
Aishanee Shah
34 lines
815 B
TypeScript
34 lines
815 B
TypeScript
/**
|
|
* @license
|
|
* Copyright 2026 Google LLC
|
|
* SPDX-License-Identifier: Apache-2.0
|
|
*/
|
|
|
|
/**
|
|
* Single source of truth for mapping model IDs to tool families.
|
|
*/
|
|
|
|
import { isGemini3Model } from '../../config/models.js';
|
|
import { type ToolFamily } from './types.js';
|
|
|
|
/**
|
|
* Resolves the ToolFamily for a given model ID.
|
|
* Defaults to 'default-legacy' if the model is not recognized or not provided.
|
|
*
|
|
* @param modelId The model identifier (e.g., 'gemini-2.5-pro', 'gemini-3-flash-preview')
|
|
* @returns The resolved ToolFamily
|
|
*/
|
|
export function getToolFamily(modelId?: string): ToolFamily {
|
|
if (!modelId) {
|
|
return 'default-legacy';
|
|
}
|
|
|
|
// Explicit mapping for Gemini 3 family
|
|
if (isGemini3Model(modelId)) {
|
|
return 'gemini-3';
|
|
}
|
|
|
|
// Fallback for all other models
|
|
return 'default-legacy';
|
|
}
|