添加gemini-2.0-flash

2025-03-11 10:01:08 +08:00 · 2025-03-11 10:01:08 +08:00 · d525e543bd
parent 045cdb15d8
commit d525e543bd
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": {