Skip to content

Commit 52e085a

Browse files
committed
v1.6 添加 gpt-4o-mini 模型
1. 添加 gpt-4o-mini 模型 2. 默认模型修改为 gpt-4o-mini 3. gpt-4o 修改为动态模型 chatgpt-4o-latest 4. gpt-4-1106-preview 模型修改为 gpt-4-turbo
1 parent 64b2623 commit 52e085a

2 files changed

Lines changed: 9 additions & 8 deletions

File tree

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@
3737
# For a discussion on single-sourcing the version across setup.py and the
3838
# project code, see
3939
# https://packaging.python.org/en/latest/single_source_version.html
40-
version='1.5', # Required. eg. 1.2.3
40+
version='1.6', # Required. eg. 1.2.3
4141

4242
# This is a one-line description or tagline of what your project does. This
4343
# corresponds to the "Summary" metadata field:

src/transgpt/trans_chatgpt.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,11 @@
1616
HTTPS_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

2425
ARG_ROLE = 'role'
2526
ARG_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

Comments
 (0)