gnh1201/welsonjs
1
0
Fork 0
mirror of https://github.com/gnh1201/welsonjs.git synced 2025-02-06 06:54:58 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update language-inference-engine.js

Browse Source
This commit is contained in:
Namhyeon Go 2025-01-29 17:22:55 +09:00
parent fb947cc777
commit 8699d89694
1 changed files with 51 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

67
lib/language-inference-engine.js
View File

@ -38,7 +38,8 @@ var engineProfiles = {
"role": "user", "role": "user",
"content": message "content": message
}], }],
"temperature": temperature "temperature": temperature,
"stream": false
}; };
}, },
"callback": function(response) { "callback": function(response) {
@ -75,7 +76,8 @@ var engineProfiles = {
"content": message "content": message
} }
], ],
"temperature": temperature "temperature": temperature,
"stream": false
}; };
}, },
"callback": function(response) { "callback": function(response) {
@ -115,7 +117,8 @@ var engineProfiles = {
"content": message "content": message
} }
], ],
"temperature": temperature "temperature": temperature,
"stream": false
}; };
}, },
"callback": function(response) { "callback": function(response) {
@ -151,7 +154,8 @@ var engineProfiles = {
"content": message "content": message
} }
], ],
"temperature": temperature "temperature": temperature,
"stream": false
} }
}, },
"callback": function(response) { "callback": function(response) {
@ -201,6 +205,37 @@ var engineProfiles = {
} }
} }
}, },
"mistral": {
"type": "llm",
"defaultModel": "ministral-8b-latest",
"url": "https://api.mistral.ai/v1/chat/completions",
"wrap": function(model, message, temperature) {
"model": model,
"messages": [
{
"role": "system",
"content": biasMessage
},
{
"role": "user",
"content": message
}
],
"temperature": temperature,
"stream": false
},
"callback": function(response) {
if ("error" in response) {
return ["Error: " + response.error.message];
} else {
return response.choices.reduce(function(a, x) {
a.push(x.message.content);
return a;
}, []);
}
}
},
"deepseek": { "deepseek": {
"type": "llm", "type": "llm",
"defaultModel": "deepseek-chat", "defaultModel": "deepseek-chat",
@ -223,17 +258,17 @@ var engineProfiles = {
], ],
"temperature": temperature, "temperature": temperature,
"stream": false "stream": false
} },
}, "callback": function(response) {
"callback": function(response) { if ("error" in response) {
if ("error" in response) { return ["Error: " + response.error.message];
return ["Error: " + response.error.message]; } else {
} else { return response.choices.reduce(function(a, x) {
return response.choices.reduce(function(a, x) { a.push(x.message.content);
a.push(x.message.content);
return a;
return a; }, []);
}, []); }
} }
} }
}; };
@ -299,7 +334,7 @@ exports.create = function() {
return new LanguageInferenceEngine(); return new LanguageInferenceEngine();
}; };
exports.VERSIONINFO = "Language Inference Engine (NLP/LLM) integration version 0.1.2"; exports.VERSIONINFO = "Language Inference Engine integration version 0.1.3";
exports.AUTHOR = "abuse@catswords.net"; exports.AUTHOR = "abuse@catswords.net";
exports.global = global; exports.global = global;
exports.require = global.require; exports.require = global.require;