From ca64a592f5ca3b5f5bb7cdb136602a2031ba349a Mon Sep 17 00:00:00 2001
From: Frank Lee <71361592+Forest-Lee@users.noreply.github.com>
Date: Mon, 10 Jun 2024 22:17:51 +0800
Subject: [PATCH] Update zhipu models (#1852)

---
 config.py                  |  1 +
 request_llms/bridge_all.py | 32 ++++++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+)

diff --git a/config.py b/config.py
index d00b3563..fc0129f4 100644
--- a/config.py
+++ b/config.py
@@ -41,6 +41,7 @@ AVAIL_LLM_MODELS = ["gpt-4-1106-preview", "gpt-4-turbo-preview", "gpt-4-vision-p
 # --- --- --- ---
 # P.S. 其他可用的模型还包括
 # AVAIL_LLM_MODELS = [
+#   "glm-4-0520", "glm-4-air", "glm-4-airx", "glm-4-flash",
 #   "qianfan", "deepseekcoder",
 #   "spark", "sparkv2", "sparkv3", "sparkv3.5",
 #   "qwen-turbo", "qwen-plus", "qwen-max", "qwen-local",
diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py
index 06e69583..60a93e0a 100644
--- a/request_llms/bridge_all.py
+++ b/request_llms/bridge_all.py
@@ -290,6 +290,38 @@ model_info = {
         "tokenizer": tokenizer_gpt35,
         "token_cnt": get_token_num_gpt35,
     },
+    "glm-4-0520": {
+        "fn_with_ui": zhipu_ui,
+        "fn_without_ui": zhipu_noui,
+        "endpoint": None,
+        "max_token": 10124 * 8,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
+    "glm-4-air": {
+        "fn_with_ui": zhipu_ui,
+        "fn_without_ui": zhipu_noui,
+        "endpoint": None,
+        "max_token": 10124 * 8,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
+    "glm-4-airx": {
+        "fn_with_ui": zhipu_ui,
+        "fn_without_ui": zhipu_noui,
+        "endpoint": None,
+        "max_token": 10124 * 8,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
+    "glm-4-flash": {
+         "fn_with_ui": zhipu_ui,
+        "fn_without_ui": zhipu_noui,
+        "endpoint": None,
+        "max_token": 10124 * 8,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,       
+    },
     "glm-4v": {
         "fn_with_ui": zhipu_ui,
         "fn_without_ui": zhipu_noui,