yuangs 2.29.0 → 2.31.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 +284 -337
- package/dist/agent/index.d.ts +2 -0
- package/dist/agent/index.js +2 -0
- package/dist/agent/index.js.map +1 -1
- package/dist/agent/llmAdapter.js +1 -1
- package/dist/agent/llmAdapter.js.map +1 -1
- package/dist/agent/policy/engine.d.ts +14 -0
- package/dist/agent/policy/engine.js +76 -0
- package/dist/agent/policy/engine.js.map +1 -0
- package/dist/agent/policy/index.d.ts +3 -0
- package/dist/agent/policy/index.js +20 -0
- package/dist/agent/policy/index.js.map +1 -0
- package/dist/agent/policy/policies/noDangerousShell.d.ts +7 -0
- package/dist/agent/policy/policies/noDangerousShell.js +45 -0
- package/dist/agent/policy/policies/noDangerousShell.js.map +1 -0
- package/dist/agent/policy/types.d.ts +23 -0
- package/dist/agent/policy/types.js +3 -0
- package/dist/agent/policy/types.js.map +1 -0
- package/dist/agent/replay/events.d.ts +21 -0
- package/dist/agent/replay/events.js +3 -0
- package/dist/agent/replay/events.js.map +1 -0
- package/dist/agent/replay/index.d.ts +3 -0
- package/dist/agent/replay/index.js +20 -0
- package/dist/agent/replay/index.js.map +1 -0
- package/dist/agent/replay/recorder.d.ts +11 -0
- package/dist/agent/replay/recorder.js +51 -0
- package/dist/agent/replay/recorder.js.map +1 -0
- package/dist/agent/replay/replayer.d.ts +21 -0
- package/dist/agent/replay/replayer.js +65 -0
- package/dist/agent/replay/replayer.js.map +1 -0
- package/dist/agent/selectModel.js +1 -1
- package/dist/agent/selectModel.js.map +1 -1
- package/dist/agent/skills.d.ts +5 -0
- package/dist/agent/skills.js +7 -3
- package/dist/agent/skills.js.map +1 -1
- package/dist/api/index.d.ts +1 -0
- package/dist/api/index.js +18 -0
- package/dist/api/index.js.map +1 -0
- package/dist/api/registryAPI.d.ts +22 -0
- package/dist/api/registryAPI.js +66 -0
- package/dist/api/registryAPI.js.map +1 -0
- package/dist/audit/index.d.ts +1 -0
- package/dist/audit/index.js +18 -0
- package/dist/audit/index.js.map +1 -0
- package/dist/audit/timeline.d.ts +75 -0
- package/dist/audit/timeline.js +254 -0
- package/dist/audit/timeline.js.map +1 -0
- package/dist/cli.js +19 -10
- package/dist/cli.js.map +1 -1
- package/dist/commands/explainCommands.d.ts +2 -0
- package/dist/commands/explainCommands.js +36 -0
- package/dist/commands/explainCommands.js.map +1 -0
- package/dist/commands/handleAICommand.js +1 -1
- package/dist/commands/handleAICommand.js.map +1 -1
- package/dist/commands/registryCommands.d.ts +2 -0
- package/dist/commands/registryCommands.js +243 -0
- package/dist/commands/registryCommands.js.map +1 -0
- package/dist/commands/replayCommands.d.ts +2 -0
- package/dist/commands/replayCommands.js +75 -0
- package/dist/commands/replayCommands.js.map +1 -0
- package/dist/commands/skillsCommands.d.ts +2 -0
- package/dist/commands/skillsCommands.js +114 -0
- package/dist/commands/skillsCommands.js.map +1 -0
- package/dist/core/capabilitySystem.js +2 -2
- package/dist/core/capabilitySystem.js.map +1 -1
- package/dist/core/executionRecord.d.ts +8 -0
- package/dist/core/executionRecord.js +2 -0
- package/dist/core/executionRecord.js.map +1 -1
- package/dist/core/explain.d.ts +8 -0
- package/dist/core/explain.js +84 -0
- package/dist/core/explain.js.map +1 -0
- package/dist/core/replayDiff.d.ts +55 -0
- package/dist/core/replayDiff.js +205 -0
- package/dist/core/replayDiff.js.map +1 -0
- package/dist/core/replayEngine.d.ts +3 -0
- package/dist/core/replayEngine.js +23 -1
- package/dist/core/replayEngine.js.map +1 -1
- package/dist/registry/errors.d.ts +21 -0
- package/dist/registry/errors.js +35 -0
- package/dist/registry/errors.js.map +1 -0
- package/dist/registry/index.d.ts +3 -0
- package/dist/registry/index.js +20 -0
- package/dist/registry/index.js.map +1 -0
- package/dist/registry/manifest.d.ts +43 -0
- package/dist/registry/manifest.js +32 -0
- package/dist/registry/manifest.js.map +1 -0
- package/dist/registry/registry.d.ts +20 -0
- package/dist/registry/registry.js +201 -0
- package/dist/registry/registry.js.map +1 -0
- package/dist/risk/explainer.d.ts +39 -0
- package/dist/risk/explainer.js +214 -0
- package/dist/risk/explainer.js.map +1 -0
- package/dist/risk/index.d.ts +1 -0
- package/dist/risk/index.js +18 -0
- package/dist/risk/index.js.map +1 -0
- package/package.json +2 -1
package/dist/agent/index.d.ts
CHANGED
package/dist/agent/index.js
CHANGED
|
@@ -31,4 +31,6 @@ var executor_1 = require("./executor");
|
|
|
31
31
|
Object.defineProperty(exports, "ToolExecutor", { enumerable: true, get: function () { return executor_1.ToolExecutor; } });
|
|
32
32
|
var contextManager_1 = require("./contextManager");
|
|
33
33
|
Object.defineProperty(exports, "ContextManager", { enumerable: true, get: function () { return contextManager_1.ContextManager; } });
|
|
34
|
+
__exportStar(require("./policy"), exports);
|
|
35
|
+
__exportStar(require("./replay"), exports);
|
|
34
36
|
//# sourceMappingURL=index.js.map
|
package/dist/agent/index.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,iDAAgD;AAAvC,8GAAA,aAAa,OAAA;AACtB,0CAAwB;AACxB,0CAAwB;AACxB,6BAAsC;AAA7B,oGAAA,aAAa,OAAA;AACtB,+BAA2C;AAAlC,yGAAA,iBAAiB,OAAA;AAC1B,2CAA0C;AAAjC,wGAAA,UAAU,OAAA;AACnB,2CAAiD;AAAxC,+GAAA,iBAAiB,OAAA;AAC1B,uCAA0C;AAAjC,wGAAA,YAAY,OAAA;AACrB,mDAAkD;AAAzC,gHAAA,cAAc,OAAA"}
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/agent/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,iDAAgD;AAAvC,8GAAA,aAAa,OAAA;AACtB,0CAAwB;AACxB,0CAAwB;AACxB,6BAAsC;AAA7B,oGAAA,aAAa,OAAA;AACtB,+BAA2C;AAAlC,yGAAA,iBAAiB,OAAA;AAC1B,2CAA0C;AAAjC,wGAAA,UAAU,OAAA;AACnB,2CAAiD;AAAxC,+GAAA,iBAAiB,OAAA;AAC1B,uCAA0C;AAAjC,wGAAA,YAAY,OAAA;AACrB,mDAAkD;AAAzC,gHAAA,cAAc,OAAA;AACvB,2CAAyB;AACzB,2CAAyB"}
|
package/dist/agent/llmAdapter.js
CHANGED
|
@@ -35,7 +35,7 @@ If the task is complete and no more actions are needed, output:
|
|
|
35
35
|
};
|
|
36
36
|
// Use configured model from user settings
|
|
37
37
|
const config = (0, client_1.getUserConfig)();
|
|
38
|
-
const model = config.defaultModel || '
|
|
38
|
+
const model = config.defaultModel || 'Assistant';
|
|
39
39
|
const result = await (0, llm_1.runLLM)({
|
|
40
40
|
prompt,
|
|
41
41
|
model,
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"llmAdapter.js","sourceRoot":"","sources":["../../src/agent/llmAdapter.ts"],"names":[],"mappings":";;;AACA,+BAA+B;AAG/B,yCAA6C;AAE7C,MAAa,UAAU;IACrB,MAAM,CAAC,KAAK,CAAC,KAAK,CAChB,QAA4B,EAC5B,IAAyC,EACzC,YAAkB;QAElB,MAAM,MAAM,GAAgB;YAC1B,MAAM,EAAE;;;;;;;;;;;;;;;;;;;;;;;EAuBZ;YACI,QAAQ;YACR,YAAY;SACb,CAAC;QAEF,0CAA0C;QAC1C,MAAM,MAAM,GAAG,IAAA,sBAAa,GAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,IAAI,
|
|
1
|
+
{"version":3,"file":"llmAdapter.js","sourceRoot":"","sources":["../../src/agent/llmAdapter.ts"],"names":[],"mappings":";;;AACA,+BAA+B;AAG/B,yCAA6C;AAE7C,MAAa,UAAU;IACrB,MAAM,CAAC,KAAK,CAAC,KAAK,CAChB,QAA4B,EAC5B,IAAyC,EACzC,YAAkB;QAElB,MAAM,MAAM,GAAgB;YAC1B,MAAM,EAAE;;;;;;;;;;;;;;;;;;;;;;;EAuBZ;YACI,QAAQ;YACR,YAAY;SACb,CAAC;QAEF,0CAA0C;QAC1C,MAAM,MAAM,GAAG,IAAA,sBAAa,GAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,MAAM,CAAC,YAAY,IAAI,WAAW,CAAC;QAEjD,MAAM,MAAM,GAAG,MAAM,IAAA,YAAM,EAAC;YAC1B,MAAM;YACN,KAAK;YACL,MAAM,EAAE,KAAK;SACd,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,GAAW;QACrC,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,GAAG,CAAC,KAAK,CAAC,0BAA0B,CAAC,IAAI,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACpF,IAAI,SAAS,EAAE,CAAC;gBACd,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC;gBAExD,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;oBACnB,OAAO;wBACL,GAAG;wBACH,UAAU,EAAE,MAAM;wBAClB,MAAM,EAAE,IAAI;wBACZ,IAAI,EAAE,QAAQ;qBACf,CAAC;gBACJ,CAAC;gBAED,OAAO;oBACL,GAAG;oBACH,UAAU,EAAE,MAAM;oBAClB,MAAM,EAAE,KAAK;oBACb,IAAI,EAAE,MAAM,CAAC,WAAW,IAAI,WAAW;oBACvC,OAAO,EAAE;wBACP,SAAS,EAAE,MAAM,CAAC,SAAS;wBAC3B,UAAU,EAAE,MAAM,CAAC,UAAU;wBAC7B,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,IAAI,EAAE,MAAM,CAAC,IAAI;wBACjB,OAAO,EAAE,MAAM,CAAC,OAAO;qBACxB;oBACD,SAAS,EAAE,MAAM,CAAC,SAAS,IAAI,EAAE;iBAClC,CAAC;YACJ,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,0DAA0D,CAAC,CAAC;QAC3E,CAAC;QAED,OAAO;YACL,GAAG;YACH,UAAU,EAAE,EAAE;YACd,MAAM,EAAE,KAAK;YACb,IAAI,EAAE,QAAQ;YACd,SAAS,EAAE,GAAG;SACf,CAAC;IACJ,CAAC;CACF;AA1FD,gCA0FC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import { Policy, PolicyContext, PolicyResult } from './types';
|
|
2
|
+
import { RiskLevel } from '../state';
|
|
3
|
+
export declare class PolicyEngine {
|
|
4
|
+
private policies;
|
|
5
|
+
registerPolicy(policy: Policy): void;
|
|
6
|
+
unregisterPolicy(name: string): void;
|
|
7
|
+
evaluate(context: PolicyContext): Promise<PolicyResult>;
|
|
8
|
+
evaluateRisk(action: {
|
|
9
|
+
type: string;
|
|
10
|
+
payload: any;
|
|
11
|
+
}): RiskLevel;
|
|
12
|
+
private containsDangerousCommand;
|
|
13
|
+
}
|
|
14
|
+
export declare const policyEngine: PolicyEngine;
|
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.policyEngine = exports.PolicyEngine = void 0;
|
|
4
|
+
class PolicyEngine {
|
|
5
|
+
policies = new Map();
|
|
6
|
+
registerPolicy(policy) {
|
|
7
|
+
this.policies.set(policy.name, policy);
|
|
8
|
+
}
|
|
9
|
+
unregisterPolicy(name) {
|
|
10
|
+
this.policies.delete(name);
|
|
11
|
+
}
|
|
12
|
+
async evaluate(context) {
|
|
13
|
+
let finalResult = {
|
|
14
|
+
allowed: true,
|
|
15
|
+
reason: 'All policies passed'
|
|
16
|
+
};
|
|
17
|
+
for (const [name, policy] of this.policies) {
|
|
18
|
+
const result = await policy.evaluate(context);
|
|
19
|
+
if (!result.allowed) {
|
|
20
|
+
return {
|
|
21
|
+
allowed: false,
|
|
22
|
+
reason: `Policy "${name}" blocked: ${result.reason}`,
|
|
23
|
+
requiresEscalation: result.requiresEscalation || false,
|
|
24
|
+
suggestedActions: result.suggestedActions
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
if (result.requiresEscalation) {
|
|
28
|
+
finalResult.requiresEscalation = true;
|
|
29
|
+
finalResult.suggestedActions = result.suggestedActions;
|
|
30
|
+
}
|
|
31
|
+
}
|
|
32
|
+
return finalResult;
|
|
33
|
+
}
|
|
34
|
+
evaluateRisk(action) {
|
|
35
|
+
const { type, payload } = action;
|
|
36
|
+
if (type === 'tool_call') {
|
|
37
|
+
const toolName = payload.tool_name;
|
|
38
|
+
const lowRiskTools = ['read_file', 'list_files', 'web_search'];
|
|
39
|
+
if (lowRiskTools.includes(toolName)) {
|
|
40
|
+
return 'low';
|
|
41
|
+
}
|
|
42
|
+
const mediumRiskTools = ['write_file', 'shell'];
|
|
43
|
+
if (mediumRiskTools.includes(toolName)) {
|
|
44
|
+
const cmd = payload.parameters?.command || payload.command || '';
|
|
45
|
+
if (this.containsDangerousCommand(cmd)) {
|
|
46
|
+
return 'high';
|
|
47
|
+
}
|
|
48
|
+
return 'medium';
|
|
49
|
+
}
|
|
50
|
+
return 'medium';
|
|
51
|
+
}
|
|
52
|
+
if (type === 'shell_cmd') {
|
|
53
|
+
const cmd = payload.command || '';
|
|
54
|
+
if (this.containsDangerousCommand(cmd)) {
|
|
55
|
+
return 'high';
|
|
56
|
+
}
|
|
57
|
+
return 'medium';
|
|
58
|
+
}
|
|
59
|
+
return 'low';
|
|
60
|
+
}
|
|
61
|
+
containsDangerousCommand(cmd) {
|
|
62
|
+
const dangerousPatterns = [
|
|
63
|
+
/rm\s+-rf\s+\//,
|
|
64
|
+
/rm\s+-rf\s+~/,
|
|
65
|
+
/>\s*\/dev\/null/,
|
|
66
|
+
/dd\s+if=/,
|
|
67
|
+
/mkfs/,
|
|
68
|
+
/format/,
|
|
69
|
+
/sudo\s+rm/
|
|
70
|
+
];
|
|
71
|
+
return dangerousPatterns.some(pattern => pattern.test(cmd));
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
exports.PolicyEngine = PolicyEngine;
|
|
75
|
+
exports.policyEngine = new PolicyEngine();
|
|
76
|
+
//# sourceMappingURL=engine.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"engine.js","sourceRoot":"","sources":["../../../src/agent/policy/engine.ts"],"names":[],"mappings":";;;AAGA,MAAa,YAAY;IACf,QAAQ,GAAwB,IAAI,GAAG,EAAE,CAAC;IAElD,cAAc,CAAC,MAAc;QAC3B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;IACzC,CAAC;IAED,gBAAgB,CAAC,IAAY;QAC3B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,OAAsB;QACnC,IAAI,WAAW,GAAiB;YAC9B,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,qBAAqB;SAC9B,CAAC;QAEF,KAAK,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC3C,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YAE9C,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;gBACpB,OAAO;oBACL,OAAO,EAAE,KAAK;oBACd,MAAM,EAAE,WAAW,IAAI,cAAc,MAAM,CAAC,MAAM,EAAE;oBACpD,kBAAkB,EAAE,MAAM,CAAC,kBAAkB,IAAI,KAAK;oBACtD,gBAAgB,EAAE,MAAM,CAAC,gBAAgB;iBAC1C,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC,kBAAkB,EAAE,CAAC;gBAC9B,WAAW,CAAC,kBAAkB,GAAG,IAAI,CAAC;gBACtC,WAAW,CAAC,gBAAgB,GAAG,MAAM,CAAC,gBAAgB,CAAC;YACzD,CAAC;QACH,CAAC;QAED,OAAO,WAAW,CAAC;IACrB,CAAC;IAED,YAAY,CAAC,MAAsC;QACjD,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,MAAM,CAAC;QAEjC,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACzB,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;YAEnC,MAAM,YAAY,GAAG,CAAC,WAAW,EAAE,YAAY,EAAE,YAAY,CAAC,CAAC;YAC/D,IAAI,YAAY,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACpC,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,eAAe,GAAG,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;YAChD,IAAI,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACvC,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,EAAE,OAAO,IAAI,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;gBACjE,IAAI,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,EAAE,CAAC;oBACvC,OAAO,MAAM,CAAC;gBAChB,CAAC;gBACD,OAAO,QAAQ,CAAC;YAClB,CAAC;YAED,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACzB,MAAM,GAAG,GAAG,OAAO,CAAC,OAAO,IAAI,EAAE,CAAC;YAClC,IAAI,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,EAAE,CAAC;gBACvC,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,wBAAwB,CAAC,GAAW;QAC1C,MAAM,iBAAiB,GAAG;YACxB,eAAe;YACf,cAAc;YACd,iBAAiB;YACjB,UAAU;YACV,MAAM;YACN,QAAQ;YACR,WAAW;SACZ,CAAC;QAEF,OAAO,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;IAC9D,CAAC;CACF;AArFD,oCAqFC;AAEY,QAAA,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
14
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
|
15
|
+
};
|
|
16
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
17
|
+
__exportStar(require("./types"), exports);
|
|
18
|
+
__exportStar(require("./engine"), exports);
|
|
19
|
+
__exportStar(require("./policies/noDangerousShell"), exports);
|
|
20
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agent/policy/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,0CAAwB;AACxB,2CAAyB;AACzB,8DAA4C"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import { Policy, PolicyContext, PolicyResult } from '../types';
|
|
2
|
+
export declare class NoDangerousShellPolicy implements Policy {
|
|
3
|
+
name: string;
|
|
4
|
+
description: string;
|
|
5
|
+
evaluate(context: PolicyContext): PolicyResult;
|
|
6
|
+
}
|
|
7
|
+
export declare const noDangerousShellPolicy: NoDangerousShellPolicy;
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.noDangerousShellPolicy = exports.NoDangerousShellPolicy = void 0;
|
|
4
|
+
class NoDangerousShellPolicy {
|
|
5
|
+
name = 'no-dangerous-shell';
|
|
6
|
+
description = 'Prevents execution of dangerous shell commands';
|
|
7
|
+
evaluate(context) {
|
|
8
|
+
const { action } = context;
|
|
9
|
+
if (action.type === 'shell_cmd') {
|
|
10
|
+
const command = action.payload?.command || '';
|
|
11
|
+
const dangerousPatterns = [
|
|
12
|
+
{ pattern: /rm\s+-rf\s+\//, name: 'rm -rf /', risk: 'high' },
|
|
13
|
+
{ pattern: /rm\s+-rf\s+~/, name: 'rm -rf ~', risk: 'high' },
|
|
14
|
+
{ pattern: />\s*\/dev\/null/, name: 'Redirect to /dev/null', risk: 'medium' },
|
|
15
|
+
{ pattern: /dd\s+if=/, name: 'dd command', risk: 'high' },
|
|
16
|
+
{ pattern: /mkfs/, name: 'mkfs (filesystem creation)', risk: 'high' },
|
|
17
|
+
{ pattern: /format/, name: 'format command', risk: 'high' },
|
|
18
|
+
{ pattern: /sudo\s+rm/, name: 'sudo rm', risk: 'high' },
|
|
19
|
+
{ pattern: /chmod\s+777\s+\/(?!dev)/, name: 'chmod 777 on system', risk: 'high' },
|
|
20
|
+
{ pattern: /:\s*~\(\)/, name: 'fork bomb', risk: 'high' }
|
|
21
|
+
];
|
|
22
|
+
for (const { pattern, name, risk } of dangerousPatterns) {
|
|
23
|
+
if (pattern.test(command)) {
|
|
24
|
+
return {
|
|
25
|
+
allowed: false,
|
|
26
|
+
reason: `Dangerous command detected: ${name} (${risk} risk)`,
|
|
27
|
+
requiresEscalation: risk === 'high',
|
|
28
|
+
suggestedActions: [
|
|
29
|
+
`Review the command: "${command}"`,
|
|
30
|
+
'Consider using safer alternatives',
|
|
31
|
+
'Break down the operation into smaller, safer steps'
|
|
32
|
+
]
|
|
33
|
+
};
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
return {
|
|
38
|
+
allowed: true,
|
|
39
|
+
reason: 'No dangerous patterns detected'
|
|
40
|
+
};
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
exports.NoDangerousShellPolicy = NoDangerousShellPolicy;
|
|
44
|
+
exports.noDangerousShellPolicy = new NoDangerousShellPolicy();
|
|
45
|
+
//# sourceMappingURL=noDangerousShell.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"noDangerousShell.js","sourceRoot":"","sources":["../../../../src/agent/policy/policies/noDangerousShell.ts"],"names":[],"mappings":";;;AAGA,MAAa,sBAAsB;IACjC,IAAI,GAAG,oBAAoB,CAAC;IAC5B,WAAW,GAAG,gDAAgD,CAAC;IAE/D,QAAQ,CAAC,OAAsB;QAC7B,MAAM,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;QAE3B,IAAI,MAAM,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAChC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,EAAE,OAAO,IAAI,EAAE,CAAC;YAE9C,MAAM,iBAAiB,GAAG;gBACxB,EAAE,OAAO,EAAE,eAAe,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC5D,EAAE,OAAO,EAAE,cAAc,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC3D,EAAE,OAAO,EAAE,iBAAiB,EAAE,IAAI,EAAE,uBAAuB,EAAE,IAAI,EAAE,QAAQ,EAAE;gBAC7E,EAAE,OAAO,EAAE,UAAU,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,MAAM,EAAE;gBACzD,EAAE,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,4BAA4B,EAAE,IAAI,EAAE,MAAM,EAAE;gBACrE,EAAE,OAAO,EAAE,QAAQ,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,MAAM,EAAE;gBAC3D,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,MAAM,EAAE;gBACvD,EAAE,OAAO,EAAE,yBAAyB,EAAE,IAAI,EAAE,qBAAqB,EAAE,IAAI,EAAE,MAAM,EAAE;gBACjF,EAAE,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,MAAM,EAAE;aAC1D,CAAC;YAEF,KAAK,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,iBAAiB,EAAE,CAAC;gBACxD,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC;oBAC1B,OAAO;wBACL,OAAO,EAAE,KAAK;wBACd,MAAM,EAAE,+BAA+B,IAAI,KAAK,IAAI,QAAQ;wBAC5D,kBAAkB,EAAE,IAAI,KAAK,MAAM;wBACnC,gBAAgB,EAAE;4BAChB,wBAAwB,OAAO,GAAG;4BAClC,mCAAmC;4BACnC,oDAAoD;yBACrD;qBACF,CAAC;gBACJ,CAAC;YACH,CAAC;QACH,CAAC;QAED,OAAO;YACL,OAAO,EAAE,IAAI;YACb,MAAM,EAAE,gCAAgC;SACzC,CAAC;IACJ,CAAC;CACF;AA3CD,wDA2CC;AAEY,QAAA,sBAAsB,GAAG,IAAI,sBAAsB,EAAE,CAAC"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
export interface PolicyContext {
|
|
2
|
+
action: {
|
|
3
|
+
type: string;
|
|
4
|
+
payload: any;
|
|
5
|
+
};
|
|
6
|
+
user?: {
|
|
7
|
+
permissions: string[];
|
|
8
|
+
};
|
|
9
|
+
environment?: {
|
|
10
|
+
isProduction: boolean;
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
export interface PolicyResult {
|
|
14
|
+
allowed: boolean;
|
|
15
|
+
reason?: string;
|
|
16
|
+
requiresEscalation?: boolean;
|
|
17
|
+
suggestedActions?: string[];
|
|
18
|
+
}
|
|
19
|
+
export interface Policy {
|
|
20
|
+
name: string;
|
|
21
|
+
description: string;
|
|
22
|
+
evaluate(context: PolicyContext): PolicyResult | Promise<PolicyResult>;
|
|
23
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/agent/policy/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
export type EventType = 'state_transition' | 'llm_call' | 'tool_execution' | 'governance_decision' | 'observation_recorded' | 'evaluation_result' | 'error_occurred';
|
|
2
|
+
export interface RuntimeEvent {
|
|
3
|
+
id: string;
|
|
4
|
+
timestamp: number;
|
|
5
|
+
executionId: string;
|
|
6
|
+
type: EventType;
|
|
7
|
+
data: {
|
|
8
|
+
from?: string;
|
|
9
|
+
to?: string;
|
|
10
|
+
action?: any;
|
|
11
|
+
decision?: any;
|
|
12
|
+
result?: any;
|
|
13
|
+
error?: string;
|
|
14
|
+
};
|
|
15
|
+
metadata?: Record<string, any>;
|
|
16
|
+
}
|
|
17
|
+
export interface EventRecorder {
|
|
18
|
+
record(event: RuntimeEvent): void;
|
|
19
|
+
flush(): Promise<void>;
|
|
20
|
+
getEvents(executionId?: string): RuntimeEvent[];
|
|
21
|
+
}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"events.js","sourceRoot":"","sources":["../../../src/agent/replay/events.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
14
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
|
15
|
+
};
|
|
16
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
17
|
+
__exportStar(require("./events"), exports);
|
|
18
|
+
__exportStar(require("./recorder"), exports);
|
|
19
|
+
__exportStar(require("./replayer"), exports);
|
|
20
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/agent/replay/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,2CAAyB;AACzB,6CAA2B;AAC3B,6CAA2B"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import { RuntimeEvent, EventRecorder } from './events';
|
|
2
|
+
export declare class FileEventRecorder implements EventRecorder {
|
|
3
|
+
private events;
|
|
4
|
+
private logFile;
|
|
5
|
+
private flushInterval;
|
|
6
|
+
constructor(logDir?: string);
|
|
7
|
+
record(event: RuntimeEvent): Promise<void>;
|
|
8
|
+
flush(): Promise<void>;
|
|
9
|
+
getEvents(executionId?: string): RuntimeEvent[];
|
|
10
|
+
}
|
|
11
|
+
export declare const createEvent: (executionId: string, type: RuntimeEvent["type"], data: RuntimeEvent["data"], metadata?: RuntimeEvent["metadata"]) => RuntimeEvent;
|
|
@@ -0,0 +1,51 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
3
|
+
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
|
+
};
|
|
5
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.createEvent = exports.FileEventRecorder = void 0;
|
|
7
|
+
const promises_1 = __importDefault(require("fs/promises"));
|
|
8
|
+
const path_1 = __importDefault(require("path"));
|
|
9
|
+
const crypto_1 = require("crypto");
|
|
10
|
+
class FileEventRecorder {
|
|
11
|
+
events = [];
|
|
12
|
+
logFile;
|
|
13
|
+
flushInterval = 1000;
|
|
14
|
+
constructor(logDir = '.yuangs_events') {
|
|
15
|
+
this.logFile = path_1.default.join(logDir, `events_${Date.now()}.jsonl`);
|
|
16
|
+
}
|
|
17
|
+
async record(event) {
|
|
18
|
+
this.events.push(event);
|
|
19
|
+
if (this.events.length >= this.flushInterval) {
|
|
20
|
+
await this.flush();
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
async flush() {
|
|
24
|
+
if (this.events.length === 0)
|
|
25
|
+
return;
|
|
26
|
+
const logDir = path_1.default.dirname(this.logFile);
|
|
27
|
+
await promises_1.default.mkdir(logDir, { recursive: true });
|
|
28
|
+
const content = this.events
|
|
29
|
+
.map(e => JSON.stringify(e))
|
|
30
|
+
.join('\n') + '\n';
|
|
31
|
+
await promises_1.default.appendFile(this.logFile, content, 'utf8');
|
|
32
|
+
this.events = [];
|
|
33
|
+
}
|
|
34
|
+
getEvents(executionId) {
|
|
35
|
+
if (!executionId) {
|
|
36
|
+
return [...this.events];
|
|
37
|
+
}
|
|
38
|
+
return this.events.filter(e => e.executionId === executionId);
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
exports.FileEventRecorder = FileEventRecorder;
|
|
42
|
+
const createEvent = (executionId, type, data, metadata) => ({
|
|
43
|
+
id: (0, crypto_1.randomUUID)(),
|
|
44
|
+
timestamp: Date.now(),
|
|
45
|
+
executionId,
|
|
46
|
+
type,
|
|
47
|
+
data,
|
|
48
|
+
metadata
|
|
49
|
+
});
|
|
50
|
+
exports.createEvent = createEvent;
|
|
51
|
+
//# sourceMappingURL=recorder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"recorder.js","sourceRoot":"","sources":["../../../src/agent/replay/recorder.ts"],"names":[],"mappings":";;;;;;AACA,2DAA6B;AAC7B,gDAAwB;AACxB,mCAAoC;AAEpC,MAAa,iBAAiB;IACpB,MAAM,GAAmB,EAAE,CAAC;IAC5B,OAAO,CAAS;IAChB,aAAa,GAAW,IAAI,CAAC;IAErC,YAAY,SAAiB,gBAAgB;QAC3C,IAAI,CAAC,OAAO,GAAG,cAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;IACjE,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAmB;QAC9B,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAExB,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;YAC7C,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;QACrB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,KAAK;QACT,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAErC,MAAM,MAAM,GAAG,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC1C,MAAM,kBAAE,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAE5C,MAAM,OAAO,GAAG,IAAI,CAAC,MAAM;aACxB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;aAC3B,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAErB,MAAM,kBAAE,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,EAAE,CAAC;IACnB,CAAC;IAED,SAAS,CAAC,WAAoB;QAC5B,IAAI,CAAC,WAAW,EAAE,CAAC;YACjB,OAAO,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1B,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW,KAAK,WAAW,CAAC,CAAC;IAChE,CAAC;CACF;AAtCD,8CAsCC;AAEM,MAAM,WAAW,GAAG,CACzB,WAAmB,EACnB,IAA0B,EAC1B,IAA0B,EAC1B,QAAmC,EACrB,EAAE,CAAC,CAAC;IAClB,EAAE,EAAE,IAAA,mBAAU,GAAE;IAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;IACrB,WAAW;IACX,IAAI;IACJ,IAAI;IACJ,QAAQ;CACT,CAAC,CAAC;AAZU,QAAA,WAAW,eAYrB"}
|
|
@@ -0,0 +1,21 @@
|
|
|
1
|
+
import { RuntimeEvent } from './events';
|
|
2
|
+
export interface ReplayerOptions {
|
|
3
|
+
speed?: number;
|
|
4
|
+
stopOnError?: boolean;
|
|
5
|
+
dryRun?: boolean;
|
|
6
|
+
}
|
|
7
|
+
export declare class EventReplayer {
|
|
8
|
+
private events;
|
|
9
|
+
private currentIndex;
|
|
10
|
+
private options;
|
|
11
|
+
constructor(events: RuntimeEvent[], options?: ReplayerOptions);
|
|
12
|
+
hasNext(): boolean;
|
|
13
|
+
next(): RuntimeEvent | null;
|
|
14
|
+
reset(): void;
|
|
15
|
+
replay(onEvent: (event: RuntimeEvent, options: Required<ReplayerOptions>) => Promise<void>): Promise<void>;
|
|
16
|
+
getSummary(): {
|
|
17
|
+
total: number;
|
|
18
|
+
completed: number;
|
|
19
|
+
errors: number;
|
|
20
|
+
};
|
|
21
|
+
}
|
|
@@ -0,0 +1,65 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
3
|
+
exports.EventReplayer = void 0;
|
|
4
|
+
class EventReplayer {
|
|
5
|
+
events = [];
|
|
6
|
+
currentIndex = 0;
|
|
7
|
+
options;
|
|
8
|
+
constructor(events, options = {}) {
|
|
9
|
+
this.events = events;
|
|
10
|
+
this.options = {
|
|
11
|
+
speed: options.speed || 1,
|
|
12
|
+
stopOnError: options.stopOnError !== undefined ? options.stopOnError : true,
|
|
13
|
+
dryRun: options.dryRun || false
|
|
14
|
+
};
|
|
15
|
+
}
|
|
16
|
+
hasNext() {
|
|
17
|
+
return this.currentIndex < this.events.length;
|
|
18
|
+
}
|
|
19
|
+
next() {
|
|
20
|
+
if (!this.hasNext()) {
|
|
21
|
+
return null;
|
|
22
|
+
}
|
|
23
|
+
return this.events[this.currentIndex++];
|
|
24
|
+
}
|
|
25
|
+
reset() {
|
|
26
|
+
this.currentIndex = 0;
|
|
27
|
+
}
|
|
28
|
+
async replay(onEvent) {
|
|
29
|
+
this.reset();
|
|
30
|
+
let hasError = false;
|
|
31
|
+
while (this.hasNext() && !hasError) {
|
|
32
|
+
const event = this.next();
|
|
33
|
+
if (!event)
|
|
34
|
+
break;
|
|
35
|
+
try {
|
|
36
|
+
await onEvent(event, this.options);
|
|
37
|
+
if (event.type === 'error_occurred') {
|
|
38
|
+
hasError = true;
|
|
39
|
+
if (this.options.stopOnError) {
|
|
40
|
+
break;
|
|
41
|
+
}
|
|
42
|
+
}
|
|
43
|
+
if (this.options.speed > 1) {
|
|
44
|
+
const delay = 100 / this.options.speed;
|
|
45
|
+
await new Promise(resolve => setTimeout(resolve, delay));
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
catch (error) {
|
|
49
|
+
console.error(`[Replay] Error at event ${event.id}:`, error.message);
|
|
50
|
+
hasError = true;
|
|
51
|
+
}
|
|
52
|
+
}
|
|
53
|
+
return;
|
|
54
|
+
}
|
|
55
|
+
getSummary() {
|
|
56
|
+
const errors = this.events.filter(e => e.type === 'error_occurred').length;
|
|
57
|
+
return {
|
|
58
|
+
total: this.events.length,
|
|
59
|
+
completed: this.currentIndex,
|
|
60
|
+
errors
|
|
61
|
+
};
|
|
62
|
+
}
|
|
63
|
+
}
|
|
64
|
+
exports.EventReplayer = EventReplayer;
|
|
65
|
+
//# sourceMappingURL=replayer.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"replayer.js","sourceRoot":"","sources":["../../../src/agent/replay/replayer.ts"],"names":[],"mappings":";;;AAQA,MAAa,aAAa;IAChB,MAAM,GAAmB,EAAE,CAAC;IAC5B,YAAY,GAAW,CAAC,CAAC;IACzB,OAAO,CAA4B;IAE3C,YAAY,MAAsB,EAAE,UAA2B,EAAE;QAC/D,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,OAAO,GAAG;YACb,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,CAAC;YACzB,WAAW,EAAE,OAAO,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI;YAC3E,MAAM,EAAE,OAAO,CAAC,MAAM,IAAI,KAAK;SAChC,CAAC;IACJ,CAAC;IAED,OAAO;QACL,OAAO,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC;IAChD,CAAC;IAED,IAAI;QACF,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,CAAC;YACpB,OAAO,IAAI,CAAC;QACd,CAAC;QAED,OAAO,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;IAC1C,CAAC;IAED,KAAK;QACH,IAAI,CAAC,YAAY,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,OAAmF;QAC9F,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,IAAI,QAAQ,GAAG,KAAK,CAAC;QAErB,OAAO,IAAI,CAAC,OAAO,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACnC,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;YAE1B,IAAI,CAAC,KAAK;gBAAE,MAAM;YAElB,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;gBAEnC,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;oBACpC,QAAQ,GAAG,IAAI,CAAC;oBAChB,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;wBAC7B,MAAM;oBACR,CAAC;gBACH,CAAC;gBAED,IAAI,IAAI,CAAC,OAAO,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;oBAC3B,MAAM,KAAK,GAAG,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;oBACvC,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC3D,CAAC;YACH,CAAC;YAAC,OAAO,KAAU,EAAE,CAAC;gBACpB,OAAO,CAAC,KAAK,CAAC,2BAA2B,KAAK,CAAC,EAAE,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;gBACrE,QAAQ,GAAG,IAAI,CAAC;YAClB,CAAC;QACH,CAAC;QAED,OAAO;IACT,CAAC;IAED,UAAU;QAKR,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC,CAAC,MAAM,CAAC;QAE3E,OAAO;YACL,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,MAAM;YACzB,SAAS,EAAE,IAAI,CAAC,YAAY;YAC5B,MAAM;SACP,CAAC;IACJ,CAAC;CACF;AA3ED,sCA2EC"}
|
|
@@ -6,7 +6,7 @@ function selectModel(intent, override) {
|
|
|
6
6
|
if (override)
|
|
7
7
|
return override;
|
|
8
8
|
const config = (0, client_1.getUserConfig)();
|
|
9
|
-
const defaultModel = config.defaultModel || '
|
|
9
|
+
const defaultModel = config.defaultModel || 'Assistant';
|
|
10
10
|
return defaultModel;
|
|
11
11
|
}
|
|
12
12
|
//# sourceMappingURL=selectModel.js.map
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"selectModel.js","sourceRoot":"","sources":["../../src/agent/selectModel.ts"],"names":[],"mappings":";;AAGA,kCAUC;AAZD,yCAA6C;AAE7C,SAAgB,WAAW,CACvB,MAAmB,EACnB,QAAiB;IAEjB,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAE9B,MAAM,MAAM,GAAG,IAAA,sBAAa,GAAE,CAAC;IAC/B,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,
|
|
1
|
+
{"version":3,"file":"selectModel.js","sourceRoot":"","sources":["../../src/agent/selectModel.ts"],"names":[],"mappings":";;AAGA,kCAUC;AAZD,yCAA6C;AAE7C,SAAgB,WAAW,CACvB,MAAmB,EACnB,QAAiB;IAEjB,IAAI,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAE9B,MAAM,MAAM,GAAG,IAAA,sBAAa,GAAE,CAAC;IAC/B,MAAM,YAAY,GAAG,MAAM,CAAC,YAAY,IAAI,WAAW,CAAC;IAExD,OAAO,YAAY,CAAC;AACxB,CAAC"}
|
package/dist/agent/skills.d.ts
CHANGED
|
@@ -11,7 +11,12 @@ export interface Skill {
|
|
|
11
11
|
confidence: number;
|
|
12
12
|
lastUsed: number;
|
|
13
13
|
createdAt: number;
|
|
14
|
+
enabled: boolean;
|
|
14
15
|
}
|
|
16
|
+
/**
|
|
17
|
+
* 计算技能分 (0 ~ 1)
|
|
18
|
+
*/
|
|
19
|
+
export declare function computeSkillScore(skill: Skill, now?: number): number;
|
|
15
20
|
/**
|
|
16
21
|
* 更新技能状态 (执行后调用)
|
|
17
22
|
*/
|
package/dist/agent/skills.js
CHANGED
|
@@ -3,6 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
|
3
3
|
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
4
4
|
};
|
|
5
5
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
6
|
+
exports.computeSkillScore = computeSkillScore;
|
|
6
7
|
exports.updateSkillStatus = updateSkillStatus;
|
|
7
8
|
exports.learnSkillFromRecord = learnSkillFromRecord;
|
|
8
9
|
exports.getRelevantSkills = getRelevantSkills;
|
|
@@ -95,7 +96,8 @@ function learnSkillFromRecord(record, success = true) {
|
|
|
95
96
|
failureCount: 0,
|
|
96
97
|
confidence: 0.5,
|
|
97
98
|
lastUsed: now,
|
|
98
|
-
createdAt: now
|
|
99
|
+
createdAt: now,
|
|
100
|
+
enabled: true
|
|
99
101
|
});
|
|
100
102
|
// 每学习一次,尝试清理一次“冷”技能
|
|
101
103
|
reapColdSkills();
|
|
@@ -109,9 +111,11 @@ function getRelevantSkills(input, limit = 3) {
|
|
|
109
111
|
return skillLibrary
|
|
110
112
|
// 1. 基础筛选: 剔除评分过低的技能 (硬淘汰阈值 0.3)
|
|
111
113
|
.filter(s => computeSkillScore(s, now) >= 0.3)
|
|
112
|
-
// 2.
|
|
114
|
+
// 2. 过滤已禁用的技能
|
|
115
|
+
.filter(s => s.enabled !== false)
|
|
116
|
+
// 3. 排序: 按综合分排序
|
|
113
117
|
.sort((a, b) => computeSkillScore(b, now) - computeSkillScore(a, now))
|
|
114
|
-
//
|
|
118
|
+
// 4. 取上限
|
|
115
119
|
.slice(0, limit);
|
|
116
120
|
}
|
|
117
121
|
/**
|
package/dist/agent/skills.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"skills.js","sourceRoot":"","sources":["../../src/agent/skills.ts"],"names":[],"mappings":";;;;;
|
|
1
|
+
{"version":3,"file":"skills.js","sourceRoot":"","sources":["../../src/agent/skills.ts"],"names":[],"mappings":";;;;;AA4DA,8CAUC;AAKD,8CAgBC;AAKD,oDAgCC;AAKD,8CAYC;AAKD,wCA2BC;AAED,oCAEC;AArLD,4CAAoB;AACpB,gDAAwB;AACxB,4CAAoB;AAGpB,kDAA0B;AAsB1B,MAAM,WAAW,GAAG,cAAI,CAAC,IAAI,CAAC,YAAE,CAAC,OAAO,EAAE,EAAE,qBAAqB,CAAC,CAAC;AACnE,IAAI,YAAY,GAAY,EAAE,CAAC;AAE/B,4BAA4B;AAE5B,SAAS,UAAU;IACf,IAAI,YAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;QAC7B,IAAI,CAAC;YACD,MAAM,IAAI,GAAG,YAAE,CAAC,YAAY,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;YACnD,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACpC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,MAAM,CAAC,8BAA8B,WAAW,mBAAmB,CAAC,CAAC,CAAC;YAC1F,YAAY,GAAG,EAAE,CAAC;QACtB,CAAC;IACL,CAAC;AACL,CAAC;AAED,SAAS,UAAU;IACf,IAAI,CAAC;QACD,YAAE,CAAC,aAAa,CAAC,WAAW,EAAE,IAAI,CAAC,SAAS,CAAC,YAAY,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC;IACzE,CAAC;IAAC,OAAO,CAAC,EAAE,CAAC;QACT,OAAO,CAAC,KAAK,CAAC,eAAK,CAAC,GAAG,CAAC,4BAA4B,WAAW,EAAE,CAAC,CAAC,CAAC;IACxE,CAAC;AACL,CAAC;AAED,qBAAqB;AACrB,UAAU,EAAE,CAAC;AAEb,yCAAyC;AAEzC;;GAEG;AACH,SAAgB,iBAAiB,CAAC,KAAY,EAAE,MAAc,IAAI,CAAC,GAAG,EAAE;IACpE,MAAM,SAAS,GAAG,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IAC1D,MAAM,WAAW,GAAG,SAAS,KAAK,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK,CAAC,YAAY,GAAG,SAAS,CAAC;IAE3E,8BAA8B;IAC9B,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;IAChE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;IAE3C,oCAAoC;IACpC,OAAO,CAAC,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,GAAG,KAAK,CAAC,UAAU,CAAC,CAAC;AACjF,CAAC;AAED;;GAEG;AACH,SAAgB,iBAAiB,CAAC,OAAe,EAAE,OAAgB;IAC/D,MAAM,KAAK,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,CAAC,CAAC;IACvD,IAAI,CAAC,KAAK;QAAE,OAAO;IAEnB,KAAK,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAC5B,IAAI,OAAO,EAAE,CAAC;QACV,KAAK,CAAC,YAAY,EAAE,CAAC;QACrB,gBAAgB;QAChB,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC;IAC5D,CAAC;SAAM,CAAC;QACJ,KAAK,CAAC,YAAY,EAAE,CAAC;QACrB,0BAA0B;QAC1B,KAAK,CAAC,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,UAAU,GAAG,GAAG,CAAC,CAAC;IAC3D,CAAC;IAED,UAAU,EAAE,CAAC,CAAC,kBAAkB;AACpC,CAAC;AAED;;GAEG;AACH,SAAgB,oBAAoB,CAAC,MAAuB,EAAE,UAAmB,IAAI;IACjF,IAAI,MAAM,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,IAAI;QAAE,OAAO;IAE7D,MAAM,aAAa,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,SAAS,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;IAErF,IAAI,aAAa,EAAE,CAAC;QAChB,iBAAiB,CAAC,aAAa,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC7C,OAAO;IACX,CAAC;IAED,iBAAiB;IACjB,IAAI,CAAC,OAAO;QAAE,OAAO;IAErB,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,YAAY,CAAC,IAAI,CAAC;QACd,EAAE,EAAE,MAAM,CAAC,EAAE;QACb,IAAI,EAAE,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI;QAChC,WAAW,EAAE,YAAY,MAAM,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,EAAE;QACrD,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,QAAQ;QAChC,YAAY,EAAE,MAAM,CAAC,SAAS,CAAC,IAAI;QACnC,YAAY,EAAE,CAAC;QACf,YAAY,EAAE,CAAC;QACf,UAAU,EAAE,GAAG;QACf,QAAQ,EAAE,GAAG;QACb,SAAS,EAAE,GAAG;QACd,OAAO,EAAE,IAAI;KAChB,CAAC,CAAC;IAEH,oBAAoB;IACpB,cAAc,EAAE,CAAC;IAEjB,UAAU,EAAE,CAAC,CAAC,kBAAkB;AACpC,CAAC;AAED;;GAEG;AACH,SAAgB,iBAAiB,CAAC,KAAa,EAAE,QAAgB,CAAC;IAC9D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAEvB,OAAO,YAAY;QACf,iCAAiC;SAChC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,IAAI,GAAG,CAAC;QAC9C,cAAc;SACb,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,KAAK,KAAK,CAAC;QACjC,gBAAgB;SACf,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;QACtE,SAAS;SACR,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;AACzB,CAAC;AAED;;GAEG;AACH,SAAgB,cAAc;IAC1B,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IACvB,MAAM,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC;IAEzC,YAAY,GAAG,YAAY,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;QACvC,MAAM,KAAK,GAAG,iBAAiB,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;QAC5C,MAAM,QAAQ,GAAG,CAAC,GAAG,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,CAAC,CAAC;QAEhE,eAAe;QACf,eAAe;QACf,IAAI,KAAK,GAAG,IAAI,IAAI,QAAQ,GAAG,EAAE;YAAE,OAAO,KAAK,CAAC;QAChD,mBAAmB;QACnB,IAAI,KAAK,CAAC,YAAY,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,CAAC,KAAK,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC,CAAC,GAAG,GAAG;YAAE,OAAO,KAAK,CAAC;QAEnH,OAAO,IAAI,CAAC;IAChB,CAAC,CAAC,CAAC;IAEH,SAAS;IACT,IAAI,YAAY,CAAC,MAAM,GAAG,GAAG,EAAE,CAAC;QAC5B,kBAAkB;QAClB,YAAY,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,iBAAiB,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QACnF,YAAY,CAAC,KAAK,EAAE,CAAC;IACzB,CAAC;IAED,IAAI,YAAY,CAAC,MAAM,KAAK,YAAY,EAAE,CAAC;QACvC,UAAU,EAAE,CAAC,CAAC,8BAA8B;IAChD,CAAC;AACL,CAAC;AAED,SAAgB,YAAY;IACxB,OAAO,CAAC,GAAG,YAAY,CAAC,CAAC;AAC7B,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from './registryAPI';
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
"use strict";
|
|
2
|
+
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
3
|
+
if (k2 === undefined) k2 = k;
|
|
4
|
+
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
5
|
+
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
6
|
+
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
7
|
+
}
|
|
8
|
+
Object.defineProperty(o, k2, desc);
|
|
9
|
+
}) : (function(o, m, k, k2) {
|
|
10
|
+
if (k2 === undefined) k2 = k;
|
|
11
|
+
o[k2] = m[k];
|
|
12
|
+
}));
|
|
13
|
+
var __exportStar = (this && this.__exportStar) || function(m, exports) {
|
|
14
|
+
for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
|
|
15
|
+
};
|
|
16
|
+
Object.defineProperty(exports, "__esModule", { value: true });
|
|
17
|
+
__exportStar(require("./registryAPI"), exports);
|
|
18
|
+
//# sourceMappingURL=index.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/api/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,gDAA8B"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { MacroManifest, MacroPublishOptions, Capability } from '../registry/manifest';
|
|
2
|
+
import type { RiskAssessment } from '../risk/explainer';
|
|
3
|
+
export declare class RegistryAPI {
|
|
4
|
+
private registry;
|
|
5
|
+
private riskExplainer;
|
|
6
|
+
constructor(storagePath?: string);
|
|
7
|
+
initialize(): Promise<void>;
|
|
8
|
+
publishMacro(id: string, version: string, description: string, requires: Capability[], author: string, options?: MacroPublishOptions): Promise<MacroManifest>;
|
|
9
|
+
getMacro(id: string, version?: string): Promise<MacroManifest | null>;
|
|
10
|
+
listMacros(filters?: {
|
|
11
|
+
state?: 'draft' | 'approved' | 'deprecated';
|
|
12
|
+
author?: string;
|
|
13
|
+
tags?: string[];
|
|
14
|
+
}): Promise<MacroManifest[]>;
|
|
15
|
+
approveMacro(id: string, version: string, approvedBy: string): Promise<MacroManifest>;
|
|
16
|
+
deprecateMacro(id: string, version?: string): Promise<MacroManifest>;
|
|
17
|
+
assessMacroRisk(id: string, version?: string): Promise<RiskAssessment | null>;
|
|
18
|
+
compareMacroVersions(id: string, oldVersion: string, newVersion: string): Promise<import("../registry/manifest").MacroDiffResult>;
|
|
19
|
+
getMacroVersions(id: string): Promise<MacroManifest[]>;
|
|
20
|
+
explainCapability(capability: Capability): Promise<string>;
|
|
21
|
+
expandCapabilities(capabilities: Capability[]): Promise<Capability[]>;
|
|
22
|
+
}
|