显式增加azure-gpt-4选项

2023-09-23 23:06:58 +08:00
parent dc5bb9741a
commit e24f077b68
2 changed files with 10 additions and 1 deletions
--- a/request_llm/bridge_all.py
+++ b/request_llm/bridge_all.py
@@ -145,6 +145,15 @@ model_info = {
        "token_cnt": get_token_num_gpt35,
    },

+    "azure-gpt-4":{
+        "fn_with_ui": chatgpt_ui,
+        "fn_without_ui": chatgpt_noui,
+        "endpoint": azure_endpoint,
+        "max_token": 8192,
+        "tokenizer": tokenizer_gpt35,
+        "token_cnt": get_token_num_gpt35,
+    },
+
    # api_2d
    "api2d-gpt-3.5-turbo": {
        "fn_with_ui": chatgpt_ui,