interintel 1.0.21 → 1.0.22

package/functions/chat-functions.js

@@ -6,6 +6,5 @@ async function askQuestion(rl, prompt) {
   });
 }
 
-module.exports = {
-  askQuestion,
-}
+export { askQuestion };
+

package/functions/file-functions.js

@@ -1,52 +1,50 @@
-const fs = require('fs');
-const path = require('path');
-const { aiVersion } = require('../interintel.config');
+import fs from 'fs';
+import path from 'path';
 
 // READING FOR INITAL REFERENCE
-function readSpecificFiles(configFilePath) {
-  
+async function readSpecificFiles(configFilePath) {
   try {
-    // Read the content of the config file
-    const configContent = fs.readFileSync(configFilePath, 'utf8');    
-    // Parse the config file content as JavaScript
-    const config = eval(configContent);
-    // Extract the file paths from the config object
-    const filePaths = config.filePaths;
-    const configDir = path.dirname(configFilePath);
+    // Dynamically import the config file
+    const absoluteConfigPath = path.resolve(configFilePath);
+    const configModule = await import('file://' + absoluteConfigPath);
+    const config = configModule.default;
 
+    const filePaths = config.filePaths;
     let allContent = 'I am sharing information from my file system for reference in our chat.\n';
-  
-    filePaths.forEach((filePath) => {
+
+    for (const filePath of filePaths) {
       try {
         // Construct the absolute path
-        const absolutePath = path.resolve(configDir, filePath);
+        const absolutePath = path.resolve(process.cwd(), filePath);
         const fileContent = fs.readFileSync(absolutePath, 'utf8');
 
-
-  
         // Read the file content and add it to allContent
         allContent += `\nStart File Name: ${filePath}\n File Content:\n${fileContent}\n End File Name: ${filePath}`;
       } catch (error) {
-        console.error(`Error reading file ${filePath}: ${error.message}`.bgRed);
+        console.error(`Error reading file ${filePath}: ${error.message}`);
       }
-    });
+    }
 
     // Add console.log statements to communicate to the user
-    console.log(`${aiVersion} sent reference files:`.yellow, `${logFileNames(filePaths)}`.yellow);
-      return allContent;
+    console.log(
+      `${config.aiVersion} sent reference files:`.yellow,
+      `${logFileNames(filePaths)}`.yellow
+    );
+    return allContent;
   } catch (error) {
-    console.error(`Error reading config file: ${error.message}`.bgRed);
+    console.error(`Error reading config file: ${error.message}`);
     return '';
   }
 }
 
 function writeFileFromPrompt(promptFileName, contentToWrite, baseDir) {
   try {
-
     if (!promptFileName.includes('.')) {
-      throw new Error("Invalid file name. Please include a file name with an extension (e.g., 'output.txt').");
+      throw new Error(
+        "Invalid file name. Please include a file name with an extension (e.g., 'output.txt')."
+      );
     }
-    
+
     const projectRoot = process.cwd();
     const fullPath = path.join(projectRoot, `interintel/session-samples/${promptFileName}`);
     const directoryPath = path.dirname(fullPath);
@@ -58,7 +56,6 @@ function writeFileFromPrompt(promptFileName, contentToWrite, baseDir) {
     fs.writeFileSync(fullPath, contentToWrite + '\n');
     console.log(`Content written to ${fullPath}`.yellow);
     return true;
-
   } catch (error) {
     console.error(`Error writing file: ${error.message}`.bgRed);
     return false;
@@ -69,7 +66,7 @@ function writeFileFromPrompt(promptFileName, contentToWrite, baseDir) {
 function logFileNames(filePaths) {
   let fileNames = [];
 
-  console.log("")
+  console.log('');
   console.log(`System message`.bgYellow + `: `.yellow);
   filePaths.forEach((filePath) => {
     const fileName = path.basename(filePath);
@@ -84,6 +81,4 @@ function appendToFile(filePath, data) {
 }
 
 // Export the function and the array
-module.exports = {
-  readSpecificFiles, writeFileFromPrompt
-};
+export { readSpecificFiles, writeFileFromPrompt };

package/functions/handleWriteFile.js

@@ -1,5 +1,7 @@
-const chatCompletion = require('../ollama.js');
-const { writeFileFromPrompt } = require('./file-functions.js');
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { chatCompletion } from '../serviceInterface.js';
+import { writeFileFromPrompt } from './file-functions.js';
 
 async function handleWriteFile(config, messages, currentState, userInput, promptFileName) {
   let contentToWrite = '';
@@ -34,8 +36,12 @@ async function handleWriteFile(config, messages, currentState, userInput, prompt
       );
 
       // Extract the response content
-      let contentToWrite = (config.aiService === 'openai') ? 
-        completionResponse.choices[0].message.content : completionResponse;
+      let contentToWrite =
+        config.aiService === 'openai'
+          ? completionResponse.choices[0].message.content
+          : completionResponse;
+
+      const __dirname = path.dirname(fileURLToPath(import.meta.url));
 
       await writeFileFromPrompt(promptFileName, contentToWrite, __dirname); // Assuming this function handles file writing
 
@@ -63,4 +69,4 @@ async function handleWriteFile(config, messages, currentState, userInput, prompt
   return { currentState, messages, promptFileName, contentToWrite, response: '' };
 }
 
-module.exports = { handleWriteFile };
+export { handleWriteFile };

package/index.js

@@ -1,14 +1,16 @@
-const path = require('path');
-const readline = require('readline');
+import path from 'path';
+import readline from 'readline';
+import dotenv from 'dotenv';
+import colors from 'colors';
 const configPath = path.join(process.cwd(), 'interintel.config.js');
-const config = require(configPath);
-require('dotenv').config();
-require('colors');
 
-const { readSpecificFiles } = require('./functions/file-functions.js');
-const { askQuestion } = require('./functions/chat-functions.js');
-const { handleWriteFile } = require('./functions/handleWriteFile.js');
-const chatCompletion = require('./ollama.js');
+import config from './interintel.config.js';
+import { readSpecificFiles } from './functions/file-functions.js';
+import { askQuestion } from './functions/chat-functions.js';
+import { handleWriteFile } from './functions/handleWriteFile.js';
+import { chatCompletion } from './serviceInterface.js';
+
+dotenv.config();
 
 const rl = readline.createInterface({
   input: process.stdin,
@@ -16,7 +18,7 @@ const rl = readline.createInterface({
 });
 
 async function main() {
-  let initialContent = readSpecificFiles(configPath);
+  let initialContent = await readSpecificFiles(configPath);
   let messages = [{ role: 'system', content: initialContent }];
 
   let currentState = null;
@@ -34,12 +36,7 @@ async function main() {
     }
 
     if (userMessage.toLowerCase().startsWith('//writefile') && currentState === null) {
-      let result = await handleWriteFile(
-        config,
-        messages,
-        currentState,
-        ''
-      );
+      let result = await handleWriteFile(config, messages, currentState, '');
       ({ currentState, messages, promptFileName, response } = result); // Update messages array
       console.log(response.yellow);
     } else if (currentState === 'awaitingFileName') {
@@ -71,9 +68,9 @@ async function main() {
       });
       const completion = await chatCompletion(config.aiService, messages, config.aiVersion);
 
-      let botMessage;
+      let botMessage = '';
 
-      if (config.aiService === 'openai') {
+      if (config.aiService === 'openai' || config.aiService === 'mistral') {
         botMessage = completion.choices[0].message.content;
       } else if (config.aiService === 'ollama') {
         // Adjust this line based on how Ollama's response is structured
@@ -86,7 +83,7 @@ async function main() {
       const completion = await chatCompletion(config.aiService, messages, config.aiVersion);
 
       let botMessage;
-      if (config.aiService === 'openai') {
+      if (config.aiService === 'openai' || config.aiService === 'mistral') {
         botMessage = completion.choices[0].message.content;
       } else if (config.aiService === 'ollama') {
         // Adjust based on Ollama's response format
@@ -99,6 +96,4 @@ async function main() {
   }
 }
 
-exports.main = function () {
-  main();
-};
+export { main };

package/interintel.config.js

@@ -1,10 +1,11 @@
-require('dotenv').config();
+import dotenv from 'dotenv';
+dotenv.config();
 
 const config = {
-  apiKey: `${process.env.OPENAI_API_KEY}`,
-  aiService: 'ollama',
-  aiVersion: `mistral`,
-  filePaths: ['./functions/openai-functions.js', './README.md'],
+  apiKey: `${process.env.MISTRAL_API_KEY}`,
+  aiService: 'mistral',
+  aiVersion: `mistral-tiny`,
+  filePaths: ['serviceInterface.js'],
 };
 
-module.exports = config;
+export default config;

package/mistral.js

@@ -0,0 +1,14 @@
+import MistralClient from '@mistralai/mistralai'
+import dotenv from 'dotenv';
+dotenv.config();
+
+const apiKey = process.env.MISTRAL_API_KEY;
+
+const client = new MistralClient(apiKey);
+
+const completion = await client.chat({
+  model: 'mistral-medium',
+  messages: [{role: 'user', content: 'When were you last fine tuned? Please keep to under 25 words'}],
+});
+
+console.log('Chat:', completion.choices[0].message.content);

package/package.json

@@ -1,12 +1,14 @@
 {
   "dependencies": {
+    "@mistralai/mistralai": "^0.0.8",
     "colors": "^1.4.0",
     "dotenv": "^16.3.1",
     "openai": "^4.24.0"
   },
   "name": "interintel",
   "description": "The application `Interintel` is a command line interface (CLI) application implemented in Node.js. It essentially is an interactive communication tool between the user and an AI model, only openai models for now.",
-  "version": "1.0.21",
+  "version": "1.0.22",
+  "type": "module",
   "main": "index.js",
   "directories": {
     "doc": "docs"

package/{ollama.js → serviceInterface.js}

@@ -1,23 +1,25 @@
-const path = require('path');
-const fetch = require('node-fetch');
-const OpenAI = require('openai');
+import path from 'path';
+import fetch from 'node-fetch';
+import OpenAI from 'openai';
+import MistralClient from '@mistralai/mistralai';
+
 const configPath = path.join(process.cwd(), 'interintel.config.js');
-const config = require(configPath);
+
+let config;
+try {
+  const importedModule = await import(configPath);
+  config = importedModule.default;
+} catch (error) {
+  console.error('Failed to import config:', error);
+}
+
+const mistralClient = new MistralClient(config.apiKey);
 
 const openai = new OpenAI({
   apiKey: config.apiKey,
   model: config.aiVersion,
 });
 
-let ai = 'ollama';
-let messages = [
-  {
-    role: 'assistant',
-    content: 'please use a respectful tone',
-  },
-];
-let model = 'mistral';
-
 async function chatCompletion(aiService, messages, model) {
   try {
     let response;
@@ -30,6 +32,16 @@ async function chatCompletion(aiService, messages, model) {
       });
 
       return response;
+      
+    } else if (aiService === 'mistral') {
+      let chatResponse;
+
+      chatResponse = await mistralClient.chat({
+        model: model, // or a specific model you wish to use
+        messages: messages,
+      });
+
+      return chatResponse;
     } else if (aiService === 'ollama') {
       // Ollama specific code
       let data = {
@@ -57,4 +69,4 @@ async function chatCompletion(aiService, messages, model) {
   }
 }
 
-module.exports = chatCompletion;
+export { chatCompletion };