sema-core 1.0.2

package/dist/util/agentStats.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.calculateStats = calculateStats;
+exports.formatSummary = formatSummary;
+exports.extractResultText = extractResultText;
+const tokens_1 = require("./tokens");
+// 从消息列表中统计 tokens 和工具使用
+function calculateStats(messages, startTime) {
+    const { inputTokens, outputTokens } = (0, tokens_1.countTokens)(messages);
+    const totalTokens = inputTokens + outputTokens;
+    let toolUseCount = 0;
+    for (const msg of messages) {
+        if (msg.type === 'assistant' && msg.message.content) {
+            for (const block of msg.message.content) {
+                if (block.type === 'tool_use') {
+                    toolUseCount++;
+                }
+            }
+        }
+    }
+    return {
+        inputTokens,
+        outputTokens,
+        totalTokens,
+        toolUseCount,
+        durationMs: Date.now() - startTime
+    };
+}
+// 格式化摘要信息
+function formatSummary(stats, status) {
+    const durationSec = Math.floor(stats.durationMs / 1000);
+    const toolsText = `${stats.toolUseCount} tool${stats.toolUseCount !== 1 ? 's' : ''} use`;
+    const tokensText = stats.totalTokens >= 1000
+        ? `${(stats.totalTokens / 1000).toFixed(1)}k tokens`
+        : `${stats.totalTokens} tokens`;
+    const durationText = durationSec < 60
+        ? `${durationSec}s`
+        : `${Math.floor(durationSec / 60)}m ${durationSec % 60}s`;
+    const prefix = status === 'completed' ? 'Done' : 'Interrupted';
+    return `${prefix}(${toolsText} · ${tokensText} · ${durationText})`;
+}
+// 从消息列表中提取最后一次 assistant 响应的文本内容
+function extractResultText(messages) {
+    for (let i = messages.length - 1; i >= 0; i--) {
+        const message = messages[i];
+        if (message.type === 'assistant' && message.message.content) {
+            let currentMessageText = '';
+            for (const block of message.message.content) {
+                if (block.type === 'text') {
+                    currentMessageText += block.text;
+                }
+            }
+            if (currentMessageText) {
+                return currentMessageText;
+            }
+        }
+    }
+    return 'Agent completed without returning text output.';
+}
+//# sourceMappingURL=agentStats.js.map

package/dist/util/agentStats.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agentStats.js","sourceRoot":"","sources":["../../src/util/agentStats.ts"],"names":[],"mappings":";;AAYA,wCAsBC;AAGD,sCAYC;AAGD,8CAgBC;AApED,qCAAsC;AAWtC,wBAAwB;AACxB,SAAgB,cAAc,CAAC,QAAe,EAAE,SAAiB;IAC/D,MAAM,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAA,oBAAW,EAAC,QAAQ,CAAC,CAAA;IAC3D,MAAM,WAAW,GAAG,WAAW,GAAG,YAAY,CAAA;IAE9C,IAAI,YAAY,GAAG,CAAC,CAAA;IACpB,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YACpD,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBACxC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;oBAC9B,YAAY,EAAE,CAAA;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO;QACL,WAAW;QACX,YAAY;QACZ,WAAW;QACX,YAAY;QACZ,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;KACnC,CAAA;AACH,CAAC;AAED,UAAU;AACV,SAAgB,aAAa,CAAC,KAAiB,EAAE,MAAmC;IAClF,MAAM,WAAW,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,CAAA;IACvD,MAAM,SAAS,GAAG,GAAG,KAAK,CAAC,YAAY,QAAQ,KAAK,CAAC,YAAY,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,MAAM,CAAA;IACxF,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,IAAI,IAAI;QAC1C,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU;QACpD,CAAC,CAAC,GAAG,KAAK,CAAC,WAAW,SAAS,CAAA;IACjC,MAAM,YAAY,GAAG,WAAW,GAAG,EAAE;QACnC,CAAC,CAAC,GAAG,WAAW,GAAG;QACnB,CAAC,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,EAAE,CAAC,KAAK,WAAW,GAAG,EAAE,GAAG,CAAA;IAE3D,MAAM,MAAM,GAAG,MAAM,KAAK,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,aAAa,CAAA;IAC9D,OAAO,GAAG,MAAM,IAAI,SAAS,MAAM,UAAU,MAAM,YAAY,GAAG,CAAA;AACpE,CAAC;AAED,iCAAiC;AACjC,SAAgB,iBAAiB,CAAC,QAAe;IAC/C,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAA;QAC3B,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;YAC5D,IAAI,kBAAkB,GAAG,EAAE,CAAA;YAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC;gBAC5C,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;oBAC1B,kBAAkB,IAAI,KAAK,CAAC,IAAI,CAAA;gBAClC,CAAC;YACH,CAAC;YACD,IAAI,kBAAkB,EAAE,CAAC;gBACvB,OAAO,kBAAkB,CAAA;YAC3B,CAAC;QACH,CAAC;IACH,CAAC;IACD,OAAO,gDAAgD,CAAA;AACzD,CAAC"}

package/dist/util/cacheLLM.d.ts

@@ -0,0 +1,47 @@
+import { AssistantMessage, UserMessage } from '../types/message';
+declare class FileLLMCache {
+    private readonly maxEntries;
+    private readonly cacheFilePath;
+    constructor();
+    /**
+     * 确保缓存目录存在
+     */
+    private ensureCacheDir;
+    /**
+     * 生成缓存键 - 基于消息内容生成简单hash
+     */
+    private generateKey;
+    /**
+     * 读取缓存文件
+     */
+    private readCacheFile;
+    /**
+     * 写入缓存文件
+     */
+    private writeCacheFile;
+    /**
+     * 获取缓存
+     */
+    get(messages: (UserMessage | AssistantMessage)[], systemPrompt: string[] | Array<{
+        type: 'text';
+        text: string;
+    }>, modelName: string, enableThinking?: boolean): AssistantMessage | null;
+    /**
+     * 设置缓存
+     */
+    set(messages: (UserMessage | AssistantMessage)[], systemPrompt: string[] | Array<{
+        type: 'text';
+        text: string;
+    }>, modelName: string, response: AssistantMessage, enableThinking?: boolean): void;
+    /**
+     * 清空缓存
+     */
+    clear(): void;
+    /**
+     * 获取缓存大小
+     */
+    size(): number;
+}
+export declare const llmCache: FileLLMCache;
+export {};
+//# sourceMappingURL=cacheLLM.d.ts.map

package/dist/util/cacheLLM.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"cacheLLM.d.ts","sourceRoot":"","sources":["../../src/util/cacheLLM.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAA;AAWhE,cAAM,YAAY;IAChB,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAM;IACjC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAQ;;IAOtC;;OAEG;IACH,OAAO,CAAC,cAAc;IAOtB;;OAEG;IACH,OAAO,CAAC,WAAW;IAoBnB;;OAEG;IACH,OAAO,CAAC,aAAa;IAarB;;OAEG;IACH,OAAO,CAAC,cAAc;IAQtB;;OAEG;IACH,GAAG,CACD,QAAQ,EAAE,CAAC,WAAW,GAAG,gBAAgB,CAAC,EAAE,EAC5C,YAAY,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,EAC9D,SAAS,EAAE,MAAM,EACjB,cAAc,GAAE,OAAe,GAC9B,gBAAgB,GAAG,IAAI;IAQ1B;;OAEG;IACH,GAAG,CACD,QAAQ,EAAE,CAAC,WAAW,GAAG,gBAAgB,CAAC,EAAE,EAC5C,YAAY,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC,EAC9D,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,gBAAgB,EAC1B,cAAc,GAAE,OAAe,GAC9B,IAAI;IAsBP;;OAEG;IACH,KAAK,IAAI,IAAI;IAUb;;OAEG;IACH,IAAI,IAAI,MAAM;CAIf;AAGD,eAAO,MAAM,QAAQ,cAAqB,CAAA"}

package/dist/util/cacheLLM.js

@@ -0,0 +1,154 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.llmCache = void 0;
+const crypto_1 = __importDefault(require("crypto"));
+const fs = __importStar(require("fs"));
+const savePath_1 = require("./savePath");
+// llm缓存实现
+class FileLLMCache {
+    constructor() {
+        this.maxEntries = 100;
+        this.cacheFilePath = (0, savePath_1.getLLMCacheFilePath)();
+        this.ensureCacheDir();
+    }
+    /**
+     * 确保缓存目录存在
+     */
+    ensureCacheDir() {
+        const cacheDir = (0, savePath_1.getCacheDir)();
+        if (!fs.existsSync(cacheDir)) {
+            fs.mkdirSync(cacheDir, { recursive: true });
+        }
+    }
+    /**
+     * 生成缓存键 - 基于消息内容生成简单hash
+     */
+    generateKey(messages, systemPrompt, modelName, enableThinking = false) {
+        // 统一处理系统提示格式
+        const normalizedSystemPrompt = Array.isArray(systemPrompt) && systemPrompt.length > 0 && typeof systemPrompt[0] === 'object' && 'type' in systemPrompt[0]
+            ? systemPrompt.map(item => item.text)
+            : systemPrompt;
+        const content = JSON.stringify({
+            messages: messages.map(msg => msg.message.content),
+            systemPrompt: normalizedSystemPrompt,
+            modelName,
+            enableThinking
+        });
+        return crypto_1.default.createHash('md5').update(content).digest('hex');
+    }
+    /**
+     * 读取缓存文件
+     */
+    readCacheFile() {
+        try {
+            if (!fs.existsSync(this.cacheFilePath)) {
+                return [];
+            }
+            const data = fs.readFileSync(this.cacheFilePath, 'utf8');
+            return JSON.parse(data) || [];
+        }
+        catch (error) {
+            // 文件损坏或不存在，返回空数组
+            return [];
+        }
+    }
+    /**
+     * 写入缓存文件
+     */
+    writeCacheFile(entries) {
+        try {
+            fs.writeFileSync(this.cacheFilePath, JSON.stringify(entries, null, 2), 'utf8');
+        }
+        catch (error) {
+            // 写入失败，忽略错误
+        }
+    }
+    /**
+     * 获取缓存
+     */
+    get(messages, systemPrompt, modelName, enableThinking = false) {
+        const key = this.generateKey(messages, systemPrompt, modelName, enableThinking);
+        const entries = this.readCacheFile();
+        const entry = entries.find(e => e.key === key);
+        return entry ? entry.response : null;
+    }
+    /**
+     * 设置缓存
+     */
+    set(messages, systemPrompt, modelName, response, enableThinking = false) {
+        const key = this.generateKey(messages, systemPrompt, modelName, enableThinking);
+        let entries = this.readCacheFile();
+        // 移除已存在的相同key
+        entries = entries.filter(e => e.key !== key);
+        // 添加新条目
+        entries.unshift({
+            key,
+            response,
+            timestamp: Date.now()
+        });
+        // 保持最多n条记录
+        if (entries.length > this.maxEntries) {
+            entries = entries.slice(0, this.maxEntries);
+        }
+        this.writeCacheFile(entries);
+    }
+    /**
+     * 清空缓存
+     */
+    clear() {
+        try {
+            if (fs.existsSync(this.cacheFilePath)) {
+                fs.unlinkSync(this.cacheFilePath);
+            }
+        }
+        catch (error) {
+            // 删除失败，忽略错误
+        }
+    }
+    /**
+     * 获取缓存大小
+     */
+    size() {
+        const entries = this.readCacheFile();
+        return entries.length;
+    }
+}
+// 全局缓存实例
+exports.llmCache = new FileLLMCache();
+//# sourceMappingURL=cacheLLM.js.map

package/dist/util/cacheLLM.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cacheLLM.js","sourceRoot":"","sources":["../../src/util/cacheLLM.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,oDAA2B;AAC3B,uCAAwB;AAExB,yCAA6D;AAS7D,UAAU;AACV,MAAM,YAAY;IAIhB;QAHiB,eAAU,GAAG,GAAG,CAAA;QAI/B,IAAI,CAAC,aAAa,GAAG,IAAA,8BAAmB,GAAE,CAAA;QAC1C,IAAI,CAAC,cAAc,EAAE,CAAA;IACvB,CAAC;IAED;;OAEG;IACK,cAAc;QACpB,MAAM,QAAQ,GAAG,IAAA,sBAAW,GAAE,CAAA;QAC9B,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC7B,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAA;QAC7C,CAAC;IACH,CAAC;IAED;;OAEG;IACK,WAAW,CACjB,QAA4C,EAC5C,YAA8D,EAC9D,SAAiB,EACjB,iBAA0B,KAAK;QAE/B,aAAa;QACb,MAAM,sBAAsB,GAAG,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,YAAY,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,MAAM,IAAI,YAAY,CAAC,CAAC,CAAC;YACvJ,CAAC,CAAE,YAAsD,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;YAChF,CAAC,CAAC,YAAwB,CAAA;QAE5B,MAAM,OAAO,GAAG,IAAI,CAAC,SAAS,CAAC;YAC7B,QAAQ,EAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,OAAO,CAAC;YAClD,YAAY,EAAE,sBAAsB;YACpC,SAAS;YACT,cAAc;SACf,CAAC,CAAA;QACF,OAAO,gBAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAA;IAC/D,CAAC;IAED;;OAEG;IACK,aAAa;QACnB,IAAI,CAAC;YACH,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;gBACvC,OAAO,EAAE,CAAA;YACX,CAAC;YACD,MAAM,IAAI,GAAG,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,CAAC,CAAA;YACxD,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,EAAE,CAAA;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,iBAAiB;YACjB,OAAO,EAAE,CAAA;QACX,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,OAAqB;QAC1C,IAAI,CAAC;YACH,EAAE,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,MAAM,CAAC,CAAA;QAChF,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,GAAG,CACD,QAA4C,EAC5C,YAA8D,EAC9D,SAAiB,EACjB,iBAA0B,KAAK;QAE/B,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,cAAc,CAAC,CAAA;QAC/E,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAA;QAEpC,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAA;QAC9C,OAAO,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAA;IACtC,CAAC;IAED;;OAEG;IACH,GAAG,CACD,QAA4C,EAC5C,YAA8D,EAC9D,SAAiB,EACjB,QAA0B,EAC1B,iBAA0B,KAAK;QAE/B,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,YAAY,EAAE,SAAS,EAAE,cAAc,CAAC,CAAA;QAC/E,IAAI,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAA;QAElC,cAAc;QACd,OAAO,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,KAAK,GAAG,CAAC,CAAA;QAE5C,QAAQ;QACR,OAAO,CAAC,OAAO,CAAC;YACd,GAAG;YACH,QAAQ;YACR,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAA;QAEF,WAAW;QACX,IAAI,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;YACrC,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,UAAU,CAAC,CAAA;QAC7C,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAA;IAC9B,CAAC;IAED;;OAEG;IACH,KAAK;QACH,IAAI,CAAC;YACH,IAAI,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC;gBACtC,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,aAAa,CAAC,CAAA;YACnC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,YAAY;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,IAAI;QACF,MAAM,OAAO,GAAG,IAAI,CAAC,aAAa,EAAE,CAAA;QACpC,OAAO,OAAO,CAAC,MAAM,CAAA;IACvB,CAAC;CACF;AAED,SAAS;AACI,QAAA,QAAQ,GAAG,IAAI,YAAY,EAAE,CAAA"}

package/dist/util/commands.d.ts

@@ -0,0 +1,16 @@
+export type CommandPrefixResult = {
+    commandPrefix: string | null;
+    commandInjectionDetected: false;
+} | {
+    commandInjectionDetected: true;
+};
+export type CommandSubcommandPrefixResult = CommandPrefixResult & {
+    subcommandPrefixes: Map<string, CommandPrefixResult>;
+};
+/**
+ * 根据shell操作符将命令字符串拆分为单个命令
+ */
+export declare function splitCommand(command: string): string[];
+export declare const getCommandSubcommandPrefix: ((command: string, abortSignal: AbortSignal) => Promise<CommandSubcommandPrefixResult | null>) & import("lodash").MemoizedFunction;
+export declare function isUnsafeCompoundCommand(command: string): boolean;
+//# sourceMappingURL=commands.d.ts.map

package/dist/util/commands.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"commands.d.ts","sourceRoot":"","sources":["../../src/util/commands.ts"],"names":[],"mappings":"AAQA,MAAM,MAAM,mBAAmB,GAC3B;IACE,aAAa,EAAE,MAAM,GAAG,IAAI,CAAA;IAC5B,wBAAwB,EAAE,KAAK,CAAA;CAChC,GACD;IAAE,wBAAwB,EAAE,IAAI,CAAA;CAAE,CAAA;AAGtC,MAAM,MAAM,6BAA6B,GAAG,mBAAmB,GAAG;IAChE,kBAAkB,EAAE,GAAG,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAA;CACrD,CAAA;AAED;;GAEG;AACH,wBAAgB,YAAY,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,EAAE,CAkDtD;AAED,eAAO,MAAM,0BAA0B,aAE1B,MAAM,eACF,WAAW,KACvB,OAAO,CAAC,6BAA6B,GAAG,IAAI,CAAC,qCA+BjD,CAAA;AAuJD,wBAAgB,uBAAuB,CAAC,OAAO,EAAE,MAAM,GAAG,OAAO,CAEhE"}

package/dist/util/commands.js

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getCommandSubcommandPrefix = void 0;
+exports.splitCommand = splitCommand;
+exports.isUnsafeCompoundCommand = isUnsafeCompoundCommand;
+const lodash_es_1 = require("lodash-es");
+const shell_quote_1 = require("shell-quote");
+const queryLLM_1 = require("../services/api/queryLLM");
+const message_1 = require("../constants/message");
+const SINGLE_QUOTE = '__SINGLE_QUOTE__';
+const DOUBLE_QUOTE = '__DOUBLE_QUOTE__';
+/**
+ * 根据shell操作符将命令字符串拆分为单个命令
+ */
+function splitCommand(command) {
+    const parts = [];
+    // 1. 合并相邻的字符串
+    // 为引号添加特殊标记，防止后续解析时被误处理
+    for (const part of (0, shell_quote_1.parse)(command
+        .replaceAll('"', `"${DOUBLE_QUOTE}`)
+        .replaceAll("'", `'${SINGLE_QUOTE}`), varName => `$${varName}`)) {
+        if (typeof part === 'string') {
+            if (parts.length > 0 && typeof parts[parts.length - 1] === 'string') {
+                parts[parts.length - 1] += ' ' + part;
+                continue;
+            }
+        }
+        parts.push(part);
+    }
+    // 2. 将令牌映射为字符串
+    const stringParts = parts
+        .map(part => {
+        if (typeof part === 'string') {
+            return part;
+        }
+        if ('comment' in part) {
+            return '#' + part.comment;
+        }
+        if ('op' in part && part.op === 'glob') {
+            return part.pattern;
+        }
+        if ('op' in part) {
+            return part.op;
+        }
+        return null;
+    })
+        .filter(_ => _ !== null);
+    // 3. 将引号映射回原始形式
+    const quotedParts = stringParts.map(part => {
+        return part
+            .replaceAll(`${SINGLE_QUOTE}`, "'")
+            .replaceAll(`${DOUBLE_QUOTE}`, '"');
+    });
+    // 4. 过滤掉分隔符
+    return quotedParts.filter(part => !COMMAND_LIST_SEPARATORS.has(part));
+}
+exports.getCommandSubcommandPrefix = (0, lodash_es_1.memoize)(async (command, abortSignal) => {
+    const subcommands = splitCommand(command);
+    const [fullCommandPrefix, ...subcommandPrefixesResults] = await Promise.all([
+        getCommandPrefix(command, abortSignal),
+        ...subcommands.map(async (subcommand) => ({
+            subcommand,
+            prefix: await getCommandPrefix(subcommand, abortSignal),
+        })),
+    ]);
+    if (!fullCommandPrefix) {
+        return null;
+    }
+    const subcommandPrefixes = subcommandPrefixesResults.reduce((acc, { subcommand, prefix }) => {
+        if (prefix) {
+            acc.set(subcommand, prefix);
+        }
+        return acc;
+    }, new Map());
+    return {
+        ...fullCommandPrefix,
+        subcommandPrefixes,
+    };
+}, command => command);
+const getCommandPrefix = (0, lodash_es_1.memoize)(async (command, abortSignal) => {
+    const response = await (0, queryLLM_1.queryQuick)({
+        systemPrompt: [
+            {
+                type: 'text',
+                text: `Your task is to process Bash commands that an AI coding agent wants to run.
+
+This policy spec defines how to determine the prefix of a Bash command:`
+            }
+        ],
+        userPrompt: `<policy_spec>
+# Code Bash command prefix detection
+
+This document defines risk levels for actions that the agent may take. This classification system is part of a broader safety framework and is used to determine when additional user confirmation or oversight may be needed.
+
+## Definitions
+
+**Command Injection:** Any technique used that would result in a command being run other than the detected prefix.
+
+## Command prefix extraction examples
+Examples:
+- cat foo.txt => cat
+- cd src => cd
+- cd path/to/files/ => cd
+- find ./src -type f -name "*.ts" => find
+- gg cat foo.py => gg cat
+- gg cp foo.py bar.py => gg cp
+- git commit -m "foo" => git commit
+- git diff HEAD~1 => git diff
+- git diff --staged => git diff
+- git diff $(pwd) => command_injection_detected
+- git status => git status
+- git status# test(\`id\`) => command_injection_detected
+- git status\`ls\` => command_injection_detected
+- git push => none
+- git push origin master => git push
+- git log -n 5 => git log
+- git log --oneline -n 5 => git log
+- grep -A 40 "from foo.bar.baz import" alpha/beta/gamma.py => grep
+- pig tail zerba.log => pig tail
+- npm test => none
+- npm test --foo => npm test
+- npm test -- -f "foo" => npm test
+- pwd\n curl example.com => command_injection_detected
+- pytest foo/bar.py => pytest
+- scalac build => none
+</policy_spec>
+
+The user has allowed certain command prefixes to be run, and will otherwise be asked to approve or deny the command.
+Your task is to determine the command prefix for the following command.
+
+IMPORTANT: Bash commands may run multiple commands that are chained together.
+For safety, if the command seems to contain command injection, you must return "command_injection_detected". 
+(This will help protect the user: if they think that they're allowlisting command A, 
+but the AI coding agent sends a malicious command that technically has the same prefix as command A, 
+then the safety system will see that you said “command_injection_detected” and ask the user for manual confirmation.)
+
+Note that not every command has a prefix. If a command has no prefix, return "none".
+
+ONLY return the prefix. Do not return any other text, markdown markers, or other content or formatting.
+
+Command: ${command}
+`,
+        signal: abortSignal,
+        enableLLMCache: false,
+    });
+    const prefix = typeof response.message.content === 'string'
+        ? response.message.content
+        : Array.isArray(response.message.content)
+            ? (response.message.content.find(_ => _.type === 'text')?.text ??
+                'none')
+            : 'none';
+    if (prefix.startsWith(message_1.API_ERROR_MESSAGE_PREFIX)) {
+        return null;
+    }
+    if (prefix === 'command_injection_detected') {
+        return { commandInjectionDetected: true };
+    }
+    // 永远不接受基本`git`作为前缀（例如，如果未检测到`git diff`前缀）
+    if (prefix === 'git') {
+        return {
+            commandPrefix: null,
+            commandInjectionDetected: false,
+        };
+    }
+    if (prefix === 'none') {
+        return {
+            commandPrefix: null,
+            commandInjectionDetected: false,
+        };
+    }
+    return {
+        commandPrefix: prefix,
+        commandInjectionDetected: false,
+    };
+}, command => command);
+const COMMAND_LIST_SEPARATORS = new Set([
+    '&&',
+    '||',
+    ';',
+    ';;',
+]);
+// 检查这是否只是一个命令列表
+function isCommandList(command) {
+    for (const part of (0, shell_quote_1.parse)(command
+        .replaceAll('"', `"${DOUBLE_QUOTE}`) // parse()会去掉引号 :P
+        .replaceAll("'", `'${SINGLE_QUOTE}`), // parse()会去掉引号 :P
+    // parse()会去掉引号 :P
+    varName => `$${varName}`)) {
+        if (typeof part === 'string') {
+            // 字符串是安全的
+            continue;
+        }
+        if ('comment' in part) {
+            // 不信任注释，它们可能包含命令注入
+            return false;
+        }
+        if ('op' in part) {
+            if (part.op === 'glob') {
+                // 通配符是安全的
+                continue;
+            }
+            else if (COMMAND_LIST_SEPARATORS.has(part.op)) {
+                // 命令列表分隔符是安全的
+                continue;
+            }
+            // 其他操作符不安全
+            return false;
+        }
+    }
+    // 在整个命令中未找到不安全的操作符
+    return true;
+}
+function isUnsafeCompoundCommand(command) {
+    return splitCommand(command).length > 1 && !isCommandList(command);
+}
+//# sourceMappingURL=commands.js.map

package/dist/util/commands.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"commands.js","sourceRoot":"","sources":["../../src/util/commands.ts"],"names":[],"mappings":";;;AAuBA,oCAkDC;AA4LD,0DAEC;AAvQD,yCAAmC;AACnC,6CAAqE;AACrE,uDAAqD;AACrD,kDAA+D;AAE/D,MAAM,YAAY,GAAG,kBAAkB,CAAA;AACvC,MAAM,YAAY,GAAG,kBAAkB,CAAA;AAcvC;;GAEG;AACH,SAAgB,YAAY,CAAC,OAAe;IAC1C,MAAM,KAAK,GAAiB,EAAE,CAAA;IAE9B,cAAc;IACd,wBAAwB;IACxB,KAAK,MAAM,IAAI,IAAI,IAAA,mBAAK,EACtB,OAAO;SACJ,UAAU,CAAC,GAAG,EAAE,IAAI,YAAY,EAAE,CAAC;SACnC,UAAU,CAAC,GAAG,EAAE,IAAI,YAAY,EAAE,CAAC,EACtC,OAAO,CAAC,EAAE,CAAC,IAAI,OAAO,EAAE,CACzB,EAAE,CAAC;QACF,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,IAAI,OAAO,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,KAAK,QAAQ,EAAE,CAAC;gBACpE,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,IAAI,GAAG,GAAG,IAAI,CAAA;gBACrC,SAAQ;YACV,CAAC;QACH,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAA;IAClB,CAAC;IAED,eAAe;IACf,MAAM,WAAW,GAAG,KAAK;SACtB,GAAG,CAAC,IAAI,CAAC,EAAE;QACV,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAA;QACb,CAAC;QACD,IAAI,SAAS,IAAI,IAAI,EAAE,CAAC;YACtB,OAAO,GAAG,GAAG,IAAI,CAAC,OAAO,CAAA;QAC3B,CAAC;QACD,IAAI,IAAI,IAAI,IAAI,IAAI,IAAI,CAAC,EAAE,KAAK,MAAM,EAAE,CAAC;YACvC,OAAO,IAAI,CAAC,OAAO,CAAA;QACrB,CAAC;QACD,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YACjB,OAAO,IAAI,CAAC,EAAE,CAAA;QAChB,CAAC;QACD,OAAO,IAAI,CAAA;IACb,CAAC,CAAC;SACD,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAA;IAE1B,gBAAgB;IAChB,MAAM,WAAW,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QACzC,OAAO,IAAI;aACR,UAAU,CAAC,GAAG,YAAY,EAAE,EAAE,GAAG,CAAC;aAClC,UAAU,CAAC,GAAG,YAAY,EAAE,EAAE,GAAG,CAAC,CAAA;IACvC,CAAC,CAAC,CAAA;IAEF,YAAY;IACZ,OAAO,WAAW,CAAC,MAAM,CACvB,IAAI,CAAC,EAAE,CAAC,CAAE,uBAAuC,CAAC,GAAG,CAAC,IAAI,CAAC,CAC5D,CAAA;AACH,CAAC;AAEY,QAAA,0BAA0B,GAAG,IAAA,mBAAO,EAC/C,KAAK,EACH,OAAe,EACf,WAAwB,EACuB,EAAE;IACjD,MAAM,WAAW,GAAG,YAAY,CAAC,OAAO,CAAC,CAAA;IAEzC,MAAM,CAAC,iBAAiB,EAAE,GAAG,yBAAyB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CACzE;QACE,gBAAgB,CAAC,OAAO,EAAE,WAAW,CAAC;QACtC,GAAG,WAAW,CAAC,GAAG,CAAC,KAAK,EAAC,UAAU,EAAC,EAAE,CAAC,CAAC;YACtC,UAAU;YACV,MAAM,EAAE,MAAM,gBAAgB,CAAC,UAAU,EAAE,WAAW,CAAC;SACxD,CAAC,CAAC;KACJ,CACF,CAAA;IACD,IAAI,CAAC,iBAAiB,EAAE,CAAC;QACvB,OAAO,IAAI,CAAA;IACb,CAAC;IACD,MAAM,kBAAkB,GAAG,yBAAyB,CAAC,MAAM,CACzD,CAAC,GAAG,EAAE,EAAE,UAAU,EAAE,MAAM,EAAE,EAAE,EAAE;QAC9B,IAAI,MAAM,EAAE,CAAC;YACX,GAAG,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAA;QAC7B,CAAC;QACD,OAAO,GAAG,CAAA;IACZ,CAAC,EACD,IAAI,GAAG,EAA+B,CACvC,CAAA;IAED,OAAO;QACL,GAAG,iBAAiB;QACpB,kBAAkB;KACnB,CAAA;AACH,CAAC,EACD,OAAO,CAAC,EAAE,CAAC,OAAO,CACnB,CAAA;AAED,MAAM,gBAAgB,GAAG,IAAA,mBAAO,EAC9B,KAAK,EACH,OAAe,EACf,WAAwB,EACa,EAAE;IACvC,MAAM,QAAQ,GAAG,MAAM,IAAA,qBAAU,EAAC;QAChC,YAAY,EAAE;YACZ;gBACE,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE;;wEAEwD;aAC/D;SACF;QACD,UAAU,EAAE;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;WAmDP,OAAO;CACjB;QACK,MAAM,EAAE,WAAW;QACnB,cAAc,EAAE,KAAK;KACtB,CAAC,CAAA;IAEF,MAAM,MAAM,GACV,OAAO,QAAQ,CAAC,OAAO,CAAC,OAAO,KAAK,QAAQ;QAC1C,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO;QAC1B,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC;YACvC,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,EAAE,IAAI;gBAC5D,MAAM,CAAC;YACT,CAAC,CAAC,MAAM,CAAA;IAEd,IAAI,MAAM,CAAC,UAAU,CAAC,kCAAwB,CAAC,EAAE,CAAC;QAChD,OAAO,IAAI,CAAA;IACb,CAAC;IAED,IAAI,MAAM,KAAK,4BAA4B,EAAE,CAAC;QAC5C,OAAO,EAAE,wBAAwB,EAAE,IAAI,EAAE,CAAA;IAC3C,CAAC;IAED,0CAA0C;IAC1C,IAAI,MAAM,KAAK,KAAK,EAAE,CAAC;QACrB,OAAO;YACL,aAAa,EAAE,IAAI;YACnB,wBAAwB,EAAE,KAAK;SAChC,CAAA;IACH,CAAC;IAED,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;QACtB,OAAO;YACL,aAAa,EAAE,IAAI;YACnB,wBAAwB,EAAE,KAAK;SAChC,CAAA;IACH,CAAC;IAED,OAAO;QACL,aAAa,EAAE,MAAM;QACrB,wBAAwB,EAAE,KAAK;KAChC,CAAA;AACH,CAAC,EACD,OAAO,CAAC,EAAE,CAAC,OAAO,CACnB,CAAA;AAED,MAAM,uBAAuB,GAAG,IAAI,GAAG,CAAkB;IACvD,IAAI;IACJ,IAAI;IACJ,GAAG;IACH,IAAI;CACL,CAAC,CAAA;AAEF,gBAAgB;AAChB,SAAS,aAAa,CAAC,OAAe;IACpC,KAAK,MAAM,IAAI,IAAI,IAAA,mBAAK,EACtB,OAAO;SACJ,UAAU,CAAC,GAAG,EAAE,IAAI,YAAY,EAAE,CAAC,CAAC,kBAAkB;SACtD,UAAU,CAAC,GAAG,EAAE,IAAI,YAAY,EAAE,CAAC,EAAE,kBAAkB;IAC1D,AADwC,kBAAkB;IAC1D,OAAO,CAAC,EAAE,CAAC,IAAI,OAAO,EAAE,CACzB,EAAE,CAAC;QACF,IAAI,OAAO,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC7B,UAAU;YACV,SAAQ;QACV,CAAC;QACD,IAAI,SAAS,IAAI,IAAI,EAAE,CAAC;YACtB,mBAAmB;YACnB,OAAO,KAAK,CAAA;QACd,CAAC;QACD,IAAI,IAAI,IAAI,IAAI,EAAE,CAAC;YACjB,IAAI,IAAI,CAAC,EAAE,KAAK,MAAM,EAAE,CAAC;gBACvB,UAAU;gBACV,SAAQ;YACV,CAAC;iBAAM,IAAI,uBAAuB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;gBAChD,cAAc;gBACd,SAAQ;YACV,CAAC;YACD,WAAW;YACX,OAAO,KAAK,CAAA;QACd,CAAC;IACH,CAAC;IACD,mBAAmB;IACnB,OAAO,IAAI,CAAA;AACb,CAAC;AAED,SAAgB,uBAAuB,CAAC,OAAe;IACrD,OAAO,YAAY,CAAC,OAAO,CAAC,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,aAAa,CAAC,OAAO,CAAC,CAAA;AACpE,CAAC"}

package/dist/util/compact.d.ts

@@ -0,0 +1,25 @@
+import { Message } from '../types/message';
+/**
+ * 执行上下文压缩（公开接口）
+ *
+ * 该函数直接执行压缩，不检查阈值。适用于：
+ * - 用户手动触发 /compact 命令
+ * - 自动压缩检查通过后调用
+
+ * 改进的错误处理机制：
+ * - 压缩失败时会自动使用消息截断作为备用策略
+ * - 确保系统在各种异常情况下都能保持功能正常
+ * - 提供多层降级方案以避免完全失败
+ */
+export declare function compactMessages(messages: Message[], abortController: AbortController): Promise<Message[]>;
+/**
+ * 自动上下文压缩的主要入口函数
+ *
+ * 该函数在每次查询前被调用，用于检查对话是否已超出容量需要压缩。
+ * 会自动分离最后的用户消息（如果有），压缩历史消息后再添加回用户消息。
+ */
+export declare function checkAutoCompact(messages: Message[], abortController: AbortController): Promise<{
+    messages: Message[];
+    wasCompacted: boolean;
+}>;
+//# sourceMappingURL=compact.d.ts.map

package/dist/util/compact.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"compact.d.ts","sourceRoot":"","sources":["../../src/util/compact.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,OAAO,EAAE,MAAM,kBAAkB,CAAA;AA6R1C;;;;;;;;;;;GAWG;AACH,wBAAsB,eAAe,CACnC,QAAQ,EAAE,OAAO,EAAE,EACnB,eAAe,EAAE,eAAe,GAC/B,OAAO,CAAC,OAAO,EAAE,CAAC,CA8CpB;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CACpC,QAAQ,EAAE,OAAO,EAAE,EACnB,eAAe,EAAE,eAAe,GAC/B,OAAO,CAAC;IAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IAAC,YAAY,EAAE,OAAO,CAAA;CAAE,CAAC,CA2CzD"}