Merge pull request #93 from intelligentnode/81-add-support-for-gemini…

…-model

81 add support for gemini model

IntelliNode/README.md

@@ -41,11 +41,11 @@ const { Chatbot, ChatGPTInput } = require('intellinode');
 ```
 2. call:
 ```js
-// set the system mode and the user message.
+// set chatGPT system mode and the user message.
 const input = new ChatGPTInput('You are a helpful assistant.');
 input.addUserMessage('What is the distance between the Earth and the Moon?');
 
-// get the responses from the chatbot
+// get chatGPT responses.
 const bot = new Chatbot(apiKey);
 const responses = await bot.chat(input);
 ```

IntelliNode/model/input/ChatModelInput.js

@@ -201,6 +201,10 @@ class GeminiInput extends ChatModelInput {
     });
   }
 
+  addAssistantMessage(text) {
+    this.addModelMessage(text);
+  }
+
   getChatInput() {
     return {
       contents: this.messages

IntelliNode/package.json

@@ -1,6 +1,6 @@
 {
   "name": "intellinode",
-  "version": "1.7.6",
+  "version": "1.7.7",
   "description": "Integrate and evaluate various AI models, such as ChatGPT, Llama, Diffusion, Cohere, Gemini and Hugging Face.",
   "main": "index.js",
   "keywords": [

IntelliNode/test/integration/Chatbot.test.js

@@ -271,7 +271,7 @@ async function testGeminiChatCase2() {
 
     // add inputs
     geminiInput.addUserMessage("Write the first line of a story about a magic backpack.");
-    geminiInput.addModelMessage("In the bustling city of Meadow brook, lived a young girl named Sophie. She was a bright and curious soul with an imaginative mind.");
+    geminiInput.addAssistantMessage("In the bustling city of Meadow brook, lived a young girl named Sophie. She was a bright and curious soul with an imaginative mind.");
     geminiInput.addUserMessage("Can you set it in a quiet village in 1600s France?");
 
     const responses = await geminiBot.chat(geminiInput);

README.md

@@ -29,7 +29,7 @@ Unified prompt, evaluation, and production integration to any large model
 
 # Intelligent Node
 
-IntelliNode is a javascript module that integrates cutting-edge AI into your project. With its intuitive functions, you can easily feed data to models like **ChatGPT**, **LLaMA**, **WaveNet**, and **Stable diffusion** and receive generated text, speech, or images. It also offers high-level functions such as semantic search, multi-model evaluation, and chatbot capabilities.
+IntelliNode is a javascript module that integrates cutting-edge AI into your project. With its intuitive functions, you can easily feed data to models like **ChatGPT**, **LLaMA**, **WaveNet**, **Gemini** and **Stable diffusion** and receive generated text, speech, or images. It also offers high-level functions such as semantic search, multi-model evaluation, and chatbot capabilities.
 
 # Access the module
 ## Install
@@ -63,11 +63,11 @@ const { Chatbot, ChatGPTInput } = require('intellinode');
 ```
 call:
 ```js
-// set the system mode and the user message.
+// set chatGPT system mode and the user message.
 const input = new ChatGPTInput('You are a helpful assistant.');
 input.addUserMessage('What is the distance between the Earth and the Moon?');
 
-// get the responses from the chatbot
+// get chatGPT responses.
 const chatbot = new Chatbot(OPENAI_API_KEY, 'openai');
 const responses = await chatbot.chat(input);
 ```