From 63c61e62042a44484ce53d183b22ea9382e1d086 Mon Sep 17 00:00:00 2001
From: davidfir3 <40458162+davidfir3@users.noreply.github.com>
Date: Tue, 25 Mar 2025 00:13:18 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0gemini-2.0-flash=20(#2180)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Co-authored-by: 柯仕锋 <12029064@zju.edu.cn>
---
 request_llms/bridge_all.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py
index a2bb3a01..52ed2fa7 100644
--- a/request_llms/bridge_all.py
+++ b/request_llms/bridge_all.py
@@ -530,6 +530,15 @@ model_info = {
         "tokenizer": tokenizer_gpt35,
         "token_cnt": get_token_num_gpt35,
     },
+    "gemini-2.0-flash": {
+        "fn_with_ui": genai_ui,
+        "fn_without_ui": genai_noui,
+        "endpoint": gemini_endpoint,
+        "has_multimodal_capacity": True,
+        "max_token": 1024 * 204800,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
 
     # cohere
     "cohere-command-r-plus": {