ElvisWang111 commited on
Commit
d8f115d
·
verified ·
1 Parent(s): e1c640c

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +19 -3
config.py CHANGED
@@ -1,32 +1,48 @@
1
  # 大模型配置
2
  MODEL_CONFIGS = {
3
-
4
  "GPT-4o": {
5
  "api_base": "https://api.openai.com/v1",
6
  "model_name": "gpt-4o",
 
 
7
  },
8
  "GPT-5": {
9
  "api_base": "https://api.openai.com/v1",
10
  "model_name": "gpt-5",
 
 
11
  },
12
  "Claude": {
13
  "api_base": "https://api.anthropic.com",
14
  "model_name": "claude-3-5-sonnet-latest",
 
 
15
  },
16
  "通义千问": {
17
  "api_base": "https://dashscope.aliyuncs.com/compatible-mode/v1",
18
  "model_name": "qwen-max",
 
 
19
  },
20
  "DeepSeek": {
21
  "api_base": "https://api.deepseek.com/v1",
22
  "model_name": "deepseek-chat",
 
 
23
  },
24
  "智谱AI": {
25
- "api_base": "https://open.bigmodel.cn/api/paas/v4/chat/completions",
26
  "model_name": "glm-4v-plus-0111",
 
 
27
  },
28
  "豆包": {
29
  "api_base": "https://ark.cn-beijing.volces.com/api/v3/",
30
  "model_name": "doubao-seed-1-6-251015",
 
 
31
  }
32
- }
 
 
 
 
1
  # 大模型配置
2
  MODEL_CONFIGS = {
 
3
  "GPT-4o": {
4
  "api_base": "https://api.openai.com/v1",
5
  "model_name": "gpt-4o",
6
+ "api_type": "openai", # OpenAI 兼容 API
7
+ "is_preset": True,
8
  },
9
  "GPT-5": {
10
  "api_base": "https://api.openai.com/v1",
11
  "model_name": "gpt-5",
12
+ "api_type": "openai", # OpenAI 兼容 API
13
+ "is_preset": True,
14
  },
15
  "Claude": {
16
  "api_base": "https://api.anthropic.com",
17
  "model_name": "claude-3-5-sonnet-latest",
18
+ "api_type": "claude", # Claude 有自己的 API
19
+ "is_preset": True,
20
  },
21
  "通义千问": {
22
  "api_base": "https://dashscope.aliyuncs.com/compatible-mode/v1",
23
  "model_name": "qwen-max",
24
+ "api_type": "openai", # OpenAI 兼容 API
25
+ "is_preset": True,
26
  },
27
  "DeepSeek": {
28
  "api_base": "https://api.deepseek.com/v1",
29
  "model_name": "deepseek-chat",
30
+ "api_type": "openai", # OpenAI 兼容 API
31
+ "is_preset": True,
32
  },
33
  "智谱AI": {
34
+ "api_base": "https://open.bigmodel.cn/api/paas/v4",
35
  "model_name": "glm-4v-plus-0111",
36
+ "api_type": "zhipu", # 智谱 AI 有自己的客户端
37
+ "is_preset": True,
38
  },
39
  "豆包": {
40
  "api_base": "https://ark.cn-beijing.volces.com/api/v3/",
41
  "model_name": "doubao-seed-1-6-251015",
42
+ "api_type": "openai", # OpenAI 兼容 API
43
+ "is_preset": True,
44
  }
45
+ }
46
+
47
+ # OpenAI API 兼容模型的标识(用于菜单显示和配置文件键名)
48
+ CUSTOM_MODEL_KEY = "OpenAI API 兼容模型"