@dommaker/harness 0.8.1 → 0.9.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/README.md +126 -183
- package/bin/harness.js +29 -12
- package/dist/agents/index.d.ts +3 -0
- package/dist/agents/index.d.ts.map +1 -0
- package/dist/agents/index.js +19 -0
- package/dist/agents/index.js.map +1 -0
- package/dist/agents/lifecycle.d.ts +68 -0
- package/dist/agents/lifecycle.d.ts.map +1 -0
- package/dist/agents/lifecycle.js +181 -0
- package/dist/agents/lifecycle.js.map +1 -0
- package/dist/agents/types.d.ts +42 -0
- package/dist/agents/types.d.ts.map +1 -0
- package/dist/agents/types.js +6 -0
- package/dist/agents/types.js.map +1 -0
- package/dist/architecture/cross-project-checker.d.ts +19 -1
- package/dist/architecture/cross-project-checker.d.ts.map +1 -1
- package/dist/architecture/cross-project-checker.js +72 -60
- package/dist/architecture/cross-project-checker.js.map +1 -1
- package/dist/cli/commands/acceptance.d.ts.map +1 -1
- package/dist/cli/commands/acceptance.js +1 -0
- package/dist/cli/commands/acceptance.js.map +1 -1
- package/dist/cli/commands/check.d.ts.map +1 -1
- package/dist/cli/commands/check.js +2 -4
- package/dist/cli/commands/check.js.map +1 -1
- package/dist/cli/commands/command.d.ts +19 -0
- package/dist/cli/commands/command.d.ts.map +1 -0
- package/dist/cli/commands/command.js +81 -0
- package/dist/cli/commands/command.js.map +1 -0
- package/dist/cli/commands/index.d.ts +1 -0
- package/dist/cli/commands/index.d.ts.map +1 -1
- package/dist/cli/commands/index.js +3 -1
- package/dist/cli/commands/index.js.map +1 -1
- package/dist/cli/commands/passes-gate.d.ts.map +1 -1
- package/dist/cli/commands/passes-gate.js +2 -4
- package/dist/cli/commands/passes-gate.js.map +1 -1
- package/dist/cli/commands/review.d.ts.map +1 -1
- package/dist/cli/commands/review.js +27 -63
- package/dist/cli/commands/review.js.map +1 -1
- package/dist/cli/commands/security.d.ts +1 -1
- package/dist/cli/commands/security.d.ts.map +1 -1
- package/dist/cli/commands/security.js +40 -94
- package/dist/cli/commands/security.js.map +1 -1
- package/dist/cli/commands/status.d.ts.map +1 -1
- package/dist/cli/commands/status.js +2 -2
- package/dist/cli/commands/status.js.map +1 -1
- package/dist/constraints/definitions.d.ts +39 -0
- package/dist/constraints/definitions.d.ts.map +1 -0
- package/dist/constraints/definitions.js +71 -0
- package/dist/constraints/definitions.js.map +1 -0
- package/dist/constraints/index.d.ts +5 -0
- package/dist/constraints/index.d.ts.map +1 -0
- package/dist/constraints/index.js +21 -0
- package/dist/constraints/index.js.map +1 -0
- package/dist/constraints/quality.d.ts +11 -0
- package/dist/constraints/quality.d.ts.map +1 -0
- package/dist/constraints/quality.js +257 -0
- package/dist/constraints/quality.js.map +1 -0
- package/dist/constraints/registry.d.ts +64 -0
- package/dist/constraints/registry.d.ts.map +1 -0
- package/dist/constraints/registry.js +167 -0
- package/dist/constraints/registry.js.map +1 -0
- package/dist/constraints/safety.d.ts +13 -0
- package/dist/constraints/safety.d.ts.map +1 -0
- package/dist/constraints/safety.js +103 -0
- package/dist/constraints/safety.js.map +1 -0
- package/dist/constraints/types.d.ts +53 -0
- package/dist/constraints/types.d.ts.map +1 -0
- package/dist/constraints/types.js +8 -0
- package/dist/constraints/types.js.map +1 -0
- package/dist/context/compaction.d.ts +57 -0
- package/dist/context/compaction.d.ts.map +1 -0
- package/dist/context/compaction.js +166 -0
- package/dist/context/compaction.js.map +1 -0
- package/dist/context/file-budget.d.ts +40 -0
- package/dist/context/file-budget.d.ts.map +1 -0
- package/dist/context/file-budget.js +147 -0
- package/dist/context/file-budget.js.map +1 -0
- package/dist/context/index.d.ts +15 -0
- package/dist/context/index.d.ts.map +1 -0
- package/dist/context/index.js +31 -0
- package/dist/context/index.js.map +1 -0
- package/dist/context/knowledge-injector.d.ts +49 -0
- package/dist/context/knowledge-injector.d.ts.map +1 -0
- package/dist/context/knowledge-injector.js +122 -0
- package/dist/context/knowledge-injector.js.map +1 -0
- package/dist/context/progressive-loader.d.ts +0 -1
- package/dist/context/progressive-loader.d.ts.map +1 -1
- package/dist/context/progressive-loader.js +15 -32
- package/dist/context/progressive-loader.js.map +1 -1
- package/dist/context/session-manager.d.ts +67 -0
- package/dist/context/session-manager.d.ts.map +1 -0
- package/dist/context/session-manager.js +275 -0
- package/dist/context/session-manager.js.map +1 -0
- package/dist/context/token-pipeline.d.ts +55 -0
- package/dist/context/token-pipeline.d.ts.map +1 -0
- package/dist/context/token-pipeline.js +220 -0
- package/dist/context/token-pipeline.js.map +1 -0
- package/dist/context/tool-output-budget.d.ts +59 -0
- package/dist/context/tool-output-budget.d.ts.map +1 -0
- package/dist/context/tool-output-budget.js +177 -0
- package/dist/context/tool-output-budget.js.map +1 -0
- package/dist/context/types.d.ts +113 -0
- package/dist/context/types.d.ts.map +1 -0
- package/dist/context/types.js +30 -0
- package/dist/context/types.js.map +1 -0
- package/dist/core/constraints/checker.d.ts +9 -1
- package/dist/core/constraints/checker.d.ts.map +1 -1
- package/dist/core/constraints/checker.js +72 -155
- package/dist/core/constraints/checker.js.map +1 -1
- package/dist/core/constraints/definitions.d.ts.map +1 -1
- package/dist/core/constraints/definitions.js +8 -10
- package/dist/core/constraints/definitions.js.map +1 -1
- package/dist/core/constraints/interceptor.d.ts.map +1 -1
- package/dist/core/constraints/interceptor.js +2 -6
- package/dist/core/constraints/interceptor.js.map +1 -1
- package/dist/core/project-config-loader.d.ts.map +1 -1
- package/dist/core/project-config-loader.js +12 -30
- package/dist/core/project-config-loader.js.map +1 -1
- package/dist/core/session/clean-state.d.ts.map +1 -1
- package/dist/core/session/clean-state.js +6 -8
- package/dist/core/session/clean-state.js.map +1 -1
- package/dist/core/session/startup.d.ts.map +1 -1
- package/dist/core/session/startup.js +3 -5
- package/dist/core/session/startup.js.map +1 -1
- package/dist/core/spec/validator.d.ts.map +1 -1
- package/dist/core/spec/validator.js +2 -4
- package/dist/core/spec/validator.js.map +1 -1
- package/dist/core/validators/checkpoint.d.ts.map +1 -1
- package/dist/core/validators/checkpoint.js +3 -5
- package/dist/core/validators/checkpoint.js.map +1 -1
- package/dist/core/validators/passes-gate.d.ts +0 -4
- package/dist/core/validators/passes-gate.d.ts.map +1 -1
- package/dist/core/validators/passes-gate.js +5 -13
- package/dist/core/validators/passes-gate.js.map +1 -1
- package/dist/dashboard/data.d.ts +39 -0
- package/dist/dashboard/data.d.ts.map +1 -0
- package/dist/dashboard/data.js +89 -0
- package/dist/dashboard/data.js.map +1 -0
- package/dist/dashboard/index.d.ts +4 -0
- package/dist/dashboard/index.d.ts.map +1 -0
- package/dist/dashboard/index.js +20 -0
- package/dist/dashboard/index.js.map +1 -0
- package/dist/dashboard/stats.d.ts +20 -0
- package/dist/dashboard/stats.d.ts.map +1 -0
- package/dist/dashboard/stats.js +115 -0
- package/dist/dashboard/stats.js.map +1 -0
- package/dist/dashboard/types.d.ts +98 -0
- package/dist/dashboard/types.d.ts.map +1 -0
- package/dist/dashboard/types.js +8 -0
- package/dist/dashboard/types.js.map +1 -0
- package/dist/failure/recorder.d.ts +4 -0
- package/dist/failure/recorder.d.ts.map +1 -1
- package/dist/failure/recorder.js +9 -11
- package/dist/failure/recorder.js.map +1 -1
- package/dist/gates/acceptance.d.ts.map +1 -1
- package/dist/gates/acceptance.js +2 -4
- package/dist/gates/acceptance.js.map +1 -1
- package/dist/gates/contract.d.ts.map +1 -1
- package/dist/gates/contract.js +0 -3
- package/dist/gates/contract.js.map +1 -1
- package/dist/gates/performance.d.ts.map +1 -1
- package/dist/gates/performance.js +3 -5
- package/dist/gates/performance.js.map +1 -1
- package/dist/gates/review.d.ts.map +1 -1
- package/dist/gates/review.js +4 -6
- package/dist/gates/review.js.map +1 -1
- package/dist/gates/security.d.ts.map +1 -1
- package/dist/gates/security.js +2 -4
- package/dist/gates/security.js.map +1 -1
- package/dist/index.d.ts +9 -2
- package/dist/index.d.ts.map +1 -1
- package/dist/index.js +34 -3
- package/dist/index.js.map +1 -1
- package/dist/knowledge/import.d.ts +95 -0
- package/dist/knowledge/import.d.ts.map +1 -0
- package/dist/knowledge/import.js +429 -0
- package/dist/knowledge/import.js.map +1 -0
- package/dist/knowledge/index.d.ts +15 -0
- package/dist/knowledge/index.d.ts.map +1 -0
- package/dist/knowledge/index.js +40 -0
- package/dist/knowledge/index.js.map +1 -0
- package/dist/knowledge/ingest.d.ts +30 -0
- package/dist/knowledge/ingest.d.ts.map +1 -0
- package/dist/knowledge/ingest.js +94 -0
- package/dist/knowledge/ingest.js.map +1 -0
- package/dist/knowledge/lifecycle-hooks.d.ts +62 -0
- package/dist/knowledge/lifecycle-hooks.d.ts.map +1 -0
- package/dist/knowledge/lifecycle-hooks.js +140 -0
- package/dist/knowledge/lifecycle-hooks.js.map +1 -0
- package/dist/knowledge/lifecycle.d.ts +43 -0
- package/dist/knowledge/lifecycle.d.ts.map +1 -0
- package/dist/knowledge/lifecycle.js +141 -0
- package/dist/knowledge/lifecycle.js.map +1 -0
- package/dist/knowledge/lint.d.ts +57 -0
- package/dist/knowledge/lint.d.ts.map +1 -0
- package/dist/knowledge/lint.js +235 -0
- package/dist/knowledge/lint.js.map +1 -0
- package/dist/knowledge/query.d.ts +33 -0
- package/dist/knowledge/query.d.ts.map +1 -0
- package/dist/knowledge/query.js +107 -0
- package/dist/knowledge/query.js.map +1 -0
- package/dist/knowledge/reference-tracker.d.ts +31 -0
- package/dist/knowledge/reference-tracker.d.ts.map +1 -0
- package/dist/knowledge/reference-tracker.js +124 -0
- package/dist/knowledge/reference-tracker.js.map +1 -0
- package/dist/knowledge/store.d.ts +42 -0
- package/dist/knowledge/store.d.ts.map +1 -0
- package/dist/knowledge/store.js +278 -0
- package/dist/knowledge/store.js.map +1 -0
- package/dist/knowledge/types.d.ts +103 -0
- package/dist/knowledge/types.d.ts.map +1 -0
- package/dist/knowledge/types.js +17 -0
- package/dist/knowledge/types.js.map +1 -0
- package/dist/llm/adapter.d.ts +36 -0
- package/dist/llm/adapter.d.ts.map +1 -0
- package/dist/llm/adapter.js +118 -0
- package/dist/llm/adapter.js.map +1 -0
- package/dist/llm/index.d.ts +3 -0
- package/dist/llm/index.d.ts.map +1 -0
- package/dist/llm/index.js +19 -0
- package/dist/llm/index.js.map +1 -0
- package/dist/llm/types.d.ts +62 -0
- package/dist/llm/types.d.ts.map +1 -0
- package/dist/llm/types.js +8 -0
- package/dist/llm/types.js.map +1 -0
- package/dist/monitoring/context-tracker.d.ts +38 -0
- package/dist/monitoring/context-tracker.d.ts.map +1 -0
- package/dist/monitoring/context-tracker.js +162 -0
- package/dist/monitoring/context-tracker.js.map +1 -0
- package/dist/monitoring/index.d.ts +3 -0
- package/dist/monitoring/index.d.ts.map +1 -1
- package/dist/monitoring/index.js +6 -0
- package/dist/monitoring/index.js.map +1 -1
- package/dist/monitoring/knowledge-doctor.d.ts +78 -0
- package/dist/monitoring/knowledge-doctor.d.ts.map +1 -0
- package/dist/monitoring/knowledge-doctor.js +266 -0
- package/dist/monitoring/knowledge-doctor.js.map +1 -0
- package/dist/monitoring/knowledge-evolver.d.ts +72 -0
- package/dist/monitoring/knowledge-evolver.d.ts.map +1 -0
- package/dist/monitoring/knowledge-evolver.js +205 -0
- package/dist/monitoring/knowledge-evolver.js.map +1 -0
- package/dist/monitoring/performance-analyzer.d.ts.map +1 -1
- package/dist/monitoring/performance-analyzer.js +10 -4
- package/dist/monitoring/performance-analyzer.js.map +1 -1
- package/dist/monitoring/trace-analyzer.d.ts.map +1 -1
- package/dist/monitoring/trace-analyzer.js +12 -5
- package/dist/monitoring/trace-analyzer.js.map +1 -1
- package/dist/safety/index.d.ts +6 -0
- package/dist/safety/index.d.ts.map +1 -0
- package/dist/safety/index.js +22 -0
- package/dist/safety/index.js.map +1 -0
- package/dist/safety/input-guardrail.d.ts +36 -0
- package/dist/safety/input-guardrail.d.ts.map +1 -0
- package/dist/safety/input-guardrail.js +118 -0
- package/dist/safety/input-guardrail.js.map +1 -0
- package/dist/safety/output-guardrail.d.ts +40 -0
- package/dist/safety/output-guardrail.d.ts.map +1 -0
- package/dist/safety/output-guardrail.js +150 -0
- package/dist/safety/output-guardrail.js.map +1 -0
- package/dist/safety/sandbox.d.ts +45 -0
- package/dist/safety/sandbox.d.ts.map +1 -0
- package/dist/safety/sandbox.js +134 -0
- package/dist/safety/sandbox.js.map +1 -0
- package/dist/safety/tool-guardrail.d.ts +51 -0
- package/dist/safety/tool-guardrail.d.ts.map +1 -0
- package/dist/safety/tool-guardrail.js +147 -0
- package/dist/safety/tool-guardrail.js.map +1 -0
- package/dist/safety/types.d.ts +104 -0
- package/dist/safety/types.d.ts.map +1 -0
- package/dist/safety/types.js +6 -0
- package/dist/safety/types.js.map +1 -0
- package/dist/tools/core/index.d.ts +13 -0
- package/dist/tools/core/index.d.ts.map +1 -0
- package/dist/tools/core/index.js +198 -0
- package/dist/tools/core/index.js.map +1 -0
- package/dist/tools/index.d.ts +4 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +20 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/registry.d.ts +59 -0
- package/dist/tools/registry.d.ts.map +1 -0
- package/dist/tools/registry.js +115 -0
- package/dist/tools/registry.js.map +1 -0
- package/dist/tools/types.d.ts +38 -0
- package/dist/tools/types.d.ts.map +1 -0
- package/dist/tools/types.js +6 -0
- package/dist/tools/types.js.map +1 -0
- package/dist/types/constraint.d.ts +4 -0
- package/dist/types/constraint.d.ts.map +1 -1
- package/dist/types/constraint.js.map +1 -1
- package/dist/utils/exec.d.ts +11 -1
- package/dist/utils/exec.d.ts.map +1 -1
- package/dist/utils/exec.js +19 -1
- package/dist/utils/exec.js.map +1 -1
- package/dist/verification/index.d.ts +4 -0
- package/dist/verification/index.d.ts.map +1 -0
- package/dist/verification/index.js +20 -0
- package/dist/verification/index.js.map +1 -0
- package/dist/verification/loop.d.ts +30 -0
- package/dist/verification/loop.d.ts.map +1 -0
- package/dist/verification/loop.js +114 -0
- package/dist/verification/loop.js.map +1 -0
- package/dist/verification/rules-based.d.ts +40 -0
- package/dist/verification/rules-based.d.ts.map +1 -0
- package/dist/verification/rules-based.js +130 -0
- package/dist/verification/rules-based.js.map +1 -0
- package/dist/verification/types.d.ts +69 -0
- package/dist/verification/types.d.ts.map +1 -0
- package/dist/verification/types.js +6 -0
- package/dist/verification/types.js.map +1 -0
- package/package.json +1 -1
- package/dist/cli/commands/diagnose.d.ts +0 -16
- package/dist/cli/commands/diagnose.d.ts.map +0 -1
- package/dist/cli/commands/diagnose.js +0 -195
- package/dist/cli/commands/diagnose.js.map +0 -1
- package/dist/cli/commands/propose.d.ts +0 -18
- package/dist/cli/commands/propose.d.ts.map +0 -1
- package/dist/cli/commands/propose.js +0 -331
- package/dist/cli/commands/propose.js.map +0 -1
- package/dist/cli/commands/traces.d.ts +0 -15
- package/dist/cli/commands/traces.d.ts.map +0 -1
- package/dist/cli/commands/traces.js +0 -167
- package/dist/cli/commands/traces.js.map +0 -1
- package/dist/extensions/long-running/constraints.d.ts +0 -38
- package/dist/extensions/long-running/constraints.d.ts.map +0 -1
- package/dist/extensions/long-running/constraints.js +0 -153
- package/dist/extensions/long-running/constraints.js.map +0 -1
- package/dist/extensions/long-running/index.d.ts +0 -30
- package/dist/extensions/long-running/index.d.ts.map +0 -1
- package/dist/extensions/long-running/index.js +0 -50
- package/dist/extensions/long-running/index.js.map +0 -1
- package/dist/extensions/long-running/types.d.ts +0 -154
- package/dist/extensions/long-running/types.d.ts.map +0 -1
- package/dist/extensions/long-running/types.js +0 -9
- package/dist/extensions/long-running/types.js.map +0 -1
- package/dist/presets/long-running.d.ts +0 -33
- package/dist/presets/long-running.d.ts.map +0 -1
- package/dist/presets/long-running.js +0 -52
- package/dist/presets/long-running.js.map +0 -1
|
@@ -0,0 +1,275 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Session Manager
|
|
4
|
+
*
|
|
5
|
+
* 解耦 Session(持久事件日志)/ Harness(窗口视图)/ Sandbox(执行环境)
|
|
6
|
+
* Session 事件存储:.harness/sessions/{id}/events.jsonl
|
|
7
|
+
* Checkpoint 存储:.harness/sessions/{id}/checkpoints/
|
|
8
|
+
*/
|
|
9
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
12
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
13
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
14
|
+
}
|
|
15
|
+
Object.defineProperty(o, k2, desc);
|
|
16
|
+
}) : (function(o, m, k, k2) {
|
|
17
|
+
if (k2 === undefined) k2 = k;
|
|
18
|
+
o[k2] = m[k];
|
|
19
|
+
}));
|
|
20
|
+
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
21
|
+
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
22
|
+
}) : function(o, v) {
|
|
23
|
+
o["default"] = v;
|
|
24
|
+
});
|
|
25
|
+
var __importStar = (this && this.__importStar) || (function () {
|
|
26
|
+
var ownKeys = function(o) {
|
|
27
|
+
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
28
|
+
var ar = [];
|
|
29
|
+
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
30
|
+
return ar;
|
|
31
|
+
};
|
|
32
|
+
return ownKeys(o);
|
|
33
|
+
};
|
|
34
|
+
return function (mod) {
|
|
35
|
+
if (mod && mod.__esModule) return mod;
|
|
36
|
+
var result = {};
|
|
37
|
+
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
38
|
+
__setModuleDefault(result, mod);
|
|
39
|
+
return result;
|
|
40
|
+
};
|
|
41
|
+
})();
|
|
42
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
43
|
+
exports.SessionManager = void 0;
|
|
44
|
+
const fs = __importStar(require("fs"));
|
|
45
|
+
const path = __importStar(require("path"));
|
|
46
|
+
const token_pipeline_1 = require("./token-pipeline");
|
|
47
|
+
const context_tracker_1 = require("../monitoring/context-tracker");
|
|
48
|
+
class SessionManager {
|
|
49
|
+
basePath;
|
|
50
|
+
sessions = new Map();
|
|
51
|
+
pipeline;
|
|
52
|
+
tracker;
|
|
53
|
+
constructor(basePath) {
|
|
54
|
+
this.basePath = basePath || process.cwd();
|
|
55
|
+
this.pipeline = new token_pipeline_1.TokenPipeline();
|
|
56
|
+
this.tracker = new context_tracker_1.ContextTracker(this.basePath);
|
|
57
|
+
}
|
|
58
|
+
/**
|
|
59
|
+
* 创建会话
|
|
60
|
+
*/
|
|
61
|
+
createSession(id) {
|
|
62
|
+
const handle = {
|
|
63
|
+
id,
|
|
64
|
+
events: [],
|
|
65
|
+
createdAt: new Date().toISOString(),
|
|
66
|
+
lastActiveAt: new Date().toISOString(),
|
|
67
|
+
};
|
|
68
|
+
this.sessions.set(id, handle);
|
|
69
|
+
// 创建会话目录
|
|
70
|
+
const sessionDir = this.getSessionDir(id);
|
|
71
|
+
try {
|
|
72
|
+
fs.mkdirSync(sessionDir, { recursive: true });
|
|
73
|
+
}
|
|
74
|
+
catch {
|
|
75
|
+
// 目录创建失败,静默处理
|
|
76
|
+
}
|
|
77
|
+
return handle;
|
|
78
|
+
}
|
|
79
|
+
/**
|
|
80
|
+
* 追加事件到会话
|
|
81
|
+
*/
|
|
82
|
+
appendToSession(id, event) {
|
|
83
|
+
const handle = this.sessions.get(id);
|
|
84
|
+
if (!handle) {
|
|
85
|
+
throw new Error(`会话 ${id} 不存在`);
|
|
86
|
+
}
|
|
87
|
+
handle.events.push(event);
|
|
88
|
+
handle.lastActiveAt = new Date().toISOString();
|
|
89
|
+
// 持久化到 JSONL
|
|
90
|
+
this.appendEvent(id, event);
|
|
91
|
+
}
|
|
92
|
+
/**
|
|
93
|
+
* 获取当前轮窗口视图
|
|
94
|
+
*
|
|
95
|
+
* 将 session 事件转换为 ContextSource,通过 TokenPipeline 生成 prompt
|
|
96
|
+
*/
|
|
97
|
+
getWindowView(id, budget) {
|
|
98
|
+
const handle = this.sessions.get(id);
|
|
99
|
+
if (!handle) {
|
|
100
|
+
throw new Error(`会话 ${id} 不存在`);
|
|
101
|
+
}
|
|
102
|
+
// 将事件转换为 ContextSource
|
|
103
|
+
const sources = this.eventsToSources(handle.events);
|
|
104
|
+
// 通过 pipeline 生成窗口视图
|
|
105
|
+
const output = this.pipeline.run({ sources, budget });
|
|
106
|
+
// 记录快照
|
|
107
|
+
this.tracker.record(output.snapshot);
|
|
108
|
+
return output;
|
|
109
|
+
}
|
|
110
|
+
/**
|
|
111
|
+
* 生成 checkpoint
|
|
112
|
+
*/
|
|
113
|
+
checkpointSession(id) {
|
|
114
|
+
const handle = this.sessions.get(id);
|
|
115
|
+
if (!handle) {
|
|
116
|
+
throw new Error(`会话 ${id} 不存在`);
|
|
117
|
+
}
|
|
118
|
+
const checkpoint = {
|
|
119
|
+
id: `cp-${Date.now()}`,
|
|
120
|
+
sessionId: id,
|
|
121
|
+
timestamp: new Date().toISOString(),
|
|
122
|
+
eventCount: handle.events.length,
|
|
123
|
+
summary: this.generateCheckpointSummary(handle.events),
|
|
124
|
+
};
|
|
125
|
+
// 持久化 checkpoint
|
|
126
|
+
this.saveCheckpoint(id, checkpoint);
|
|
127
|
+
return checkpoint;
|
|
128
|
+
}
|
|
129
|
+
/**
|
|
130
|
+
* 从 checkpoint 恢复会话
|
|
131
|
+
*/
|
|
132
|
+
restoreSession(checkpointId) {
|
|
133
|
+
// 搜索所有会话的 checkpoints
|
|
134
|
+
const sessionsDir = path.join(this.basePath, '.harness', 'sessions');
|
|
135
|
+
try {
|
|
136
|
+
const sessionIds = fs.readdirSync(sessionsDir);
|
|
137
|
+
for (const sessionId of sessionIds) {
|
|
138
|
+
const checkpointPath = path.join(sessionsDir, sessionId, 'checkpoints', `${checkpointId}.json`);
|
|
139
|
+
if (fs.existsSync(checkpointPath)) {
|
|
140
|
+
const checkpointData = JSON.parse(fs.readFileSync(checkpointPath, 'utf-8'));
|
|
141
|
+
// 恢复会话
|
|
142
|
+
const handle = this.createSession(sessionId);
|
|
143
|
+
// 从 events.jsonl 恢复事件
|
|
144
|
+
const eventsPath = path.join(sessionsDir, sessionId, 'events.jsonl');
|
|
145
|
+
if (fs.existsSync(eventsPath)) {
|
|
146
|
+
const content = fs.readFileSync(eventsPath, 'utf-8');
|
|
147
|
+
const lines = content.trim().split('\n').filter(Boolean);
|
|
148
|
+
// 只恢复 checkpoint 之前的事件
|
|
149
|
+
const events = lines
|
|
150
|
+
.slice(0, checkpointData.eventCount)
|
|
151
|
+
.map(line => {
|
|
152
|
+
try {
|
|
153
|
+
return JSON.parse(line);
|
|
154
|
+
}
|
|
155
|
+
catch {
|
|
156
|
+
return null;
|
|
157
|
+
}
|
|
158
|
+
})
|
|
159
|
+
.filter((e) => e !== null);
|
|
160
|
+
handle.events = events;
|
|
161
|
+
}
|
|
162
|
+
return handle;
|
|
163
|
+
}
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
catch {
|
|
167
|
+
// 恢复失败
|
|
168
|
+
}
|
|
169
|
+
throw new Error(`Checkpoint ${checkpointId} 不存在`);
|
|
170
|
+
}
|
|
171
|
+
/**
|
|
172
|
+
* 获取会话信息
|
|
173
|
+
*/
|
|
174
|
+
getSession(id) {
|
|
175
|
+
return this.sessions.get(id);
|
|
176
|
+
}
|
|
177
|
+
/**
|
|
178
|
+
* 获取 tracker
|
|
179
|
+
*/
|
|
180
|
+
getTracker() {
|
|
181
|
+
return this.tracker;
|
|
182
|
+
}
|
|
183
|
+
/**
|
|
184
|
+
* 将事件转换为 ContextSource
|
|
185
|
+
*/
|
|
186
|
+
eventsToSources(events) {
|
|
187
|
+
return events.map(event => {
|
|
188
|
+
let priority;
|
|
189
|
+
let type;
|
|
190
|
+
switch (event.type) {
|
|
191
|
+
case 'system':
|
|
192
|
+
priority = 1;
|
|
193
|
+
type = 'system_prompt';
|
|
194
|
+
break;
|
|
195
|
+
case 'user_message':
|
|
196
|
+
priority = 6;
|
|
197
|
+
type = 'user_message';
|
|
198
|
+
break;
|
|
199
|
+
case 'assistant_message':
|
|
200
|
+
priority = 5;
|
|
201
|
+
type = 'session_event';
|
|
202
|
+
break;
|
|
203
|
+
case 'tool_call':
|
|
204
|
+
case 'tool_result':
|
|
205
|
+
priority = 5;
|
|
206
|
+
type = 'tool_output';
|
|
207
|
+
break;
|
|
208
|
+
case 'checkpoint':
|
|
209
|
+
priority = 4;
|
|
210
|
+
type = 'session_event';
|
|
211
|
+
break;
|
|
212
|
+
default:
|
|
213
|
+
priority = 5;
|
|
214
|
+
type = 'session_event';
|
|
215
|
+
}
|
|
216
|
+
return {
|
|
217
|
+
type,
|
|
218
|
+
id: event.id,
|
|
219
|
+
content: event.content,
|
|
220
|
+
priority,
|
|
221
|
+
metadata: event.metadata,
|
|
222
|
+
};
|
|
223
|
+
});
|
|
224
|
+
}
|
|
225
|
+
/**
|
|
226
|
+
* 生成 checkpoint 摘要
|
|
227
|
+
*/
|
|
228
|
+
generateCheckpointSummary(events) {
|
|
229
|
+
const userMessages = events.filter(e => e.type === 'user_message');
|
|
230
|
+
const toolCalls = events.filter(e => e.type === 'tool_call');
|
|
231
|
+
const parts = [];
|
|
232
|
+
parts.push(`事件总数: ${events.length}`);
|
|
233
|
+
parts.push(`用户消息: ${userMessages.length}`);
|
|
234
|
+
parts.push(`工具调用: ${toolCalls.length}`);
|
|
235
|
+
if (userMessages.length > 0) {
|
|
236
|
+
parts.push(`最近目标: ${userMessages[userMessages.length - 1].content.slice(0, 200)}`);
|
|
237
|
+
}
|
|
238
|
+
return parts.join('\n');
|
|
239
|
+
}
|
|
240
|
+
/**
|
|
241
|
+
* 持久化事件到 JSONL
|
|
242
|
+
*/
|
|
243
|
+
appendEvent(sessionId, event) {
|
|
244
|
+
try {
|
|
245
|
+
const eventsPath = path.join(this.getSessionDir(sessionId), 'events.jsonl');
|
|
246
|
+
const line = JSON.stringify(event) + '\n';
|
|
247
|
+
fs.appendFileSync(eventsPath, line, 'utf-8');
|
|
248
|
+
}
|
|
249
|
+
catch {
|
|
250
|
+
// 持久化失败,静默处理
|
|
251
|
+
}
|
|
252
|
+
}
|
|
253
|
+
/**
|
|
254
|
+
* 保存 checkpoint
|
|
255
|
+
*/
|
|
256
|
+
saveCheckpoint(sessionId, checkpoint) {
|
|
257
|
+
try {
|
|
258
|
+
const checkpointDir = path.join(this.getSessionDir(sessionId), 'checkpoints');
|
|
259
|
+
fs.mkdirSync(checkpointDir, { recursive: true });
|
|
260
|
+
const checkpointPath = path.join(checkpointDir, `${checkpoint.id}.json`);
|
|
261
|
+
fs.writeFileSync(checkpointPath, JSON.stringify(checkpoint, null, 2), 'utf-8');
|
|
262
|
+
}
|
|
263
|
+
catch {
|
|
264
|
+
// 保存失败,静默处理
|
|
265
|
+
}
|
|
266
|
+
}
|
|
267
|
+
/**
|
|
268
|
+
* 获取会话目录
|
|
269
|
+
*/
|
|
270
|
+
getSessionDir(sessionId) {
|
|
271
|
+
return path.join(this.basePath, '.harness', 'sessions', sessionId);
|
|
272
|
+
}
|
|
273
|
+
}
|
|
274
|
+
exports.SessionManager = SessionManager;
|
|
275
|
+
//# sourceMappingURL=session-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session-manager.js","sourceRoot":"","sources":["../../src/context/session-manager.ts"],"names":[],"mappings":";AAAA;;;;;;GAMG;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,uCAAyB;AACzB,2CAA6B;AAC7B,qDAAiD;AACjD,mEAA+D;AAU/D,MAAa,cAAc;IACjB,QAAQ,CAAS;IACjB,QAAQ,GAA+B,IAAI,GAAG,EAAE,CAAC;IACjD,QAAQ,CAAgB;IACxB,OAAO,CAAiB;IAEhC,YAAY,QAAiB;QAC3B,IAAI,CAAC,QAAQ,GAAG,QAAQ,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;QAC1C,IAAI,CAAC,QAAQ,GAAG,IAAI,8BAAa,EAAE,CAAC;QACpC,IAAI,CAAC,OAAO,GAAG,IAAI,gCAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,EAAU;QACtB,MAAM,MAAM,GAAkB;YAC5B,EAAE;YACF,MAAM,EAAE,EAAE;YACV,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,YAAY,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;SACvC,CAAC;QAEF,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,EAAE,MAAM,CAAC,CAAC;QAE9B,SAAS;QACT,MAAM,UAAU,GAAG,IAAI,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;QAC1C,IAAI,CAAC;YACH,EAAE,CAAC,SAAS,CAAC,UAAU,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAChD,CAAC;QAAC,MAAM,CAAC;YACP,cAAc;QAChB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,eAAe,CAAC,EAAU,EAAE,KAAmB;QAC7C,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,MAAM,CAAC,YAAY,GAAG,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAE/C,aAAa;QACb,IAAI,CAAC,WAAW,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;IAC9B,CAAC;IAED;;;;OAIG;IACH,aAAa,CAAC,EAAU,EAAE,MAA6B;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,uBAAuB;QACvB,MAAM,OAAO,GAAG,IAAI,CAAC,eAAe,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEpD,qBAAqB;QACrB,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,CAAC;QAEtD,OAAO;QACP,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAErC,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,EAAU;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACrC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAClC,CAAC;QAED,MAAM,UAAU,GAAsB;YACpC,EAAE,EAAE,MAAM,IAAI,CAAC,GAAG,EAAE,EAAE;YACtB,SAAS,EAAE,EAAE;YACb,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,MAAM;YAChC,OAAO,EAAE,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,MAAM,CAAC;SACvD,CAAC;QAEF,iBAAiB;QACjB,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QAEpC,OAAO,UAAU,CAAC;IACpB,CAAC;IAED;;OAEG;IACH,cAAc,CAAC,YAAoB;QACjC,sBAAsB;QACtB,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;QAErE,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,EAAE,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;YAE/C,KAAK,MAAM,SAAS,IAAI,UAAU,EAAE,CAAC;gBACnC,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,aAAa,EAAE,GAAG,YAAY,OAAO,CAAC,CAAC;gBAEhG,IAAI,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;oBAClC,MAAM,cAAc,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAsB,CAAC;oBAEjG,OAAO;oBACP,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;oBAE7C,sBAAsB;oBACtB,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;oBACrE,IAAI,EAAE,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;wBAC9B,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;wBACrD,MAAM,KAAK,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;wBAEzD,uBAAuB;wBACvB,MAAM,MAAM,GAAG,KAAK;6BACjB,KAAK,CAAC,CAAC,EAAE,cAAc,CAAC,UAAU,CAAC;6BACnC,GAAG,CAAC,IAAI,CAAC,EAAE;4BACV,IAAI,CAAC;gCACH,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAiB,CAAC;4BAC1C,CAAC;4BAAC,MAAM,CAAC;gCACP,OAAO,IAAI,CAAC;4BACd,CAAC;wBACH,CAAC,CAAC;6BACD,MAAM,CAAC,CAAC,CAAC,EAAqB,EAAE,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC;wBAEhD,MAAM,CAAC,MAAM,GAAG,MAAM,CAAC;oBACzB,CAAC;oBAED,OAAO,MAAM,CAAC;gBAChB,CAAC;YACH,CAAC;QACH,CAAC;QAAC,MAAM,CAAC;YACP,OAAO;QACT,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,cAAc,YAAY,MAAM,CAAC,CAAC;IACpD,CAAC;IAED;;OAEG;IACH,UAAU,CAAC,EAAU;QACnB,OAAO,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,UAAU;QACR,OAAO,IAAI,CAAC,OAAO,CAAC;IACtB,CAAC;IAED;;OAEG;IACK,eAAe,CAAC,MAAsB;QAC5C,OAAO,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;YACxB,IAAI,QAAgB,CAAC;YACrB,IAAI,IAA2B,CAAC;YAEhC,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;gBACnB,KAAK,QAAQ;oBACX,QAAQ,GAAG,CAAC,CAAC;oBACb,IAAI,GAAG,eAAe,CAAC;oBACvB,MAAM;gBACR,KAAK,cAAc;oBACjB,QAAQ,GAAG,CAAC,CAAC;oBACb,IAAI,GAAG,cAAc,CAAC;oBACtB,MAAM;gBACR,KAAK,mBAAmB;oBACtB,QAAQ,GAAG,CAAC,CAAC;oBACb,IAAI,GAAG,eAAe,CAAC;oBACvB,MAAM;gBACR,KAAK,WAAW,CAAC;gBACjB,KAAK,aAAa;oBAChB,QAAQ,GAAG,CAAC,CAAC;oBACb,IAAI,GAAG,aAAa,CAAC;oBACrB,MAAM;gBACR,KAAK,YAAY;oBACf,QAAQ,GAAG,CAAC,CAAC;oBACb,IAAI,GAAG,eAAe,CAAC;oBACvB,MAAM;gBACR;oBACE,QAAQ,GAAG,CAAC,CAAC;oBACb,IAAI,GAAG,eAAe,CAAC;YAC3B,CAAC;YAED,OAAO;gBACL,IAAI;gBACJ,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,QAAQ;gBACR,QAAQ,EAAE,KAAK,CAAC,QAAQ;aACzB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,yBAAyB,CAAC,MAAsB;QACtD,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,cAAc,CAAC,CAAC;QACnE,MAAM,SAAS,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,WAAW,CAAC,CAAC;QAE7D,MAAM,KAAK,GAAa,EAAE,CAAC;QAC3B,KAAK,CAAC,IAAI,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;QACrC,KAAK,CAAC,IAAI,CAAC,SAAS,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,SAAS,SAAS,CAAC,MAAM,EAAE,CAAC,CAAC;QAExC,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,KAAK,CAAC,IAAI,CAAC,SAAS,YAAY,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;QACrF,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;IAED;;OAEG;IACK,WAAW,CAAC,SAAiB,EAAE,KAAmB;QACxD,IAAI,CAAC;YACH,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,cAAc,CAAC,CAAC;YAC5E,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,GAAG,IAAI,CAAC;YAC1C,EAAE,CAAC,cAAc,CAAC,UAAU,EAAE,IAAI,EAAE,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,MAAM,CAAC;YACP,aAAa;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACK,cAAc,CAAC,SAAiB,EAAE,UAA6B;QACrE,IAAI,CAAC;YACH,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,EAAE,aAAa,CAAC,CAAC;YAC9E,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YAEjD,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,GAAG,UAAU,CAAC,EAAE,OAAO,CAAC,CAAC;YACzE,EAAE,CAAC,aAAa,CAAC,cAAc,EAAE,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QACjF,CAAC;QAAC,MAAM,CAAC;YACP,YAAY;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,SAAiB;QACrC,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,UAAU,EAAE,SAAS,CAAC,CAAC;IACrE,CAAC;CACF;AApQD,wCAoQC"}
|
|
@@ -0,0 +1,55 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Token 流水线
|
|
3
|
+
*
|
|
4
|
+
* 五步:收集 → 排序 → 压缩 → 预算 → 组装
|
|
5
|
+
* 每次 LLM 调用前运行
|
|
6
|
+
*/
|
|
7
|
+
import type { ContextSource, TokenBudgetAllocation, PipelineInput, PipelineOutput } from './types';
|
|
8
|
+
interface CollectedGroup {
|
|
9
|
+
priority: number;
|
|
10
|
+
sources: ContextSource[];
|
|
11
|
+
}
|
|
12
|
+
export declare class TokenPipeline {
|
|
13
|
+
/**
|
|
14
|
+
* 完整流水线:收集 → 排序 → 压缩 → 预算 → 组装
|
|
15
|
+
*/
|
|
16
|
+
run(input: PipelineInput): PipelineOutput;
|
|
17
|
+
/**
|
|
18
|
+
* Step 1: 收集 — 从 sources 按 priority 分组
|
|
19
|
+
*/
|
|
20
|
+
collect(sources: ContextSource[]): CollectedGroup[];
|
|
21
|
+
/**
|
|
22
|
+
* Step 2: 排序 — 按 P1→P6,组内保持原序
|
|
23
|
+
*/
|
|
24
|
+
sort(groups: CollectedGroup[]): ContextSource[];
|
|
25
|
+
/**
|
|
26
|
+
* Step 3: 压缩 — 对 P3-P5 应用 token 预算裁剪
|
|
27
|
+
*
|
|
28
|
+
* P1(系统提示词)和 P6(用户消息)不压缩
|
|
29
|
+
* P2(工具定义)按需懒加载,这里不处理
|
|
30
|
+
*/
|
|
31
|
+
compress(sorted: ContextSource[], budget: TokenBudgetAllocation): {
|
|
32
|
+
compressed: ContextSource[];
|
|
33
|
+
dropped: Array<{
|
|
34
|
+
type: string;
|
|
35
|
+
id: string;
|
|
36
|
+
reason: string;
|
|
37
|
+
}>;
|
|
38
|
+
};
|
|
39
|
+
/**
|
|
40
|
+
* Step 4: 预算 — 按 TokenBudgetAllocation 分配各段
|
|
41
|
+
*/
|
|
42
|
+
applyBudget(compressed: ContextSource[], budget: TokenBudgetAllocation): ContextSource[];
|
|
43
|
+
/**
|
|
44
|
+
* Step 5: 组装 — 拼装最终 prompt
|
|
45
|
+
*
|
|
46
|
+
* P1 开头 + P6 结尾,中间 P2-P5
|
|
47
|
+
*/
|
|
48
|
+
assemble(budgeted: ContextSource[]): string;
|
|
49
|
+
/**
|
|
50
|
+
* 构建上下文使用快照
|
|
51
|
+
*/
|
|
52
|
+
private buildSnapshot;
|
|
53
|
+
}
|
|
54
|
+
export {};
|
|
55
|
+
//# sourceMappingURL=token-pipeline.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"token-pipeline.d.ts","sourceRoot":"","sources":["../../src/context/token-pipeline.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,KAAK,EACV,aAAa,EACb,qBAAqB,EACrB,aAAa,EACb,cAAc,EAEf,MAAM,SAAS,CAAC;AAEjB,UAAU,cAAc;IACtB,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,aAAa,EAAE,CAAC;CAC1B;AAED,qBAAa,aAAa;IACxB;;OAEG;IACH,GAAG,CAAC,KAAK,EAAE,aAAa,GAAG,cAAc;IAwBzC;;OAEG;IACH,OAAO,CAAC,OAAO,EAAE,aAAa,EAAE,GAAG,cAAc,EAAE;IAcnD;;OAEG;IACH,IAAI,CAAC,MAAM,EAAE,cAAc,EAAE,GAAG,aAAa,EAAE;IAU/C;;;;;OAKG;IACH,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,EAAE,MAAM,EAAE,qBAAqB,GAAG;QAChE,UAAU,EAAE,aAAa,EAAE,CAAC;QAC5B,OAAO,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAC;YAAC,EAAE,EAAE,MAAM,CAAC;YAAC,MAAM,EAAE,MAAM,CAAA;SAAE,CAAC,CAAC;KAC9D;IAkED;;OAEG;IACH,WAAW,CAAC,UAAU,EAAE,aAAa,EAAE,EAAE,MAAM,EAAE,qBAAqB,GAAG,aAAa,EAAE;IAqBxF;;;;OAIG;IACH,QAAQ,CAAC,QAAQ,EAAE,aAAa,EAAE,GAAG,MAAM;IAgC3C;;OAEG;IACH,OAAO,CAAC,aAAa;CA8CtB"}
|
|
@@ -0,0 +1,220 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
/**
|
|
3
|
+
* Token 流水线
|
|
4
|
+
*
|
|
5
|
+
* 五步:收集 → 排序 → 压缩 → 预算 → 组装
|
|
6
|
+
* 每次 LLM 调用前运行
|
|
7
|
+
*/
|
|
8
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
9
|
+
exports.TokenPipeline = void 0;
|
|
10
|
+
const token_budget_1 = require("./token-budget");
|
|
11
|
+
class TokenPipeline {
|
|
12
|
+
/**
|
|
13
|
+
* 完整流水线:收集 → 排序 → 压缩 → 预算 → 组装
|
|
14
|
+
*/
|
|
15
|
+
run(input) {
|
|
16
|
+
const { sources, budget } = input;
|
|
17
|
+
// Step 1: 收集 — 按 priority 分组
|
|
18
|
+
const collected = this.collect(sources);
|
|
19
|
+
// Step 2: 排序 — 按 P1→P6 排序
|
|
20
|
+
const sorted = this.sort(collected);
|
|
21
|
+
// Step 3: 压缩 — 对 P3-P5 应用预算裁剪
|
|
22
|
+
const { compressed, dropped } = this.compress(sorted, budget);
|
|
23
|
+
// Step 4: 预算 — 按分配裁剪
|
|
24
|
+
const budgeted = this.applyBudget(compressed, budget);
|
|
25
|
+
// Step 5: 组装 — 拼装最终 prompt
|
|
26
|
+
const prompt = this.assemble(budgeted);
|
|
27
|
+
// 生成快照
|
|
28
|
+
const snapshot = this.buildSnapshot(budgeted, dropped);
|
|
29
|
+
return { prompt, snapshot, dropped };
|
|
30
|
+
}
|
|
31
|
+
/**
|
|
32
|
+
* Step 1: 收集 — 从 sources 按 priority 分组
|
|
33
|
+
*/
|
|
34
|
+
collect(sources) {
|
|
35
|
+
const groups = new Map();
|
|
36
|
+
for (const source of sources) {
|
|
37
|
+
const existing = groups.get(source.priority) || [];
|
|
38
|
+
existing.push(source);
|
|
39
|
+
groups.set(source.priority, existing);
|
|
40
|
+
}
|
|
41
|
+
return Array.from(groups.entries())
|
|
42
|
+
.map(([priority, sources]) => ({ priority, sources }))
|
|
43
|
+
.sort((a, b) => a.priority - b.priority);
|
|
44
|
+
}
|
|
45
|
+
/**
|
|
46
|
+
* Step 2: 排序 — 按 P1→P6,组内保持原序
|
|
47
|
+
*/
|
|
48
|
+
sort(groups) {
|
|
49
|
+
const sorted = [];
|
|
50
|
+
for (const group of groups) {
|
|
51
|
+
sorted.push(...group.sources);
|
|
52
|
+
}
|
|
53
|
+
return sorted;
|
|
54
|
+
}
|
|
55
|
+
/**
|
|
56
|
+
* Step 3: 压缩 — 对 P3-P5 应用 token 预算裁剪
|
|
57
|
+
*
|
|
58
|
+
* P1(系统提示词)和 P6(用户消息)不压缩
|
|
59
|
+
* P2(工具定义)按需懒加载,这里不处理
|
|
60
|
+
*/
|
|
61
|
+
compress(sorted, budget) {
|
|
62
|
+
const compressed = [];
|
|
63
|
+
const dropped = [];
|
|
64
|
+
// 各段预算
|
|
65
|
+
const budgetMap = {
|
|
66
|
+
1: budget.systemPrompt,
|
|
67
|
+
2: budget.toolDefinitions,
|
|
68
|
+
3: budget.knowledge,
|
|
69
|
+
4: budget.notes,
|
|
70
|
+
5: budget.history,
|
|
71
|
+
6: Infinity, // 用户消息不限
|
|
72
|
+
};
|
|
73
|
+
// 按优先级分组处理
|
|
74
|
+
const byPriority = new Map();
|
|
75
|
+
for (const item of sorted) {
|
|
76
|
+
const existing = byPriority.get(item.priority) || [];
|
|
77
|
+
existing.push(item);
|
|
78
|
+
byPriority.set(item.priority, existing);
|
|
79
|
+
}
|
|
80
|
+
for (const [priority, items] of byPriority) {
|
|
81
|
+
const sectionBudget = budgetMap[priority] || budget.history;
|
|
82
|
+
if (priority === 1 || priority === 6) {
|
|
83
|
+
// P1/P6 不压缩
|
|
84
|
+
compressed.push(...items);
|
|
85
|
+
continue;
|
|
86
|
+
}
|
|
87
|
+
// 按 token 预算裁剪
|
|
88
|
+
let usedTokens = 0;
|
|
89
|
+
for (const item of items) {
|
|
90
|
+
const tokens = token_budget_1.TokenEstimator.estimateText(item.content);
|
|
91
|
+
if (usedTokens + tokens <= sectionBudget) {
|
|
92
|
+
compressed.push(item);
|
|
93
|
+
usedTokens += tokens;
|
|
94
|
+
}
|
|
95
|
+
else {
|
|
96
|
+
// 尝试截断
|
|
97
|
+
const remaining = sectionBudget - usedTokens;
|
|
98
|
+
if (remaining > 50) {
|
|
99
|
+
const ratio = remaining / tokens;
|
|
100
|
+
const truncatedContent = item.content.slice(0, Math.floor(item.content.length * ratio));
|
|
101
|
+
compressed.push({ ...item, content: truncatedContent });
|
|
102
|
+
dropped.push({
|
|
103
|
+
type: item.type,
|
|
104
|
+
id: item.id,
|
|
105
|
+
reason: `截断: ${tokens} → ${Math.floor(tokens * ratio)} tokens`,
|
|
106
|
+
});
|
|
107
|
+
}
|
|
108
|
+
else {
|
|
109
|
+
dropped.push({
|
|
110
|
+
type: item.type,
|
|
111
|
+
id: item.id,
|
|
112
|
+
reason: `超出预算: 需要 ${tokens} tokens,剩余 ${remaining}`,
|
|
113
|
+
});
|
|
114
|
+
}
|
|
115
|
+
usedTokens += tokens; // 仍然计入已用
|
|
116
|
+
}
|
|
117
|
+
}
|
|
118
|
+
}
|
|
119
|
+
return { compressed, dropped };
|
|
120
|
+
}
|
|
121
|
+
/**
|
|
122
|
+
* Step 4: 预算 — 按 TokenBudgetAllocation 分配各段
|
|
123
|
+
*/
|
|
124
|
+
applyBudget(compressed, budget) {
|
|
125
|
+
const tokenBudget = new token_budget_1.TokenBudget(budget.total);
|
|
126
|
+
const result = [];
|
|
127
|
+
for (const item of compressed) {
|
|
128
|
+
const tokens = token_budget_1.TokenEstimator.estimateText(item.content);
|
|
129
|
+
if (tokenBudget.canAfford(tokens)) {
|
|
130
|
+
tokenBudget.consume(tokens);
|
|
131
|
+
result.push(item);
|
|
132
|
+
}
|
|
133
|
+
else if (item.priority === 1 || item.priority === 6) {
|
|
134
|
+
// P1/P6 强制保留
|
|
135
|
+
tokenBudget.forceConsume(tokens);
|
|
136
|
+
result.push(item);
|
|
137
|
+
}
|
|
138
|
+
// 其他优先级的超预算项被丢弃
|
|
139
|
+
}
|
|
140
|
+
return result;
|
|
141
|
+
}
|
|
142
|
+
/**
|
|
143
|
+
* Step 5: 组装 — 拼装最终 prompt
|
|
144
|
+
*
|
|
145
|
+
* P1 开头 + P6 结尾,中间 P2-P5
|
|
146
|
+
*/
|
|
147
|
+
assemble(budgeted) {
|
|
148
|
+
const sections = {
|
|
149
|
+
1: [], 2: [], 3: [], 4: [], 5: [], 6: [],
|
|
150
|
+
};
|
|
151
|
+
for (const item of budgeted) {
|
|
152
|
+
sections[item.priority] = sections[item.priority] || [];
|
|
153
|
+
sections[item.priority].push(item.content);
|
|
154
|
+
}
|
|
155
|
+
const parts = [];
|
|
156
|
+
// P1: 系统提示词(开头)
|
|
157
|
+
if (sections[1].length > 0) {
|
|
158
|
+
parts.push(sections[1].join('\n'));
|
|
159
|
+
}
|
|
160
|
+
// P2-P5: 中间部分
|
|
161
|
+
for (const priority of [2, 3, 4, 5]) {
|
|
162
|
+
if (sections[priority].length > 0) {
|
|
163
|
+
parts.push(sections[priority].join('\n'));
|
|
164
|
+
}
|
|
165
|
+
}
|
|
166
|
+
// P6: 用户消息(结尾)
|
|
167
|
+
if (sections[6].length > 0) {
|
|
168
|
+
parts.push(sections[6].join('\n'));
|
|
169
|
+
}
|
|
170
|
+
return parts.join('\n\n');
|
|
171
|
+
}
|
|
172
|
+
/**
|
|
173
|
+
* 构建上下文使用快照
|
|
174
|
+
*/
|
|
175
|
+
buildSnapshot(budgeted, dropped) {
|
|
176
|
+
const breakdown = {
|
|
177
|
+
systemPrompt: 0,
|
|
178
|
+
messages: 0,
|
|
179
|
+
toolOutputs: 0,
|
|
180
|
+
knowledge: 0,
|
|
181
|
+
other: 0,
|
|
182
|
+
};
|
|
183
|
+
for (const item of budgeted) {
|
|
184
|
+
const tokens = token_budget_1.TokenEstimator.estimateText(item.content);
|
|
185
|
+
switch (item.type) {
|
|
186
|
+
case 'system_prompt':
|
|
187
|
+
breakdown.systemPrompt += tokens;
|
|
188
|
+
break;
|
|
189
|
+
case 'user_message':
|
|
190
|
+
case 'session_event':
|
|
191
|
+
breakdown.messages += tokens;
|
|
192
|
+
break;
|
|
193
|
+
case 'tool_output':
|
|
194
|
+
case 'tool_definition':
|
|
195
|
+
breakdown.toolOutputs += tokens;
|
|
196
|
+
break;
|
|
197
|
+
case 'knowledge':
|
|
198
|
+
breakdown.knowledge += tokens;
|
|
199
|
+
break;
|
|
200
|
+
default:
|
|
201
|
+
breakdown.other += tokens;
|
|
202
|
+
}
|
|
203
|
+
}
|
|
204
|
+
return {
|
|
205
|
+
timestamp: new Date().toISOString(),
|
|
206
|
+
totalTokens: breakdown.systemPrompt + breakdown.messages + breakdown.toolOutputs + breakdown.knowledge + breakdown.other,
|
|
207
|
+
breakdown,
|
|
208
|
+
truncatedItems: dropped.filter(d => d.reason.startsWith('截断')).map(d => ({
|
|
209
|
+
type: d.type,
|
|
210
|
+
id: d.id,
|
|
211
|
+
originalTokens: 0,
|
|
212
|
+
keptTokens: 0,
|
|
213
|
+
})),
|
|
214
|
+
offloadedItems: [],
|
|
215
|
+
compactionTriggered: false,
|
|
216
|
+
};
|
|
217
|
+
}
|
|
218
|
+
}
|
|
219
|
+
exports.TokenPipeline = TokenPipeline;
|
|
220
|
+
//# sourceMappingURL=token-pipeline.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"token-pipeline.js","sourceRoot":"","sources":["../../src/context/token-pipeline.ts"],"names":[],"mappings":";AAAA;;;;;GAKG;;;AAEH,iDAA6D;AAc7D,MAAa,aAAa;IACxB;;OAEG;IACH,GAAG,CAAC,KAAoB;QACtB,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,GAAG,KAAK,CAAC;QAElC,6BAA6B;QAC7B,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAExC,0BAA0B;QAC1B,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAEpC,8BAA8B;QAC9B,MAAM,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAE9D,qBAAqB;QACrB,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;QAEtD,2BAA2B;QAC3B,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAEvC,OAAO;QACP,MAAM,QAAQ,GAAG,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAEvD,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACH,OAAO,CAAC,OAAwB;QAC9B,MAAM,MAAM,GAAG,IAAI,GAAG,EAA2B,CAAC;QAElD,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACnD,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACtB,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;aAChC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;aACrD,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;IAC7C,CAAC;IAED;;OAEG;IACH,IAAI,CAAC,MAAwB;QAC3B,MAAM,MAAM,GAAoB,EAAE,CAAC;QAEnC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC3B,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,OAAO,CAAC,CAAC;QAChC,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;;OAKG;IACH,QAAQ,CAAC,MAAuB,EAAE,MAA6B;QAI7D,MAAM,UAAU,GAAoB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAwD,EAAE,CAAC;QAExE,OAAO;QACP,MAAM,SAAS,GAA2B;YACxC,CAAC,EAAE,MAAM,CAAC,YAAY;YACtB,CAAC,EAAE,MAAM,CAAC,eAAe;YACzB,CAAC,EAAE,MAAM,CAAC,SAAS;YACnB,CAAC,EAAE,MAAM,CAAC,KAAK;YACf,CAAC,EAAE,MAAM,CAAC,OAAO;YACjB,CAAC,EAAE,QAAQ,EAAE,SAAS;SACvB,CAAC;QAEF,WAAW;QACX,MAAM,UAAU,GAAG,IAAI,GAAG,EAA2B,CAAC;QACtD,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;YAC1B,MAAM,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACrD,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpB,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAED,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,UAAU,EAAE,CAAC;YAC3C,MAAM,aAAa,GAAG,SAAS,CAAC,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC;YAE5D,IAAI,QAAQ,KAAK,CAAC,IAAI,QAAQ,KAAK,CAAC,EAAE,CAAC;gBACrC,YAAY;gBACZ,UAAU,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;gBAC1B,SAAS;YACX,CAAC;YAED,eAAe;YACf,IAAI,UAAU,GAAG,CAAC,CAAC;YACnB,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;gBACzB,MAAM,MAAM,GAAG,6BAAc,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAEzD,IAAI,UAAU,GAAG,MAAM,IAAI,aAAa,EAAE,CAAC;oBACzC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;oBACtB,UAAU,IAAI,MAAM,CAAC;gBACvB,CAAC;qBAAM,CAAC;oBACN,OAAO;oBACP,MAAM,SAAS,GAAG,aAAa,GAAG,UAAU,CAAC;oBAC7C,IAAI,SAAS,GAAG,EAAE,EAAE,CAAC;wBACnB,MAAM,KAAK,GAAG,SAAS,GAAG,MAAM,CAAC;wBACjC,MAAM,gBAAgB,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC;wBACxF,UAAU,CAAC,IAAI,CAAC,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;wBACxD,OAAO,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,EAAE,EAAE,IAAI,CAAC,EAAE;4BACX,MAAM,EAAE,OAAO,MAAM,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,GAAG,KAAK,CAAC,SAAS;yBAC/D,CAAC,CAAC;oBACL,CAAC;yBAAM,CAAC;wBACN,OAAO,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,IAAI,CAAC,IAAI;4BACf,EAAE,EAAE,IAAI,CAAC,EAAE;4BACX,MAAM,EAAE,YAAY,MAAM,cAAc,SAAS,EAAE;yBACpD,CAAC,CAAC;oBACL,CAAC;oBACD,UAAU,IAAI,MAAM,CAAC,CAAC,SAAS;gBACjC,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,UAA2B,EAAE,MAA6B;QACpE,MAAM,WAAW,GAAG,IAAI,0BAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAClD,MAAM,MAAM,GAAoB,EAAE,CAAC;QAEnC,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,6BAAc,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEzD,IAAI,WAAW,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC;gBAClC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBAC5B,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpB,CAAC;iBAAM,IAAI,IAAI,CAAC,QAAQ,KAAK,CAAC,IAAI,IAAI,CAAC,QAAQ,KAAK,CAAC,EAAE,CAAC;gBACtD,aAAa;gBACb,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;gBACjC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACpB,CAAC;YACD,gBAAgB;QAClB,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED;;;;OAIG;IACH,QAAQ,CAAC,QAAyB;QAChC,MAAM,QAAQ,GAA6B;YACzC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,EAAE;SACzC,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACxD,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC7C,CAAC;QAED,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,gBAAgB;QAChB,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,cAAc;QACd,KAAK,MAAM,QAAQ,IAAI,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;YACpC,IAAI,QAAQ,CAAC,QAAQ,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAClC,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YAC5C,CAAC;QACH,CAAC;QAED,eAAe;QACf,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC3B,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QACrC,CAAC;QAED,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,aAAa,CAAC,QAAyB,EAAE,OAA4D;QAC3G,MAAM,SAAS,GAAG;YAChB,YAAY,EAAE,CAAC;YACf,QAAQ,EAAE,CAAC;YACX,WAAW,EAAE,CAAC;YACd,SAAS,EAAE,CAAC;YACZ,KAAK,EAAE,CAAC;SACT,CAAC;QAEF,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;YAC5B,MAAM,MAAM,GAAG,6BAAc,CAAC,YAAY,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAEzD,QAAQ,IAAI,CAAC,IAAI,EAAE,CAAC;gBAClB,KAAK,eAAe;oBAClB,SAAS,CAAC,YAAY,IAAI,MAAM,CAAC;oBACjC,MAAM;gBACR,KAAK,cAAc,CAAC;gBACpB,KAAK,eAAe;oBAClB,SAAS,CAAC,QAAQ,IAAI,MAAM,CAAC;oBAC7B,MAAM;gBACR,KAAK,aAAa,CAAC;gBACnB,KAAK,iBAAiB;oBACpB,SAAS,CAAC,WAAW,IAAI,MAAM,CAAC;oBAChC,MAAM;gBACR,KAAK,WAAW;oBACd,SAAS,CAAC,SAAS,IAAI,MAAM,CAAC;oBAC9B,MAAM;gBACR;oBACE,SAAS,CAAC,KAAK,IAAI,MAAM,CAAC;YAC9B,CAAC;QACH,CAAC;QAED,OAAO;YACL,SAAS,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE;YACnC,WAAW,EAAE,SAAS,CAAC,YAAY,GAAG,SAAS,CAAC,QAAQ,GAAG,SAAS,CAAC,WAAW,GAAG,SAAS,CAAC,SAAS,GAAG,SAAS,CAAC,KAAK;YACxH,SAAS;YACT,cAAc,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;gBACvE,IAAI,EAAE,CAAC,CAAC,IAAI;gBACZ,EAAE,EAAE,CAAC,CAAC,EAAE;gBACR,cAAc,EAAE,CAAC;gBACjB,UAAU,EAAE,CAAC;aACd,CAAC,CAAC;YACH,cAAc,EAAE,EAAE;YAClB,mBAAmB,EAAE,KAAK;SAC3B,CAAC;IACJ,CAAC;CACF;AAnPD,sCAmPC"}
|
|
@@ -0,0 +1,59 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* 工具输出预算
|
|
3
|
+
*
|
|
4
|
+
* 控制工具输出对上下文的占用
|
|
5
|
+
* 策略:full / preview / overflow / dedup
|
|
6
|
+
*/
|
|
7
|
+
import type { ToolOutputBudgetConfig } from './types';
|
|
8
|
+
export type OutputStrategy = 'full' | 'preview' | 'overflow' | 'dedup';
|
|
9
|
+
export interface OutputCheckResult {
|
|
10
|
+
allowed: boolean;
|
|
11
|
+
strategy: OutputStrategy;
|
|
12
|
+
reason?: string;
|
|
13
|
+
}
|
|
14
|
+
export interface OutputApplyResult {
|
|
15
|
+
content: string;
|
|
16
|
+
offloaded?: {
|
|
17
|
+
path: string;
|
|
18
|
+
};
|
|
19
|
+
originalTokens: number;
|
|
20
|
+
keptTokens: number;
|
|
21
|
+
}
|
|
22
|
+
export declare class ToolOutputBudget {
|
|
23
|
+
private config;
|
|
24
|
+
private seenHashes;
|
|
25
|
+
constructor(config?: Partial<ToolOutputBudgetConfig>);
|
|
26
|
+
/**
|
|
27
|
+
* 检查工具输出应该使用什么策略
|
|
28
|
+
*/
|
|
29
|
+
checkOutput(output: string, totalBudget?: number): OutputCheckResult;
|
|
30
|
+
/**
|
|
31
|
+
* 应用预算策略
|
|
32
|
+
*/
|
|
33
|
+
applyBudget(output: string, strategy: OutputStrategy, totalBudget?: number): OutputApplyResult;
|
|
34
|
+
/**
|
|
35
|
+
* 生成预览内容
|
|
36
|
+
*/
|
|
37
|
+
private generatePreview;
|
|
38
|
+
/**
|
|
39
|
+
* 写入溢出文件
|
|
40
|
+
*/
|
|
41
|
+
private writeToOverflow;
|
|
42
|
+
/**
|
|
43
|
+
* 计算内容 hash
|
|
44
|
+
*/
|
|
45
|
+
private computeHash;
|
|
46
|
+
/**
|
|
47
|
+
* 记录已见 hash
|
|
48
|
+
*/
|
|
49
|
+
private recordHash;
|
|
50
|
+
/**
|
|
51
|
+
* 清除去重缓存
|
|
52
|
+
*/
|
|
53
|
+
clearDedupCache(): void;
|
|
54
|
+
/**
|
|
55
|
+
* 获取配置
|
|
56
|
+
*/
|
|
57
|
+
getConfig(): ToolOutputBudgetConfig;
|
|
58
|
+
}
|
|
59
|
+
//# sourceMappingURL=tool-output-budget.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tool-output-budget.d.ts","sourceRoot":"","sources":["../../src/context/tool-output-budget.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,SAAS,CAAC;AAGtD,MAAM,MAAM,cAAc,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU,GAAG,OAAO,CAAC;AAEvE,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,OAAO,CAAC;IACjB,QAAQ,EAAE,cAAc,CAAC;IACzB,MAAM,CAAC,EAAE,MAAM,CAAC;CACjB;AAED,MAAM,WAAW,iBAAiB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,SAAS,CAAC,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,CAAC;IAC7B,cAAc,EAAE,MAAM,CAAC;IACvB,UAAU,EAAE,MAAM,CAAC;CACpB;AAED,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,MAAM,CAAyB;IACvC,OAAO,CAAC,UAAU,CAAmC;gBAEzC,MAAM,CAAC,EAAE,OAAO,CAAC,sBAAsB,CAAC;IAIpD;;OAEG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,iBAAiB;IAsCpE;;OAEG;IACH,WAAW,CAAC,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,EAAE,WAAW,CAAC,EAAE,MAAM,GAAG,iBAAiB;IAkC9F;;OAEG;IACH,OAAO,CAAC,eAAe;IAavB;;OAEG;IACH,OAAO,CAAC,eAAe;IAiBvB;;OAEG;IACH,OAAO,CAAC,WAAW;IAInB;;OAEG;IACH,OAAO,CAAC,UAAU;IAKlB;;OAEG;IACH,eAAe,IAAI,IAAI;IAIvB;;OAEG;IACH,SAAS,IAAI,sBAAsB;CAGpC"}
|