Skip to content

Commit 291178c

Browse files
committed
feat: 支持最多5个自定义API接入
1 parent e514fa4 commit 291178c

File tree

3 files changed

+101
-13
lines changed

3 files changed

+101
-13
lines changed

package.json

Lines changed: 96 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -112,23 +112,112 @@
112112
"enum": [
113113
"deepseek-chat",
114114
"deepseek-reasoner",
115-
"custom"
115+
"custom1",
116+
"custom2",
117+
"custom3",
118+
"custom4",
119+
"custom5"
116120
]
117121
},
118-
"codeReDesign.customModelName": {
122+
"codeReDesign.custom1APIKey": {
123+
"type": "string",
124+
"default": null,
125+
"description": "Custom DeepSeek API Key 1"
126+
},
127+
"codeReDesign.custom1BaseURL": {
128+
"type": "string",
129+
"default": "",
130+
"description": "Custom DeepSeek API Base URL 1"
131+
},
132+
"codeReDesign.custom1ModelName": {
133+
"type": "string",
134+
"default": "",
135+
"description": "Custom DeepSeek Model Name 1"
136+
},
137+
"codeReDesign.custom1ModelNickname": {
138+
"type": "string",
139+
"default": "",
140+
"description": "Custom DeepSeek Model Nickname 1"
141+
},
142+
"codeReDesign.custom2APIKey": {
143+
"type": "string",
144+
"default": null,
145+
"description": "Custom DeepSeek API Key 2"
146+
},
147+
"codeReDesign.custom2BaseURL": {
119148
"type": "string",
120149
"default": "",
121-
"description": "Custom DeepSeek Model Name"
150+
"description": "Custom DeepSeek API Base URL 2"
122151
},
123-
"codeReDesign.customApiBaseURL": {
152+
"codeReDesign.custom2ModelName": {
124153
"type": "string",
125154
"default": "",
126-
"description": "Custom DeepSeek API Base URL"
155+
"description": "Custom DeepSeek Model Name 2"
127156
},
128-
"codeReDesign.customApiKey":{
157+
"codeReDesign.custom2ModelNickname": {
158+
"type": "string",
159+
"default": "",
160+
"description": "Custom DeepSeek Model Nickname 2"
161+
},
162+
"codeReDesign.custom3APIKey": {
129163
"type": "string",
130164
"default": null,
131-
"description": "Custom DeepSeek API Key"
165+
"description": "Custom DeepSeek API Key 3"
166+
},
167+
"codeReDesign.custom3BaseURL": {
168+
"type": "string",
169+
"default": "",
170+
"description": "Custom DeepSeek API Base URL 3"
171+
},
172+
"codeReDesign.custom3ModelName": {
173+
"type": "string",
174+
"default": "",
175+
"description": "Custom DeepSeek Model Name 3"
176+
},
177+
"codeReDesign.custom3ModelNickname": {
178+
"type": "string",
179+
"default": "",
180+
"description": "Custom DeepSeek Model Nickname 3"
181+
},
182+
"codeReDesign.custom4APIKey": {
183+
"type": "string",
184+
"default": null,
185+
"description": "Custom DeepSeek API Key 4"
186+
},
187+
"codeReDesign.custom4BaseURL": {
188+
"type": "string",
189+
"default": "",
190+
"description": "Custom DeepSeek API Base URL 4"
191+
},
192+
"codeReDesign.custom4ModelName": {
193+
"type": "string",
194+
"default": "",
195+
"description": "Custom DeepSeek Model Name 4"
196+
},
197+
"codeReDesign.custom4ModelNickname": {
198+
"type": "string",
199+
"default": "",
200+
"description": "Custom DeepSeek Model Nickname 4"
201+
},
202+
"codeReDesign.custom5APIKey": {
203+
"type": "string",
204+
"default": null,
205+
"description": "Custom DeepSeek API Key 5"
206+
},
207+
"codeReDesign.custom5BaseURL": {
208+
"type": "string",
209+
"default": "",
210+
"description": "Custom DeepSeek API Base URL 5"
211+
},
212+
"codeReDesign.custom5ModelName": {
213+
"type": "string",
214+
"default": "",
215+
"description": "Custom DeepSeek Model Name 5"
216+
},
217+
"codeReDesign.custom5ModelNickname": {
218+
"type": "string",
219+
"default": "",
220+
"description": "Custom DeepSeek Model Nickname 5"
132221
}
133222
}
134223
}

prompt/customapi.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
1-
�Զ���ģ�͵� APIKey ��Ϊ���Զ�������
2-
��������Զ����ģ������Ϊ5����BaseURL�� ModelName��APIKey��
1+
�Զ���ģ������Ϊ5����BaseURL�� ModelName��APIKey��
32
���ҿ���ѡ����Щģ���е�һ�� ��Custom1��Custom2, ...��
43
���Ը�ģ��ȡnickname�����Dz����ظ�
54
ѡ���ʱ�������NickName����ʾNickname��ѡ�񣬲�Ȼ����CostomXXX

src/deepseekApi.ts

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,10 +11,10 @@ function getDeepSeekModelConfig(): { modelName: string, apiBaseURL: string, apiK
1111
const modelConfig = config.get<string>('modelConfig') || 'deepseek-chat';
1212
const apiKey = config.get<string>('deepSeekApiKey') || null;
1313

14-
if (modelConfig === 'custom') {
15-
const customModelName = config.get<string>('customModelName') || '';
16-
const customApiBaseURL = config.get<string>('customApiBaseURL') || '';
17-
const apiKey = config.get<string>('customApiKey') || null;
14+
if (modelConfig.startsWith('custom')) {
15+
const customModelName = config.get<string>(`${modelConfig}ModelName`) || '';
16+
const customApiBaseURL = config.get<string>(`${modelConfig}BaseURL`) || '';
17+
const apiKey = config.get<string>(`${modelConfig}APIKey`) || null;
1818
return {
1919
modelName: customModelName,
2020
apiBaseURL: customApiBaseURL,

0 commit comments

Comments
 (0)