diff --git a/.changeset/tame-chefs-tickle.md b/.changeset/tame-chefs-tickle.md new file mode 100644 index 0000000000000000000000000000000000000000..2b3ba211e6ccff57c253ae15c413675eb52e3382 --- /dev/null +++ b/.changeset/tame-chefs-tickle.md @@ -0,0 +1,5 @@ +--- +"llamaindex": patch +--- + +feat: Gemini GA release models diff --git a/packages/core/src/llm/gemini/base.ts b/packages/core/src/llm/gemini/base.ts index 5cc072bca8b2206b9193fabb5eae683392ad2be4..65491fd98bfc5d63c00c39680c213ca405cce8e7 100644 --- a/packages/core/src/llm/gemini/base.ts +++ b/packages/core/src/llm/gemini/base.ts @@ -38,6 +38,8 @@ export const GEMINI_MODEL_INFO_MAP: Record<GEMINI_MODEL, GeminiModelInfo> = { // multi-modal/multi turn [GEMINI_MODEL.GEMINI_PRO_1_5_PRO_PREVIEW]: { contextWindow: 10 ** 6 }, [GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_PREVIEW]: { contextWindow: 10 ** 6 }, + [GEMINI_MODEL.GEMINI_PRO_1_5]: { contextWindow: 2 * 10 ** 6 }, + [GEMINI_MODEL.GEMINI_PRO_1_5_FLASH]: { contextWindow: 10 ** 6 }, }; const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [ @@ -45,6 +47,8 @@ const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [ GEMINI_MODEL.GEMINI_PRO_VISION, GEMINI_MODEL.GEMINI_PRO_1_5_PRO_PREVIEW, GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_PREVIEW, + GEMINI_MODEL.GEMINI_PRO_1_5, + GEMINI_MODEL.GEMINI_PRO_1_5_FLASH, ]; const DEFAULT_GEMINI_PARAMS = { diff --git a/packages/core/src/llm/gemini/types.ts b/packages/core/src/llm/gemini/types.ts index 1ef3a881a4407fe6af498c66ee4403a5757cc230..998ec12ea023a9ff431c87d86e6de7f850bae95f 100644 --- a/packages/core/src/llm/gemini/types.ts +++ b/packages/core/src/llm/gemini/types.ts @@ -53,6 +53,8 @@ export enum GEMINI_MODEL { GEMINI_PRO_LATEST = "gemini-1.5-pro-latest", GEMINI_PRO_1_5_PRO_PREVIEW = "gemini-1.5-pro-preview-0514", GEMINI_PRO_1_5_FLASH_PREVIEW = "gemini-1.5-flash-preview-0514", + GEMINI_PRO_1_5 = "gemini-1.5-pro-001", + GEMINI_PRO_1_5_FLASH = "gemini-1.5-flash-001", } export interface GeminiModelInfo {