添加gemini-2.0-flash (#2180)

Co-authored-by: 柯仕锋 <12029064@zju.edu.cn>
2025-03-25 00:13:18 +08:00
parent 82aac97980
commit 63c61e6204
1 changed files with 9 additions and 0 deletions
--- 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": {