diff --git a/.changeset/kind-bananas-shave.md b/.changeset/kind-bananas-shave.md new file mode 100644 index 0000000000000000000000000000000000000000..e615aaa1e85c34f26c900e6976476503201fa767 --- /dev/null +++ b/.changeset/kind-bananas-shave.md @@ -0,0 +1,5 @@ +--- +"llamaindex": patch +--- + +Add Gemini 2.0 Flash Experimental diff --git a/packages/llamaindex/src/llm/gemini/base.ts b/packages/llamaindex/src/llm/gemini/base.ts index 0046e11c4d1f230703ee1615bd506585946dedcf..50937887f5e3b21eecd1e150ef3fc395f4ca7c1b 100644 --- a/packages/llamaindex/src/llm/gemini/base.ts +++ b/packages/llamaindex/src/llm/gemini/base.ts @@ -53,6 +53,7 @@ export const GEMINI_MODEL_INFO_MAP: Record<GEMINI_MODEL, GeminiModelInfo> = { [GEMINI_MODEL.GEMINI_PRO_1_5_FLASH]: { contextWindow: 10 ** 6 }, [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 }, }; const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [ @@ -66,6 +67,7 @@ const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [ GEMINI_MODEL.GEMINI_PRO_FLASH_LATEST, GEMINI_MODEL.GEMINI_PRO_1_5_LATEST, GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_LATEST, + GEMINI_MODEL.GEMINI_2_0_FLASH_EXPERIMENTAL, ]; const DEFAULT_GEMINI_PARAMS = { diff --git a/packages/llamaindex/src/llm/gemini/types.ts b/packages/llamaindex/src/llm/gemini/types.ts index fa5c8395281dad65c5715b0556d0c4a646faf4e5..35a7074c47bca2c885fbc860fe326ffe3ed0d793 100644 --- a/packages/llamaindex/src/llm/gemini/types.ts +++ b/packages/llamaindex/src/llm/gemini/types.ts @@ -64,6 +64,7 @@ export enum GEMINI_MODEL { // Note: should be switched to -latest suffix when google supports it 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", } export interface GeminiModelInfo {