ishangsf/DeepCodeGeniusWeb-vscode
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #446 from devchat-ai/feature/set-default-model-params-#248

Browse Source 
Implement Default Model Parameters in VSCode Settings
This commit is contained in:
boob.yang 2024-02-06 11:41:03 +08:00 committed by GitHub
commit ad76c5026a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 49 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
src/extension.ts
View File

@ -272,6 +272,54 @@ async function updateInvalidDefaultModel() {
// "gpt-3.5-turbo-1106",
// "gpt-3.5-turbo-16k",
async function configSetModelDefaultParams() {
const modelParams = {
"Model.gpt-3-5": {
"max_input_tokens": 13000
},
"Model.gpt-4": {
"max_input_tokens": 6000
},
"Model.gpt-4-turbo": {
"max_input_tokens": 32000
},
"Model.claude-2": {
"max_input_tokens": 32000
},
"Model.xinghuo-2": {
"max_input_tokens": 6000
},
"Model.chatglm_pro": {
"max_input_tokens": 8000
},
"Model.ERNIE-Bot": {
"max_input_tokens": 8000
},
"Model.CodeLlama-70b": {
"max_input_tokens": 4000
},
"Model.Mixtral-8x7B": {
"max_input_tokens": 4000
},
"Model.Minimax-abab6": {
"max_input_tokens": 4000
},
"Model.llama-2-70b-chat": {
"max_input_tokens": 4000
}
};
// set default params
for (const model of Object.keys(modelParams)) {
const modelConfig: any = UiUtilWrapper.getConfiguration("devchat", model);
if (!modelConfig["max_input_tokens"]) {
modelConfig["max_input_tokens"] = modelParams[model]["max_input_tokens"];
await vscode.workspace.getConfiguration("devchat").update(model, modelConfig, vscode.ConfigurationTarget.Global);
}
}
}
async function activate(context: vscode.ExtensionContext) {
ExtensionContextHolder.context = context;
@ -285,6 +333,7 @@ async function activate(context: vscode.ExtensionContext) {
await updateInvalidSettings();
await updateInvalidDefaultModel();
await configUpdateto240205();
await configSetModelDefaultParams();
regLanguageContext();