nightshift-mcp 2.0.1 → 2.1.0

package/dist/run-logger.js

@@ -0,0 +1,219 @@
+import * as fs from "fs";
+import * as path from "path";
+// ============================================
+// Token Parsing
+// ============================================
+/**
+ * Parse token usage from agent output. Each agent reports differently.
+ */
+export function parseTokensFromOutput(agent, output) {
+    if (!output)
+        return undefined;
+    switch (agent) {
+        case "codex": {
+            // Codex reports tokens at the end of output, e.g. "108,712" or "tokens used: 14079"
+            // Also: "tokens used**" pattern seen in logs
+            const tokensUsedMatch = output.match(/tokens?\s*used[*:]*\s*([\d,]+)/i);
+            if (tokensUsedMatch) {
+                const total = parseInt(tokensUsedMatch[1].replace(/,/g, ""), 10);
+                if (!isNaN(total))
+                    return { total, model: "gpt-5.4" };
+            }
+            // Bare large number at end of output (Codex sometimes just prints the count)
+            const bareNumberMatch = output.match(/\n([\d,]{4,})\s*$/);
+            if (bareNumberMatch) {
+                const total = parseInt(bareNumberMatch[1].replace(/,/g, ""), 10);
+                if (!isNaN(total) && total > 100)
+                    return { total, model: "gpt-5.4" };
+            }
+            return undefined;
+        }
+        case "ollama": {
+            // Ollama API returns: prompt_eval_count, eval_count in JSON responses
+            const promptEvalMatch = output.match(/"prompt_eval_count"\s*:\s*(\d+)/);
+            const evalMatch = output.match(/"eval_count"\s*:\s*(\d+)/);
+            if (promptEvalMatch || evalMatch) {
+                const input = promptEvalMatch ? parseInt(promptEvalMatch[1], 10) : undefined;
+                const outputTokens = evalMatch ? parseInt(evalMatch[1], 10) : undefined;
+                const total = (input || 0) + (outputTokens || 0);
+                const model = process.env.NIGHTSHIFT_OLLAMA_MODEL || "qwen3:8b";
+                return { input, output: outputTokens, total: total || undefined, model };
+            }
+            return undefined;
+        }
+        case "gemini": {
+            // Gemini CLI may report: "Token count: N" or usage stats
+            const tokenCountMatch = output.match(/token[s]?\s*(?:count|usage|used)[:\s]*([\d,]+)/i);
+            if (tokenCountMatch) {
+                const total = parseInt(tokenCountMatch[1].replace(/,/g, ""), 10);
+                if (!isNaN(total))
+                    return { total };
+            }
+            return undefined;
+        }
+        case "claude": {
+            // Claude Code --print mode may report token usage
+            const inputMatch = output.match(/input[_ ]tokens?[:\s]*([\d,]+)/i);
+            const outputMatch = output.match(/output[_ ]tokens?[:\s]*([\d,]+)/i);
+            if (inputMatch || outputMatch) {
+                const input = inputMatch ? parseInt(inputMatch[1].replace(/,/g, ""), 10) : undefined;
+                const outputTokens = outputMatch ? parseInt(outputMatch[1].replace(/,/g, ""), 10) : undefined;
+                const total = (input || 0) + (outputTokens || 0);
+                return { input, output: outputTokens, total: total || undefined };
+            }
+            return undefined;
+        }
+        default:
+            return undefined;
+    }
+}
+// ============================================
+// Rate Limit Detection
+// ============================================
+/**
+ * Detect rate limiting from agent output.
+ */
+export function detectRateLimit(output) {
+    if (!output)
+        return undefined;
+    // HTTP 429
+    const has429 = /\b429\b/.test(output) && /rate|limit|quota|capacity|exhausted/i.test(output);
+    // Explicit rate limit messages
+    const hasRateLimit = /rate.?limit|quota.?exceeded|too many requests|capacity.*exhausted|exhausted.*capacity|throttl/i.test(output);
+    // Retry-after header or message
+    const retryMatch = output.match(/retry.?after[:\s]*(\d+)/i)
+        || output.match(/reset after (\d+)s/i)
+        || output.match(/wait (\d+)\s*s/i);
+    if (!has429 && !hasRateLimit)
+        return undefined;
+    const retryAfter = retryMatch ? parseInt(retryMatch[1], 10) : undefined;
+    // Try to identify provider
+    let provider;
+    if (/openai|gpt/i.test(output))
+        provider = "openai";
+    else if (/anthropic|claude/i.test(output))
+        provider = "anthropic";
+    else if (/google|gemini/i.test(output))
+        provider = "google";
+    else if (/ollama/i.test(output))
+        provider = "ollama";
+    return { detected: true, retryAfter, provider };
+}
+// ============================================
+// Tool Usage Parsing
+// ============================================
+/**
+ * Parse tool/command usage from agent output.
+ * Detects MCP tool calls, shell commands, file operations, etc.
+ */
+export function parseToolsUsed(output) {
+    if (!output)
+        return undefined;
+    const tools = new Set();
+    // MCP tool calls (e.g., "nightshift - read_prd", "Tool: read_file")
+    const mcpToolMatches = output.matchAll(/(?:Tool|tool_use|nightshift)\s*[-:]?\s*(\w+)/g);
+    for (const match of mcpToolMatches) {
+        const toolName = match[1];
+        // Filter out noise words
+        if (toolName && toolName.length > 2 && !/^(the|and|for|was|use|has)$/i.test(toolName)) {
+            tools.add(toolName);
+        }
+    }
+    // Claude Code tool patterns: "Read(file)", "Edit(file)", "Bash(cmd)", "Write(file)"
+    const claudeToolMatches = output.matchAll(/\b(Read|Edit|Write|Bash|Glob|Grep|Agent|WebSearch|WebFetch)\s*\(/g);
+    for (const match of claudeToolMatches) {
+        tools.add(match[1]);
+    }
+    // Codex tool patterns: "exec", "read_file", "write_file", "shell"
+    const codexToolMatches = output.matchAll(/\b(exec|read_file|write_file|shell|list_dir|patch_file)\b/g);
+    for (const match of codexToolMatches) {
+        tools.add(match[1]);
+    }
+    // Gemini tool patterns: "ReadFile", "WriteFile", "ExecuteCommand"
+    const geminiToolMatches = output.matchAll(/\b(ReadFile|WriteFile|ExecuteCommand|SearchFiles|ListDirectory)\b/g);
+    for (const match of geminiToolMatches) {
+        tools.add(match[1]);
+    }
+    // Shell commands (common patterns)
+    const shellMatches = output.matchAll(/(?:^|\n)\s*\$?\s*(npm|npx|git|tsc|vitest|eslint|prettier|node)\b/gm);
+    for (const match of shellMatches) {
+        tools.add(`shell:${match[1]}`);
+    }
+    return tools.size > 0 ? Array.from(tools) : undefined;
+}
+// ============================================
+// Run Logger
+// ============================================
+/**
+ * Append a run log entry to the project's runs.jsonl file.
+ */
+export function appendRunLog(projectPath, entry) {
+    const robotChatDir = path.join(projectPath, ".robot-chat");
+    if (!fs.existsSync(robotChatDir)) {
+        fs.mkdirSync(robotChatDir, { recursive: true });
+    }
+    const logFile = path.join(robotChatDir, "runs.jsonl");
+    const line = JSON.stringify(entry) + "\n";
+    try {
+        fs.appendFileSync(logFile, line, "utf-8");
+    }
+    catch {
+        // Non-critical — best effort logging
+    }
+}
+/**
+ * Read all run log entries from the project's runs.jsonl file.
+ */
+export function readRunLog(projectPath, limit) {
+    const logFile = path.join(projectPath, ".robot-chat", "runs.jsonl");
+    if (!fs.existsSync(logFile))
+        return [];
+    try {
+        const content = fs.readFileSync(logFile, "utf-8");
+        const lines = content.trim().split("\n").filter(Boolean);
+        const entries = lines.map((line) => JSON.parse(line));
+        return limit ? entries.slice(-limit) : entries;
+    }
+    catch {
+        return [];
+    }
+}
+/**
+ * Get aggregate stats from the run log.
+ */
+export function getRunStats(projectPath) {
+    const entries = readRunLog(projectPath);
+    const byAgent = {};
+    let totalTokens = 0;
+    let totalDuration = 0;
+    let rateLimitCount = 0;
+    for (const entry of entries) {
+        if (!byAgent[entry.agent]) {
+            byAgent[entry.agent] = { runs: 0, totalTokens: 0, avgDuration: 0, totalDuration: 0 };
+        }
+        const agentStats = byAgent[entry.agent];
+        agentStats.runs++;
+        agentStats.totalDuration += entry.duration_seconds;
+        agentStats.avgDuration = Math.round(agentStats.totalDuration / agentStats.runs);
+        if (entry.tokens?.total) {
+            agentStats.totalTokens += entry.tokens.total;
+            totalTokens += entry.tokens.total;
+        }
+        totalDuration += entry.duration_seconds;
+        if (entry.status === "rate_limited")
+            rateLimitCount++;
+    }
+    // Clean up totalDuration from output
+    const result = {};
+    for (const [agent, stats] of Object.entries(byAgent)) {
+        result[agent] = { runs: stats.runs, totalTokens: stats.totalTokens, avgDuration: stats.avgDuration };
+    }
+    return {
+        totalRuns: entries.length,
+        byAgent: result,
+        totalTokens,
+        totalDuration,
+        rateLimitCount,
+    };
+}
+//# sourceMappingURL=run-logger.js.map

package/dist/run-logger.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"run-logger.js","sourceRoot":"","sources":["../src/run-logger.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAkC7B,+CAA+C;AAC/C,gBAAgB;AAChB,+CAA+C;AAE/C;;GAEG;AACH,MAAM,UAAU,qBAAqB,CAAC,KAAgB,EAAE,MAAc;IACpE,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,QAAQ,KAAK,EAAE,CAAC;QACd,KAAK,OAAO,CAAC,CAAC,CAAC;YACb,oFAAoF;YACpF,6CAA6C;YAC7C,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACxE,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;oBAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YACxD,CAAC;YACD,6EAA6E;YAC7E,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC1D,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,GAAG;oBAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;YACvE,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,sEAAsE;YACtE,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACxE,MAAM,SAAS,GAAG,MAAM,CAAC,KAAK,CAAC,0BAA0B,CAAC,CAAC;YAC3D,IAAI,eAAe,IAAI,SAAS,EAAE,CAAC;gBACjC,MAAM,KAAK,GAAG,eAAe,CAAC,CAAC,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC7E,MAAM,YAAY,GAAG,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACxE,MAAM,KAAK,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC;gBACjD,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,uBAAuB,IAAI,UAAU,CAAC;gBAChE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,IAAI,SAAS,EAAE,KAAK,EAAE,CAAC;YAC3E,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,yDAAyD;YACzD,MAAM,eAAe,GAAG,MAAM,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAC;YACxF,IAAI,eAAe,EAAE,CAAC;gBACpB,MAAM,KAAK,GAAG,QAAQ,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACjE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC;oBAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YACtC,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,KAAK,QAAQ,CAAC,CAAC,CAAC;YACd,kDAAkD;YAClD,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,CAAC;YACnE,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,kCAAkC,CAAC,CAAC;YACrE,IAAI,UAAU,IAAI,WAAW,EAAE,CAAC;gBAC9B,MAAM,KAAK,GAAG,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBACrF,MAAM,YAAY,GAAG,WAAW,CAAC,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;gBAC9F,MAAM,KAAK,GAAG,CAAC,KAAK,IAAI,CAAC,CAAC,GAAG,CAAC,YAAY,IAAI,CAAC,CAAC,CAAC;gBACjD,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,KAAK,IAAI,SAAS,EAAE,CAAC;YACpE,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QAED;YACE,OAAO,SAAS,CAAC;IACrB,CAAC;AACH,CAAC;AAED,+CAA+C;AAC/C,uBAAuB;AACvB,+CAA+C;AAE/C;;GAEG;AACH,MAAM,UAAU,eAAe,CAAC,MAAc;IAC5C,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,WAAW;IACX,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,sCAAsC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAE7F,+BAA+B;IAC/B,MAAM,YAAY,GAAG,gGAAgG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAEnI,gCAAgC;IAChC,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,0BAA0B,CAAC;WACtD,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC;WACnC,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IAErC,IAAI,CAAC,MAAM,IAAI,CAAC,YAAY;QAAE,OAAO,SAAS,CAAC;IAE/C,MAAM,UAAU,GAAG,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IAExE,2BAA2B;IAC3B,IAAI,QAA4B,CAAC;IACjC,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,QAAQ,CAAC;SAC/C,IAAI,mBAAmB,CAAC,IAAI,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,WAAW,CAAC;SAC7D,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,QAAQ,CAAC;SACvD,IAAI,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC;QAAE,QAAQ,GAAG,QAAQ,CAAC;IAErD,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,CAAC;AAClD,CAAC;AAED,+CAA+C;AAC/C,qBAAqB;AACrB,+CAA+C;AAE/C;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,MAAc;IAC3C,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,MAAM,KAAK,GAAG,IAAI,GAAG,EAAU,CAAC;IAEhC,oEAAoE;IACpE,MAAM,cAAc,GAAG,MAAM,CAAC,QAAQ,CAAC,+CAA+C,CAAC,CAAC;IACxF,KAAK,MAAM,KAAK,IAAI,cAAc,EAAE,CAAC;QACnC,MAAM,QAAQ,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC1B,yBAAyB;QACzB,IAAI,QAAQ,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,8BAA8B,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YACtF,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IAED,oFAAoF;IACpF,MAAM,iBAAiB,GAAG,MAAM,CAAC,QAAQ,CAAC,mEAAmE,CAAC,CAAC;IAC/G,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE,CAAC;QACtC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAED,kEAAkE;IAClE,MAAM,gBAAgB,GAAG,MAAM,CAAC,QAAQ,CAAC,4DAA4D,CAAC,CAAC;IACvG,KAAK,MAAM,KAAK,IAAI,gBAAgB,EAAE,CAAC;QACrC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAED,kEAAkE;IAClE,MAAM,iBAAiB,GAAG,MAAM,CAAC,QAAQ,CAAC,oEAAoE,CAAC,CAAC;IAChH,KAAK,MAAM,KAAK,IAAI,iBAAiB,EAAE,CAAC;QACtC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACtB,CAAC;IAED,mCAAmC;IACnC,MAAM,YAAY,GAAG,MAAM,CAAC,QAAQ,CAAC,oEAAoE,CAAC,CAAC;IAC3G,KAAK,MAAM,KAAK,IAAI,YAAY,EAAE,CAAC;QACjC,KAAK,CAAC,GAAG,CAAC,SAAS,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;IACjC,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;AACxD,CAAC;AAED,+CAA+C;AAC/C,aAAa;AACb,+CAA+C;AAE/C;;GAEG;AACH,MAAM,UAAU,YAAY,CAAC,WAAmB,EAAE,KAAkB;IAClE,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,aAAa,CAAC,CAAC;IAC3D,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,EAAE,CAAC;QACjC,EAAE,CAAC,SAAS,CAAC,YAAY,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,YAAY,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;IAE1C,IAAI,CAAC;QACH,EAAE,CAAC,cAAc,CAAC,OAAO,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAAC,MAAM,CAAC;QACP,qCAAqC;IACvC,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,UAAU,CAAC,WAAmB,EAAE,KAAc;IAC5D,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,aAAa,EAAE,YAAY,CAAC,CAAC;IACpE,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IAEvC,IAAI,CAAC;QACH,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAClD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QACzD,MAAM,OAAO,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAgB,CAAC,CAAC;QACrE,OAAO,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC;IACjD,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,EAAE,CAAC;IACZ,CAAC;AACH,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,WAAW,CAAC,WAAmB;IAO7C,MAAM,OAAO,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;IACxC,MAAM,OAAO,GAAsG,EAAE,CAAC;IACtH,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,aAAa,GAAG,CAAC,CAAC;IACtB,IAAI,cAAc,GAAG,CAAC,CAAC;IAEvB,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,WAAW,EAAE,CAAC,EAAE,aAAa,EAAE,CAAC,EAAE,CAAC;QACvF,CAAC;QACD,MAAM,UAAU,GAAG,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACxC,UAAU,CAAC,IAAI,EAAE,CAAC;QAClB,UAAU,CAAC,aAAa,IAAI,KAAK,CAAC,gBAAgB,CAAC;QACnD,UAAU,CAAC,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,UAAU,CAAC,aAAa,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC;QAEhF,IAAI,KAAK,CAAC,MAAM,EAAE,KAAK,EAAE,CAAC;YACxB,UAAU,CAAC,WAAW,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;YAC7C,WAAW,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;QACpC,CAAC;QAED,aAAa,IAAI,KAAK,CAAC,gBAAgB,CAAC;QAExC,IAAI,KAAK,CAAC,MAAM,KAAK,cAAc;YAAE,cAAc,EAAE,CAAC;IACxD,CAAC;IAED,qCAAqC;IACrC,MAAM,MAAM,GAA+E,EAAE,CAAC;IAC9F,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;QACrD,MAAM,CAAC,KAAK,CAAC,GAAG,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC;IACvG,CAAC;IAED,OAAO;QACL,SAAS,EAAE,OAAO,CAAC,MAAM;QACzB,OAAO,EAAE,MAAM;QACf,WAAW;QACX,aAAa;QACb,cAAc;KACf,CAAC;AACJ,CAAC"}

package/dist/tools/agents.d.ts

@@ -8,5 +8,8 @@ export declare const getAgentStatusTool: import("../tool-registry.js").ToolDefin
 export declare const listRunningAgentsTool: import("../tool-registry.js").ToolDefinition;
 export declare const getDaemonStatusTool: import("../tool-registry.js").ToolDefinition;
 export declare const stopDaemonTool: import("../tool-registry.js").ToolDefinition;
+export declare const getRunStatsTool: import("../tool-registry.js").ToolDefinition;
+export declare const getPipelineTool: import("../tool-registry.js").ToolDefinition;
+export declare const setPipelineTool: import("../tool-registry.js").ToolDefinition;
 export declare const agentTools: import("../tool-registry.js").ToolDefinition[];
 //# sourceMappingURL=agents.d.ts.map

package/dist/tools/agents.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agents.d.ts","sourceRoot":"","sources":["../../src/tools/agents.ts"],"names":[],"mappings":"AAkEA,eAAO,MAAM,mBAAmB,8CAsC9B,CAAC;AAEH,eAAO,MAAM,cAAc,8CAqFzB,CAAC;AAEH,eAAO,MAAM,wBAAwB,8CAoDnC,CAAC;AAEH,eAAO,MAAM,aAAa,8CAqLxB,CAAC;AAEH,eAAO,MAAM,gBAAgB,8CAoK3B,CAAC;AAEH,eAAO,MAAM,WAAW,8CAsJtB,CAAC;AAEH,eAAO,MAAM,kBAAkB,8CAwC7B,CAAC;AAEH,eAAO,MAAM,qBAAqB,8CAwChC,CAAC;AAEH,eAAO,MAAM,mBAAmB,8CAsB9B,CAAC;AAEH,eAAO,MAAM,cAAc,8CA4CzB,CAAC;AAGH,eAAO,MAAM,UAAU,gDAWtB,CAAC"}
+{"version":3,"file":"agents.d.ts","sourceRoot":"","sources":["../../src/tools/agents.ts"],"names":[],"mappings":"AA0EA,eAAO,MAAM,mBAAmB,8CAsC9B,CAAC;AAEH,eAAO,MAAM,cAAc,8CAiGzB,CAAC;AAEH,eAAO,MAAM,wBAAwB,8CAuDnC,CAAC;AAEH,eAAO,MAAM,aAAa,8CA6LxB,CAAC;AAEH,eAAO,MAAM,gBAAgB,8CAoK3B,CAAC;AAEH,eAAO,MAAM,WAAW,8CAsJtB,CAAC;AAEH,eAAO,MAAM,kBAAkB,8CAyC7B,CAAC;AAEH,eAAO,MAAM,qBAAqB,8CAyChC,CAAC;AAEH,eAAO,MAAM,mBAAmB,8CAsB9B,CAAC;AAEH,eAAO,MAAM,cAAc,8CA4CzB,CAAC;AAEH,eAAO,MAAM,eAAe,8CAgC1B,CAAC;AAEH,eAAO,MAAM,eAAe,8CA8B1B,CAAC;AAEH,eAAO,MAAM,eAAe,8CA0D1B,CAAC;AAGH,eAAO,MAAM,UAAU,gDActB,CAAC"}

package/dist/tools/agents.js

@@ -2,6 +2,8 @@ import { z } from "zod";
 import { defineTool } from "../tool-registry.js";
 import { spawnAgent, spawnAgentBackground, getAvailableAgents, getAgentStatus, canAgentRun, getTrackedAgentStatus, listTrackedAgents, updateTrackedAgent, } from "../agent-spawner.js";
 import { getNightshiftCliCommand } from "../platform.js";
+import { getRunStats, readRunLog } from "../run-logger.js";
+import { savePipelineOverride, clearPipelineOverride, resolveExecutablePipeline, listPipelines, getTagRouting, } from "../pipeline.js";
 import { ProjectOrchestrator } from "../orchestrator.js";
 import { isDaemonRunning, ensureDaemonRunning, stopDaemon, } from "../daemon-manager.js";
 const AGENT_TYPES = ["claude", "codex", "gemini", "vibe", "goose", "ollama"];
@@ -106,8 +108,10 @@ Agent selection guide (only available agents will work — check list_available_
         agent: z.enum(AGENT_TYPES).describe("Which agent to spawn. IMPORTANT: only use agents confirmed available by list_available_agents or preflight check. Do NOT guess."),
         prompt: z.string().describe("The prompt/task to send to the agent"),
         timeout: z.number().optional().describe("Timeout in seconds (default: 300). Use 600 for research tasks, 900 for complex implementation. See tool description for guidance."),
+        role: z.string().optional().describe("Agent role for audit trail (e.g., 'planner', 'coder', 'fixer', 'verifier', 'reviewer')"),
+        storyId: z.string().optional().describe("Story ID being worked on (for run log tracking)"),
     }),
-    handler: async ({ agent, prompt, timeout }, ctx) => {
+    handler: async ({ agent, prompt, timeout, role, storyId }, ctx) => {
         try {
             // Pre-flight: verify agent is available before attempting spawn
             const preflight = await preflightAgentCheck(agent);
@@ -128,18 +132,23 @@ Agent selection guide (only available agents will work — check list_available_
                 prompt,
                 projectPath: ctx.projectPath,
                 timeout: timeout ? timeout * 1000 : undefined,
+                role,
+                storyId,
             });
             // End trace span
             if (spanId) {
                 ctx.traceManager?.endSpan(spanId, result.success ? "completed" : "failed", {
                     exitCode: result.exitCode,
+                    tokens: result.tokens,
+                    rateLimited: result.rateLimitInfo?.detected,
                 });
             }
             // Post result to chat
+            const tokenSummary = result.tokens?.total ? ` (${result.tokens.total.toLocaleString()} tokens)` : "";
             ctx.chatManager.writeMessage({
                 agent: "NightShift",
                 type: result.success ? "INFO" : "ERROR",
-                content: `${agent} agent ${result.success ? "completed" : "failed"}: ${result.output.substring(0, 500)}${result.output.length > 500 ? "..." : ""}`,
+                content: `${agent}${role ? ` [${role}]` : ""} agent ${result.success ? "completed" : "failed"}${tokenSummary}: ${result.output.substring(0, 500)}${result.output.length > 500 ? "..." : ""}`,
             });
             return {
                 content: [{
@@ -147,8 +156,13 @@ Agent selection guide (only available agents will work — check list_available_
                         text: JSON.stringify({
                             success: result.success,
                             agent,
+                            role,
                             output: result.output,
                             exitCode: result.exitCode,
+                            duration_seconds: result.duration_seconds,
+                            tokens: result.tokens,
+                            rateLimitInfo: result.rateLimitInfo,
+                            toolsUsed: result.toolsUsed,
                             error: result.error,
                         }, null, 2),
                     }],
@@ -171,8 +185,9 @@ export const spawnAgentBackgroundTool = defineTool({
     inputSchema: z.object({
         agent: z.enum(AGENT_TYPES).describe("Which agent to spawn. MUST be an available agent — check list_available_agents first."),
         prompt: z.string().describe("The prompt/task to send to the agent"),
+        role: z.string().optional().describe("Agent role for audit trail (e.g., 'planner', 'coder', 'fixer', 'verifier')"),
     }),
-    handler: async ({ agent, prompt }, ctx) => {
+    handler: async ({ agent, prompt, role }, ctx) => {
         try {
             // Pre-flight: verify agent is available before attempting spawn
             const preflight = await preflightAgentCheck(agent);
@@ -187,18 +202,20 @@ export const spawnAgentBackgroundTool = defineTool({
                 agent: agent,
                 prompt,
                 projectPath: ctx.projectPath,
+                role,
             });
             // Post to chat
             ctx.chatManager.writeMessage({
                 agent: "NightShift",
                 type: "INFO",
-                content: `Spawned ${agent} in background (PID: ${result.pid})\nOutput: ${result.outputFile}`,
+                content: `Spawned ${agent}${role ? ` [${role}]` : ""} in background (PID: ${result.pid})\nOutput: ${result.outputFile}`,
             });
             return {
                 content: [{
                         type: "text",
                         text: JSON.stringify({
                             agent,
+                            role,
                             pid: result.pid,
                             outputFile: result.outputFile,
                             status: "running in background",
@@ -225,8 +242,9 @@ IMPORTANT: Only delegate to agents that are actually available. Use list_availab
         agent: z.enum(AGENT_TYPES).describe("Which agent to delegate to. MUST be an available agent — check list_available_agents first."),
         storyId: z.string().optional().describe("Specific story ID (defaults to next available)"),
         background: z.boolean().optional().describe("Run in background (default: false)"),
+        role: z.string().optional().describe("Agent role for audit trail (default: 'implementer')"),
     }),
-    handler: async ({ agent, storyId, background }, ctx) => {
+    handler: async ({ agent, storyId, background, role }, ctx) => {
         try {
             // Pre-flight: verify agent is available before attempting delegation
             const preflight = await preflightAgentCheck(agent);
@@ -312,11 +330,14 @@ Begin implementation now.`;
                 promptLength: delegationPrompt.length,
                 background: !!background,
             });
+            const effectiveRole = role || "implementer";
             if (background) {
                 const result = spawnAgentBackground({
                     agent: agent,
                     prompt: delegationPrompt,
                     projectPath: ctx.projectPath,
+                    role: effectiveRole,
+                    storyId: story.id,
                 });
                 // Tag the tracked agent with the story ID
                 if (result.pid) {
@@ -328,6 +349,7 @@ Begin implementation now.`;
                             text: JSON.stringify({
                                 status: "delegated",
                                 agent,
+                                role: effectiveRole,
                                 story: story.id,
                                 title: story.title,
                                 background: true,
@@ -344,6 +366,8 @@ Begin implementation now.`;
                     prompt: delegationPrompt,
                     projectPath: ctx.projectPath,
                     timeout: 10 * 60 * 1000, // 10 minutes for story work
+                    role: effectiveRole,
+                    storyId: story.id,
                 });
                 // End trace span
                 if (spanId) {
@@ -708,6 +732,7 @@ export const getAgentStatusTool = defineTool({
                         pid,
                         status: status.running ? "running" : "exited",
                         agent: status.agent,
+                        role: status.role,
                         storyId: status.storyId,
                         outputFile: status.outputFile,
                         startTime: status.startTime ? new Date(status.startTime).toISOString() : undefined,
@@ -746,6 +771,7 @@ export const listRunningAgentsTool = defineTool({
                         agents: agents.map((a) => ({
                             pid: a.pid,
                             agent: a.agent,
+                            role: a.role,
                             storyId: a.storyId,
                             status: a.running ? "running" : "exited",
                             elapsedSeconds: a.elapsedSeconds,
@@ -819,6 +845,120 @@ export const stopDaemonTool = defineTool({
         };
     },
 });
+export const getRunStatsTool = defineTool({
+    name: "get_run_stats",
+    category: "agents",
+    description: "Get token usage and performance stats from the run log",
+    fullDescription: "Get aggregate statistics from the run log including token usage per agent, average durations, rate limit counts, and tools used. Use this to optimize your agent pipeline and track costs.",
+    inputSchema: z.object({
+        limit: z.number().optional().describe("Number of recent runs to show in detail (default: 10)"),
+    }),
+    handler: async ({ limit }, ctx) => {
+        const stats = getRunStats(ctx.projectPath);
+        const recentRuns = readRunLog(ctx.projectPath, limit ?? 10);
+        return {
+            content: [{
+                    type: "text",
+                    text: JSON.stringify({
+                        summary: stats,
+                        recentRuns: recentRuns.map((r) => ({
+                            timestamp: r.timestamp,
+                            agent: r.agent,
+                            role: r.role,
+                            storyId: r.storyId,
+                            status: r.status,
+                            duration_seconds: r.duration_seconds,
+                            tokens: r.tokens,
+                            toolsUsed: r.toolsUsed,
+                            rateLimitInfo: r.rateLimitInfo,
+                        })),
+                    }, null, 2),
+                }],
+        };
+    },
+});
+export const getPipelineTool = defineTool({
+    name: "get_pipeline",
+    category: "agents",
+    description: "View configured pipelines and tag routing rules",
+    fullDescription: `View all configured agent pipelines and tag routing rules. Shows built-in defaults merged with any project config (nightshift.config.json) and runtime overrides.
+
+Pipelines define the sequence of agents for different task types. Stories are automatically routed to pipelines based on their tags.`,
+    inputSchema: z.object({
+        storyTags: z.array(z.string()).optional().describe("Preview which pipeline would be selected for these tags"),
+    }),
+    handler: async ({ storyTags }, ctx) => {
+        const pipelines = listPipelines(ctx.projectPath);
+        const routing = getTagRouting(ctx.projectPath);
+        const result = { pipelines, tagRouting: routing };
+        if (storyTags && storyTags.length > 0) {
+            const resolved = await resolveExecutablePipeline(ctx.projectPath, storyTags);
+            result.preview = {
+                tags: storyTags,
+                selectedPipeline: resolved.name,
+                executableSteps: resolved.steps,
+                skippedSteps: resolved.skipped,
+            };
+        }
+        return {
+            content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
+        };
+    },
+});
+export const setPipelineTool = defineTool({
+    name: "set_pipeline",
+    category: "agents",
+    description: "Configure or update an agent pipeline at runtime",
+    fullDescription: `Create or update an agent pipeline at runtime. Changes persist for this project session (stored in .robot-chat/pipeline-override.json).
+
+Example: set_pipeline({ name: "default", steps: [{ agent: "gemini", role: "planner" }, { agent: "codex", role: "implementer" }] })
+
+Use reset=true to clear all runtime overrides and revert to defaults + config file.`,
+    inputSchema: z.object({
+        name: z.string().optional().describe("Pipeline name to create/update (e.g., 'default', 'frontend', 'backend', or any custom name)"),
+        steps: z.array(z.object({
+            agent: z.enum(AGENT_TYPES).describe("Agent for this step"),
+            role: z.string().describe("Role name (planner, implementer, verifier, drafter, fixer, reviewer, researcher)"),
+            optional: z.boolean().optional().describe("Skip if agent unavailable (default: true)"),
+            timeout: z.number().optional().describe("Timeout in seconds for this step"),
+        })).optional().describe("Pipeline steps in execution order"),
+        tagRouting: z.record(z.string()).optional().describe("Map story tags to pipeline names, e.g. { 'frontend': 'frontend', 'api': 'backend' }"),
+        reset: z.boolean().optional().describe("Clear all runtime overrides, revert to defaults"),
+    }),
+    handler: async ({ name, steps, tagRouting, reset }, ctx) => {
+        if (reset) {
+            clearPipelineOverride(ctx.projectPath);
+            return {
+                content: [{ type: "text", text: "Pipeline overrides cleared. Using defaults + config file." }],
+            };
+        }
+        if (!name && !tagRouting) {
+            return {
+                content: [{ type: "text", text: "Provide a pipeline name + steps, tagRouting rules, or reset=true." }],
+                isError: true,
+            };
+        }
+        const overrides = {};
+        if (name && steps) {
+            overrides.pipelines = { [name]: steps };
+        }
+        if (tagRouting) {
+            overrides.tagRouting = tagRouting;
+        }
+        savePipelineOverride(ctx.projectPath, overrides);
+        // Show the resolved state after update
+        const allPipelines = listPipelines(ctx.projectPath);
+        return {
+            content: [{
+                    type: "text",
+                    text: JSON.stringify({
+                        message: `Pipeline${name ? ` "${name}"` : ""} updated.`,
+                        currentPipelines: allPipelines,
+                    }, null, 2),
+                }],
+        };
+    },
+});
 // Export all agent tools as an array
 export const agentTools = [
     listAvailableAgents,
@@ -831,5 +971,8 @@ export const agentTools = [
     listRunningAgentsTool,
     getDaemonStatusTool,
     stopDaemonTool,
+    getRunStatsTool,
+    getPipelineTool,
+    setPipelineTool,
 ];
 //# sourceMappingURL=agents.js.map