From 55823dbdf6b39785ac840d735f24081d6e72557b Mon Sep 17 00:00:00 2001 From: Kevin Hu Date: Sat, 8 Feb 2025 10:19:51 +0800 Subject: [PATCH] Refresh Gemini model list. (#4780) ### What problem does this PR solve? #4761 ### Type of change - [x] New Feature (non-breaking change which adds functionality) --- conf/llm_factories.json | 28 +++++++++++++++++----------- 1 file changed, 17 insertions(+), 11 deletions(-) diff --git a/conf/llm_factories.json b/conf/llm_factories.json index 6c2af0be..246711f3 100644 --- a/conf/llm_factories.json +++ b/conf/llm_factories.json @@ -947,27 +947,27 @@ "status": "1", "llm": [ { - "llm_name": "gemini-1.5-pro-latest", + "llm_name": "gemini-2.0-flash-001", "tags": "LLM,CHAT,1024K", "max_tokens": 1048576, "model_type": "chat" }, { - "llm_name": "gemini-1.5-flash-latest", - "tags": "LLM,CHAT,1024K", + "llm_name": "gemini-1.5-flash", + "tags": "LLM,IMAGE2TEXT,1024K", "max_tokens": 1048576, - "model_type": "chat" + "model_type": "image2text" }, { - "llm_name": "gemini-2.0-flash-exp", - "tags": "LLM,CHAT,1024K", - "max_tokens": 1048576, - "model_type": "chat" + "llm_name": "gemini-1.5-pro", + "tags": "LLM,IMAGE2TEXT,2048K", + "max_tokens": 2097152, + "model_type": "image2text" }, { - "llm_name": "gemini-1.0-pro-vision-latest", - "tags": "LLM,IMAGE2TEXT,12K", - "max_tokens": 12288, + "llm_name": "gemini-1.5-flash-8b", + "tags": "LLM,IMAGE2TEXT,1024K", + "max_tokens": 1048576, "model_type": "image2text" }, { @@ -975,6 +975,12 @@ "tags": "TEXT EMBEDDING", "max_tokens": 2048, "model_type": "embedding" + }, + { + "llm_name": "embedding-001", + "tags": "TEXT EMBEDDING", + "max_tokens": 2048, + "model_type": "embedding" } ] },