Skip to content
Snippets Groups Projects
Unverified Commit dd596a0c authored by Peter Goldstein's avatar Peter Goldstein Committed by GitHub
Browse files

Add Gemini Flash 2.0 Experimental (#1563)

parent 9e648a0b
No related branches found
No related tags found
No related merge requests found
---
"llamaindex": patch
---
Add Gemini 2.0 Flash Experimental
...@@ -53,6 +53,7 @@ export const GEMINI_MODEL_INFO_MAP: Record<GEMINI_MODEL, GeminiModelInfo> = { ...@@ -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_FLASH]: { contextWindow: 10 ** 6 },
[GEMINI_MODEL.GEMINI_PRO_1_5_LATEST]: { contextWindow: 2 * 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_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[] = [ const SUPPORT_TOOL_CALL_MODELS: GEMINI_MODEL[] = [
...@@ -66,6 +67,7 @@ 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_FLASH_LATEST,
GEMINI_MODEL.GEMINI_PRO_1_5_LATEST, GEMINI_MODEL.GEMINI_PRO_1_5_LATEST,
GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_LATEST, GEMINI_MODEL.GEMINI_PRO_1_5_FLASH_LATEST,
GEMINI_MODEL.GEMINI_2_0_FLASH_EXPERIMENTAL,
]; ];
const DEFAULT_GEMINI_PARAMS = { const DEFAULT_GEMINI_PARAMS = {
......
...@@ -64,6 +64,7 @@ export enum GEMINI_MODEL { ...@@ -64,6 +64,7 @@ export enum GEMINI_MODEL {
// Note: should be switched to -latest suffix when google supports it // 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_LATEST = "gemini-1.5-pro-002",
GEMINI_PRO_1_5_FLASH_LATEST = "gemini-1.5-flash-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 { export interface GeminiModelInfo {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment