@researchcomputer/agents-sdk 0.1.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 +60 -0
- package/dist/agents/messages.d.ts +29 -0
- package/dist/agents/messages.d.ts.map +1 -0
- package/dist/agents/messages.js +56 -0
- package/dist/agents/messages.js.map +1 -0
- package/dist/agents/messages.test.d.ts +2 -0
- package/dist/agents/messages.test.d.ts.map +1 -0
- package/dist/agents/messages.test.js +80 -0
- package/dist/agents/messages.test.js.map +1 -0
- package/dist/agents/subagent.d.ts +16 -0
- package/dist/agents/subagent.d.ts.map +1 -0
- package/dist/agents/subagent.js +46 -0
- package/dist/agents/subagent.js.map +1 -0
- package/dist/agents/swarm.d.ts +22 -0
- package/dist/agents/swarm.d.ts.map +1 -0
- package/dist/agents/swarm.js +140 -0
- package/dist/agents/swarm.js.map +1 -0
- package/dist/agents/tools.d.ts +6 -0
- package/dist/agents/tools.d.ts.map +1 -0
- package/dist/agents/tools.js +71 -0
- package/dist/agents/tools.js.map +1 -0
- package/dist/agents/tools.test.d.ts +2 -0
- package/dist/agents/tools.test.d.ts.map +1 -0
- package/dist/agents/tools.test.js +70 -0
- package/dist/agents/tools.test.js.map +1 -0
- package/dist/bridge/config.d.ts +21 -0
- package/dist/bridge/config.d.ts.map +1 -0
- package/dist/bridge/config.js +35 -0
- package/dist/bridge/config.js.map +1 -0
- package/dist/bridge/config.test.d.ts +2 -0
- package/dist/bridge/config.test.d.ts.map +1 -0
- package/dist/bridge/config.test.js +79 -0
- package/dist/bridge/config.test.js.map +1 -0
- package/dist/bridge/server.d.ts +3 -0
- package/dist/bridge/server.d.ts.map +1 -0
- package/dist/bridge/server.js +316 -0
- package/dist/bridge/server.js.map +1 -0
- package/dist/context/compression.d.ts +27 -0
- package/dist/context/compression.d.ts.map +1 -0
- package/dist/context/compression.js +117 -0
- package/dist/context/compression.js.map +1 -0
- package/dist/context/compression.test.d.ts +2 -0
- package/dist/context/compression.test.d.ts.map +1 -0
- package/dist/context/compression.test.js +170 -0
- package/dist/context/compression.test.js.map +1 -0
- package/dist/context/converter.d.ts +11 -0
- package/dist/context/converter.d.ts.map +1 -0
- package/dist/context/converter.js +40 -0
- package/dist/context/converter.js.map +1 -0
- package/dist/context/converter.test.d.ts +2 -0
- package/dist/context/converter.test.d.ts.map +1 -0
- package/dist/context/converter.test.js +110 -0
- package/dist/context/converter.test.js.map +1 -0
- package/dist/context/run-context.d.ts +9 -0
- package/dist/context/run-context.d.ts.map +1 -0
- package/dist/context/run-context.js +14 -0
- package/dist/context/run-context.js.map +1 -0
- package/dist/context/run-context.test.d.ts +2 -0
- package/dist/context/run-context.test.d.ts.map +1 -0
- package/dist/context/run-context.test.js +34 -0
- package/dist/context/run-context.test.js.map +1 -0
- package/dist/context/system-prompt.d.ts +24 -0
- package/dist/context/system-prompt.d.ts.map +1 -0
- package/dist/context/system-prompt.js +69 -0
- package/dist/context/system-prompt.js.map +1 -0
- package/dist/context/system-prompt.test.d.ts +2 -0
- package/dist/context/system-prompt.test.d.ts.map +1 -0
- package/dist/context/system-prompt.test.js +134 -0
- package/dist/context/system-prompt.test.js.map +1 -0
- package/dist/errors.d.ts +24 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +47 -0
- package/dist/errors.js.map +1 -0
- package/dist/errors.test.d.ts +2 -0
- package/dist/errors.test.d.ts.map +1 -0
- package/dist/errors.test.js +68 -0
- package/dist/errors.test.js.map +1 -0
- package/dist/factory.d.ts +58 -0
- package/dist/factory.d.ts.map +1 -0
- package/dist/factory.js +282 -0
- package/dist/factory.js.map +1 -0
- package/dist/factory.test.d.ts +2 -0
- package/dist/factory.test.d.ts.map +1 -0
- package/dist/factory.test.js +63 -0
- package/dist/factory.test.js.map +1 -0
- package/dist/fork.test.d.ts +2 -0
- package/dist/fork.test.d.ts.map +1 -0
- package/dist/fork.test.js +456 -0
- package/dist/fork.test.js.map +1 -0
- package/dist/index.d.ts +29 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +32 -0
- package/dist/index.js.map +1 -0
- package/dist/mcp/client.d.ts +3 -0
- package/dist/mcp/client.d.ts.map +1 -0
- package/dist/mcp/client.js +75 -0
- package/dist/mcp/client.js.map +1 -0
- package/dist/mcp/schema-convert.d.ts +3 -0
- package/dist/mcp/schema-convert.d.ts.map +1 -0
- package/dist/mcp/schema-convert.js +73 -0
- package/dist/mcp/schema-convert.js.map +1 -0
- package/dist/mcp/schema-convert.test.d.ts +2 -0
- package/dist/mcp/schema-convert.test.d.ts.map +1 -0
- package/dist/mcp/schema-convert.test.js +95 -0
- package/dist/mcp/schema-convert.test.js.map +1 -0
- package/dist/mcp/tools.d.ts +14 -0
- package/dist/mcp/tools.d.ts.map +1 -0
- package/dist/mcp/tools.js +23 -0
- package/dist/mcp/tools.js.map +1 -0
- package/dist/mcp/tools.test.d.ts +2 -0
- package/dist/mcp/tools.test.d.ts.map +1 -0
- package/dist/mcp/tools.test.js +45 -0
- package/dist/mcp/tools.test.js.map +1 -0
- package/dist/memory/memory.d.ts +3 -0
- package/dist/memory/memory.d.ts.map +1 -0
- package/dist/memory/memory.js +130 -0
- package/dist/memory/memory.js.map +1 -0
- package/dist/memory/memory.test.d.ts +2 -0
- package/dist/memory/memory.test.d.ts.map +1 -0
- package/dist/memory/memory.test.js +94 -0
- package/dist/memory/memory.test.js.map +1 -0
- package/dist/middleware/hooks.d.ts +18 -0
- package/dist/middleware/hooks.d.ts.map +1 -0
- package/dist/middleware/hooks.js +71 -0
- package/dist/middleware/hooks.js.map +1 -0
- package/dist/middleware/hooks.test.d.ts +2 -0
- package/dist/middleware/hooks.test.d.ts.map +1 -0
- package/dist/middleware/hooks.test.js +172 -0
- package/dist/middleware/hooks.test.js.map +1 -0
- package/dist/middleware/index.d.ts +7 -0
- package/dist/middleware/index.d.ts.map +1 -0
- package/dist/middleware/index.js +5 -0
- package/dist/middleware/index.js.map +1 -0
- package/dist/middleware/permission-middleware.d.ts +11 -0
- package/dist/middleware/permission-middleware.d.ts.map +1 -0
- package/dist/middleware/permission-middleware.js +56 -0
- package/dist/middleware/permission-middleware.js.map +1 -0
- package/dist/middleware/permission-middleware.test.d.ts +2 -0
- package/dist/middleware/permission-middleware.test.d.ts.map +1 -0
- package/dist/middleware/permission-middleware.test.js +170 -0
- package/dist/middleware/permission-middleware.test.js.map +1 -0
- package/dist/middleware/permissions.d.ts +14 -0
- package/dist/middleware/permissions.d.ts.map +1 -0
- package/dist/middleware/permissions.js +118 -0
- package/dist/middleware/permissions.js.map +1 -0
- package/dist/middleware/permissions.test.d.ts +2 -0
- package/dist/middleware/permissions.test.d.ts.map +1 -0
- package/dist/middleware/permissions.test.js +187 -0
- package/dist/middleware/permissions.test.js.map +1 -0
- package/dist/middleware/pipeline.d.ts +13 -0
- package/dist/middleware/pipeline.d.ts.map +1 -0
- package/dist/middleware/pipeline.js +28 -0
- package/dist/middleware/pipeline.js.map +1 -0
- package/dist/middleware/pipeline.test.d.ts +2 -0
- package/dist/middleware/pipeline.test.d.ts.map +1 -0
- package/dist/middleware/pipeline.test.js +189 -0
- package/dist/middleware/pipeline.test.js.map +1 -0
- package/dist/observability/cost-tracker.d.ts +3 -0
- package/dist/observability/cost-tracker.d.ts.map +1 -0
- package/dist/observability/cost-tracker.js +26 -0
- package/dist/observability/cost-tracker.js.map +1 -0
- package/dist/observability/cost-tracker.test.d.ts +2 -0
- package/dist/observability/cost-tracker.test.d.ts.map +1 -0
- package/dist/observability/cost-tracker.test.js +49 -0
- package/dist/observability/cost-tracker.test.js.map +1 -0
- package/dist/observability/trace.d.ts +2 -0
- package/dist/observability/trace.d.ts.map +1 -0
- package/dist/observability/trace.js +5 -0
- package/dist/observability/trace.js.map +1 -0
- package/dist/session/session.d.ts +3 -0
- package/dist/session/session.d.ts.map +1 -0
- package/dist/session/session.js +62 -0
- package/dist/session/session.js.map +1 -0
- package/dist/session/session.test.d.ts +2 -0
- package/dist/session/session.test.d.ts.map +1 -0
- package/dist/session/session.test.js +84 -0
- package/dist/session/session.test.js.map +1 -0
- package/dist/skills.d.ts +14 -0
- package/dist/skills.d.ts.map +1 -0
- package/dist/skills.js +31 -0
- package/dist/skills.js.map +1 -0
- package/dist/skills.test.d.ts +2 -0
- package/dist/skills.test.d.ts.map +1 -0
- package/dist/skills.test.js +61 -0
- package/dist/skills.test.js.map +1 -0
- package/dist/tools/ask-user.d.ts +10 -0
- package/dist/tools/ask-user.d.ts.map +1 -0
- package/dist/tools/ask-user.js +27 -0
- package/dist/tools/ask-user.js.map +1 -0
- package/dist/tools/bash.d.ts +11 -0
- package/dist/tools/bash.d.ts.map +1 -0
- package/dist/tools/bash.js +54 -0
- package/dist/tools/bash.js.map +1 -0
- package/dist/tools/bash.test.d.ts +2 -0
- package/dist/tools/bash.test.d.ts.map +1 -0
- package/dist/tools/bash.test.js +63 -0
- package/dist/tools/bash.test.js.map +1 -0
- package/dist/tools/edit.d.ts +10 -0
- package/dist/tools/edit.d.ts.map +1 -0
- package/dist/tools/edit.js +65 -0
- package/dist/tools/edit.js.map +1 -0
- package/dist/tools/edit.test.d.ts +2 -0
- package/dist/tools/edit.test.d.ts.map +1 -0
- package/dist/tools/edit.test.js +60 -0
- package/dist/tools/edit.test.js.map +1 -0
- package/dist/tools/glob.d.ts +8 -0
- package/dist/tools/glob.d.ts.map +1 -0
- package/dist/tools/glob.js +43 -0
- package/dist/tools/glob.js.map +1 -0
- package/dist/tools/glob.test.d.ts +2 -0
- package/dist/tools/glob.test.d.ts.map +1 -0
- package/dist/tools/glob.test.js +54 -0
- package/dist/tools/glob.test.js.map +1 -0
- package/dist/tools/grep.d.ts +9 -0
- package/dist/tools/grep.d.ts.map +1 -0
- package/dist/tools/grep.js +74 -0
- package/dist/tools/grep.js.map +1 -0
- package/dist/tools/grep.test.d.ts +2 -0
- package/dist/tools/grep.test.d.ts.map +1 -0
- package/dist/tools/grep.test.js +48 -0
- package/dist/tools/grep.test.js.map +1 -0
- package/dist/tools/index.d.ts +25 -0
- package/dist/tools/index.d.ts.map +1 -0
- package/dist/tools/index.js +40 -0
- package/dist/tools/index.js.map +1 -0
- package/dist/tools/index.test.d.ts +2 -0
- package/dist/tools/index.test.d.ts.map +1 -0
- package/dist/tools/index.test.js +102 -0
- package/dist/tools/index.test.js.map +1 -0
- package/dist/tools/notebook-edit.d.ts +9 -0
- package/dist/tools/notebook-edit.d.ts.map +1 -0
- package/dist/tools/notebook-edit.js +55 -0
- package/dist/tools/notebook-edit.js.map +1 -0
- package/dist/tools/read.d.ts +9 -0
- package/dist/tools/read.d.ts.map +1 -0
- package/dist/tools/read.js +48 -0
- package/dist/tools/read.js.map +1 -0
- package/dist/tools/read.test.d.ts +2 -0
- package/dist/tools/read.test.d.ts.map +1 -0
- package/dist/tools/read.test.js +68 -0
- package/dist/tools/read.test.js.map +1 -0
- package/dist/tools/util.d.ts +19 -0
- package/dist/tools/util.d.ts.map +1 -0
- package/dist/tools/util.js +44 -0
- package/dist/tools/util.js.map +1 -0
- package/dist/tools/util.test.d.ts +2 -0
- package/dist/tools/util.test.d.ts.map +1 -0
- package/dist/tools/util.test.js +75 -0
- package/dist/tools/util.test.js.map +1 -0
- package/dist/tools/web-fetch.d.ts +7 -0
- package/dist/tools/web-fetch.d.ts.map +1 -0
- package/dist/tools/web-fetch.js +35 -0
- package/dist/tools/web-fetch.js.map +1 -0
- package/dist/tools/web-search.d.ts +7 -0
- package/dist/tools/web-search.d.ts.map +1 -0
- package/dist/tools/web-search.js +20 -0
- package/dist/tools/web-search.js.map +1 -0
- package/dist/tools/write.d.ts +8 -0
- package/dist/tools/write.d.ts.map +1 -0
- package/dist/tools/write.js +33 -0
- package/dist/tools/write.js.map +1 -0
- package/dist/tools/write.test.d.ts +2 -0
- package/dist/tools/write.test.d.ts.map +1 -0
- package/dist/tools/write.test.js +49 -0
- package/dist/tools/write.test.js.map +1 -0
- package/dist/types.d.ts +257 -0
- package/dist/types.d.ts.map +1 -0
- package/dist/types.js +2 -0
- package/dist/types.js.map +1 -0
- package/dist/types.test.d.ts +2 -0
- package/dist/types.test.d.ts.map +1 -0
- package/dist/types.test.js +69 -0
- package/dist/types.test.js.map +1 -0
- package/package.json +51 -0
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
export function createCostTracker() {
|
|
2
|
+
let totalTokens = 0;
|
|
3
|
+
let totalCost = 0;
|
|
4
|
+
const models = new Map();
|
|
5
|
+
return {
|
|
6
|
+
record(usage, modelId) {
|
|
7
|
+
totalTokens += usage.totalTokens;
|
|
8
|
+
totalCost += usage.cost.total;
|
|
9
|
+
if (modelId) {
|
|
10
|
+
const existing = models.get(modelId) ?? { tokens: 0, cost: 0 };
|
|
11
|
+
existing.tokens += usage.totalTokens;
|
|
12
|
+
existing.cost += usage.cost.total;
|
|
13
|
+
models.set(modelId, existing);
|
|
14
|
+
}
|
|
15
|
+
},
|
|
16
|
+
total() { return { tokens: totalTokens, cost: totalCost }; },
|
|
17
|
+
perModel() {
|
|
18
|
+
const copy = new Map();
|
|
19
|
+
for (const [key, value] of models) {
|
|
20
|
+
copy.set(key, { ...value });
|
|
21
|
+
}
|
|
22
|
+
return copy;
|
|
23
|
+
},
|
|
24
|
+
};
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=cost-tracker.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-tracker.js","sourceRoot":"","sources":["../../src/observability/cost-tracker.ts"],"names":[],"mappings":"AAGA,MAAM,UAAU,iBAAiB;IAC/B,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,MAAM,MAAM,GAAG,IAAI,GAAG,EAA4C,CAAC;IACnE,OAAO;QACL,MAAM,CAAC,KAAY,EAAE,OAAgB;YACnC,WAAW,IAAI,KAAK,CAAC,WAAW,CAAC;YACjC,SAAS,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;YAC9B,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC;gBAC/D,QAAQ,CAAC,MAAM,IAAI,KAAK,CAAC,WAAW,CAAC;gBACrC,QAAQ,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC;gBAClC,MAAM,CAAC,GAAG,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAChC,CAAC;QACH,CAAC;QACD,KAAK,KAAK,OAAO,EAAE,MAAM,EAAE,WAAW,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;QAC5D,QAAQ;YACN,MAAM,IAAI,GAAG,IAAI,GAAG,EAA4C,CAAC;YACjE,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,EAAE,CAAC;gBAClC,IAAI,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,GAAG,KAAK,EAAE,CAAC,CAAC;YAC9B,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-tracker.test.d.ts","sourceRoot":"","sources":["../../src/observability/cost-tracker.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,49 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { createCostTracker } from './cost-tracker.js';
|
|
3
|
+
function makeUsage(totalTokens, totalCost) {
|
|
4
|
+
return {
|
|
5
|
+
input: totalTokens * 0.5,
|
|
6
|
+
output: totalTokens * 0.5,
|
|
7
|
+
cacheRead: 0,
|
|
8
|
+
cacheWrite: 0,
|
|
9
|
+
totalTokens,
|
|
10
|
+
cost: { input: totalCost * 0.5, output: totalCost * 0.5, cacheRead: 0, cacheWrite: 0, total: totalCost },
|
|
11
|
+
};
|
|
12
|
+
}
|
|
13
|
+
describe('createCostTracker', () => {
|
|
14
|
+
it('starts with zero totals', () => {
|
|
15
|
+
const tracker = createCostTracker();
|
|
16
|
+
expect(tracker.total()).toEqual({ tokens: 0, cost: 0 });
|
|
17
|
+
});
|
|
18
|
+
it('records usage and accumulates totals', () => {
|
|
19
|
+
const tracker = createCostTracker();
|
|
20
|
+
tracker.record(makeUsage(100, 0.01));
|
|
21
|
+
tracker.record(makeUsage(200, 0.02));
|
|
22
|
+
expect(tracker.total()).toEqual({ tokens: 300, cost: 0.03 });
|
|
23
|
+
});
|
|
24
|
+
it('tracks per-model usage when modelId is provided', () => {
|
|
25
|
+
const tracker = createCostTracker();
|
|
26
|
+
tracker.record(makeUsage(100, 0.01), 'gpt-4');
|
|
27
|
+
tracker.record(makeUsage(50, 0.005), 'gpt-4');
|
|
28
|
+
tracker.record(makeUsage(200, 0.02), 'claude-3');
|
|
29
|
+
const perModel = tracker.perModel();
|
|
30
|
+
expect(perModel.get('gpt-4')).toEqual({ tokens: 150, cost: 0.015 });
|
|
31
|
+
expect(perModel.get('claude-3')).toEqual({ tokens: 200, cost: 0.02 });
|
|
32
|
+
});
|
|
33
|
+
it('does not track model when modelId is omitted', () => {
|
|
34
|
+
const tracker = createCostTracker();
|
|
35
|
+
tracker.record(makeUsage(100, 0.01));
|
|
36
|
+
expect(tracker.perModel().size).toBe(0);
|
|
37
|
+
expect(tracker.total().tokens).toBe(100);
|
|
38
|
+
});
|
|
39
|
+
it('perModel returns a copy', () => {
|
|
40
|
+
const tracker = createCostTracker();
|
|
41
|
+
tracker.record(makeUsage(100, 0.01), 'gpt-4');
|
|
42
|
+
const map1 = tracker.perModel();
|
|
43
|
+
tracker.record(makeUsage(50, 0.005), 'gpt-4');
|
|
44
|
+
const map2 = tracker.perModel();
|
|
45
|
+
expect(map1.get('gpt-4').tokens).toBe(100);
|
|
46
|
+
expect(map2.get('gpt-4').tokens).toBe(150);
|
|
47
|
+
});
|
|
48
|
+
});
|
|
49
|
+
//# sourceMappingURL=cost-tracker.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"cost-tracker.test.js","sourceRoot":"","sources":["../../src/observability/cost-tracker.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAGtD,SAAS,SAAS,CAAC,WAAmB,EAAE,SAAiB;IACvD,OAAO;QACL,KAAK,EAAE,WAAW,GAAG,GAAG;QACxB,MAAM,EAAE,WAAW,GAAG,GAAG;QACzB,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW;QACX,IAAI,EAAE,EAAE,KAAK,EAAE,SAAS,GAAG,GAAG,EAAE,MAAM,EAAE,SAAS,GAAG,GAAG,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,SAAS,EAAE;KACzG,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;IACjC,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;IAC1D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;QAC9C,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;QACrC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,GAAG,EAAE;QACzD,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;QAC9C,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;QAC9C,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,UAAU,CAAC,CAAC;QACjD,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QACpC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACpE,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,MAAM,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC,CAAC;QACrC,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACjC,MAAM,OAAO,GAAG,iBAAiB,EAAE,CAAC;QACpC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QAChC,OAAO,CAAC,MAAM,CAAC,SAAS,CAAC,EAAE,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;QAC9C,MAAM,IAAI,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QAChC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAE,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.d.ts","sourceRoot":"","sources":["../../src/observability/trace.ts"],"names":[],"mappings":"AAEA,wBAAgB,eAAe,IAAI,MAAM,CAExC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"trace.js","sourceRoot":"","sources":["../../src/observability/trace.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAEzC,MAAM,UAAU,eAAe;IAC7B,OAAO,UAAU,EAAE,CAAC;AACtB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.d.ts","sourceRoot":"","sources":["../../src/session/session.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,cAAc,EAAmB,MAAM,aAAa,CAAC;AAGnE,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,MAAM,GAAG,cAAc,CA0DhE"}
|
|
@@ -0,0 +1,62 @@
|
|
|
1
|
+
import * as fs from 'node:fs/promises';
|
|
2
|
+
import * as path from 'node:path';
|
|
3
|
+
import { SessionLoadError } from '../errors.js';
|
|
4
|
+
export function createSessionManager(dir) {
|
|
5
|
+
return {
|
|
6
|
+
async save(snapshot) {
|
|
7
|
+
await fs.mkdir(dir, { recursive: true });
|
|
8
|
+
const safeId = snapshot.id.replace(/[^a-zA-Z0-9_-]/g, '_');
|
|
9
|
+
const filePath = path.join(dir, `${safeId}.json`);
|
|
10
|
+
await fs.writeFile(filePath, JSON.stringify(snapshot, null, 2), 'utf-8');
|
|
11
|
+
},
|
|
12
|
+
async load(id) {
|
|
13
|
+
const safeId = id.replace(/[^a-zA-Z0-9_-]/g, '_');
|
|
14
|
+
const filePath = path.join(dir, `${safeId}.json`);
|
|
15
|
+
let content;
|
|
16
|
+
try {
|
|
17
|
+
content = await fs.readFile(filePath, 'utf-8');
|
|
18
|
+
}
|
|
19
|
+
catch (err) {
|
|
20
|
+
if (err.code === 'ENOENT')
|
|
21
|
+
return null;
|
|
22
|
+
throw err;
|
|
23
|
+
}
|
|
24
|
+
let parsed;
|
|
25
|
+
try {
|
|
26
|
+
parsed = JSON.parse(content);
|
|
27
|
+
}
|
|
28
|
+
catch {
|
|
29
|
+
throw new SessionLoadError(`Corrupt session file: ${id}.json`);
|
|
30
|
+
}
|
|
31
|
+
if (parsed.version !== 1) {
|
|
32
|
+
throw new SessionLoadError(`Unsupported session version: ${parsed.version}`);
|
|
33
|
+
}
|
|
34
|
+
return parsed;
|
|
35
|
+
},
|
|
36
|
+
async list() {
|
|
37
|
+
let entries;
|
|
38
|
+
try {
|
|
39
|
+
entries = await fs.readdir(dir);
|
|
40
|
+
}
|
|
41
|
+
catch {
|
|
42
|
+
return [];
|
|
43
|
+
}
|
|
44
|
+
const sessions = [];
|
|
45
|
+
for (const entry of entries) {
|
|
46
|
+
if (!entry.endsWith('.json'))
|
|
47
|
+
continue;
|
|
48
|
+
try {
|
|
49
|
+
const content = await fs.readFile(path.join(dir, entry), 'utf-8');
|
|
50
|
+
const parsed = JSON.parse(content);
|
|
51
|
+
sessions.push({ id: parsed.id, updatedAt: parsed.updatedAt });
|
|
52
|
+
}
|
|
53
|
+
catch {
|
|
54
|
+
// skip corrupt files
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
sessions.sort((a, b) => b.updatedAt - a.updatedAt);
|
|
58
|
+
return sessions;
|
|
59
|
+
},
|
|
60
|
+
};
|
|
61
|
+
}
|
|
62
|
+
//# sourceMappingURL=session.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.js","sourceRoot":"","sources":["../../src/session/session.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAEhD,MAAM,UAAU,oBAAoB,CAAC,GAAW;IAC9C,OAAO;QACL,KAAK,CAAC,IAAI,CAAC,QAAyB;YAClC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzC,MAAM,MAAM,GAAG,QAAQ,CAAC,EAAE,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YAC3D,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;YAClD,MAAM,EAAE,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;QAC3E,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,EAAU;YACnB,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,iBAAiB,EAAE,GAAG,CAAC,CAAC;YAClD,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,MAAM,OAAO,CAAC,CAAC;YAClD,IAAI,OAAe,CAAC;YACpB,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACjD,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBAClB,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;oBAAE,OAAO,IAAI,CAAC;gBACvC,MAAM,GAAG,CAAC;YACZ,CAAC;YAED,IAAI,MAAW,CAAC;YAChB,IAAI,CAAC;gBACH,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;YAC/B,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,IAAI,gBAAgB,CAAC,yBAAyB,EAAE,OAAO,CAAC,CAAC;YACjE,CAAC;YAED,IAAI,MAAM,CAAC,OAAO,KAAK,CAAC,EAAE,CAAC;gBACzB,MAAM,IAAI,gBAAgB,CAAC,gCAAgC,MAAM,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/E,CAAC;YAED,OAAO,MAAyB,CAAC;QACnC,CAAC;QAED,KAAK,CAAC,IAAI;YACR,IAAI,OAAiB,CAAC;YACtB,IAAI,CAAC;gBACH,OAAO,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;YAClC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,CAAC;YACZ,CAAC;YAED,MAAM,QAAQ,GAAwC,EAAE,CAAC;YACzD,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC;oBAAE,SAAS;gBACvC,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;oBAClE,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;oBACnC,QAAQ,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,EAAE,EAAE,SAAS,EAAE,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;gBAChE,CAAC;gBAAC,MAAM,CAAC;oBACP,qBAAqB;gBACvB,CAAC;YACH,CAAC;YAED,QAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC,CAAC;YACnD,OAAO,QAAQ,CAAC;QAClB,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.test.d.ts","sourceRoot":"","sources":["../../src/session/session.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,84 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import * as fs from 'node:fs/promises';
|
|
3
|
+
import * as os from 'node:os';
|
|
4
|
+
import * as path from 'node:path';
|
|
5
|
+
import { createSessionManager } from './session.js';
|
|
6
|
+
import { SessionLoadError } from '../errors.js';
|
|
7
|
+
function makeSnapshot(overrides = {}) {
|
|
8
|
+
return {
|
|
9
|
+
version: 1,
|
|
10
|
+
id: 'session-1',
|
|
11
|
+
messages: [],
|
|
12
|
+
modelId: 'test-model',
|
|
13
|
+
providerName: 'test-provider',
|
|
14
|
+
systemPromptHash: 'abc123',
|
|
15
|
+
memoryRefs: [],
|
|
16
|
+
createdAt: 1000,
|
|
17
|
+
updatedAt: 2000,
|
|
18
|
+
...overrides,
|
|
19
|
+
};
|
|
20
|
+
}
|
|
21
|
+
describe('SessionManager', () => {
|
|
22
|
+
let tmpDir;
|
|
23
|
+
beforeEach(async () => {
|
|
24
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'session-test-'));
|
|
25
|
+
});
|
|
26
|
+
afterEach(async () => {
|
|
27
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
28
|
+
});
|
|
29
|
+
describe('save and load roundtrip', () => {
|
|
30
|
+
it('saves and loads a session snapshot', async () => {
|
|
31
|
+
const manager = createSessionManager(tmpDir);
|
|
32
|
+
const snapshot = makeSnapshot();
|
|
33
|
+
await manager.save(snapshot);
|
|
34
|
+
const loaded = await manager.load('session-1');
|
|
35
|
+
expect(loaded).toEqual(snapshot);
|
|
36
|
+
});
|
|
37
|
+
it('creates directory if needed', async () => {
|
|
38
|
+
const subdir = path.join(tmpDir, 'nested', 'sessions');
|
|
39
|
+
const manager = createSessionManager(subdir);
|
|
40
|
+
const snapshot = makeSnapshot();
|
|
41
|
+
await manager.save(snapshot);
|
|
42
|
+
const loaded = await manager.load('session-1');
|
|
43
|
+
expect(loaded).toEqual(snapshot);
|
|
44
|
+
});
|
|
45
|
+
});
|
|
46
|
+
describe('load', () => {
|
|
47
|
+
it('returns null for missing session', async () => {
|
|
48
|
+
const manager = createSessionManager(tmpDir);
|
|
49
|
+
const result = await manager.load('nonexistent');
|
|
50
|
+
expect(result).toBeNull();
|
|
51
|
+
});
|
|
52
|
+
it('throws SessionLoadError for corrupt JSON', async () => {
|
|
53
|
+
await fs.writeFile(path.join(tmpDir, 'bad.json'), 'not json', 'utf-8');
|
|
54
|
+
const manager = createSessionManager(tmpDir);
|
|
55
|
+
await expect(manager.load('bad')).rejects.toThrow(SessionLoadError);
|
|
56
|
+
});
|
|
57
|
+
it('throws SessionLoadError for wrong version', async () => {
|
|
58
|
+
const bad = { ...makeSnapshot(), version: 99 };
|
|
59
|
+
await fs.writeFile(path.join(tmpDir, 'v99.json'), JSON.stringify(bad), 'utf-8');
|
|
60
|
+
const manager = createSessionManager(tmpDir);
|
|
61
|
+
await expect(manager.load('v99')).rejects.toThrow(SessionLoadError);
|
|
62
|
+
});
|
|
63
|
+
});
|
|
64
|
+
describe('list', () => {
|
|
65
|
+
it('lists sessions sorted by updatedAt desc', async () => {
|
|
66
|
+
const manager = createSessionManager(tmpDir);
|
|
67
|
+
await manager.save(makeSnapshot({ id: 'old', updatedAt: 1000 }));
|
|
68
|
+
await manager.save(makeSnapshot({ id: 'new', updatedAt: 3000 }));
|
|
69
|
+
await manager.save(makeSnapshot({ id: 'mid', updatedAt: 2000 }));
|
|
70
|
+
const list = await manager.list();
|
|
71
|
+
expect(list).toEqual([
|
|
72
|
+
{ id: 'new', updatedAt: 3000 },
|
|
73
|
+
{ id: 'mid', updatedAt: 2000 },
|
|
74
|
+
{ id: 'old', updatedAt: 1000 },
|
|
75
|
+
]);
|
|
76
|
+
});
|
|
77
|
+
it('returns empty array for non-existent directory', async () => {
|
|
78
|
+
const manager = createSessionManager(path.join(tmpDir, 'nope'));
|
|
79
|
+
const list = await manager.list();
|
|
80
|
+
expect(list).toEqual([]);
|
|
81
|
+
});
|
|
82
|
+
});
|
|
83
|
+
});
|
|
84
|
+
//# sourceMappingURL=session.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"session.test.js","sourceRoot":"","sources":["../../src/session/session.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,QAAQ,CAAC;AACrE,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,EAAE,MAAM,SAAS,CAAC;AAC9B,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAClC,OAAO,EAAE,oBAAoB,EAAE,MAAM,cAAc,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAGhD,SAAS,YAAY,CAAC,YAAsC,EAAE;IAC5D,OAAO;QACL,OAAO,EAAE,CAAC;QACV,EAAE,EAAE,WAAW;QACf,QAAQ,EAAE,EAAE;QACZ,OAAO,EAAE,YAAY;QACrB,YAAY,EAAE,eAAe;QAC7B,gBAAgB,EAAE,QAAQ;QAC1B,UAAU,EAAE,EAAE;QACd,SAAS,EAAE,IAAI;QACf,SAAS,EAAE,IAAI;QACf,GAAG,SAAS;KACb,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,MAAc,CAAC;IAEnB,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,eAAe,CAAC,CAAC,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,KAAK,IAAI,EAAE;QACnB,MAAM,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,YAAY,EAAE,CAAC;YAChC,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6BAA6B,EAAE,KAAK,IAAI,EAAE;YAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC,CAAC;YACvD,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,YAAY,EAAE,CAAC;YAChC,MAAM,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7B,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACnC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACjD,MAAM,CAAC,MAAM,CAAC,CAAC,QAAQ,EAAE,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,UAAU,EAAE,OAAO,CAAC,CAAC;YACvE,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,KAAK,IAAI,EAAE;YACzD,MAAM,GAAG,GAAG,EAAE,GAAG,YAAY,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;YAC/C,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,UAAU,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,OAAO,CAAC,CAAC;YAChF,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,MAAM,EAAE,GAAG,EAAE;QACpB,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,OAAO,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAC7C,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACjE,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YACjE,MAAM,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAEjE,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC;gBACnB,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE;gBAC9B,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE;gBAC9B,EAAE,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE;aAC/B,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,OAAO,GAAG,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;YAChE,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,IAAI,EAAE,CAAC;YAClC,MAAM,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC3B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
package/dist/skills.d.ts
ADDED
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { HookHandler, McpServerConfig, PermissionRule, ResolvedSkill, SdkTool } from './types.js';
|
|
2
|
+
import type { CodingAgentConfig } from './factory.js';
|
|
3
|
+
export interface ComposeCodingAgentConfigOptions {
|
|
4
|
+
defaultTools?: SdkTool<any, any>[];
|
|
5
|
+
}
|
|
6
|
+
export interface ComposedCodingAgentConfig extends CodingAgentConfig {
|
|
7
|
+
hooks: HookHandler[];
|
|
8
|
+
mcpServers: McpServerConfig[];
|
|
9
|
+
permissionRules: PermissionRule[];
|
|
10
|
+
skills: ResolvedSkill[];
|
|
11
|
+
tools: SdkTool<any, any>[];
|
|
12
|
+
}
|
|
13
|
+
export declare function composeCodingAgentConfig(config: CodingAgentConfig, options?: ComposeCodingAgentConfigOptions): ComposedCodingAgentConfig;
|
|
14
|
+
//# sourceMappingURL=skills.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skills.d.ts","sourceRoot":"","sources":["../src/skills.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,WAAW,EAAE,eAAe,EAAE,cAAc,EAAE,aAAa,EAAE,OAAO,EAAE,MAAM,YAAY,CAAC;AACvG,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEtD,MAAM,WAAW,+BAA+B;IAC9C,YAAY,CAAC,EAAE,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;CACpC;AAED,MAAM,WAAW,yBAA0B,SAAQ,iBAAiB;IAClE,KAAK,EAAE,WAAW,EAAE,CAAC;IACrB,UAAU,EAAE,eAAe,EAAE,CAAC;IAC9B,eAAe,EAAE,cAAc,EAAE,CAAC;IAClC,MAAM,EAAE,aAAa,EAAE,CAAC;IACxB,KAAK,EAAE,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC;CAC5B;AAED,wBAAgB,wBAAwB,CACtC,MAAM,EAAE,iBAAiB,EACzB,OAAO,GAAE,+BAAoC,GAC5C,yBAAyB,CAmB3B"}
|
package/dist/skills.js
ADDED
|
@@ -0,0 +1,31 @@
|
|
|
1
|
+
export function composeCodingAgentConfig(config, options = {}) {
|
|
2
|
+
const skills = config.skills ?? [];
|
|
3
|
+
const skillTools = skills.flatMap((skill) => skill.tools ?? []);
|
|
4
|
+
const skillHooks = skills.flatMap((skill) => skill.hooks ?? []);
|
|
5
|
+
const skillMcpServers = skills.flatMap((skill) => skill.mcpServers ?? []);
|
|
6
|
+
const skillPermissionRules = skills.flatMap((skill) => skill.permissionRules ?? []);
|
|
7
|
+
const tools = config.tools
|
|
8
|
+
? mergeByKey([...skillTools, ...config.tools], (tool) => tool.name)
|
|
9
|
+
: mergeByKey([...(options.defaultTools ?? []), ...skillTools], (tool) => tool.name);
|
|
10
|
+
return {
|
|
11
|
+
...config,
|
|
12
|
+
hooks: [...skillHooks, ...(config.hooks ?? [])],
|
|
13
|
+
mcpServers: mergeByKey([...skillMcpServers, ...(config.mcpServers ?? [])], (server) => server.name),
|
|
14
|
+
permissionRules: [...skillPermissionRules, ...(config.permissionRules ?? [])],
|
|
15
|
+
skills,
|
|
16
|
+
tools,
|
|
17
|
+
};
|
|
18
|
+
}
|
|
19
|
+
function mergeByKey(items, getKey) {
|
|
20
|
+
const orderedKeys = [];
|
|
21
|
+
const merged = new Map();
|
|
22
|
+
for (const item of items) {
|
|
23
|
+
const key = getKey(item);
|
|
24
|
+
if (!merged.has(key)) {
|
|
25
|
+
orderedKeys.push(key);
|
|
26
|
+
}
|
|
27
|
+
merged.set(key, item);
|
|
28
|
+
}
|
|
29
|
+
return orderedKeys.map((key) => merged.get(key));
|
|
30
|
+
}
|
|
31
|
+
//# sourceMappingURL=skills.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skills.js","sourceRoot":"","sources":["../src/skills.ts"],"names":[],"mappings":"AAeA,MAAM,UAAU,wBAAwB,CACtC,MAAyB,EACzB,UAA2C,EAAE;IAE7C,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC;IACnC,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAChE,MAAM,UAAU,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;IAChE,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC;IAC1E,MAAM,oBAAoB,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;IAEpF,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK;QACxB,CAAC,CAAC,UAAU,CAAC,CAAC,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;QACnE,CAAC,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,YAAY,IAAI,EAAE,CAAC,EAAE,GAAG,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAEtF,OAAO;QACL,GAAG,MAAM;QACT,KAAK,EAAE,CAAC,GAAG,UAAU,EAAE,GAAG,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;QAC/C,UAAU,EAAE,UAAU,CAAC,CAAC,GAAG,eAAe,EAAE,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC;QACnG,eAAe,EAAE,CAAC,GAAG,oBAAoB,EAAE,GAAG,CAAC,MAAM,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC;QAC7E,MAAM;QACN,KAAK;KACN,CAAC;AACJ,CAAC;AAED,SAAS,UAAU,CAAI,KAAU,EAAE,MAA2B;IAC5D,MAAM,WAAW,GAAa,EAAE,CAAC;IACjC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAa,CAAC;IAEpC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QACzB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YACrB,WAAW,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACxB,CAAC;QACD,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;IACxB,CAAC;IAED,OAAO,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAE,CAAC,CAAC;AACpD,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skills.test.d.ts","sourceRoot":"","sources":["../src/skills.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { describe, expect, it } from 'vitest';
|
|
2
|
+
import { Type } from '@sinclair/typebox';
|
|
3
|
+
import { composeCodingAgentConfig } from './skills.js';
|
|
4
|
+
import { getModel } from '@mariozechner/pi-ai';
|
|
5
|
+
function createTool(name, description) {
|
|
6
|
+
return {
|
|
7
|
+
name,
|
|
8
|
+
label: name,
|
|
9
|
+
description,
|
|
10
|
+
capabilities: [],
|
|
11
|
+
parameters: Type.Object({}),
|
|
12
|
+
async execute() {
|
|
13
|
+
return { content: [{ type: 'text', text: name }], details: { name } };
|
|
14
|
+
},
|
|
15
|
+
};
|
|
16
|
+
}
|
|
17
|
+
describe('composeCodingAgentConfig', () => {
|
|
18
|
+
it('merges skill prompt, tools, hooks, MCP servers, and permission rules', () => {
|
|
19
|
+
const hook = {
|
|
20
|
+
event: 'SessionStart',
|
|
21
|
+
async handler() { },
|
|
22
|
+
};
|
|
23
|
+
const rule = {
|
|
24
|
+
target: { type: 'tool', name: 'Read' },
|
|
25
|
+
behavior: 'allow',
|
|
26
|
+
source: 'project',
|
|
27
|
+
};
|
|
28
|
+
const skill = {
|
|
29
|
+
id: 'typescript',
|
|
30
|
+
promptSections: ['Prefer TypeScript.'],
|
|
31
|
+
tools: [createTool('SkillTool', 'Provided by skill')],
|
|
32
|
+
hooks: [hook],
|
|
33
|
+
mcpServers: [{ name: 'skill-mcp', transport: 'sse', url: 'https://example.invalid/sse' }],
|
|
34
|
+
permissionRules: [rule],
|
|
35
|
+
};
|
|
36
|
+
const composed = composeCodingAgentConfig({
|
|
37
|
+
model: getModel('openai', 'gpt-4o-mini'),
|
|
38
|
+
skills: [skill],
|
|
39
|
+
}, {
|
|
40
|
+
defaultTools: [createTool('Read', 'Default read tool')],
|
|
41
|
+
});
|
|
42
|
+
expect(composed.skills).toEqual([skill]);
|
|
43
|
+
expect(composed.tools.map((tool) => tool.name)).toEqual(['Read', 'SkillTool']);
|
|
44
|
+
expect(composed.hooks).toEqual([hook]);
|
|
45
|
+
expect(composed.mcpServers.map((server) => server.name)).toEqual(['skill-mcp']);
|
|
46
|
+
expect(composed.permissionRules).toEqual([rule]);
|
|
47
|
+
});
|
|
48
|
+
it('lets explicit tools override skill tools with the same name', () => {
|
|
49
|
+
const composed = composeCodingAgentConfig({
|
|
50
|
+
model: getModel('openai', 'gpt-4o-mini'),
|
|
51
|
+
skills: [{
|
|
52
|
+
id: 'formatter',
|
|
53
|
+
tools: [createTool('Format', 'Skill formatter')],
|
|
54
|
+
}],
|
|
55
|
+
tools: [createTool('Format', 'Explicit formatter')],
|
|
56
|
+
});
|
|
57
|
+
expect(composed.tools).toHaveLength(1);
|
|
58
|
+
expect(composed.tools[0]?.description).toBe('Explicit formatter');
|
|
59
|
+
});
|
|
60
|
+
});
|
|
61
|
+
//# sourceMappingURL=skills.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"skills.test.js","sourceRoot":"","sources":["../src/skills.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAEzC,OAAO,EAAE,wBAAwB,EAAE,MAAM,aAAa,CAAC;AACvD,OAAO,EAAE,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAE/C,SAAS,UAAU,CAAC,IAAY,EAAE,WAAmB;IACnD,OAAO;QACL,IAAI;QACJ,KAAK,EAAE,IAAI;QACX,WAAW;QACX,YAAY,EAAE,EAAE;QAChB,UAAU,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;QAC3B,KAAK,CAAC,OAAO;YACX,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,EAAE,CAAC;QACxE,CAAC;KACF,CAAC;AACJ,CAAC;AAED,QAAQ,CAAC,0BAA0B,EAAE,GAAG,EAAE;IACxC,EAAE,CAAC,sEAAsE,EAAE,GAAG,EAAE;QAC9E,MAAM,IAAI,GAAgB;YACxB,KAAK,EAAE,cAAc;YACrB,KAAK,CAAC,OAAO,KAAI,CAAC;SACnB,CAAC;QACF,MAAM,IAAI,GAAmB;YAC3B,MAAM,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE;YACtC,QAAQ,EAAE,OAAO;YACjB,MAAM,EAAE,SAAS;SAClB,CAAC;QACF,MAAM,KAAK,GAAkB;YAC3B,EAAE,EAAE,YAAY;YAChB,cAAc,EAAE,CAAC,oBAAoB,CAAC;YACtC,KAAK,EAAE,CAAC,UAAU,CAAC,WAAW,EAAE,mBAAmB,CAAC,CAAC;YACrD,KAAK,EAAE,CAAC,IAAI,CAAC;YACb,UAAU,EAAE,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,SAAS,EAAE,KAAK,EAAE,GAAG,EAAE,6BAA6B,EAAE,CAAC;YACzF,eAAe,EAAE,CAAC,IAAI,CAAC;SACxB,CAAC;QAEF,MAAM,QAAQ,GAAG,wBAAwB,CACvC;YACE,KAAK,EAAE,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC;YACxC,MAAM,EAAE,CAAC,KAAK,CAAC;SAChB,EACD;YACE,YAAY,EAAE,CAAC,UAAU,CAAC,MAAM,EAAE,mBAAmB,CAAC,CAAC;SACxD,CACF,CAAC;QAEF,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC;QAC/E,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC;QAChF,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6DAA6D,EAAE,GAAG,EAAE;QACrE,MAAM,QAAQ,GAAG,wBAAwB,CAAC;YACxC,KAAK,EAAE,QAAQ,CAAC,QAAQ,EAAE,aAAa,CAAC;YACxC,MAAM,EAAE,CAAC;oBACP,EAAE,EAAE,WAAW;oBACf,KAAK,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,iBAAiB,CAAC,CAAC;iBACjD,CAAC;YACF,KAAK,EAAE,CAAC,UAAU,CAAC,QAAQ,EAAE,oBAAoB,CAAC,CAAC;SACpD,CAAC,CAAC;QAEH,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACvC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,WAAW,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IACpE,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { SdkTool } from '../types.js';
|
|
2
|
+
declare const AskUserParams: import("@sinclair/typebox").TObject<{
|
|
3
|
+
question: import("@sinclair/typebox").TString;
|
|
4
|
+
}>;
|
|
5
|
+
export interface AskUserToolOptions {
|
|
6
|
+
onQuestion?: (question: string) => Promise<string>;
|
|
7
|
+
}
|
|
8
|
+
export declare function createAskUserTool(options?: AskUserToolOptions): SdkTool<typeof AskUserParams>;
|
|
9
|
+
export {};
|
|
10
|
+
//# sourceMappingURL=ask-user.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ask-user.d.ts","sourceRoot":"","sources":["../../src/tools/ask-user.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAE3C,QAAA,MAAM,aAAa;;EAEjB,CAAC;AAEH,MAAM,WAAW,kBAAkB;IACjC,UAAU,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAC;CACpD;AAED,wBAAgB,iBAAiB,CAAC,OAAO,CAAC,EAAE,kBAAkB,GAAG,OAAO,CAAC,OAAO,aAAa,CAAC,CAqB7F"}
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
import { Type } from '@sinclair/typebox';
|
|
2
|
+
const AskUserParams = Type.Object({
|
|
3
|
+
question: Type.String(),
|
|
4
|
+
});
|
|
5
|
+
export function createAskUserTool(options) {
|
|
6
|
+
return {
|
|
7
|
+
name: 'AskUser',
|
|
8
|
+
label: 'Ask user a question',
|
|
9
|
+
description: 'Asks the user a question and returns their response.',
|
|
10
|
+
parameters: AskUserParams,
|
|
11
|
+
capabilities: [],
|
|
12
|
+
async execute(_toolCallId, params) {
|
|
13
|
+
if (options?.onQuestion) {
|
|
14
|
+
const answer = await options.onQuestion(params.question);
|
|
15
|
+
return {
|
|
16
|
+
content: [{ type: 'text', text: answer }],
|
|
17
|
+
details: { question: params.question },
|
|
18
|
+
};
|
|
19
|
+
}
|
|
20
|
+
return {
|
|
21
|
+
content: [{ type: 'text', text: 'User interaction is not available.' }],
|
|
22
|
+
details: { question: params.question, available: false },
|
|
23
|
+
};
|
|
24
|
+
},
|
|
25
|
+
};
|
|
26
|
+
}
|
|
27
|
+
//# sourceMappingURL=ask-user.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"ask-user.js","sourceRoot":"","sources":["../../src/tools/ask-user.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AAGzC,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC;IAChC,QAAQ,EAAE,IAAI,CAAC,MAAM,EAAE;CACxB,CAAC,CAAC;AAMH,MAAM,UAAU,iBAAiB,CAAC,OAA4B;IAC5D,OAAO;QACL,IAAI,EAAE,SAAS;QACf,KAAK,EAAE,qBAAqB;QAC5B,WAAW,EAAE,sDAAsD;QACnE,UAAU,EAAE,aAAa;QACzB,YAAY,EAAE,EAAE;QAChB,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM;YAC/B,IAAI,OAAO,EAAE,UAAU,EAAE,CAAC;gBACxB,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;gBACzD,OAAO;oBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;oBACzC,OAAO,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE;iBACvC,CAAC;YACJ,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,oCAAoC,EAAE,CAAC;gBACvE,OAAO,EAAE,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE;aACzD,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
import type { SdkTool, ToolOptions } from '../types.js';
|
|
2
|
+
declare const BashParams: import("@sinclair/typebox").TObject<{
|
|
3
|
+
command: import("@sinclair/typebox").TString;
|
|
4
|
+
timeout: import("@sinclair/typebox").TOptional<import("@sinclair/typebox").TNumber>;
|
|
5
|
+
}>;
|
|
6
|
+
export interface BashToolOptions extends ToolOptions {
|
|
7
|
+
timeout?: number;
|
|
8
|
+
}
|
|
9
|
+
export declare function createBashTool(options?: BashToolOptions): SdkTool<typeof BashParams>;
|
|
10
|
+
export {};
|
|
11
|
+
//# sourceMappingURL=bash.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bash.d.ts","sourceRoot":"","sources":["../../src/tools/bash.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAOxD,QAAA,MAAM,UAAU;;;EAGd,CAAC;AAEH,MAAM,WAAW,eAAgB,SAAQ,WAAW;IAClD,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,wBAAgB,cAAc,CAAC,OAAO,CAAC,EAAE,eAAe,GAAG,OAAO,CAAC,OAAO,UAAU,CAAC,CAqDpF"}
|
|
@@ -0,0 +1,54 @@
|
|
|
1
|
+
import { Type } from '@sinclair/typebox';
|
|
2
|
+
import { execFile } from 'node:child_process';
|
|
3
|
+
import { ToolExecutionError } from '../errors.js';
|
|
4
|
+
import { truncateOutput } from './util.js';
|
|
5
|
+
const MAX_OUTPUT = 100 * 1024; // 100KB
|
|
6
|
+
const DEFAULT_TIMEOUT = 120 * 1000; // 120 seconds
|
|
7
|
+
const BashParams = Type.Object({
|
|
8
|
+
command: Type.String(),
|
|
9
|
+
timeout: Type.Optional(Type.Number()),
|
|
10
|
+
});
|
|
11
|
+
export function createBashTool(options) {
|
|
12
|
+
const cwd = options?.cwd ?? process.cwd();
|
|
13
|
+
const defaultTimeout = options?.timeout ? options.timeout * 1000 : DEFAULT_TIMEOUT;
|
|
14
|
+
return {
|
|
15
|
+
name: 'Bash',
|
|
16
|
+
label: 'Execute bash command',
|
|
17
|
+
description: 'Executes a bash command and returns its output.',
|
|
18
|
+
parameters: BashParams,
|
|
19
|
+
capabilities: ['process:spawn', 'fs:write', 'network:egress'],
|
|
20
|
+
async execute(_toolCallId, params, signal) {
|
|
21
|
+
const timeoutMs = params.timeout ? params.timeout * 1000 : defaultTimeout;
|
|
22
|
+
return new Promise((resolve, reject) => {
|
|
23
|
+
const child = execFile('bash', ['-c', params.command], {
|
|
24
|
+
cwd,
|
|
25
|
+
timeout: timeoutMs,
|
|
26
|
+
maxBuffer: MAX_OUTPUT * 2, // allow some room, we truncate later
|
|
27
|
+
signal: signal,
|
|
28
|
+
}, (error, stdout, stderr) => {
|
|
29
|
+
if (error) {
|
|
30
|
+
const exitCode = error.status ?? error.code;
|
|
31
|
+
if (error.killed || error.message?.includes('ETIMEDOUT') || error.signal === 'SIGTERM') {
|
|
32
|
+
reject(new ToolExecutionError(`Command timed out after ${timeoutMs / 1000}s`));
|
|
33
|
+
return;
|
|
34
|
+
}
|
|
35
|
+
if (error.name === 'AbortError' || error.code === 'ABORT_ERR') {
|
|
36
|
+
reject(new ToolExecutionError('Command aborted'));
|
|
37
|
+
return;
|
|
38
|
+
}
|
|
39
|
+
const output = truncateOutput((stdout || '') + (stderr ? '\n' + stderr : ''), MAX_OUTPUT);
|
|
40
|
+
reject(new ToolExecutionError(`Command exited with code ${exitCode}\n${output}`));
|
|
41
|
+
return;
|
|
42
|
+
}
|
|
43
|
+
const combined = stdout + (stderr ? '\n' + stderr : '');
|
|
44
|
+
const output = truncateOutput(combined, MAX_OUTPUT);
|
|
45
|
+
resolve({
|
|
46
|
+
content: [{ type: 'text', text: output }],
|
|
47
|
+
details: { exitCode: 0 },
|
|
48
|
+
});
|
|
49
|
+
});
|
|
50
|
+
});
|
|
51
|
+
},
|
|
52
|
+
};
|
|
53
|
+
}
|
|
54
|
+
//# sourceMappingURL=bash.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bash.js","sourceRoot":"","sources":["../../src/tools/bash.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAE,MAAM,mBAAmB,CAAC;AACzC,OAAO,EAAE,QAAQ,EAAE,MAAM,oBAAoB,CAAC;AAE9C,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAE3C,MAAM,UAAU,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,QAAQ;AACvC,MAAM,eAAe,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC,cAAc;AAElD,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;IAC7B,OAAO,EAAE,IAAI,CAAC,MAAM,EAAE;IACtB,OAAO,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;CACtC,CAAC,CAAC;AAMH,MAAM,UAAU,cAAc,CAAC,OAAyB;IACtD,MAAM,GAAG,GAAG,OAAO,EAAE,GAAG,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;IAC1C,MAAM,cAAc,GAAG,OAAO,EAAE,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,eAAe,CAAC;IAEnF,OAAO;QACL,IAAI,EAAE,MAAM;QACZ,KAAK,EAAE,sBAAsB;QAC7B,WAAW,EAAE,iDAAiD;QAC9D,UAAU,EAAE,UAAU;QACtB,YAAY,EAAE,CAAC,eAAe,EAAE,UAAU,EAAE,gBAAgB,CAAC;QAC7D,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,MAAM,EAAE,MAAO;YACxC,MAAM,SAAS,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,CAAC,cAAc,CAAC;YAE1E,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACrC,MAAM,KAAK,GAAG,QAAQ,CACpB,MAAM,EACN,CAAC,IAAI,EAAE,MAAM,CAAC,OAAO,CAAC,EACtB;oBACE,GAAG;oBACH,OAAO,EAAE,SAAS;oBAClB,SAAS,EAAE,UAAU,GAAG,CAAC,EAAE,qCAAqC;oBAChE,MAAM,EAAE,MAAM;iBACf,EACD,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,EAAE;oBACxB,IAAI,KAAK,EAAE,CAAC;wBACV,MAAM,QAAQ,GAAI,KAAa,CAAC,MAAM,IAAK,KAAa,CAAC,IAAI,CAAC;wBAC9D,IAAK,KAAa,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,WAAW,CAAC,IAAK,KAAa,CAAC,MAAM,KAAK,SAAS,EAAE,CAAC;4BACzG,MAAM,CAAC,IAAI,kBAAkB,CAAC,2BAA2B,SAAS,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC;4BAC/E,OAAO;wBACT,CAAC;wBACD,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,IAAK,KAAa,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;4BACvE,MAAM,CAAC,IAAI,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,CAAC;4BAClD,OAAO;wBACT,CAAC;wBACD,MAAM,MAAM,GAAG,cAAc,CAAC,CAAC,MAAM,IAAI,EAAE,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;wBAC1F,MAAM,CAAC,IAAI,kBAAkB,CAC3B,4BAA4B,QAAQ,KAAK,MAAM,EAAE,CAClD,CAAC,CAAC;wBACH,OAAO;oBACT,CAAC;oBAED,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,GAAG,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;oBACxD,MAAM,MAAM,GAAG,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;oBAEpD,OAAO,CAAC;wBACN,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC;wBACzC,OAAO,EAAE,EAAE,QAAQ,EAAE,CAAC,EAAE;qBACzB,CAAC,CAAC;gBACL,CAAC,CACF,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"bash.test.d.ts","sourceRoot":"","sources":["../../src/tools/bash.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,63 @@
|
|
|
1
|
+
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
|
|
2
|
+
import { createBashTool } from './bash.js';
|
|
3
|
+
import fs from 'node:fs';
|
|
4
|
+
import path from 'node:path';
|
|
5
|
+
import os from 'node:os';
|
|
6
|
+
describe('createBashTool', () => {
|
|
7
|
+
let tmpDir;
|
|
8
|
+
beforeEach(() => {
|
|
9
|
+
tmpDir = fs.mkdtempSync(path.join(os.tmpdir(), 'bash-test-'));
|
|
10
|
+
});
|
|
11
|
+
afterEach(() => {
|
|
12
|
+
fs.rmSync(tmpDir, { recursive: true, force: true });
|
|
13
|
+
});
|
|
14
|
+
it('returns a tool with correct name and capabilities', () => {
|
|
15
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
16
|
+
expect(tool.name).toBe('Bash');
|
|
17
|
+
expect(tool.capabilities).toEqual(['process:spawn', 'fs:write', 'network:egress']);
|
|
18
|
+
});
|
|
19
|
+
it('executes a command and returns stdout', async () => {
|
|
20
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
21
|
+
const result = await tool.execute('call1', { command: 'echo hello' });
|
|
22
|
+
const text = result.content[0].text;
|
|
23
|
+
expect(text).toContain('hello');
|
|
24
|
+
});
|
|
25
|
+
it('throws on non-zero exit code', async () => {
|
|
26
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
27
|
+
await expect(tool.execute('call1', { command: 'exit 1' }))
|
|
28
|
+
.rejects.toThrow();
|
|
29
|
+
});
|
|
30
|
+
it('runs in the specified cwd', async () => {
|
|
31
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
32
|
+
const result = await tool.execute('call1', { command: 'pwd' });
|
|
33
|
+
const text = result.content[0].text;
|
|
34
|
+
expect(text.trim()).toBe(fs.realpathSync(tmpDir));
|
|
35
|
+
});
|
|
36
|
+
it('truncates output over 100KB', async () => {
|
|
37
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
38
|
+
// Generate output > 100KB
|
|
39
|
+
const result = await tool.execute('call1', { command: 'python3 -c "print(\'x\' * 200000)"' });
|
|
40
|
+
const text = result.content[0].text;
|
|
41
|
+
expect(text).toContain('[truncated]');
|
|
42
|
+
});
|
|
43
|
+
it('times out with custom timeout', async () => {
|
|
44
|
+
const tool = createBashTool({ cwd: tmpDir, timeout: 1 }); // 1 second default
|
|
45
|
+
await expect(tool.execute('call1', { command: 'sleep 10', timeout: 1 }))
|
|
46
|
+
.rejects.toThrow();
|
|
47
|
+
});
|
|
48
|
+
it('respects AbortSignal', async () => {
|
|
49
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
50
|
+
const controller = new AbortController();
|
|
51
|
+
setTimeout(() => controller.abort(), 100);
|
|
52
|
+
await expect(tool.execute('call1', { command: 'sleep 10' }, controller.signal))
|
|
53
|
+
.rejects.toThrow();
|
|
54
|
+
});
|
|
55
|
+
it('includes stderr in output on success', async () => {
|
|
56
|
+
const tool = createBashTool({ cwd: tmpDir });
|
|
57
|
+
const result = await tool.execute('call1', { command: 'echo out; echo err >&2' });
|
|
58
|
+
const text = result.content[0].text;
|
|
59
|
+
expect(text).toContain('out');
|
|
60
|
+
// stderr may or may not be included depending on impl; just check no throw
|
|
61
|
+
});
|
|
62
|
+
});
|
|
63
|
+
//# sourceMappingURL=bash.test.js.map
|