From ef23e41e3a35742d96287348f4a7531dde8ebfe6 Mon Sep 17 00:00:00 2001
From: Jihoon Yi <44565599+baramofme@users.noreply.github.com>
Date: Sat, 10 May 2025 15:50:55 +0900
Subject: [PATCH] Update language-inference-engine.js

Fix gemini call broken
---
 lib/language-inference-engine.js | 37 +++++++++++++++++++++-----------
 1 file changed, 24 insertions(+), 13 deletions(-)

diff --git a/lib/language-inference-engine.js b/lib/language-inference-engine.js
index 52b822e..0cde9f6 100644
--- a/lib/language-inference-engine.js
+++ b/lib/language-inference-engine.js
@@ -185,33 +185,40 @@ var ENGINE_PROFILES = {
             }, []);
         }
     },
-    "google": {
+    "gemini": {
         "type": "llm",
         "availableModels": [
+            "gemini-2.0-flash",
             "gemini-1.5-flash"
         ],
         "headers": {
             "Content-Type": "application/json"
         },
         "url": "https://generativelanguage.googleapis.com/v1beta/models/{model}:generateContent?key={apikey}",
-        "warp": function(model, message, temperature) {
+        "wrap": function(model, message, temperature) {
             return {
-                "contents": [
-                    {
-                        "parts": [
-                            {
-                                "text": message
-                            }
-                        ]
-                    }
-                ]
+                "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": [{
+                        "text": BIAS_MESSAGE
+                    }]
+                }, {
+                    "role": "user",
+                    "parts": [{
+                        "text": message
+                    }]
+                }],
+                "generationConfig": {
+                    "temperature": temperature
+                }
             };
         },
         "callback": function(response) {
+            response = JSON.parse(response)
             if ("error" in response) {
                 return ["Error: " + response.error.message];
             } else {
-                return response.candidates.reduce(function(a, x) {
+                return response["candidates"].reduce(function(a, x) {
                     x.content.parts.forEach(function(part) {
                         if ("text" in part) {
                             a.push(part.text);
@@ -219,7 +226,6 @@ var ENGINE_PROFILES = {
                             a.push("Not supported type");
                         }
                     });
-                    
                     return a;
                 }, []);
             }
@@ -508,6 +514,11 @@ function LanguageInferenceEngine() {
         var headers = this.engineProfile.headers;
         var wrap = this.engineProfile.wrap;
         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 response = HTTP.create("MSXML")