aether-ai-agent-cli 1.2.0__tar.gz → 1.3.0__tar.gz

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aether-ai-agent-cli
-Version: 1.2.0
+Version: 1.3.0
 Summary: Aether Core AI v110 — Universal AI Gateway CLI (Python Wrapper)
 Home-page: https://github.com/Krylo-60/aether-ai-cli
 Author: Krishiv PB

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/aether_ai_agent_cli.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aether-ai-agent-cli
-Version: 1.2.0
+Version: 1.3.0
 Summary: Aether Core AI v110 — Universal AI Gateway CLI (Python Wrapper)
 Home-page: https://github.com/Krylo-60/aether-ai-cli
 Author: Krishiv PB

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/aether_ai_agent_cli.egg-info/SOURCES.txt

@@ -23,6 +23,7 @@ aether_pip/node_project/src/ai/fallback.js
 aether_pip/node_project/src/ai/google.js
 aether_pip/node_project/src/ai/providers.js
 aether_pip/node_project/src/ai/router.js
+aether_pip/node_project/src/ai/tokens.js
 aether_pip/node_project/src/ai/universal.js
 aether_pip/node_project/src/ai/xai.js
 aether_pip/node_project/src/ui/banner.js

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/aether_pip/node_project/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@krishivpb60/aether-ai-cli",
-  "version": "1.2.0",
+  "version": "1.3.0",
   "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": {

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/aether_pip/node_project/src/ai/router.js

@@ -11,6 +11,7 @@ import {
   callAnthropic,
   callCohere,
 } from "./universal.js";
+import { estimateTokens, recordTokenUsage } from "./tokens.js";
 
 /**
  * Routes a prompt through the universal AI failover mesh.
@@ -32,7 +33,10 @@ export async function routePrompt(prompt, systemPrompt, config, onToken, history
   if (mathExpr) {
     const mathResult = solveMath(mathExpr);
     if (mathResult) {
-      return { ...mathResult, provider: "local", node: 0 };
+      const pTokens = estimateTokens(systemPrompt + prompt);
+      const cTokens = estimateTokens(mathResult.text);
+      const usage = recordTokenUsage("local-math", pTokens, cTokens);
+      return { ...mathResult, provider: "local", node: 0, usage };
     }
   }
 
@@ -54,7 +58,10 @@ export async function routePrompt(prompt, systemPrompt, config, onToken, history
   // ── No providers configured → Krylo ────────────────────
   if (active.length === 0) {
     const kryloReply = generateKryloReply(prompt);
-    return { ...kryloReply, provider: "krylo-fallback", node: 0 };
+    const pTokens = estimateTokens(systemPrompt + prompt);
+    const cTokens = estimateTokens(kryloReply.text);
+    const usage = recordTokenUsage("krylo-local", pTokens, cTokens);
+    return { ...kryloReply, provider: "krylo-fallback", node: 0, usage };
   }
 
   // ── Try each provider in order ──────────────────────────
@@ -96,7 +103,11 @@ export async function routePrompt(prompt, systemPrompt, config, onToken, history
           );
       }
 
-      return { ...result, node: nodeIndex };
+      const pTokens = estimateTokens(systemPrompt + prompt + history.map(h => h.content).join(""));
+      const cTokens = estimateTokens(result.text);
+      const usage = recordTokenUsage(result.model, pTokens, cTokens);
+
+      return { ...result, node: nodeIndex, usage };
     } catch (err) {
       errors.push(`[Node ${nodeIndex} ${provider.name}] ${err.message}`);
       nodeIndex++;
@@ -105,10 +116,14 @@ export async function routePrompt(prompt, systemPrompt, config, onToken, history
 
   // ── Final Fallback: Krylo Companion ─────────────────────
   const kryloReply = generateKryloReply(prompt);
+  const pTokens = estimateTokens(systemPrompt + prompt + history.map(h => h.content).join(""));
+  const cTokens = estimateTokens(kryloReply.text);
+  const usage = recordTokenUsage("krylo-local", pTokens, cTokens);
   return {
     ...kryloReply,
     provider: "krylo-fallback",
     node: 0,
     errors,
+    usage,
   };
 }

aether_ai_agent_cli-1.3.0/aether_pip/node_project/src/ai/tokens.js

@@ -0,0 +1,101 @@
+// ═══════════════════════════════════════════════════════════
+// AETHER AI CLI — Real-time Token & Telemetry Tracker
+// Tracks input/output token counts and session usage metrics.
+// ═══════════════════════════════════════════════════════════
+
+// Session token accumulator
+let sessionTokens = {
+  prompt: 0,
+  completion: 0,
+  total: 0,
+  exchanges: 0,
+};
+
+// Model-by-model breakdown
+const modelBreakdown = {};
+
+/**
+ * Heuristically estimates the token count of a string based on character length.
+ * Standard rule of thumb: 1 token ≈ 4 characters.
+ * @param {string} text
+ * @returns {number}
+ */
+export function estimateTokens(text) {
+  if (!text) return 0;
+  return Math.ceil(text.length / 4);
+}
+
+/**
+ * Returns a copy of the current session token stats.
+ * @returns {{prompt: number, completion: number, total: number, exchanges: number}}
+ */
+export function getSessionTokenStats() {
+  return { ...sessionTokens };
+}
+
+/**
+ * Resets all accumulated session token statistics and breakdowns.
+ */
+export function resetSessionTokenStats() {
+  sessionTokens = {
+    prompt: 0,
+    completion: 0,
+    total: 0,
+    exchanges: 0,
+  };
+  // Clear object properties safely without losing reference
+  for (const key of Object.keys(modelBreakdown)) {
+    delete modelBreakdown[key];
+  }
+}
+
+/**
+ * Records token usage for a query.
+ * @param {string} model - The model name
+ * @param {number} promptTokens - Input prompt tokens count
+ * @param {number} completionTokens - Output completion tokens count
+ * @returns {{promptTokens: number, completionTokens: number, totalTokens: number}}
+ */
+export function recordTokenUsage(model, promptTokens, completionTokens) {
+  const modelName = model || "unknown-model";
+  const totalTokens = promptTokens + completionTokens;
+
+  // Update session totals
+  sessionTokens.prompt += promptTokens;
+  sessionTokens.completion += completionTokens;
+  sessionTokens.total += totalTokens;
+  sessionTokens.exchanges += 1;
+
+  // Update model breakdown
+  if (!modelBreakdown[modelName]) {
+    modelBreakdown[modelName] = {
+      prompt: 0,
+      completion: 0,
+      total: 0,
+      exchanges: 0,
+    };
+  }
+  modelBreakdown[modelName].prompt += promptTokens;
+  modelBreakdown[modelName].completion += completionTokens;
+  modelBreakdown[modelName].total += totalTokens;
+  modelBreakdown[modelName].exchanges += 1;
+
+  return {
+    promptTokens,
+    completionTokens,
+    totalTokens,
+  };
+}
+
+/**
+ * Returns the model-by-model token usage breakdown.
+ * @returns {object}
+ */
+export function getBreakdownByModel() {
+  // Deep clone modelBreakdown
+  const clone = {};
+  for (const [key, value] of Object.entries(modelBreakdown)) {
+    clone[key] = { ...value };
+  }
+  return clone;
+}

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/aether_pip/node_project/src/chat.js

@@ -46,6 +46,8 @@ import { parseFile, formatContext } from "./file-parser.js";
 import { runMainframeHack } from "./ai/fallback.js";
 import { AGENT_INSTRUCTIONS } from "./agent.js";
 import { checkForUpdates } from "./updater.js";
+import { getSessionTokenStats, getBreakdownByModel, resetSessionTokenStats } from "./ai/tokens.js";
+
 
 
 // Configure marked dynamically for terminal output
@@ -71,6 +73,10 @@ export async function startChat(options = {}) {
   
   // Run update check
   await checkForUpdates();
+
+  // Reset token stats for the new session
+  resetSessionTokenStats();
+
   
   // Set theme from configuration
   const theme = aiConfig.THEME || "cyberpunk";
@@ -130,7 +136,7 @@ export async function startChat(options = {}) {
       "/help", "/mode", "/modes", "/attach", "/files", "/clear",
       "/providers", "/export", "/status", "/copy", "/exit", "/quit",
       "/theme", "/themes", "/history-clear", "/game", "/abort", "/cmd", "/write",
-      "/commit", "/run", "/history", "/autopilot"
+      "/commit", "/run", "/history", "/autopilot", "/tokens"
     ];
     const customCmds = aiConfig.CUSTOM_COMMANDS || {};
     const commands = [...builtIn, ...Object.keys(customCmds)];
@@ -303,11 +309,19 @@ export async function startChat(options = {}) {
           }
         }
 
+        const showTokens = aiConfig.SHOW_TOKENS !== "false";
+        let tokensText = "";
+        if (showTokens && result.usage) {
+          const { promptTokens, completionTokens } = result.usage;
+          tokensText = ` • ${promptTokens.toLocaleString()} in / ${completionTokens.toLocaleString()} out tokens`;
+        }
+
         console.log(separator("─"));
         console.log(
           "  " + colors.dim(`Node ${result.node} • ${result.provider}`) +
           (result.model ? colors.dim(` • ${result.model}`) : "") +
           colors.dim(` • ${elapsedSec}s${speedText}`) +
+          colors.dim(tokensText) +
           colors.dim(` • ${Math.floor(history.length / 2)} exchanges`)
         );
         console.log("");
@@ -409,7 +423,7 @@ export async function startChat(options = {}) {
         "/", "/help", "/mode", "/modes", "/attach", "/files", "/clear",
         "/providers", "/export", "/status", "/copy", "/exit", "/quit",
         "/theme", "/themes", "/history-clear", "/game", "/abort", "/cmd",
-        "/guess", "/write", "/commit", "/run", "/history", "/autopilot"
+        "/guess", "/write", "/commit", "/run", "/history", "/autopilot", "/tokens"
       ];
       
       const customCmds = aiConfig.CUSTOM_COMMANDS || {};
@@ -556,6 +570,10 @@ async function handleCommand(input, ctx) {
       await handleAutopilotSwitch(args, ctx);
       break;
 
+    case "/tokens":
+      await handleTokensDisplay(ctx);
+      break;
+
     case "/exit":
     case "/quit":
       ctx.rl.close();
@@ -587,6 +605,7 @@ function showHelp(aiConfig) {
   console.log(keyValue("/history", "List, switch, and resume past interactive chat sessions"));
   console.log(keyValue("/history-clear", "Clear saved persistent chat history"));
   console.log(keyValue("/autopilot <mode>", "View or switch agent autopilot level (off, safe, workspace, machine)"));
+  console.log(keyValue("/tokens", "View detailed session token usage and exchanges telemetry"));
   console.log(keyValue("/game", "Start the local mainframe hacking mini-game"));
   console.log(keyValue("/copy", "Copy the last assistant response to clipboard"));
   console.log(keyValue("/cmd <list|add|remove>", "Manage custom command shortcuts"));
@@ -1126,7 +1145,7 @@ async function handleCustomCommands(args, ctx) {
     const builtIn = [
       "/help", "/mode", "/modes", "/attach", "/files", "/clear",
       "/providers", "/export", "/status", "/copy", "/exit", "/quit",
-      "/theme", "/themes", "/history-clear", "/game", "/abort", "/cmd", "/guess"
+      "/theme", "/themes", "/history-clear", "/game", "/abort", "/cmd", "/guess", "/tokens"
     ];
     
     if (builtIn.includes(name.toLowerCase())) {
@@ -1338,3 +1357,40 @@ async function handleRunCommand(args, ctx) {
     });
   });
 }
+
+/**
+ * Interactive display of session token usage statistics.
+ */
+async function handleTokensDisplay(ctx) {
+  const stats = getSessionTokenStats();
+  const breakdown = getBreakdownByModel();
+
+  console.log("\n" + separator("━"));
+  console.log(colors.accent.bold("  ★  AETHER SESSION TOKEN TELEMETRY  ★"));
+  console.log(separator("─"));
+
+  const models = Object.keys(breakdown);
+  if (models.length === 0) {
+    console.log(colors.muted("  No queries executed in this session yet."));
+  } else {
+    // Print header
+    console.log(
+      colors.brand("  " + "Model".padEnd(35) + "Prompt".padStart(10) + "Completion".padStart(12) + "Total".padStart(10))
+    );
+    console.log(colors.dim("  " + "─".repeat(67)));
+    for (const [model, data] of Object.entries(breakdown)) {
+      const truncatedModel = model.length > 33 ? model.slice(0, 30) + "..." : model;
+      console.log(
+        "  " + colors.text(truncatedModel.padEnd(35)) +
+        colors.brand(data.prompt.toLocaleString().padStart(10)) +
+        colors.brand(data.completion.toLocaleString().padStart(12)) +
+        colors.accent.bold(data.total.toLocaleString().padStart(10))
+      );
+    }
+  }
+
+  console.log(separator("─"));
+  console.log("  " + colors.accent("Total Exchanges:") + colors.text(` ${stats.exchanges}`));
+  console.log("  " + colors.accent("Total Tokens:") + colors.text(`    Prompt: ${stats.prompt.toLocaleString()} | Completion: ${stats.completion.toLocaleString()} | Sum: `) + colors.brand.bold(stats.total.toLocaleString()));
+  console.log(separator("━") + "\n");
+}

{aether_ai_agent_cli-1.2.0 → aether_ai_agent_cli-1.3.0}/aether_pip/node_project/src/config.js

@@ -175,7 +175,8 @@ export function isValidConfigKey(key) {
   // Accept any API key or model override
   const allowedSpecialKeys = [
     "THEME", "CUSTOM_COMMANDS", "AUTOPILOT", 
-    "AUTO_UPDATE", "SHOW_HIGHLIGHTS", "LAST_UPDATE_CHECK", "LAST_NOTIFIED_VERSION"
+    "AUTO_UPDATE", "SHOW_HIGHLIGHTS", "LAST_UPDATE_CHECK", "LAST_NOTIFIED_VERSION",
+    "SHOW_TOKENS"
   ];
   if (upper.endsWith("_API_KEY") || upper.endsWith("_API_KEYS") || upper.endsWith("_MODEL") || allowedSpecialKeys.includes(upper)) {
     return true;