gnh1201/welsonjs
1
0
Fork 0
mirror of https://github.com/gnh1201/welsonjs.git synced 2025-06-07 13:49:04 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Update language-inference-engine.js

Browse Source 
Fix gemini call broken
This commit is contained in:
Jihoon Yi 2025-05-10 15:50:55 +09:00 committed by GitHub
parent 5f10d50a3c
commit ef23e41e3a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 24 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -185,33 +185,40 @@ var ENGINE_PROFILES = {
}, []); }, []);
} }
}, },
"google": { "gemini": {
"type": "llm", "type": "llm",
"availableModels": [ "availableModels": [
"gemini-2.0-flash",
"gemini-1.5-flash" "gemini-1.5-flash"
], ],
"headers": { "headers": {
"Content-Type": "application/json" "Content-Type": "application/json"
}, },
"url": "https://generativelanguage.googleapis.com/v1beta/models/{model}:generateContent?key={apikey}", "url": "https://generativelanguage.googleapis.com/v1beta/models/{model}:generateContent?key={apikey}",
"warp": function(model, message, temperature) { "wrap": function(model, message, temperature) {
return { return {
"contents": [ "contents": [{
{ "role": "user", // Changed "developer" to "user" for the initial prompt. The overall prompt is still intended to guide the model, so this is reasonable.
"parts": [ "parts": [{
{ "text": BIAS_MESSAGE
}]
}, {
"role": "user",
"parts": [{
"text": message "text": message
}]
}],
"generationConfig": {
"temperature": temperature
} }
]
}
]
}; };
}, },
"callback": function(response) { "callback": function(response) {
response = JSON.parse(response)
if ("error" in response) { if ("error" in response) {
return ["Error: " + response.error.message]; return ["Error: " + response.error.message];
} else { } else {
return response.candidates.reduce(function(a, x) { return response["candidates"].reduce(function(a, x) {
x.content.parts.forEach(function(part) { x.content.parts.forEach(function(part) {
if ("text" in part) { if ("text" in part) {
a.push(part.text); a.push(part.text);
@ -219,7 +226,6 @@ var ENGINE_PROFILES = {
a.push("Not supported type"); a.push("Not supported type");
} }
}); });
return a; return a;
}, []); }, []);
} }
@ -508,6 +514,11 @@ function LanguageInferenceEngine() {
var headers = this.engineProfile.headers; var headers = this.engineProfile.headers;
var wrap = this.engineProfile.wrap; var wrap = this.engineProfile.wrap;
var url = this.engineProfile.url; var url = this.engineProfile.url;
if(this.provider === "gemini"){
url = url
.replace(/{model}/g, this.model)
.replace(/{apikey}/g, apikey);
}
var callback = this.engineProfile.callback; var callback = this.engineProfile.callback;
var response = HTTP.create("MSXML") var response = HTTP.create("MSXML")