run-llama/LlamaIndexTS
1
0
Fork 0
mirror of https://github.com/run-llama/LlamaIndexTS.git synced 2026-07-19 18:43:34 -04:00
Code Issues 138 Packages Projects Releases 10 Wiki Activity

Add Gemini Flash 2.0 Experimental (#1563)

Browse Source
This commit is contained in:
Peter Goldstein
2024-12-13 15:10:46 -05:00
committed by GitHub
parent 9e648a0b4f
commit dd596a0cf2
3 changed files with 8 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.changeset/kind-bananas-shave.md
+5
View File
@@ -0,0 +1,5 @@
---
"llamaindex": patch
---
Add Gemini 2.0 Flash Experimental
packages/llamaindex/src/llm/gemini/base.ts
+2
View File
@@ -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 = {
packages/llamaindex/src/llm/gemini/types.ts
+1
View File
@@ -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 {