igris-soul 1.1.0 → 1.1.2

package/cli.cjs

@@ -0,0 +1,23 @@
+#!/usr/bin/env node
+
+const { spawn } = require("node:child_process");
+const { fileURLToPath } = require("node:url");
+const path = require("node:path");
+
+const packageDir = path.dirname(
+  fileURLToPath(require.resolve("./package.json")),
+);
+
+const indexPath = path.join(packageDir, "index.js");
+
+const child = spawn(process.execPath, [indexPath, ...process.argv.slice(2)], {
+  stdio: "inherit",
+});
+
+child.on("exit", (code, signal) => {
+  if (signal) {
+    process.kill(process.pid, signal);
+  } else {
+    process.exit(code ?? 0);
+  }
+});

package/index.js

@@ -10,6 +10,7 @@ import {
 import { homedir } from "node:os";
 import { join } from "node:path";
 import rl from "readline/promises";
+
 import {
   HumanMessage,
   AIMessage,
@@ -18,81 +19,102 @@ import {
   createAgent,
   AIMessageChunk,
 } from "langchain";
+
 import * as z from "zod";
 
 /* =========================================================
-   IGRIS CONFIGURATION
+   IGRIS DIRECTORIES & FILES
    ========================================================= */
 
 const igrisDir = join(homedir(), ".igris");
 const envPath = join(igrisDir, ".env");
 const userDataPath = join(igrisDir, "user.json");
 
-// Create ~/.igris automatically
+/* =========================================================
+   READLINE
+   ========================================================= */
+
+const readline = rl.createInterface({
+  input: process.stdin,
+  output: process.stdout,
+});
+
+/* =========================================================
+   CREATE IGRIS DIRECTORY
+   ========================================================= */
+
 await mkdir(igrisDir, { recursive: true });
 
-// Check whether .env exists
+/* =========================================================
+   CREATE .ENV IF IT DOESN'T EXIST
+   ========================================================= */
+
 async function ensureEnvFile() {
   try {
     await access(envPath);
   } catch {
     await writeFile(
       envPath,
-      `GENAI_API_KEY=\nTAVILY_API_KEY=\n`,
+      "GENAI_API_KEY=\nTAVILY_API_KEY=\n",
       "utf8",
     );
-
-    console.log(`
-Igris configuration created.
-
-Configuration file:
-${envPath}
-
-You will now be asked for your API keys.
-`);
   }
 }
 
 await ensureEnvFile();
 
-// Load environment variables from ~/.igris/.env
+/* =========================================================
+   LOAD ENVIRONMENT VARIABLES
+   ========================================================= */
+
 config({
   path: envPath,
+  override: true,
   quiet: true,
 });
 
 /* =========================================================
-   READ / SAVE API KEYS
+   API KEY SETUP
    ========================================================= */
 
 async function getApiKey(variableName, displayName) {
   let value = process.env[variableName]?.trim();
 
+  // Already configured
   if (value) {
     return value;
   }
 
-  const key = await readline.question(
-    `${displayName}: `,
-  );
+  console.log("");
 
-  value = key.trim();
+  value = (
+    await readline.question(`${displayName}: `)
+  ).trim();
 
   if (!value) {
-    console.error(`\n${displayName} cannot be empty.`);
+    console.error(
+      `\n${displayName} cannot be empty.`,
+    );
+
+    readline.close();
     process.exit(1);
   }
 
-  // Read existing .env
   let envContent = "";
 
   try {
-    envContent = await readFile(envPath, "utf8");
+    envContent = await readFile(
+      envPath,
+      "utf8",
+    );
   } catch {
     envContent = "";
   }
 
-  const regex = new RegExp(`^${variableName}=.*$`, "m");
+  const regex = new RegExp(
+    `^${variableName}=.*$`,
+    "m",
+  );
 
   if (regex.test(envContent)) {
     envContent = envContent.replace(
@@ -100,27 +122,23 @@ async function getApiKey(variableName, displayName) {
       `${variableName}=${value}`,
     );
   } else {
-    envContent += `\n${variableName}=${value}\n`;
+    envContent += `\n${variableName}=${value}`;
   }
 
-  await writeFile(envPath, envContent.trim() + "\n", "utf8");
+  await writeFile(
+    envPath,
+    envContent.trim() + "\n",
+    "utf8",
+  );
 
+  // Make it immediately available to process.env
   process.env[variableName] = value;
 
   return value;
 }
 
 /* =========================================================
-   READLINE
-   ========================================================= */
-
-const readline = rl.createInterface({
-  input: process.stdin,
-  output: process.stdout,
-});
-
-/* =========================================================
-   API CONFIGURATION
+   GET REQUIRED API KEYS
    ========================================================= */
 
 const genaiApiKey = await getApiKey(
@@ -141,7 +159,11 @@ const tavly = tavily({
   apiKey: tavilyApiKey,
 });
 
-async function getLetestInfos({ query }) {
+/* =========================================================
+   LATEST INFORMATION TOOL
+   ========================================================= */
+
+async function getLatestInfos({ query }) {
   const response = await tavly.search(query, {
     searchDepth: "fast",
     maxResults: 3,
@@ -150,52 +172,70 @@ async function getLetestInfos({ query }) {
 
   const results = response.results;
 
-  const content = results
+  return results
     .map((result) => result.content)
     .join("\n\n");
-
-  return content;
 }
 
-const getLatestInfosTool = tool(getLetestInfos, {
-  name: "get_latest_infos",
-  description:
-    "Get the latest updates from India, including news, policies, and stock market trends.",
-  schema: z.object({
-    query: z
-      .string()
-      .describe(
-        "The query for which you want to get the latest updates from India.",
-      ),
-  }),
-});
+const getLatestInfosTool = tool(
+  getLatestInfos,
+  {
+    name: "get_latest_infos",
+
+    description:
+      "Get the latest updates from India, including news, policies, and stock market trends.",
+
+    schema: z.object({
+      query: z
+        .string()
+        .describe(
+          "The query for which you want to get the latest updates from India.",
+        ),
+    }),
+  },
+);
 
 /* =========================================================
-   USER DATA
+   USER NAME
    ========================================================= */
 
 async function getUserName() {
   try {
     const userData = JSON.parse(
-      await readFile(userDataPath, "utf8"),
+      await readFile(
+        userDataPath,
+        "utf8",
+      ),
     );
 
-    return userData.name;
+    if (userData.name?.trim()) {
+      return userData.name.trim();
+    }
   } catch {
-    const name = (
-      await readline.question(
-        "What is your name, My Lord? ",
-      )
-    ).trim();
+    // User data doesn't exist yet.
+  }
 
-    await writeFile(
-      userDataPath,
-      JSON.stringify({ name }, null, 2),
-      "utf8",
-    );
+  const name = (
+    await readline.question(
+      "What is your name, My Lord? ",
+    )
+  ).trim();
 
-    return name;
+  if (!name) {
+    return "My Lord";
   }
+
+  await writeFile(
+    userDataPath,
+    JSON.stringify(
+      { name },
+      null,
+      2,
+    ),
+    "utf8",
+  );
+
+  return name;
 }
 
 const userName = await getUserName();
@@ -244,7 +284,7 @@ Use them only when needed for styling, context, or a poetic response.
 ];
 
 /* =========================================================
-   START IGRIS
+   STARTUP MESSAGE
    ========================================================= */
 
 console.log(`
@@ -255,6 +295,7 @@ console.log(`
 
 Hi, My Lord, I am Igris, your loyal servant.
 How may I assist you today?
+
 `);
 
 /* =========================================================
@@ -263,21 +304,27 @@ How may I assist you today?
 
 try {
   while (true) {
-    const prompt = await readline.question("You: ");
+    const prompt = await readline.question(
+      "You: ",
+    );
+
+    const trimmedPrompt = prompt.trim();
 
     if (
       ["exit", "quit"].includes(
-        prompt.trim().toLowerCase(),
+        trimmedPrompt.toLowerCase(),
       )
     ) {
       break;
     }
 
-    if (!prompt.trim()) {
+    if (!trimmedPrompt) {
       continue;
     }
 
-    messages.push(new HumanMessage(prompt));
+    messages.push(
+      new HumanMessage(trimmedPrompt),
+    );
 
     const stream = await agent.stream(
       {
@@ -290,7 +337,7 @@ try {
 
     let aiResponse = "";
 
-    console.log("Igris: ");
+    console.log("Igris:");
 
     for await (const [chunk] of stream) {
       if (chunk instanceof AIMessageChunk) {
@@ -299,11 +346,20 @@ try {
       }
     }
 
-    messages.push(new AIMessage(aiResponse));
+    messages.push(
+      new AIMessage(aiResponse),
+    );
 
     process.stdout.write("\n\n");
   }
+} catch (error) {
+  console.error(
+    "\nIgris encountered an error:",
+  );
+
+  console.error(
+    error?.message || error,
+  );
 } finally {
   readline.close();
-}
-
+}

package/package.json

@@ -1,20 +1,22 @@
 {
   "name": "igris-soul",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "A terminal AI assistant powered by Gemini and Tavily.",
   "license": "ISC",
   "author": "Barshan Majumdar",
   "type": "module",
   "main": "index.js",
   "bin": {
-    "Arise": "index.js",
-    "arise": "index.js",
-    "igris-soul": "index.js",
-    "igris": "index.js"
+    "Arise": "cli.cjs",
+    "arise": "cli.cjs",
+    "igris-soul": "cli.cjs",
+    "igris": "cli.cjs"
   },
   "files": [
-    "index.js",
-    ".env.example"
+  "index.js",
+  "cli.cjs",
+  ".env.example",
+  "README.md"
   ],
   "engines": {
     "node": ">=20"