support o1-preview and o1-mini

2024-09-13 03:11:53 +00:00
parent 4e041e1d4e
commit 0d0575a639
2 changed files with 18 additions and 1 deletions
--- a/request_llms/bridge_all.py
+++ b/request_llms/bridge_all.py
@@ -248,6 +248,23 @@ model_info = {
        "token_cnt": get_token_num_gpt4,
    },

+    "o1-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,
+    },
+    "o1-mini": {
+        "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-turbo": {
        "fn_with_ui": chatgpt_ui,
        "fn_without_ui": chatgpt_noui,