Merge pull request #96 from intelligentnode/fix-gemini

Fix gemini

Author: intelligentnode

IntelliNode/function/Chatbot.js

@@ -143,6 +143,12 @@ class Chatbot {
                 const content = line.replace(/^(User|Assistant): /, '');
                 return { role, content };
             });
+        } else if (modelInput instanceof GeminiInput) {
+            messages = modelInput.messages.map(message => {
+                const role = message.role;
+                const content = message.parts.map(part => part.text).join(" ");
+                return { role, content };
+            });
         } else if (Array.isArray(modelInput.messages)) {
             messages = modelInput.messages;
         } else {
@@ -151,7 +157,7 @@ class Chatbot {
         }
 
         lastMessage = messages[messages.length - 1];
-    
+        
         if (lastMessage && lastMessage.role === "user") {
 
             const semanticResult = await this.extendedController.semanticSearch(lastMessage.content, modelInput.searchK);

IntelliNode/model/input/ChatModelInput.js

@@ -210,6 +210,19 @@ class GeminiInput extends ChatModelInput {
       contents: this.messages
     };
   }
+
+  cleanMessages() {
+    this.messages = [];
+  }
+
+  deleteLastMessage(message) {
+    if (this.messages.length > 0) {
+      this.messages.splice(-1, 1);
+      return true;
+    }
+    return false;
+  }
+
 }
 
 class ChatLLamaInput extends ChatModelInput {

IntelliNode/package.json

@@ -1,6 +1,6 @@
 {
   "name": "intellinode",
-  "version": "1.7.7",
+  "version": "1.7.8",
   "description": "Integrate and evaluate various AI models, such as ChatGPT, Llama, Diffusion, Cohere, Gemini and Hugging Face.",
   "main": "index.js",
   "keywords": [
@@ -16,8 +16,8 @@
     "speech synthesis",
     "prompt",
     "automation",
-    "Mistral",
-    "Google Gemini"
+    "mistralai",
+    "gemini"
   ],
   "author": "IntelliNode",
   "license": "Apache",

IntelliNode/test/integration/ChatbotWithData.test.js

@@ -7,17 +7,22 @@ const { ChatGPTInput,
         ChatLLamaInput,
         CohereInput,
         LLamaReplicateInput,
+        GeminiInput,
         LLamaSageInput } = require("../../model/input/ChatModelInput");
 
 const openaiKey = process.env.OPENAI_API_KEY;
 const replicateApiKey = process.env.REPLICATE_API_KEY;
 const intelliKey = process.env.INTELLI_ONE_KEY;
+const geminiApiKey = process.env.GEMINI_API_KEY;
 // openai bot
 const bot = new Chatbot(openaiKey, SupportedChatModels.OPENAI, null, 
           {oneKey:intelliKey, intelliBase: process.env.INTELLI_API_BASE});
 // llama - replicate bot
 const replicateBot = new Chatbot(replicateApiKey, SupportedChatModels.REPLICATE, null, 
           {oneKey:intelliKey, intelliBase: process.env.INTELLI_API_BASE});
+// gemini api key
+const geminiBot = new Chatbot(geminiApiKey, SupportedChatModels.GEMINI, null, 
+                              {oneKey:intelliKey, intelliBase: process.env.INTELLI_API_BASE});
 
 async function testOpenaiChatGPTCase1() {
   try {
@@ -107,6 +112,24 @@ async function testCohereChatCase() {
   assert(responses.length > 0, "Cohere chat response length should be greater than 0");
 }
 
+async function testGeminiChatCase1() {
+  try {
+    console.log('\ngemini test case 1: \n')
+
+    const input = new GeminiInput("", {searchK:4});
+    input.addUserMessage("Tell me about the Mars the Red Planet? summarize the context");
+
+    const responses = await geminiBot.chat(input);
+    
+    console.log('### the chatbot response ###')
+    responses.forEach((response) => console.log("- " + response));
+
+    assert(responses.length > 0, "testOpenaiChatGPTCase1 response length should be greater than 0");
+  } catch (error) {
+    console.error("Test case failed with exception:", error.message);
+  }
+}
+
 (async () => {
 
   console.log('### Openai model ###')
@@ -122,4 +145,7 @@ async function testCohereChatCase() {
   console.log('### Cohere model ###')
   await testCohereChatCase();
 
+  console.log('### Gemini model ###')
+  testGeminiChatCase1();
+
 })();