diff --git a/.changeset/little-onions-know.md b/.changeset/little-onions-know.md new file mode 100644 index 0000000000000000000000000000000000000000..eaad25937fe22a040552980cd7a8450728490cab --- /dev/null +++ b/.changeset/little-onions-know.md @@ -0,0 +1,5 @@ +--- +"@llamaindex/google": patch +--- + +Add Gemini 2.0 models diff --git a/packages/providers/google/src/base.ts b/packages/providers/google/src/base.ts index 50937887f5e3b21eecd1e150ef3fc395f4ca7c1b..8ffd4718665a4deef4dd4d7875a001cd2f3eb578 100644 --- a/packages/providers/google/src/base.ts +++ b/packages/providers/google/src/base.ts @@ -54,6 +54,9 @@ export const GEMINI_MODEL_INFO_MAP: Record<GEMINI_MODEL, GeminiModelInfo> = { [GEMINI_MODEL.GEMINI_PRO_1_5_LATEST]: { contextWindow: 2 * 10 ** 6 }, [GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_LATEST]: { contextWindow: 10 ** 6 }, [GEMINI_MODEL.GEMINI_2_0_FLASH_EXPERIMENTAL]: { contextWindow: 10 ** 6 }, + [GEMINI_MODEL.GEMINI_2_0_FLASH]: { contextWindow: 10 ** 6 }, + [GEMINI_MODEL.GEMINI_2_0_FLASH_LITE_PREVIEW]: { contextWindow: 10 ** 6 }, + [GEMINI_MODEL.GEMINI_2_0_FLASH_THINKING_EXP]: { contextWindow: 32768 }, }; const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [ @@ -68,6 +71,7 @@ const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [ GEMINI_MODEL.GEMINI_PRO_1_5_LATEST, GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_LATEST, GEMINI_MODEL.GEMINI_2_0_FLASH_EXPERIMENTAL, + GEMINI_MODEL.GEMINI_2_0_FLASH, ]; const DEFAULT_GEMINI_PARAMS = { diff --git a/packages/providers/google/src/types.ts b/packages/providers/google/src/types.ts index 35a7074c47bca2c885fbc860fe326ffe3ed0d793..6d90ec6d17c1b02ae373de24a510abc17156b769 100644 --- a/packages/providers/google/src/types.ts +++ b/packages/providers/google/src/types.ts @@ -65,6 +65,9 @@ export enum GEMINI_MODEL { GEMINI_PRO_1_5_LATEST = "gemini-1.5-pro-002", GEMINI_PRO_1_5_FLASH_LATEST = "gemini-1.5-flash-002", GEMINI_2_0_FLASH_EXPERIMENTAL = "gemini-2.0-flash-exp", + GEMINI_2_0_FLASH = "gemini-2.0-flash-001", + GEMINI_2_0_FLASH_LITE_PREVIEW = "gemini-2.0-flash-lite-preview-02-05", + GEMINI_2_0_FLASH_THINKING_EXP = "gemini-2.0-flash-thinking-exp-01-21", } export interface GeminiModelInfo {