@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 @@
|
|
|
1
|
+
{"version":3,"file":"client.js","sourceRoot":"","sources":["../../src/mcp/client.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,2CAA2C,CAAC;AACnE,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,yCAAyC,CAAC;AAE7E,OAAO,EAAE,kBAAkB,EAAE,MAAM,cAAc,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAOzC,MAAM,UAAU,gBAAgB;IAC9B,MAAM,WAAW,GAAG,IAAI,GAAG,EAA8B,CAAC;IAE1D,OAAO;QACL,KAAK,CAAC,OAAO,CAAC,MAAuB;YACnC,IAAI,CAAC;gBACH,IAAI,SAAc,CAAC;gBAEnB,IAAI,MAAM,CAAC,SAAS,KAAK,OAAO,EAAE,CAAC;oBACjC,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;wBACpB,MAAM,IAAI,kBAAkB,CAAC,kDAAkD,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBAChG,CAAC;oBACD,SAAS,GAAG,IAAI,oBAAoB,CAAC;wBACnC,OAAO,EAAE,MAAM,CAAC,OAAO;wBACvB,IAAI,EAAE,MAAM,CAAC,IAAI;wBACjB,GAAG,EAAE,MAAM,CAAC,GAAyC;qBACtD,CAAC,CAAC;gBACL,CAAC;qBAAM,IAAI,MAAM,CAAC,SAAS,KAAK,KAAK,EAAE,CAAC;oBACtC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;wBAChB,MAAM,IAAI,kBAAkB,CAAC,4CAA4C,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;oBAC1F,CAAC;oBACD,SAAS,GAAG,IAAI,kBAAkB,CAAC,IAAI,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAC1D,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,kBAAkB,CAAC,0BAA0B,MAAM,CAAC,SAAS,EAAE,CAAC,CAAC;gBAC7E,CAAC;gBAED,MAAM,MAAM,GAAG,IAAI,MAAM,CAAC,EAAE,IAAI,EAAE,8BAA8B,EAAE,OAAO,EAAE,OAAO,EAAE,CAAC,CAAC;gBACtF,MAAM,MAAM,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;gBAEhC,MAAM,EAAE,KAAK,EAAE,QAAQ,EAAE,GAAG,MAAM,MAAM,CAAC,SAAS,EAAE,CAAC;gBAErD,MAAM,YAAY,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAC5C,WAAW,CAAC,MAAM,CAAC,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,EAAE;oBACrD,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAChE,OAAO,MAA6D,CAAC;gBACvE,CAAC,CAAC,CACH,CAAC;gBAEF,MAAM,UAAU,GAAuB;oBACrC,IAAI,EAAE,MAAM,CAAC,IAAI;oBACjB,MAAM;oBACN,MAAM;oBACN,KAAK,EAAE,YAAY;oBACnB,KAAK,CAAC,KAAK;wBACT,MAAM,MAAM,CAAC,KAAK,EAAE,CAAC;oBACvB,CAAC;iBACF,CAAC;gBAEF,WAAW,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,UAAU,CAAC,CAAC;gBACzC,OAAO,UAAU,CAAC;YACpB,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,GAAG,YAAY,kBAAkB;oBAAE,MAAM,GAAG,CAAC;gBACjD,MAAM,IAAI,kBAAkB,CAAC,mCAAmC,MAAM,CAAC,IAAI,KAAM,GAAa,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5G,CAAC;QACH,CAAC;QAED,KAAK,CAAC,UAAU,CAAC,IAAY;YAC3B,MAAM,IAAI,GAAG,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACnC,IAAI,IAAI,EAAE,CAAC;gBACT,MAAM,IAAI,CAAC,KAAK,EAAE,CAAC;gBACnB,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,CAAC;QACH,CAAC;QAED,QAAQ;YACN,MAAM,QAAQ,GAAc,EAAE,CAAC;YAC/B,KAAK,MAAM,IAAI,IAAI,WAAW,CAAC,MAAM,EAAE,EAAE,CAAC;gBACxC,QAAQ,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC;YAC/B,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,cAAc;YACZ,OAAO,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE,CAAC,CAAC;QAC1C,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema-convert.d.ts","sourceRoot":"","sources":["../../src/mcp/schema-convert.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,aAAa,CAAC;AAE1D,wBAAgB,mBAAmB,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,sBAAsB,CAQ3F"}
|
|
@@ -0,0 +1,73 @@
|
|
|
1
|
+
import { Type } from '@sinclair/typebox';
|
|
2
|
+
export function jsonSchemaToTypeBox(jsonSchema) {
|
|
3
|
+
const warnings = [];
|
|
4
|
+
const schema = convert(jsonSchema, warnings);
|
|
5
|
+
return {
|
|
6
|
+
schema,
|
|
7
|
+
isExact: warnings.length === 0,
|
|
8
|
+
warnings,
|
|
9
|
+
};
|
|
10
|
+
}
|
|
11
|
+
function convert(schema, warnings) {
|
|
12
|
+
if (!schema || typeof schema !== 'object') {
|
|
13
|
+
warnings.push('Missing or invalid schema');
|
|
14
|
+
return Type.Any();
|
|
15
|
+
}
|
|
16
|
+
// Check for unsupported combinators
|
|
17
|
+
for (const key of ['oneOf', 'allOf', 'anyOf', '$ref']) {
|
|
18
|
+
if (key in schema) {
|
|
19
|
+
warnings.push(`Unsupported keyword: ${key}`);
|
|
20
|
+
return withDescription(Type.Any(), schema.description);
|
|
21
|
+
}
|
|
22
|
+
}
|
|
23
|
+
const type = schema.type;
|
|
24
|
+
// Implicit object: no type but has properties
|
|
25
|
+
if (!type && schema.properties) {
|
|
26
|
+
return convertObject(schema, warnings);
|
|
27
|
+
}
|
|
28
|
+
if (!type) {
|
|
29
|
+
warnings.push('Unknown or missing type');
|
|
30
|
+
return withDescription(Type.Any(), schema.description);
|
|
31
|
+
}
|
|
32
|
+
switch (type) {
|
|
33
|
+
case 'string':
|
|
34
|
+
if (schema.enum && Array.isArray(schema.enum)) {
|
|
35
|
+
return withDescription(Type.Union(schema.enum.map((v) => Type.Literal(v))), schema.description);
|
|
36
|
+
}
|
|
37
|
+
return withDescription(Type.String(), schema.description);
|
|
38
|
+
case 'number':
|
|
39
|
+
case 'integer':
|
|
40
|
+
return withDescription(Type.Number(), schema.description);
|
|
41
|
+
case 'boolean':
|
|
42
|
+
return withDescription(Type.Boolean(), schema.description);
|
|
43
|
+
case 'null':
|
|
44
|
+
return withDescription(Type.Null(), schema.description);
|
|
45
|
+
case 'array':
|
|
46
|
+
if (schema.items) {
|
|
47
|
+
return withDescription(Type.Array(convert(schema.items, warnings)), schema.description);
|
|
48
|
+
}
|
|
49
|
+
return withDescription(Type.Array(Type.Any()), schema.description);
|
|
50
|
+
case 'object':
|
|
51
|
+
return convertObject(schema, warnings);
|
|
52
|
+
default:
|
|
53
|
+
warnings.push(`Unknown type: ${type}`);
|
|
54
|
+
return withDescription(Type.Any(), schema.description);
|
|
55
|
+
}
|
|
56
|
+
}
|
|
57
|
+
function convertObject(schema, warnings) {
|
|
58
|
+
const properties = schema.properties ?? {};
|
|
59
|
+
const required = new Set(schema.required ?? []);
|
|
60
|
+
const props = {};
|
|
61
|
+
for (const [key, value] of Object.entries(properties)) {
|
|
62
|
+
const converted = convert(value, warnings);
|
|
63
|
+
props[key] = required.has(key) ? converted : Type.Optional(converted);
|
|
64
|
+
}
|
|
65
|
+
return withDescription(Type.Object(props), schema.description);
|
|
66
|
+
}
|
|
67
|
+
function withDescription(schema, description) {
|
|
68
|
+
if (description) {
|
|
69
|
+
return { ...schema, description };
|
|
70
|
+
}
|
|
71
|
+
return schema;
|
|
72
|
+
}
|
|
73
|
+
//# sourceMappingURL=schema-convert.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema-convert.js","sourceRoot":"","sources":["../../src/mcp/schema-convert.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,IAAI,EAAgB,MAAM,mBAAmB,CAAC;AAGvD,MAAM,UAAU,mBAAmB,CAAC,UAA+B;IACjE,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC;IAC7C,OAAO;QACL,MAAM;QACN,OAAO,EAAE,QAAQ,CAAC,MAAM,KAAK,CAAC;QAC9B,QAAQ;KACT,CAAC;AACJ,CAAC;AAED,SAAS,OAAO,CAAC,MAA2B,EAAE,QAAkB;IAC9D,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,EAAE,CAAC;QAC1C,QAAQ,CAAC,IAAI,CAAC,2BAA2B,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC,GAAG,EAAE,CAAC;IACpB,CAAC;IAED,oCAAoC;IACpC,KAAK,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,EAAE,CAAC;QACtD,IAAI,GAAG,IAAI,MAAM,EAAE,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,wBAAwB,GAAG,EAAE,CAAC,CAAC;YAC7C,OAAO,eAAe,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QACzD,CAAC;IACH,CAAC;IAED,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;IAEzB,8CAA8C;IAC9C,IAAI,CAAC,IAAI,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QAC/B,OAAO,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;IACzC,CAAC;IAED,IAAI,CAAC,IAAI,EAAE,CAAC;QACV,QAAQ,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACzC,OAAO,eAAe,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;IACzD,CAAC;IAED,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,QAAQ;YACX,IAAI,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;gBAC9C,OAAO,eAAe,CACpB,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAS,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,EAC3D,MAAM,CAAC,WAAW,CACnB,CAAC;YACJ,CAAC;YACD,OAAO,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAE5D,KAAK,QAAQ,CAAC;QACd,KAAK,SAAS;YACZ,OAAO,eAAe,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAE5D,KAAK,SAAS;YACZ,OAAO,eAAe,CAAC,IAAI,CAAC,OAAO,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAE7D,KAAK,MAAM;YACT,OAAO,eAAe,CAAC,IAAI,CAAC,IAAI,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAE1D,KAAK,OAAO;YACV,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBACjB,OAAO,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;YAC1F,CAAC;YACD,OAAO,eAAe,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;QAErE,KAAK,QAAQ;YACX,OAAO,aAAa,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAEzC;YACE,QAAQ,CAAC,IAAI,CAAC,iBAAiB,IAAI,EAAE,CAAC,CAAC;YACvC,OAAO,eAAe,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;IAC3D,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,MAA2B,EAAE,QAAkB;IACpE,MAAM,UAAU,GAAG,MAAM,CAAC,UAAU,IAAI,EAAE,CAAC;IAC3C,MAAM,QAAQ,GAAG,IAAI,GAAG,CAAS,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC,CAAC;IAExD,MAAM,KAAK,GAA4B,EAAE,CAAC;IAC1C,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QACtD,MAAM,SAAS,GAAG,OAAO,CAAC,KAA4B,EAAE,QAAQ,CAAC,CAAC;QAClE,KAAK,CAAC,GAAG,CAAC,GAAG,QAAQ,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACxE,CAAC;IAED,OAAO,eAAe,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,WAAW,CAAC,CAAC;AACjE,CAAC;AAED,SAAS,eAAe,CAAoB,MAAS,EAAE,WAAoB;IACzE,IAAI,WAAW,EAAE,CAAC;QAChB,OAAO,EAAE,GAAG,MAAM,EAAE,WAAW,EAAO,CAAC;IACzC,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema-convert.test.d.ts","sourceRoot":"","sources":["../../src/mcp/schema-convert.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,95 @@
|
|
|
1
|
+
import { describe, it, expect } from 'vitest';
|
|
2
|
+
import { jsonSchemaToTypeBox } from './schema-convert.js';
|
|
3
|
+
import { Value } from '@sinclair/typebox/value';
|
|
4
|
+
describe('jsonSchemaToTypeBox', () => {
|
|
5
|
+
it('converts a simple object schema', () => {
|
|
6
|
+
const result = jsonSchemaToTypeBox({
|
|
7
|
+
type: 'object',
|
|
8
|
+
properties: {
|
|
9
|
+
name: { type: 'string', description: 'The name' },
|
|
10
|
+
age: { type: 'number' },
|
|
11
|
+
},
|
|
12
|
+
required: ['name'],
|
|
13
|
+
});
|
|
14
|
+
expect(result.isExact).toBe(true);
|
|
15
|
+
expect(result.warnings).toHaveLength(0);
|
|
16
|
+
// Validate data against the converted schema
|
|
17
|
+
expect(Value.Check(result.schema, { name: 'Alice', age: 30 })).toBe(true);
|
|
18
|
+
expect(Value.Check(result.schema, { name: 'Bob' })).toBe(true);
|
|
19
|
+
expect(Value.Check(result.schema, { age: 30 })).toBe(false); // name is required
|
|
20
|
+
});
|
|
21
|
+
it('converts boolean and array types', () => {
|
|
22
|
+
const result = jsonSchemaToTypeBox({
|
|
23
|
+
type: 'object',
|
|
24
|
+
properties: {
|
|
25
|
+
active: { type: 'boolean' },
|
|
26
|
+
tags: { type: 'array', items: { type: 'string' } },
|
|
27
|
+
},
|
|
28
|
+
});
|
|
29
|
+
expect(result.isExact).toBe(true);
|
|
30
|
+
expect(Value.Check(result.schema, { active: true, tags: ['a', 'b'] })).toBe(true);
|
|
31
|
+
expect(Value.Check(result.schema, { tags: [123] })).toBe(false);
|
|
32
|
+
});
|
|
33
|
+
it('converts string enums', () => {
|
|
34
|
+
const result = jsonSchemaToTypeBox({
|
|
35
|
+
type: 'object',
|
|
36
|
+
properties: {
|
|
37
|
+
status: { type: 'string', enum: ['open', 'closed'] },
|
|
38
|
+
},
|
|
39
|
+
required: ['status'],
|
|
40
|
+
});
|
|
41
|
+
expect(result.isExact).toBe(true);
|
|
42
|
+
expect(Value.Check(result.schema, { status: 'open' })).toBe(true);
|
|
43
|
+
expect(Value.Check(result.schema, { status: 'invalid' })).toBe(false);
|
|
44
|
+
});
|
|
45
|
+
it('falls back to Any for unsupported keywords', () => {
|
|
46
|
+
const result = jsonSchemaToTypeBox({
|
|
47
|
+
type: 'object',
|
|
48
|
+
properties: {
|
|
49
|
+
data: { oneOf: [{ type: 'string' }, { type: 'number' }] },
|
|
50
|
+
},
|
|
51
|
+
});
|
|
52
|
+
expect(result.isExact).toBe(false);
|
|
53
|
+
expect(result.warnings.length).toBeGreaterThan(0);
|
|
54
|
+
expect(result.warnings[0]).toContain('oneOf');
|
|
55
|
+
});
|
|
56
|
+
it('handles empty schema', () => {
|
|
57
|
+
const result = jsonSchemaToTypeBox({});
|
|
58
|
+
expect(result.isExact).toBe(false);
|
|
59
|
+
expect(result.warnings.length).toBeGreaterThan(0);
|
|
60
|
+
});
|
|
61
|
+
it('handles null type', () => {
|
|
62
|
+
const result = jsonSchemaToTypeBox({ type: 'null' });
|
|
63
|
+
expect(result.isExact).toBe(true);
|
|
64
|
+
expect(Value.Check(result.schema, null)).toBe(true);
|
|
65
|
+
});
|
|
66
|
+
it('handles implicit object (no type but has properties)', () => {
|
|
67
|
+
const result = jsonSchemaToTypeBox({
|
|
68
|
+
properties: {
|
|
69
|
+
foo: { type: 'string' },
|
|
70
|
+
},
|
|
71
|
+
required: ['foo'],
|
|
72
|
+
});
|
|
73
|
+
expect(result.isExact).toBe(true);
|
|
74
|
+
expect(Value.Check(result.schema, { foo: 'bar' })).toBe(true);
|
|
75
|
+
expect(Value.Check(result.schema, {})).toBe(false);
|
|
76
|
+
});
|
|
77
|
+
it('preserves description in schema options', () => {
|
|
78
|
+
const result = jsonSchemaToTypeBox({
|
|
79
|
+
type: 'string',
|
|
80
|
+
description: 'A greeting message',
|
|
81
|
+
});
|
|
82
|
+
expect(result.schema.description).toBe('A greeting message');
|
|
83
|
+
});
|
|
84
|
+
it('handles integer type', () => {
|
|
85
|
+
const result = jsonSchemaToTypeBox({ type: 'integer' });
|
|
86
|
+
expect(result.isExact).toBe(true);
|
|
87
|
+
expect(Value.Check(result.schema, 42)).toBe(true);
|
|
88
|
+
});
|
|
89
|
+
it('falls back to Any for $ref', () => {
|
|
90
|
+
const result = jsonSchemaToTypeBox({ $ref: '#/definitions/Foo' });
|
|
91
|
+
expect(result.isExact).toBe(false);
|
|
92
|
+
expect(result.warnings[0]).toContain('$ref');
|
|
93
|
+
});
|
|
94
|
+
});
|
|
95
|
+
//# sourceMappingURL=schema-convert.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"schema-convert.test.js","sourceRoot":"","sources":["../../src/mcp/schema-convert.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,MAAM,QAAQ,CAAC;AAC9C,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAC1D,OAAO,EAAE,KAAK,EAAE,MAAM,yBAAyB,CAAC;AAEhD,QAAQ,CAAC,qBAAqB,EAAE,GAAG,EAAE;IACnC,EAAE,CAAC,iCAAiC,EAAE,GAAG,EAAE;QACzC,MAAM,MAAM,GAAG,mBAAmB,CAAC;YACjC,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,UAAU,EAAE;gBACjD,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aACxB;YACD,QAAQ,EAAE,CAAC,MAAM,CAAC;SACnB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAExC,6CAA6C;QAC7C,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC1E,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,mBAAmB;IAClF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,GAAG,EAAE;QAC1C,MAAM,MAAM,GAAG,mBAAmB,CAAC;YACjC,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE;gBAC3B,IAAI,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;aACnD;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClF,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uBAAuB,EAAE,GAAG,EAAE;QAC/B,MAAM,MAAM,GAAG,mBAAmB,CAAC;YACjC,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,MAAM,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE;aACrD;YACD,QAAQ,EAAE,CAAC,QAAQ,CAAC;SACrB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxE,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4CAA4C,EAAE,GAAG,EAAE;QACpD,MAAM,MAAM,GAAG,mBAAmB,CAAC;YACjC,IAAI,EAAE,QAAQ;YACd,UAAU,EAAE;gBACV,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC,EAAE;aAC1D;SACF,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;QAClD,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,MAAM,GAAG,mBAAmB,CAAC,EAAE,CAAC,CAAC;QACvC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;QAC3B,MAAM,MAAM,GAAG,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QACrD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sDAAsD,EAAE,GAAG,EAAE;QAC9D,MAAM,MAAM,GAAG,mBAAmB,CAAC;YACjC,UAAU,EAAE;gBACV,GAAG,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;aACxB;YACD,QAAQ,EAAE,CAAC,KAAK,CAAC;SAClB,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,GAAG,EAAE,KAAK,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACrD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yCAAyC,EAAE,GAAG,EAAE;QACjD,MAAM,MAAM,GAAG,mBAAmB,CAAC;YACjC,IAAI,EAAE,QAAQ;YACd,WAAW,EAAE,oBAAoB;SAClC,CAAC,CAAC;QAEH,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,CAAC;IAC/D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sBAAsB,EAAE,GAAG,EAAE;QAC9B,MAAM,MAAM,GAAG,mBAAmB,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC,CAAC;QACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;QACpC,MAAM,MAAM,GAAG,mBAAmB,CAAC,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAC/C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
import type { SdkTool } from '../types.js';
|
|
2
|
+
export interface McpToolDefinition {
|
|
3
|
+
name: string;
|
|
4
|
+
description?: string;
|
|
5
|
+
inputSchema?: Record<string, any>;
|
|
6
|
+
}
|
|
7
|
+
export type CallToolFn = (name: string, args: Record<string, unknown>) => Promise<{
|
|
8
|
+
content: Array<{
|
|
9
|
+
type: string;
|
|
10
|
+
text?: string;
|
|
11
|
+
}>;
|
|
12
|
+
}>;
|
|
13
|
+
export declare function wrapMcpTool(serverName: string, mcpTool: McpToolDefinition, callTool: CallToolFn): SdkTool;
|
|
14
|
+
//# sourceMappingURL=tools.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.d.ts","sourceRoot":"","sources":["../../src/mcp/tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,MAAM,aAAa,CAAC;AAG3C,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;CACnC;AAED,MAAM,MAAM,UAAU,GAAG,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,KAAK,OAAO,CAAC;IAAE,OAAO,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC,CAAC;AAEvI,wBAAgB,WAAW,CACzB,UAAU,EAAE,MAAM,EAClB,OAAO,EAAE,iBAAiB,EAC1B,QAAQ,EAAE,UAAU,GACnB,OAAO,CAsBT"}
|
|
@@ -0,0 +1,23 @@
|
|
|
1
|
+
import { jsonSchemaToTypeBox } from './schema-convert.js';
|
|
2
|
+
export function wrapMcpTool(serverName, mcpTool, callTool) {
|
|
3
|
+
const { schema } = jsonSchemaToTypeBox(mcpTool.inputSchema ?? {});
|
|
4
|
+
return {
|
|
5
|
+
name: `mcp__${serverName}__${mcpTool.name}`,
|
|
6
|
+
label: `MCP: ${serverName}/${mcpTool.name}`,
|
|
7
|
+
description: mcpTool.description ?? `MCP tool: ${mcpTool.name}`,
|
|
8
|
+
parameters: schema,
|
|
9
|
+
capabilities: ['mcp:call'],
|
|
10
|
+
async execute(_toolCallId, args) {
|
|
11
|
+
const result = await callTool(mcpTool.name, (args ?? {}));
|
|
12
|
+
const text = result.content
|
|
13
|
+
.filter((c) => c.type === 'text' && c.text)
|
|
14
|
+
.map((c) => c.text)
|
|
15
|
+
.join('\n');
|
|
16
|
+
return {
|
|
17
|
+
content: [{ type: 'text', text: text || 'Tool executed successfully.' }],
|
|
18
|
+
details: {},
|
|
19
|
+
};
|
|
20
|
+
},
|
|
21
|
+
};
|
|
22
|
+
}
|
|
23
|
+
//# sourceMappingURL=tools.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.js","sourceRoot":"","sources":["../../src/mcp/tools.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAU1D,MAAM,UAAU,WAAW,CACzB,UAAkB,EAClB,OAA0B,EAC1B,QAAoB;IAEpB,MAAM,EAAE,MAAM,EAAE,GAAG,mBAAmB,CAAC,OAAO,CAAC,WAAW,IAAI,EAAE,CAAC,CAAC;IAElE,OAAO;QACL,IAAI,EAAE,QAAQ,UAAU,KAAK,OAAO,CAAC,IAAI,EAAE;QAC3C,KAAK,EAAE,QAAQ,UAAU,IAAI,OAAO,CAAC,IAAI,EAAE;QAC3C,WAAW,EAAE,OAAO,CAAC,WAAW,IAAI,aAAa,OAAO,CAAC,IAAI,EAAE;QAC/D,UAAU,EAAE,MAAM;QAClB,YAAY,EAAE,CAAC,UAAU,CAAC;QAC1B,KAAK,CAAC,OAAO,CAAC,WAAmB,EAAE,IAAa;YAC9C,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,IAAI,IAAI,EAAE,CAA4B,CAAC,CAAC;YACrF,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO;iBACxB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,IAAI,CAAC,CAAC,IAAI,CAAC;iBAC1C,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;iBAClB,IAAI,CAAC,IAAI,CAAC,CAAC;YAEd,OAAO;gBACL,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,IAAI,IAAI,6BAA6B,EAAE,CAAC;gBACjF,OAAO,EAAE,EAAE;aACZ,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.test.d.ts","sourceRoot":"","sources":["../../src/mcp/tools.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,45 @@
|
|
|
1
|
+
import { describe, it, expect, vi } from 'vitest';
|
|
2
|
+
import { wrapMcpTool } from './tools.js';
|
|
3
|
+
describe('wrapMcpTool', () => {
|
|
4
|
+
it('creates a tool with correct naming convention', () => {
|
|
5
|
+
const callTool = vi.fn();
|
|
6
|
+
const tool = wrapMcpTool('myserver', { name: 'doStuff', description: 'Does stuff' }, callTool);
|
|
7
|
+
expect(tool.name).toBe('mcp__myserver__doStuff');
|
|
8
|
+
expect(tool.description).toBe('Does stuff');
|
|
9
|
+
expect(tool.capabilities).toEqual(['mcp:call']);
|
|
10
|
+
expect(tool.label).toBe('MCP: myserver/doStuff');
|
|
11
|
+
});
|
|
12
|
+
it('calls callTool with the original tool name and args', async () => {
|
|
13
|
+
const callTool = vi.fn().mockResolvedValue({
|
|
14
|
+
content: [{ type: 'text', text: 'result here' }],
|
|
15
|
+
});
|
|
16
|
+
const tool = wrapMcpTool('srv', { name: 'myTool' }, callTool);
|
|
17
|
+
const result = await tool.execute('call-1', { key: 'value' });
|
|
18
|
+
expect(callTool).toHaveBeenCalledWith('myTool', { key: 'value' });
|
|
19
|
+
expect(result.content[0]).toEqual({ type: 'text', text: 'result here' });
|
|
20
|
+
});
|
|
21
|
+
it('handles empty content response', async () => {
|
|
22
|
+
const callTool = vi.fn().mockResolvedValue({ content: [] });
|
|
23
|
+
const tool = wrapMcpTool('srv', { name: 'empty' }, callTool);
|
|
24
|
+
const result = await tool.execute('call-2', {});
|
|
25
|
+
expect(result.content[0]).toEqual({ type: 'text', text: 'Tool executed successfully.' });
|
|
26
|
+
});
|
|
27
|
+
it('converts input schema to TypeBox parameters', () => {
|
|
28
|
+
const callTool = vi.fn();
|
|
29
|
+
const tool = wrapMcpTool('srv', {
|
|
30
|
+
name: 'typed',
|
|
31
|
+
inputSchema: {
|
|
32
|
+
type: 'object',
|
|
33
|
+
properties: { query: { type: 'string' } },
|
|
34
|
+
required: ['query'],
|
|
35
|
+
},
|
|
36
|
+
}, callTool);
|
|
37
|
+
expect(tool.parameters).toBeDefined();
|
|
38
|
+
});
|
|
39
|
+
it('provides default description when none given', () => {
|
|
40
|
+
const callTool = vi.fn();
|
|
41
|
+
const tool = wrapMcpTool('srv', { name: 'noDesc' }, callTool);
|
|
42
|
+
expect(tool.description).toBe('MCP tool: noDesc');
|
|
43
|
+
});
|
|
44
|
+
});
|
|
45
|
+
//# sourceMappingURL=tools.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"tools.test.js","sourceRoot":"","sources":["../../src/mcp/tools.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,QAAQ,CAAC;AAClD,OAAO,EAAE,WAAW,EAAE,MAAM,YAAY,CAAC;AAEzC,QAAQ,CAAC,aAAa,EAAE,GAAG,EAAE;IAC3B,EAAE,CAAC,+CAA+C,EAAE,GAAG,EAAE;QACvD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,WAAW,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,YAAY,EAAE,EAAE,QAAQ,CAAC,CAAC;QAE/F,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;QACjD,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC5C,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC;QAChD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,uBAAuB,CAAC,CAAC;IACnD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;QACnE,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC;YACzC,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;SACjD,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC9D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;QAE9D,MAAM,CAAC,QAAQ,CAAC,CAAC,oBAAoB,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,OAAO,EAAE,CAAC,CAAC;QAClE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;IAC3E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;QAC9C,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5D,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC7D,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;QAEhD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,6BAA6B,EAAE,CAAC,CAAC;IAC3F,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,6CAA6C,EAAE,GAAG,EAAE;QACrD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,WAAW,CACtB,KAAK,EACL;YACE,IAAI,EAAE,OAAO;YACb,WAAW,EAAE;gBACX,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE;gBACzC,QAAQ,EAAE,CAAC,OAAO,CAAC;aACpB;SACF,EACD,QAAQ,CACT,CAAC;QAEF,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,WAAW,EAAE,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;QACtD,MAAM,QAAQ,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC;QACzB,MAAM,IAAI,GAAG,WAAW,CAAC,KAAK,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,EAAE,QAAQ,CAAC,CAAC;QAC9D,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;IACpD,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.d.ts","sourceRoot":"","sources":["../../src/memory/memory.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAU,aAAa,EAAmB,MAAM,aAAa,CAAC;AAuC1E,wBAAgB,mBAAmB,IAAI,aAAa,CAkGnD"}
|
|
@@ -0,0 +1,130 @@
|
|
|
1
|
+
import * as fs from 'node:fs/promises';
|
|
2
|
+
import * as path from 'node:path';
|
|
3
|
+
function parseMemoryFile(content) {
|
|
4
|
+
const match = content.match(/^---\n([\s\S]*?)\n---\n([\s\S]*)$/);
|
|
5
|
+
if (!match)
|
|
6
|
+
return null;
|
|
7
|
+
const frontmatter = match[1];
|
|
8
|
+
const body = match[2].trim();
|
|
9
|
+
const nameMatch = frontmatter.match(/^name:\s*(.+)$/m);
|
|
10
|
+
const descMatch = frontmatter.match(/^description:\s*(.+)$/m);
|
|
11
|
+
const typeMatch = frontmatter.match(/^type:\s*(.+)$/m);
|
|
12
|
+
if (!nameMatch || !descMatch || !typeMatch)
|
|
13
|
+
return null;
|
|
14
|
+
return {
|
|
15
|
+
name: nameMatch[1].trim(),
|
|
16
|
+
description: descMatch[1].trim(),
|
|
17
|
+
type: typeMatch[1].trim(),
|
|
18
|
+
content: body,
|
|
19
|
+
};
|
|
20
|
+
}
|
|
21
|
+
function serializeMemory(memory) {
|
|
22
|
+
return `---\nname: ${memory.name}\ndescription: ${memory.description}\ntype: ${memory.type}\n---\n\n${memory.content}\n`;
|
|
23
|
+
}
|
|
24
|
+
function sanitizeFilename(name) {
|
|
25
|
+
return name
|
|
26
|
+
.toLowerCase()
|
|
27
|
+
.replace(/[^a-z0-9_-]/g, '-')
|
|
28
|
+
.replace(/-+/g, '-')
|
|
29
|
+
.replace(/^-|-$/g, '');
|
|
30
|
+
}
|
|
31
|
+
function tokenize(text) {
|
|
32
|
+
return text.toLowerCase().split(/\W+/).filter(t => t.length > 0);
|
|
33
|
+
}
|
|
34
|
+
export function createMemoryManager() {
|
|
35
|
+
return {
|
|
36
|
+
async load(dir) {
|
|
37
|
+
let entries;
|
|
38
|
+
try {
|
|
39
|
+
entries = await fs.readdir(dir);
|
|
40
|
+
}
|
|
41
|
+
catch {
|
|
42
|
+
return [];
|
|
43
|
+
}
|
|
44
|
+
const memories = [];
|
|
45
|
+
for (const entry of entries) {
|
|
46
|
+
if (!entry.endsWith('.md'))
|
|
47
|
+
continue;
|
|
48
|
+
const content = await fs.readFile(path.join(dir, entry), 'utf-8');
|
|
49
|
+
const memory = parseMemoryFile(content);
|
|
50
|
+
if (memory)
|
|
51
|
+
memories.push(memory);
|
|
52
|
+
}
|
|
53
|
+
return memories;
|
|
54
|
+
},
|
|
55
|
+
async save(dir, memory) {
|
|
56
|
+
await fs.mkdir(dir, { recursive: true });
|
|
57
|
+
const filename = sanitizeFilename(memory.name) + '.md';
|
|
58
|
+
await fs.writeFile(path.join(dir, filename), serializeMemory(memory), 'utf-8');
|
|
59
|
+
},
|
|
60
|
+
async remove(dir, name) {
|
|
61
|
+
const filename = sanitizeFilename(name) + '.md';
|
|
62
|
+
try {
|
|
63
|
+
await fs.unlink(path.join(dir, filename));
|
|
64
|
+
}
|
|
65
|
+
catch {
|
|
66
|
+
// ignore if file doesn't exist
|
|
67
|
+
}
|
|
68
|
+
},
|
|
69
|
+
retrieve(memories, context) {
|
|
70
|
+
const maxItems = context.maxItems ?? 10;
|
|
71
|
+
const maxTokens = context.maxTokens ?? Infinity;
|
|
72
|
+
const queryTerms = tokenize(context.query);
|
|
73
|
+
// Empty query: return all memories up to limits (most recent first)
|
|
74
|
+
if (queryTerms.length === 0) {
|
|
75
|
+
const results = [];
|
|
76
|
+
let tokenBudget = 0;
|
|
77
|
+
for (let i = 0; i < memories.length; i++) {
|
|
78
|
+
if (results.length >= maxItems)
|
|
79
|
+
break;
|
|
80
|
+
const itemTokens = Math.ceil(memories[i].content.length / 4);
|
|
81
|
+
if (tokenBudget + itemTokens > maxTokens)
|
|
82
|
+
break;
|
|
83
|
+
tokenBudget += itemTokens;
|
|
84
|
+
results.push({ memory: memories[i], relevanceScore: 1, source: 'memory', updatedAt: Date.now() - i });
|
|
85
|
+
}
|
|
86
|
+
return results;
|
|
87
|
+
}
|
|
88
|
+
const scored = memories.map((memory, index) => {
|
|
89
|
+
const docText = memory.description + ' ' + memory.content;
|
|
90
|
+
const docTokens = tokenize(docText);
|
|
91
|
+
const wordCount = docTokens.length || 1;
|
|
92
|
+
let score = 0;
|
|
93
|
+
for (const term of queryTerms) {
|
|
94
|
+
let count = 0;
|
|
95
|
+
for (const token of docTokens) {
|
|
96
|
+
if (token === term)
|
|
97
|
+
count++;
|
|
98
|
+
}
|
|
99
|
+
score += count / wordCount;
|
|
100
|
+
}
|
|
101
|
+
return {
|
|
102
|
+
memory,
|
|
103
|
+
relevanceScore: score,
|
|
104
|
+
source: 'memory',
|
|
105
|
+
updatedAt: Date.now() - index, // recency tiebreaker
|
|
106
|
+
};
|
|
107
|
+
});
|
|
108
|
+
scored.sort((a, b) => {
|
|
109
|
+
if (b.relevanceScore !== a.relevanceScore)
|
|
110
|
+
return b.relevanceScore - a.relevanceScore;
|
|
111
|
+
return b.updatedAt - a.updatedAt;
|
|
112
|
+
});
|
|
113
|
+
const results = [];
|
|
114
|
+
let tokenBudget = 0;
|
|
115
|
+
for (const item of scored) {
|
|
116
|
+
if (item.relevanceScore === 0)
|
|
117
|
+
break;
|
|
118
|
+
if (results.length >= maxItems)
|
|
119
|
+
break;
|
|
120
|
+
const itemTokens = Math.ceil(item.memory.content.length / 4); // rough estimate
|
|
121
|
+
if (tokenBudget + itemTokens > maxTokens)
|
|
122
|
+
break;
|
|
123
|
+
tokenBudget += itemTokens;
|
|
124
|
+
results.push(item);
|
|
125
|
+
}
|
|
126
|
+
return results;
|
|
127
|
+
},
|
|
128
|
+
};
|
|
129
|
+
}
|
|
130
|
+
//# sourceMappingURL=memory.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.js","sourceRoot":"","sources":["../../src/memory/memory.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,kBAAkB,CAAC;AACvC,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAGlC,SAAS,eAAe,CAAC,OAAe;IACtC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACjE,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IAExB,MAAM,WAAW,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;IAC7B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAE7B,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IACvD,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;IAC9D,MAAM,SAAS,GAAG,WAAW,CAAC,KAAK,CAAC,iBAAiB,CAAC,CAAC;IAEvD,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS,IAAI,CAAC,SAAS;QAAE,OAAO,IAAI,CAAC;IAExD,OAAO;QACL,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;QACzB,WAAW,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;QAChC,IAAI,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAoB;QAC3C,OAAO,EAAE,IAAI;KACd,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,MAAc;IACrC,OAAO,cAAc,MAAM,CAAC,IAAI,kBAAkB,MAAM,CAAC,WAAW,WAAW,MAAM,CAAC,IAAI,YAAY,MAAM,CAAC,OAAO,IAAI,CAAC;AAC3H,CAAC;AAED,SAAS,gBAAgB,CAAC,IAAY;IACpC,OAAO,IAAI;SACR,WAAW,EAAE;SACb,OAAO,CAAC,cAAc,EAAE,GAAG,CAAC;SAC5B,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;SACnB,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,OAAO,IAAI,CAAC,WAAW,EAAE,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AACnE,CAAC;AAED,MAAM,UAAU,mBAAmB;IACjC,OAAO;QACL,KAAK,CAAC,IAAI,CAAC,GAAW;YACpB,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,GAAa,EAAE,CAAC;YAC9B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;gBAC5B,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC;oBAAE,SAAS;gBACrC,MAAM,OAAO,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,KAAK,CAAC,EAAE,OAAO,CAAC,CAAC;gBAClE,MAAM,MAAM,GAAG,eAAe,CAAC,OAAO,CAAC,CAAC;gBACxC,IAAI,MAAM;oBAAE,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YACpC,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAED,KAAK,CAAC,IAAI,CAAC,GAAW,EAAE,MAAc;YACpC,MAAM,EAAE,CAAC,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACzC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;YACvD,MAAM,EAAE,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,EAAE,eAAe,CAAC,MAAM,CAAC,EAAE,OAAO,CAAC,CAAC;QACjF,CAAC;QAED,KAAK,CAAC,MAAM,CAAC,GAAW,EAAE,IAAY;YACpC,MAAM,QAAQ,GAAG,gBAAgB,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC;YAChD,IAAI,CAAC;gBACH,MAAM,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC,CAAC;YAC5C,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QAED,QAAQ,CACN,QAAkB,EAClB,OAAiE;YAEjE,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,QAAQ,CAAC;YAChD,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;YAE3C,oEAAoE;YACpE,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC5B,MAAM,OAAO,GAAsB,EAAE,CAAC;gBACtC,IAAI,WAAW,GAAG,CAAC,CAAC;gBACpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;oBACzC,IAAI,OAAO,CAAC,MAAM,IAAI,QAAQ;wBAAE,MAAM;oBACtC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;oBAC7D,IAAI,WAAW,GAAG,UAAU,GAAG,SAAS;wBAAE,MAAM;oBAChD,WAAW,IAAI,UAAU,CAAC;oBAC1B,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,cAAc,EAAE,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;gBACxG,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC;YAED,MAAM,MAAM,GAAsB,QAAQ,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;gBAC/D,MAAM,OAAO,GAAG,MAAM,CAAC,WAAW,GAAG,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC;gBAC1D,MAAM,SAAS,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;gBACpC,MAAM,SAAS,GAAG,SAAS,CAAC,MAAM,IAAI,CAAC,CAAC;gBAExC,IAAI,KAAK,GAAG,CAAC,CAAC;gBACd,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;oBAC9B,IAAI,KAAK,GAAG,CAAC,CAAC;oBACd,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;wBAC9B,IAAI,KAAK,KAAK,IAAI;4BAAE,KAAK,EAAE,CAAC;oBAC9B,CAAC;oBACD,KAAK,IAAI,KAAK,GAAG,SAAS,CAAC;gBAC7B,CAAC;gBAED,OAAO;oBACL,MAAM;oBACN,cAAc,EAAE,KAAK;oBACrB,MAAM,EAAE,QAAQ;oBAChB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,EAAE,qBAAqB;iBACrD,CAAC;YACJ,CAAC,CAAC,CAAC;YAEH,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE;gBACnB,IAAI,CAAC,CAAC,cAAc,KAAK,CAAC,CAAC,cAAc;oBAAE,OAAO,CAAC,CAAC,cAAc,GAAG,CAAC,CAAC,cAAc,CAAC;gBACtF,OAAO,CAAC,CAAC,SAAS,GAAG,CAAC,CAAC,SAAS,CAAC;YACnC,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAsB,EAAE,CAAC;YACtC,IAAI,WAAW,GAAG,CAAC,CAAC;YACpB,KAAK,MAAM,IAAI,IAAI,MAAM,EAAE,CAAC;gBAC1B,IAAI,IAAI,CAAC,cAAc,KAAK,CAAC;oBAAE,MAAM;gBACrC,IAAI,OAAO,CAAC,MAAM,IAAI,QAAQ;oBAAE,MAAM;gBACtC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,iBAAiB;gBAC/E,IAAI,WAAW,GAAG,UAAU,GAAG,SAAS;oBAAE,MAAM;gBAChD,WAAW,IAAI,UAAU,CAAC;gBAC1B,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACrB,CAAC;YAED,OAAO,OAAO,CAAC;QACjB,CAAC;KACF,CAAC;AACJ,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.test.d.ts","sourceRoot":"","sources":["../../src/memory/memory.test.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,94 @@
|
|
|
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 { createMemoryManager } from './memory.js';
|
|
6
|
+
describe('MemoryManager', () => {
|
|
7
|
+
let tmpDir;
|
|
8
|
+
const manager = createMemoryManager();
|
|
9
|
+
beforeEach(async () => {
|
|
10
|
+
tmpDir = await fs.mkdtemp(path.join(os.tmpdir(), 'memory-test-'));
|
|
11
|
+
});
|
|
12
|
+
afterEach(async () => {
|
|
13
|
+
await fs.rm(tmpDir, { recursive: true, force: true });
|
|
14
|
+
});
|
|
15
|
+
const sampleMemory = {
|
|
16
|
+
name: 'test-memory',
|
|
17
|
+
description: 'A test memory for unit testing',
|
|
18
|
+
type: 'user',
|
|
19
|
+
content: 'This is the memory content about TypeScript testing.',
|
|
20
|
+
};
|
|
21
|
+
describe('save and load roundtrip', () => {
|
|
22
|
+
it('saves and loads a memory correctly', async () => {
|
|
23
|
+
await manager.save(tmpDir, sampleMemory);
|
|
24
|
+
const loaded = await manager.load(tmpDir);
|
|
25
|
+
expect(loaded).toHaveLength(1);
|
|
26
|
+
expect(loaded[0]).toEqual(sampleMemory);
|
|
27
|
+
});
|
|
28
|
+
it('saves multiple memories', async () => {
|
|
29
|
+
await manager.save(tmpDir, sampleMemory);
|
|
30
|
+
await manager.save(tmpDir, { ...sampleMemory, name: 'second', description: 'Second memory', content: 'Other content' });
|
|
31
|
+
const loaded = await manager.load(tmpDir);
|
|
32
|
+
expect(loaded).toHaveLength(2);
|
|
33
|
+
});
|
|
34
|
+
it('sanitizes filenames with special characters', async () => {
|
|
35
|
+
const mem = { ...sampleMemory, name: 'My Memory! @#$%' };
|
|
36
|
+
await manager.save(tmpDir, mem);
|
|
37
|
+
const files = await fs.readdir(tmpDir);
|
|
38
|
+
expect(files[0]).toBe('my-memory.md');
|
|
39
|
+
const loaded = await manager.load(tmpDir);
|
|
40
|
+
expect(loaded[0].name).toBe('My Memory! @#$%');
|
|
41
|
+
});
|
|
42
|
+
it('creates directory if it does not exist', async () => {
|
|
43
|
+
const subdir = path.join(tmpDir, 'nested', 'dir');
|
|
44
|
+
await manager.save(subdir, sampleMemory);
|
|
45
|
+
const loaded = await manager.load(subdir);
|
|
46
|
+
expect(loaded).toHaveLength(1);
|
|
47
|
+
});
|
|
48
|
+
it('returns empty array for non-existent directory', async () => {
|
|
49
|
+
const loaded = await manager.load(path.join(tmpDir, 'nonexistent'));
|
|
50
|
+
expect(loaded).toEqual([]);
|
|
51
|
+
});
|
|
52
|
+
});
|
|
53
|
+
describe('remove', () => {
|
|
54
|
+
it('removes a saved memory', async () => {
|
|
55
|
+
await manager.save(tmpDir, sampleMemory);
|
|
56
|
+
await manager.remove(tmpDir, sampleMemory.name);
|
|
57
|
+
const loaded = await manager.load(tmpDir);
|
|
58
|
+
expect(loaded).toHaveLength(0);
|
|
59
|
+
});
|
|
60
|
+
it('does not throw when removing non-existent memory', async () => {
|
|
61
|
+
await expect(manager.remove(tmpDir, 'nonexistent')).resolves.not.toThrow();
|
|
62
|
+
});
|
|
63
|
+
});
|
|
64
|
+
describe('retrieve', () => {
|
|
65
|
+
const memories = [
|
|
66
|
+
{ name: 'typescript', description: 'TypeScript patterns', type: 'reference', content: 'TypeScript uses strict typing and interfaces for code quality.' },
|
|
67
|
+
{ name: 'python', description: 'Python patterns', type: 'reference', content: 'Python uses dynamic typing and duck typing for flexibility.' },
|
|
68
|
+
{ name: 'testing', description: 'Testing best practices', type: 'project', content: 'Testing with vitest requires describe and it blocks.' },
|
|
69
|
+
];
|
|
70
|
+
it('returns memories matching query keywords', () => {
|
|
71
|
+
const results = manager.retrieve(memories, { query: 'TypeScript typing' });
|
|
72
|
+
expect(results.length).toBeGreaterThan(0);
|
|
73
|
+
expect(results[0].memory.name).toBe('typescript');
|
|
74
|
+
});
|
|
75
|
+
it('returns empty array for no matching query', () => {
|
|
76
|
+
const results = manager.retrieve(memories, { query: 'zzzznoMatch' });
|
|
77
|
+
expect(results).toHaveLength(0);
|
|
78
|
+
});
|
|
79
|
+
it('respects maxItems', () => {
|
|
80
|
+
const results = manager.retrieve(memories, { query: 'typing', maxItems: 1 });
|
|
81
|
+
expect(results).toHaveLength(1);
|
|
82
|
+
});
|
|
83
|
+
it('respects maxTokens budget', () => {
|
|
84
|
+
const results = manager.retrieve(memories, { query: 'typing', maxTokens: 5 });
|
|
85
|
+
expect(results.length).toBeLessThanOrEqual(1);
|
|
86
|
+
});
|
|
87
|
+
it('returns all memories for empty query', () => {
|
|
88
|
+
const results = manager.retrieve(memories, { query: '' });
|
|
89
|
+
expect(results).toHaveLength(3);
|
|
90
|
+
expect(results[0].relevanceScore).toBe(1);
|
|
91
|
+
});
|
|
92
|
+
});
|
|
93
|
+
});
|
|
94
|
+
//# sourceMappingURL=memory.test.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"memory.test.js","sourceRoot":"","sources":["../../src/memory/memory.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,mBAAmB,EAAE,MAAM,aAAa,CAAC;AAGlD,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,MAAc,CAAC;IACnB,MAAM,OAAO,GAAG,mBAAmB,EAAE,CAAC;IAEtC,UAAU,CAAC,KAAK,IAAI,EAAE;QACpB,MAAM,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC;IACpE,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,MAAM,YAAY,GAAW;QAC3B,IAAI,EAAE,aAAa;QACnB,WAAW,EAAE,gCAAgC;QAC7C,IAAI,EAAE,MAAM;QACZ,OAAO,EAAE,sDAAsD;KAChE,CAAC;IAEF,QAAQ,CAAC,yBAAyB,EAAE,GAAG,EAAE;QACvC,EAAE,CAAC,oCAAoC,EAAE,KAAK,IAAI,EAAE;YAClD,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACzC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAC1C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yBAAyB,EAAE,KAAK,IAAI,EAAE;YACvC,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACzC,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,GAAG,YAAY,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,eAAe,EAAE,OAAO,EAAE,eAAe,EAAE,CAAC,CAAC;YACxH,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,MAAM,GAAG,GAAW,EAAE,GAAG,YAAY,EAAE,IAAI,EAAE,iBAAiB,EAAE,CAAC;YACjE,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;YAChC,MAAM,KAAK,GAAG,MAAM,EAAE,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACvC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YACtC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,wCAAwC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,QAAQ,EAAE,KAAK,CAAC,CAAC;YAClD,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACzC,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gDAAgD,EAAE,KAAK,IAAI,EAAE;YAC9D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC;YACpE,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,QAAQ,EAAE,GAAG,EAAE;QACtB,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC;YACzC,MAAM,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC,IAAI,CAAC,CAAC;YAChD,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1C,MAAM,CAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,MAAM,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,aAAa,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QAC7E,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,UAAU,EAAE,GAAG,EAAE;QACxB,MAAM,QAAQ,GAAa;YACzB,EAAE,IAAI,EAAE,YAAY,EAAE,WAAW,EAAE,qBAAqB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,gEAAgE,EAAE;YACxJ,EAAE,IAAI,EAAE,QAAQ,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,6DAA6D,EAAE;YAC7I,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,wBAAwB,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,sDAAsD,EAAE;SAC7I,CAAC;QAEF,EAAE,CAAC,0CAA0C,EAAE,GAAG,EAAE;YAClD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,mBAAmB,EAAE,CAAC,CAAC;YAC3E,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,CAAC;YAC1C,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;YACrE,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mBAAmB,EAAE,GAAG,EAAE;YAC3B,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,EAAE,CAAC,CAAC;YAC7E,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2BAA2B,EAAE,GAAG,EAAE;YACnC,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;YAC9E,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,sCAAsC,EAAE,GAAG,EAAE;YAC9C,MAAM,OAAO,GAAG,OAAO,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;YAC1D,MAAM,CAAC,OAAO,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAChC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAC5C,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
import type { AgentToolResult } from '@mariozechner/pi-agent-core';
|
|
2
|
+
import type { HookHandler, HookEvent, HookContext, RunContext } from '../types.js';
|
|
3
|
+
/**
|
|
4
|
+
* Run all matching PreToolUse hooks, chaining updated args through each hook.
|
|
5
|
+
* Returns the final (possibly modified) args.
|
|
6
|
+
*/
|
|
7
|
+
export declare function runPreToolUseHooks(hooks: HookHandler[], toolName: string, args: unknown, runContext: RunContext): Promise<unknown>;
|
|
8
|
+
/**
|
|
9
|
+
* Run all matching PostToolUse hooks, chaining updated results through each hook.
|
|
10
|
+
* Returns the final (possibly modified) tool result.
|
|
11
|
+
*/
|
|
12
|
+
export declare function runPostToolUseHooks(hooks: HookHandler[], toolName: string, toolResult: AgentToolResult<any>, runContext: RunContext): Promise<AgentToolResult<any>>;
|
|
13
|
+
/**
|
|
14
|
+
* Run all matching lifecycle hooks (SessionStart, SessionEnd, Stop, etc.).
|
|
15
|
+
* Lifecycle hooks ignore the matcher field.
|
|
16
|
+
*/
|
|
17
|
+
export declare function runLifecycleHooks(hooks: HookHandler[], event: HookEvent, runContext: RunContext, extra?: Partial<Pick<HookContext, 'agentName' | 'messages'>>): Promise<void>;
|
|
18
|
+
//# sourceMappingURL=hooks.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hooks.d.ts","sourceRoot":"","sources":["../../src/middleware/hooks.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,6BAA6B,CAAC;AACnE,OAAO,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AAWnF;;;GAGG;AACH,wBAAsB,kBAAkB,CACtC,KAAK,EAAE,WAAW,EAAE,EACpB,QAAQ,EAAE,MAAM,EAChB,IAAI,EAAE,OAAO,EACb,UAAU,EAAE,UAAU,GACrB,OAAO,CAAC,OAAO,CAAC,CAoBlB;AAED;;;GAGG;AACH,wBAAsB,mBAAmB,CACvC,KAAK,EAAE,WAAW,EAAE,EACpB,QAAQ,EAAE,MAAM,EAChB,UAAU,EAAE,eAAe,CAAC,GAAG,CAAC,EAChC,UAAU,EAAE,UAAU,GACrB,OAAO,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAoB/B;AAED;;;GAGG;AACH,wBAAsB,iBAAiB,CACrC,KAAK,EAAE,WAAW,EAAE,EACpB,KAAK,EAAE,SAAS,EAChB,UAAU,EAAE,UAAU,EACtB,KAAK,CAAC,EAAE,OAAO,CAAC,IAAI,CAAC,WAAW,EAAE,WAAW,GAAG,UAAU,CAAC,CAAC,GAC3D,OAAO,CAAC,IAAI,CAAC,CAYf"}
|