@krishivpb60/aether-ai-cli 1.1.0 → 1.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@krishivpb60/aether-ai-cli",
-  "version": "1.1.0",
+  "version": "1.1.2",
   "description": "Aether Core AI — A cyberpunk command-line AI assistant with multi-mode reasoning, 12-node failover mesh, file context injection, and offline fallbacks.",
   "main": "src/cli.js",
   "bin": {

package/src/ai/router.js

@@ -26,7 +26,7 @@ import {
  * @param {object} config - Flat config object with all API keys
  * @returns {Promise<{ text: string, provider: string, model?: string, node: number, type?: string }>}
  */
-export async function routePrompt(prompt, systemPrompt, config, onToken) {
+export async function routePrompt(prompt, systemPrompt, config, onToken, history = []) {
   // ── Node 0: Local Math Solver ───────────────────────────
   const mathExpr = detectMathExpression(prompt);
   if (mathExpr) {
@@ -71,20 +71,20 @@ export async function routePrompt(prompt, systemPrompt, config, onToken) {
           result = await callOpenAICompatible(
             prompt, systemPrompt, apiKey,
             provider.baseUrl, model, provider.name,
-            onToken
+            onToken, history
           );
           break;
 
         case "custom-google":
-          result = await callGoogleGemini(prompt, systemPrompt, apiKey, model, onToken);
+          result = await callGoogleGemini(prompt, systemPrompt, apiKey, model, onToken, history);
           break;
 
         case "custom-anthropic":
-          result = await callAnthropic(prompt, systemPrompt, apiKey, model, onToken);
+          result = await callAnthropic(prompt, systemPrompt, apiKey, model, onToken, history);
           break;
 
         case "custom-cohere":
-          result = await callCohere(prompt, systemPrompt, apiKey, model, onToken);
+          result = await callCohere(prompt, systemPrompt, apiKey, model, onToken, history);
           break;
 
         default:
@@ -92,7 +92,7 @@ export async function routePrompt(prompt, systemPrompt, config, onToken) {
           result = await callOpenAICompatible(
             prompt, systemPrompt, apiKey,
             provider.baseUrl, model, provider.name,
-            onToken
+            onToken, history
           );
       }
 

package/src/ai/universal.js

@@ -13,12 +13,17 @@
  * @param {string} providerName - For error messages
  * @returns {Promise<{ text: string, provider: string, model: string }>}
  */
-export async function callOpenAICompatible(prompt, systemPrompt, apiKey, baseUrl, model, providerName, onToken) {
+export async function callOpenAICompatible(prompt, systemPrompt, apiKey, baseUrl, model, providerName, onToken, history = []) {
   const isStreaming = typeof onToken === "function";
+  const formattedHistory = history.map(h => ({
+    role: h.role === "assistant" ? "assistant" : "user",
+    content: h.content
+  }));
   const body = {
     model,
     messages: [
       { role: "system", content: systemPrompt },
+      ...formattedHistory,
       { role: "user", content: prompt },
     ],
     temperature: 0.7,
@@ -116,9 +121,10 @@ export async function callOpenAICompatible(prompt, systemPrompt, apiKey, baseUrl
  * @param {string} model - Model name
  * @returns {Promise<{ text: string, provider: string, model: string }>}
  */
-export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "gemini-2.5-flash", onToken) {
+export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "gemini-2.5-flash", onToken, history = []) {
   const BASE = "https://generativelanguage.googleapis.com/v1beta/models";
   const isStreaming = typeof onToken === "function";
+  let currentHistory = [...history];
 
   if (isStreaming) {
     let fullText = "";
@@ -128,9 +134,16 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
 
     while (continuations <= MAX) {
       const url = `${BASE}/${model}:streamGenerateContent?key=${apiKey}`;
+      const formattedHistory = currentHistory.map(h => ({
+        role: h.role === "assistant" ? "model" : "user",
+        parts: [{ text: h.content }]
+      }));
       const body = {
         systemInstruction: { parts: [{ text: systemPrompt }] },
-        contents: [{ role: "user", parts: [{ text: currentPrompt }] }],
+        contents: [
+          ...formattedHistory,
+          { role: "user", parts: [{ text: currentPrompt }] }
+        ],
         generationConfig: { temperature: 0.7, maxOutputTokens: 8192 },
       };
 
@@ -145,6 +158,8 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
         throw new Error(`Gemini API error (${response.status}): ${response.statusText}. ${errorBody}`);
       }
 
+      let streamedTextInThisTurn = "";
+
       if (!response.body || typeof response.body.getReader !== "function") {
         // Fallback to non-streaming if response body is not streamable (e.g. in unit tests)
         const data = await response.json();
@@ -164,9 +179,12 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
         if (chunkText) {
           onToken(chunkText);
           fullText += chunkText;
+          streamedTextInThisTurn += chunkText;
         }
 
         if (finishReason === "MAX_TOKENS" && continuations < MAX) {
+          currentHistory.push({ role: "user", content: currentPrompt });
+          currentHistory.push({ role: "assistant", content: streamedTextInThisTurn });
           continuations++;
           currentPrompt = "Continue your previous response from exactly where you left off.";
         } else {
@@ -217,6 +235,7 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
                     if (text) {
                       onToken(text);
                       fullText += text;
+                      streamedTextInThisTurn += text;
                     }
                   } catch (e) {
                     // Ignore parse errors
@@ -231,6 +250,8 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
         }
 
         if (finishReason === "MAX_TOKENS" && continuations < MAX) {
+          currentHistory.push({ role: "user", content: currentPrompt });
+          currentHistory.push({ role: "assistant", content: streamedTextInThisTurn });
           continuations++;
           currentPrompt = "Continue your previous response from exactly where you left off.";
         } else {
@@ -249,9 +270,16 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
 
     while (continuations <= MAX) {
       const url = `${BASE}/${model}:generateContent?key=${apiKey}`;
+      const formattedHistory = currentHistory.map(h => ({
+        role: h.role === "assistant" ? "model" : "user",
+        parts: [{ text: h.content }]
+      }));
       const body = {
         systemInstruction: { parts: [{ text: systemPrompt }] },
-        contents: [{ role: "user", parts: [{ text: currentPrompt }] }],
+        contents: [
+          ...formattedHistory,
+          { role: "user", parts: [{ text: currentPrompt }] }
+        ],
         generationConfig: { temperature: 0.7, maxOutputTokens: 8192 },
       };
 
@@ -278,6 +306,8 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
       fullText += chunkText;
 
       if (candidate.finishReason === "MAX_TOKENS" && continuations < MAX) {
+        currentHistory.push({ role: "user", content: currentPrompt });
+        currentHistory.push({ role: "assistant", content: chunkText });
         continuations++;
         currentPrompt = "Continue your previous response from exactly where you left off.";
       } else {
@@ -298,14 +328,21 @@ export async function callGoogleGemini(prompt, systemPrompt, apiKey, model = "ge
  * @param {string} model - Model name
  * @returns {Promise<{ text: string, provider: string, model: string }>}
  */
-export async function callAnthropic(prompt, systemPrompt, apiKey, model = "claude-sonnet-4-20250514", onToken) {
+export async function callAnthropic(prompt, systemPrompt, apiKey, model = "claude-sonnet-4-20250514", onToken, history = []) {
   const url = "https://api.anthropic.com/v1/messages";
   const isStreaming = typeof onToken === "function";
+  const formattedHistory = history.map(h => ({
+    role: h.role === "assistant" ? "assistant" : "user",
+    content: h.content
+  }));
   const body = {
     model,
     max_tokens: 4096,
     system: systemPrompt,
-    messages: [{ role: "user", content: prompt }],
+    messages: [
+      ...formattedHistory,
+      { role: "user", content: prompt }
+    ],
     ...(isStreaming ? { stream: true } : {}),
   };
 
@@ -384,13 +421,18 @@ export async function callAnthropic(prompt, systemPrompt, apiKey, model = "claud
  * @param {string} model - Model name
  * @returns {Promise<{ text: string, provider: string, model: string }>}
  */
-export async function callCohere(prompt, systemPrompt, apiKey, model = "command-r-plus", onToken) {
+export async function callCohere(prompt, systemPrompt, apiKey, model = "command-r-plus", onToken, history = []) {
   const url = "https://api.cohere.com/v2/chat";
   const isStreaming = typeof onToken === "function";
+  const formattedHistory = history.map(h => ({
+    role: h.role === "assistant" ? "assistant" : "user",
+    content: h.content
+  }));
   const body = {
     model,
     messages: [
       { role: "system", content: systemPrompt },
+      ...formattedHistory,
       { role: "user", content: prompt },
     ],
     ...(isStreaming ? { stream: true } : {}),

package/src/chat.js

@@ -46,8 +46,8 @@ const getMarked = () => new Marked(markedTerminal({
   showSectionPrefix: false,
   code: (c) => colors.orange(c),
   codespan: (c) => colors.accent3(c),
-  heading: (h) => colors.accent(h).bold,
-  strong: (s) => colors.magenta(s).bold,
+  heading: (h) => colors.accent.bold(h),
+  strong: (s) => colors.magenta.bold(s),
   em: chalk.italic,
   hr: (h) => colors.dim(h),
 }));
@@ -239,7 +239,7 @@ export async function startChat(options = {}) {
     };
 
     try {
-      const result = await routePrompt(fullPrompt, currentMode.systemPrompt, aiConfig, onToken);
+      const result = await routePrompt(fullPrompt, currentMode.systemPrompt, aiConfig, onToken, history);
       spinner.stop();
 
       // Store in history

package/src/cli.js

@@ -44,8 +44,8 @@ const getMarked = () => new Marked(markedTerminal({
   showSectionPrefix: false,
   code: (c) => colors.orange(c),
   codespan: (c) => colors.accent3(c),
-  heading: (h) => colors.accent(h).bold,
-  strong: (s) => colors.magenta(s).bold,
+  heading: (h) => colors.accent.bold(h),
+  strong: (s) => colors.magenta.bold(s),
   em: chalk.italic,
   hr: (h) => colors.dim(h),
 }));

package/test/router.test.js

@@ -171,4 +171,54 @@ test("Universal AI Router Suite", async (t) => {
     assert.ok(result.errors[0].includes("Node 1 Groq"));
     assert.ok(result.errors[1].includes("Node 2 OpenAI"));
   });
+
+  await t.test("routePrompt forwards chat history to OpenAI and Google payloads correctly", async () => {
+    const history = [
+      { role: "user", content: "What is your name?" },
+      { role: "assistant", content: "I am Aether." }
+    ];
+
+    // 1. Test Groq (OpenAI-compatible) receives history
+    globalThis.fetch = async (url, options) => {
+      fetchCalls.push({ url, options });
+      return {
+        ok: true,
+        json: async () => ({
+          choices: [{ message: { content: "Groq reply" } }],
+        }),
+      };
+    };
+
+    await routePrompt("How are you?", "Sys prompt", { GROQ_API_KEY: "groq-key" }, null, history);
+    assert.strictEqual(fetchCalls.length, 1);
+    const groqBody = JSON.parse(fetchCalls[0].options.body);
+    assert.deepStrictEqual(groqBody.messages, [
+      { role: "system", content: "Sys prompt" },
+      { role: "user", content: "What is your name?" },
+      { role: "assistant", content: "I am Aether." },
+      { role: "user", content: "How are you?" }
+    ]);
+
+    fetchCalls = [];
+
+    // 2. Test Google Gemini receives history
+    globalThis.fetch = async (url, options) => {
+      fetchCalls.push({ url, options });
+      return {
+        ok: true,
+        json: async () => ({
+          candidates: [{ content: { parts: [{ text: "Gemini reply" }] } }],
+        }),
+      };
+    };
+
+    await routePrompt("How are you?", "Sys prompt", { GOOGLE_API_KEYS: "google-key" }, null, history);
+    assert.strictEqual(fetchCalls.length, 1);
+    const googleBody = JSON.parse(fetchCalls[0].options.body);
+    assert.deepStrictEqual(googleBody.contents, [
+      { role: "user", parts: [{ text: "What is your name?" }] },
+      { role: "model", parts: [{ text: "I am Aether." }] },
+      { role: "user", parts: [{ text: "How are you?" }] }
+    ]);
+  });
 });