From 67a3947661c89b96b1c0cfe45662fc0b88ad97a7 Mon Sep 17 00:00:00 2001
From: binary-husky <96192199+binary-husky@users.noreply.github.com>
Date: Fri, 26 Jan 2024 16:35:35 +0800
Subject: [PATCH] Update bridge_all.py

---
 request_llms/bridge_all.py | 18 ++++++++++++++++++
 1 file changed, 18 insertions(+)

diff --git a/request_llms/bridge_all.py b/request_llms/bridge_all.py
index 45288ae1..575ac631 100644
--- a/request_llms/bridge_all.py
+++ b/request_llms/bridge_all.py
@@ -159,6 +159,24 @@ model_info = {
         "token_cnt": get_token_num_gpt4,
     },
 
+    "gpt-4-1106-preview": {
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": openai_endpoint,
+        "max_token": 128000,
+        "tokenizer": tokenizer_gpt4,
+        "token_cnt": get_token_num_gpt4,
+    },
+
+    "gpt-4-0125-preview": {
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": openai_endpoint,
+        "max_token": 128000,
+        "tokenizer": tokenizer_gpt4,
+        "token_cnt": get_token_num_gpt4,
+    },
+
     "gpt-3.5-random": {
         "fn_with_ui": chatgpt_ui,
         "fn_without_ui": chatgpt_noui,