@aalis/plugin-prompt-budget 0.1.0

package/dist/index.d.ts

@@ -0,0 +1,18 @@
+/**
+ * @aalis/plugin-prompt-budget —— 暴露 Prompt 预算自检工具
+ *
+ * 设计动机：plugin-agent-default 已 emit `token:usage` 事件（含 12 桶 breakdown），
+ * WebUI 通过 plugin-webui-server 订阅并渲染面板。但 AI 自己跑在工具循环里时，
+ * 没有看 WebUI 的机会——需要一个**主动 query** 路径，让模型怀疑"是不是 prompt 太大了"
+ * 时能立即查到当前 session 的最新预算消耗。
+ *
+ * 实现：监听 `token:usage` 事件维护 sessionId → 最新 usage 缓存，注册
+ * `prompt_budget_info` 工具供 AI 查询。零业务逻辑，纯只读 introspection。
+ */
+import type { Context } from '@aalis/core';
+import '@aalis/plugin-tools-api';
+export declare const name = "@aalis/plugin-prompt-budget";
+export declare const displayName = "Prompt \u9884\u7B97\u81EA\u68C0";
+export declare const subsystem = "agent";
+export declare function apply(ctx: Context): void;
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAEH,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAG3C,OAAO,yBAAyB,CAAC;AAEjC,eAAO,MAAM,IAAI,gCAAgC,CAAC;AAClD,eAAO,MAAM,WAAW,oCAAgB,CAAC;AACzC,eAAO,MAAM,SAAS,UAAU,CAAC;AAQjC,wBAAgB,KAAK,CAAC,GAAG,EAAE,OAAO,GAAG,IAAI,CA6ExC"}

package/dist/index.js

@@ -0,0 +1,89 @@
+/**
+ * @aalis/plugin-prompt-budget —— 暴露 Prompt 预算自检工具
+ *
+ * 设计动机：plugin-agent-default 已 emit `token:usage` 事件（含 12 桶 breakdown），
+ * WebUI 通过 plugin-webui-server 订阅并渲染面板。但 AI 自己跑在工具循环里时，
+ * 没有看 WebUI 的机会——需要一个**主动 query** 路径，让模型怀疑"是不是 prompt 太大了"
+ * 时能立即查到当前 session 的最新预算消耗。
+ *
+ * 实现：监听 `token:usage` 事件维护 sessionId → 最新 usage 缓存，注册
+ * `prompt_budget_info` 工具供 AI 查询。零业务逻辑，纯只读 introspection。
+ */
+import { useToolService } from '@aalis/plugin-tools-api';
+import '@aalis/plugin-tools-api';
+export const name = '@aalis/plugin-prompt-budget';
+export const displayName = 'Prompt 预算自检';
+export const subsystem = 'agent';
+export function apply(ctx) {
+    const tools = useToolService(ctx);
+    /** sessionId → 最新一次的 token:usage 快照（仅保留最近一次即可） */
+    const cache = new Map();
+    ctx.on('token:usage', u => {
+        if (!u || typeof u.sessionId !== 'string')
+            return;
+        cache.set(u.sessionId, { ...u, observedAt: Date.now() });
+    });
+    tools.register({
+        definition: {
+            type: 'function',
+            function: {
+                name: 'prompt_budget_info',
+                description: '查询当前 session 最近一次 LLM 调用的 prompt token 预算消耗。' +
+                    '当你怀疑"system prompt 过大""上下文快满了"或想验证压缩 / 工具裁剪是否生效时使用。' +
+                    '返回 12 桶 breakdown（system / persona / memorySummary / memoryVector / skills / ' +
+                    'platform / subtask / systemOther / history / toolResults / toolDefs / reservedForReply）' +
+                    '及 used / contextWindow / usageRatio。' +
+                    '注意：数据由上一次 LLM 调用 emit，刚启动或未调用时返回 noData。',
+                parameters: {
+                    type: 'object',
+                    properties: {
+                        sessionId: {
+                            type: 'string',
+                            description: '可选。默认使用当前调用 session（callCtx.sessionId）；传入可查询其他会话。',
+                        },
+                    },
+                    required: [],
+                    additionalProperties: false,
+                },
+            },
+        },
+        handler: async (args, callCtx) => {
+            const sessionId = args.sessionId || callCtx.sessionId;
+            const u = cache.get(sessionId);
+            if (!u) {
+                return JSON.stringify({
+                    noData: true,
+                    sessionId,
+                    advice: '本 session 尚未产生 token:usage 事件（可能是首轮调用前或本 session 从未调过 LLM）。' +
+                        '在 LLM 至少调用一次后再查询本工具即可。',
+                });
+            }
+            const ageMs = Date.now() - u.observedAt;
+            const tag = u.usageRatio >= 0.85 ? 'CRITICAL' : u.usageRatio >= 0.7 ? 'WARN' : u.usageRatio >= 0.5 ? 'INFO' : 'OK';
+            // top3 提示哪几个桶最大
+            const buckets = Object.entries(u.breakdown).sort((a, b) => b[1] - a[1]);
+            const top3 = buckets.slice(0, 3).map(([k, v]) => ({ name: k, tokens: v }));
+            const advice = u.usageRatio >= 0.85
+                ? '上下文几乎用尽。建议：调用 memory.compress / 清理 toolResults / 缩减 system prompt。'
+                : u.usageRatio >= 0.7
+                    ? '上下文压力较高。可考虑主动压缩历史或减少后续工具调用的输出体量。'
+                    : '预算健康，无需干预。';
+            return JSON.stringify({
+                sessionId,
+                ageMs,
+                observedAt: new Date(u.observedAt).toISOString(),
+                tag,
+                used: u.used,
+                contextWindow: u.contextWindow,
+                usageRatio: Number(u.usageRatio.toFixed(4)),
+                maxTokens: u.maxTokens,
+                tokenBudget: u.tokenBudget,
+                breakdown: u.breakdown,
+                top3,
+                advice,
+            });
+        },
+    });
+    ctx.logger.info('prompt_budget_info 工具已注册');
+}
+//# sourceMappingURL=index.js.map

package/dist/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;GAUG;AAIH,OAAO,EAAE,cAAc,EAAE,MAAM,yBAAyB,CAAC;AACzD,OAAO,yBAAyB,CAAC;AAEjC,MAAM,CAAC,MAAM,IAAI,GAAG,6BAA6B,CAAC;AAClD,MAAM,CAAC,MAAM,WAAW,GAAG,aAAa,CAAC;AACzC,MAAM,CAAC,MAAM,SAAS,GAAG,OAAO,CAAC;AAQjC,MAAM,UAAU,KAAK,CAAC,GAAY;IAChC,MAAM,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC;IAClC,kDAAkD;IAClD,MAAM,KAAK,GAAG,IAAI,GAAG,EAAsB,CAAC;IAE5C,GAAG,CAAC,EAAE,CAAC,aAAa,EAAE,CAAC,CAAC,EAAE;QACxB,IAAI,CAAC,CAAC,IAAI,OAAO,CAAC,CAAC,SAAS,KAAK,QAAQ;YAAE,OAAO;QAClD,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,EAAE,GAAG,CAAC,EAAE,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;IAEH,KAAK,CAAC,QAAQ,CAAC;QACb,UAAU,EAAE;YACV,IAAI,EAAE,UAAU;YAChB,QAAQ,EAAE;gBACR,IAAI,EAAE,oBAAoB;gBAC1B,WAAW,EACT,8CAA8C;oBAC9C,qDAAqD;oBACrD,+EAA+E;oBAC/E,yFAAyF;oBACzF,sCAAsC;oBACtC,0CAA0C;gBAC5C,UAAU,EAAE;oBACV,IAAI,EAAE,QAAQ;oBACd,UAAU,EAAE;wBACV,SAAS,EAAE;4BACT,IAAI,EAAE,QAAQ;4BACd,WAAW,EAAE,mDAAmD;yBACjE;qBACF;oBACD,QAAQ,EAAE,EAAE;oBACZ,oBAAoB,EAAE,KAAK;iBAC5B;aACF;SACF;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE;YAC/B,MAAM,SAAS,GAAI,IAAI,CAAC,SAAoB,IAAI,OAAO,CAAC,SAAS,CAAC;YAClE,MAAM,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;YAC/B,IAAI,CAAC,CAAC,EAAE,CAAC;gBACP,OAAO,IAAI,CAAC,SAAS,CAAC;oBACpB,MAAM,EAAE,IAAI;oBACZ,SAAS;oBACT,MAAM,EACJ,6DAA6D;wBAC7D,wBAAwB;iBAC3B,CAAC,CAAC;YACL,CAAC;YACD,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,UAAU,CAAC;YACxC,MAAM,GAAG,GACP,CAAC,CAAC,UAAU,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC;YACzG,gBAAgB;YAChB,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;YACxE,MAAM,IAAI,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC3E,MAAM,MAAM,GACV,CAAC,CAAC,UAAU,IAAI,IAAI;gBAClB,CAAC,CAAC,oEAAoE;gBACtE,CAAC,CAAC,CAAC,CAAC,UAAU,IAAI,GAAG;oBACnB,CAAC,CAAC,kCAAkC;oBACpC,CAAC,CAAC,YAAY,CAAC;YACrB,OAAO,IAAI,CAAC,SAAS,CAAC;gBACpB,SAAS;gBACT,KAAK;gBACL,UAAU,EAAE,IAAI,IAAI,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE;gBAChD,GAAG;gBACH,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,aAAa,EAAE,CAAC,CAAC,aAAa;gBAC9B,UAAU,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;gBAC3C,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,WAAW,EAAE,CAAC,CAAC,WAAW;gBAC1B,SAAS,EAAE,CAAC,CAAC,SAAS;gBACtB,IAAI;gBACJ,MAAM;aACP,CAAC,CAAC;QACL,CAAC;KACF,CAAC,CAAC;IAEH,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,0BAA0B,CAAC,CAAC;AAC9C,CAAC"}

package/package.json

@@ -0,0 +1,29 @@
+{
+  "name": "@aalis/plugin-prompt-budget",
+  "version": "0.1.0",
+  "type": "module",
+  "keywords": [
+    "aalis-plugin"
+  ],
+  "main": "dist/index.js",
+  "types": "dist/index.d.ts",
+  "files": [
+    "dist"
+  ],
+  "dependencies": {
+    "@aalis/plugin-tools-api": "^0.1.0"
+  },
+  "devDependencies": {
+    "@types/node": "^22.0.0",
+    "typescript": "^5.7.0",
+    "@aalis/core": "0.1.0",
+    "@aalis/plugin-agent-api": "0.1.0"
+  },
+  "peerDependencies": {
+    "@aalis/core": "^0.1.0"
+  },
+  "scripts": {
+    "build": "tsc",
+    "dev": "tsc --watch"
+  }
+}