1616HTTPS_PROXY = "HTTPS_PROXY"
1717
1818# GPT 接口模型定义 https://platform.openai.com/docs/models/
19- CHATGPT_35_TURBO = "gpt-3.5-turbo"
20- CHATGPT_4 = "gpt-4" # 8K
21- CHATGPT_4_TRUBO = "gpt-4-1106-preview" # 128K
22- CHATGPT_4o = "gpt-4o" # 128K
19+ CHATGPT_35_TURBO = "gpt-3.5-turbo" # 16K
20+ CHATGPT_4 = "gpt-4" # 8K
21+ CHATGPT_4_TRUBO = "gpt-4-turbo" # 128K
22+ CHATGPT_4o = "chatgpt-4o-latest" # 128K
23+ CHATGPT_4o_MINI = "gpt-4o-mini" # 128K
2324
2425ARG_ROLE = 'role'
2526ARG_OPENAI_MODEL = 'openai_model'
@@ -31,10 +32,10 @@ class ChatgptTranslation(BaseTranslation) :
3132 RETRY = 3
3233 RETRY_WAIT_SECONDS = 30
3334
34- def __init__ (self , openai_key , openai_model = CHATGPT_35_TURBO , proxy_ip = '127.0.0.1' , proxy_port = 0 ) :
35+ def __init__ (self , openai_key , openai_model = CHATGPT_4o_MINI , proxy_ip = '127.0.0.1' , proxy_port = 0 ) :
3536 BaseTranslation .__init__ (self , '' , openai_key )
3637 openai .api_key = openai_key
37- self .model = openai_model or CHATGPT_35_TURBO
38+ self .model = openai_model or CHATGPT_4o_MINI
3839 self .proxy = f"http://{ proxy_ip } :{ proxy_port } " if proxy_port > 0 else ""
3940
4041
@@ -57,7 +58,7 @@ def _ask_gpt(self, role_setting, segment) :
5758 self ._enable_proxy ()
5859 msg = [
5960 role_setting ,
60- {"role" : "user" , "content" : segment }
61+ { "role" : "user" , "content" : segment }
6162 ]
6263 rsp = self ._wait_for_ask (msg )
6364 rst = rsp .get ("choices" )[0 ]["message" ]["content" ]
0 commit comments