@primo-ai/core 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 +114 -0
- package/dist/agent.d.ts +66 -0
- package/dist/agent.d.ts.map +1 -0
- package/dist/agent.js +260 -0
- package/dist/agent.js.map +1 -0
- package/dist/checkpoint-store.d.ts +19 -0
- package/dist/checkpoint-store.d.ts.map +1 -0
- package/dist/checkpoint-store.js +79 -0
- package/dist/checkpoint-store.js.map +1 -0
- package/dist/concurrency-controller.d.ts +9 -0
- package/dist/concurrency-controller.d.ts.map +1 -0
- package/dist/concurrency-controller.js +71 -0
- package/dist/concurrency-controller.js.map +1 -0
- package/dist/config-merge.d.ts +12 -0
- package/dist/config-merge.d.ts.map +1 -0
- package/dist/config-merge.js +47 -0
- package/dist/config-merge.js.map +1 -0
- package/dist/config.d.ts +16 -0
- package/dist/config.d.ts.map +1 -0
- package/dist/config.js +144 -0
- package/dist/config.js.map +1 -0
- package/dist/context-builder.d.ts +29 -0
- package/dist/context-builder.d.ts.map +1 -0
- package/dist/context-builder.js +178 -0
- package/dist/context-builder.js.map +1 -0
- package/dist/dynamic-resolver.d.ts +7 -0
- package/dist/dynamic-resolver.d.ts.map +1 -0
- package/dist/dynamic-resolver.js +11 -0
- package/dist/dynamic-resolver.js.map +1 -0
- package/dist/errors.d.ts +30 -0
- package/dist/errors.d.ts.map +1 -0
- package/dist/errors.js +40 -0
- package/dist/errors.js.map +1 -0
- package/dist/event-bus.d.ts +8 -0
- package/dist/event-bus.d.ts.map +1 -0
- package/dist/event-bus.js +36 -0
- package/dist/event-bus.js.map +1 -0
- package/dist/event-system.d.ts +19 -0
- package/dist/event-system.d.ts.map +1 -0
- package/dist/event-system.js +58 -0
- package/dist/event-system.js.map +1 -0
- package/dist/fallback-runner.d.ts +30 -0
- package/dist/fallback-runner.d.ts.map +1 -0
- package/dist/fallback-runner.js +40 -0
- package/dist/fallback-runner.js.map +1 -0
- package/dist/gateways/builtin-gateway.d.ts +17 -0
- package/dist/gateways/builtin-gateway.d.ts.map +1 -0
- package/dist/gateways/builtin-gateway.js +58 -0
- package/dist/gateways/builtin-gateway.js.map +1 -0
- package/dist/gateways/gateway-chain.d.ts +9 -0
- package/dist/gateways/gateway-chain.d.ts.map +1 -0
- package/dist/gateways/gateway-chain.js +18 -0
- package/dist/gateways/gateway-chain.js.map +1 -0
- package/dist/gateways/openai-compatible-gateway.d.ts +10 -0
- package/dist/gateways/openai-compatible-gateway.d.ts.map +1 -0
- package/dist/gateways/openai-compatible-gateway.js +26 -0
- package/dist/gateways/openai-compatible-gateway.js.map +1 -0
- package/dist/harness.d.ts +44 -0
- package/dist/harness.d.ts.map +1 -0
- package/dist/harness.js +70 -0
- package/dist/harness.js.map +1 -0
- package/dist/hook-manager.d.ts +22 -0
- package/dist/hook-manager.d.ts.map +1 -0
- package/dist/hook-manager.js +92 -0
- package/dist/hook-manager.js.map +1 -0
- package/dist/index.d.ts +34 -0
- package/dist/index.d.ts.map +1 -0
- package/dist/index.js +34 -0
- package/dist/index.js.map +1 -0
- package/dist/llm-invoker.d.ts +34 -0
- package/dist/llm-invoker.d.ts.map +1 -0
- package/dist/llm-invoker.js +124 -0
- package/dist/llm-invoker.js.map +1 -0
- package/dist/loop-orchestrator.d.ts +48 -0
- package/dist/loop-orchestrator.d.ts.map +1 -0
- package/dist/loop-orchestrator.js +247 -0
- package/dist/loop-orchestrator.js.map +1 -0
- package/dist/model-factory.d.ts +13 -0
- package/dist/model-factory.d.ts.map +1 -0
- package/dist/model-factory.js +40 -0
- package/dist/model-factory.js.map +1 -0
- package/dist/model-profile.d.ts +17 -0
- package/dist/model-profile.d.ts.map +1 -0
- package/dist/model-profile.js +57 -0
- package/dist/model-profile.js.map +1 -0
- package/dist/model-resolver.d.ts +16 -0
- package/dist/model-resolver.d.ts.map +1 -0
- package/dist/model-resolver.js +28 -0
- package/dist/model-resolver.js.map +1 -0
- package/dist/parse-model.d.ts +6 -0
- package/dist/parse-model.d.ts.map +1 -0
- package/dist/parse-model.js +11 -0
- package/dist/parse-model.js.map +1 -0
- package/dist/pipeline.d.ts +28 -0
- package/dist/pipeline.d.ts.map +1 -0
- package/dist/pipeline.js +286 -0
- package/dist/pipeline.js.map +1 -0
- package/dist/plugin-manager.d.ts +42 -0
- package/dist/plugin-manager.d.ts.map +1 -0
- package/dist/plugin-manager.js +148 -0
- package/dist/plugin-manager.js.map +1 -0
- package/dist/processors/build-context.d.ts +3 -0
- package/dist/processors/build-context.d.ts.map +1 -0
- package/dist/processors/build-context.js +5 -0
- package/dist/processors/build-context.js.map +1 -0
- package/dist/processors/evaluate-iteration.d.ts +13 -0
- package/dist/processors/evaluate-iteration.d.ts.map +1 -0
- package/dist/processors/evaluate-iteration.js +196 -0
- package/dist/processors/evaluate-iteration.js.map +1 -0
- package/dist/processors/execute-tools.d.ts +4 -0
- package/dist/processors/execute-tools.d.ts.map +1 -0
- package/dist/processors/execute-tools.js +69 -0
- package/dist/processors/execute-tools.js.map +1 -0
- package/dist/processors/gate-tool.d.ts +3 -0
- package/dist/processors/gate-tool.d.ts.map +1 -0
- package/dist/processors/gate-tool.js +6 -0
- package/dist/processors/gate-tool.js.map +1 -0
- package/dist/processors/index.d.ts +10 -0
- package/dist/processors/index.d.ts.map +1 -0
- package/dist/processors/index.js +10 -0
- package/dist/processors/index.js.map +1 -0
- package/dist/processors/invoke-llm.d.ts +13 -0
- package/dist/processors/invoke-llm.d.ts.map +1 -0
- package/dist/processors/invoke-llm.js +138 -0
- package/dist/processors/invoke-llm.js.map +1 -0
- package/dist/processors/prepare-step.d.ts +3 -0
- package/dist/processors/prepare-step.d.ts.map +1 -0
- package/dist/processors/prepare-step.js +5 -0
- package/dist/processors/prepare-step.js.map +1 -0
- package/dist/processors/process-input.d.ts +3 -0
- package/dist/processors/process-input.d.ts.map +1 -0
- package/dist/processors/process-input.js +20 -0
- package/dist/processors/process-input.js.map +1 -0
- package/dist/processors/process-output.d.ts +3 -0
- package/dist/processors/process-output.d.ts.map +1 -0
- package/dist/processors/process-output.js +5 -0
- package/dist/processors/process-output.js.map +1 -0
- package/dist/processors/process-step-output.d.ts +3 -0
- package/dist/processors/process-step-output.d.ts.map +1 -0
- package/dist/processors/process-step-output.js +23 -0
- package/dist/processors/process-step-output.js.map +1 -0
- package/dist/processors/provider-history-compat.d.ts +7 -0
- package/dist/processors/provider-history-compat.d.ts.map +1 -0
- package/dist/processors/provider-history-compat.js +166 -0
- package/dist/processors/provider-history-compat.js.map +1 -0
- package/dist/provider-capabilities.d.ts +4 -0
- package/dist/provider-capabilities.d.ts.map +1 -0
- package/dist/provider-capabilities.js +47 -0
- package/dist/provider-capabilities.js.map +1 -0
- package/dist/retry.d.ts +6 -0
- package/dist/retry.d.ts.map +1 -0
- package/dist/retry.js +26 -0
- package/dist/retry.js.map +1 -0
- package/dist/serialize.d.ts +12 -0
- package/dist/serialize.d.ts.map +1 -0
- package/dist/serialize.js +22 -0
- package/dist/serialize.js.map +1 -0
- package/dist/session-manager.d.ts +17 -0
- package/dist/session-manager.d.ts.map +1 -0
- package/dist/session-manager.js +143 -0
- package/dist/session-manager.js.map +1 -0
- package/dist/session-persistence.d.ts +13 -0
- package/dist/session-persistence.d.ts.map +1 -0
- package/dist/session-persistence.js +63 -0
- package/dist/session-persistence.js.map +1 -0
- package/dist/session-storage.d.ts +19 -0
- package/dist/session-storage.d.ts.map +1 -0
- package/dist/session-storage.js +103 -0
- package/dist/session-storage.js.map +1 -0
- package/dist/state-machine.d.ts +13 -0
- package/dist/state-machine.d.ts.map +1 -0
- package/dist/state-machine.js +49 -0
- package/dist/state-machine.js.map +1 -0
- package/dist/storage-replay-backend.d.ts +7 -0
- package/dist/storage-replay-backend.d.ts.map +1 -0
- package/dist/storage-replay-backend.js +14 -0
- package/dist/storage-replay-backend.js.map +1 -0
- package/dist/sub-agent.d.ts +13 -0
- package/dist/sub-agent.d.ts.map +1 -0
- package/dist/sub-agent.js +87 -0
- package/dist/sub-agent.js.map +1 -0
- package/dist/task-manager.d.ts +22 -0
- package/dist/task-manager.d.ts.map +1 -0
- package/dist/task-manager.js +151 -0
- package/dist/task-manager.js.map +1 -0
- package/dist/token-counter.d.ts +9 -0
- package/dist/token-counter.d.ts.map +1 -0
- package/dist/token-counter.js +74 -0
- package/dist/token-counter.js.map +1 -0
- package/dist/tool-registry.d.ts +31 -0
- package/dist/tool-registry.d.ts.map +1 -0
- package/dist/tool-registry.js +147 -0
- package/dist/tool-registry.js.map +1 -0
- package/package.json +37 -0
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"concurrency-controller.js","sourceRoot":"","sources":["../src/concurrency-controller.ts"],"names":[],"mappings":"AAQA,MAAM,OAAO,qBAAqB;IACxB,KAAK,GAAG,IAAI,GAAG,EAAqB,CAAC;IAE7C,YAAY,QAA2B;QACrC,KAAK,MAAM,EAAE,GAAG,EAAE,aAAa,EAAE,IAAI,QAAQ,EAAE,CAAC;YAC9C,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,OAAO,EAAE,CAAC,EAAE,GAAG,EAAE,aAAa,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACvE,CAAC;IACH,CAAC;IAED,OAAO,CAAC,OAAe,EAAE,SAAkB;QACzC,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACrC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,6BAA6B,OAAO,EAAE,CAAC,CAAC,CAAC;QAC3E,CAAC;QAED,IAAI,IAAI,CAAC,OAAO,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YAC5B,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,QAAQ,GAAG,KAAK,CAAC;YACrB,OAAO,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;gBAC1B,IAAI,QAAQ;oBAAE,OAAO;gBACrB,QAAQ,GAAG,IAAI,CAAC;gBAChB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;YACzB,CAAC,CAAC,CAAC;QACL,CAAC;QAED,IAAI,SAAS,GAAwB,IAAI,CAAC;QAE1C,MAAM,cAAc,GAAG,IAAI,OAAO,CAAa,CAAC,OAAO,EAAE,EAAE;YACzD,MAAM,MAAM,GAAG,GAAG,EAAE;gBAClB,IAAI,QAAQ,GAAG,KAAK,CAAC;gBACrB,OAAO,CAAC,GAAG,EAAE;oBACX,IAAI,QAAQ;wBAAE,OAAO;oBACrB,QAAQ,GAAG,IAAI,CAAC;oBAChB,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;gBACzB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC;YACF,SAAS,GAAG,MAAM,CAAC;YACnB,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,IAAI,SAAS,IAAI,IAAI,EAAE,CAAC;YACtB,OAAO,cAAc,CAAC;QACxB,CAAC;QAED,IAAI,KAAoC,CAAC;QACzC,MAAM,cAAc,GAAG,IAAI,OAAO,CAAQ,CAAC,CAAC,EAAE,MAAM,EAAE,EAAE;YACtD,KAAK,GAAG,UAAU,CAAC,GAAG,EAAE;gBACtB,sDAAsD;gBACtD,IAAI,SAAS,EAAE,CAAC;oBACd,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC;oBAC5C,IAAI,GAAG,KAAK,CAAC,CAAC;wBAAE,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBAC9C,CAAC;gBACD,MAAM,CAAC,IAAI,KAAK,CAAC,2CAA2C,OAAO,WAAW,SAAS,IAAI,CAAC,CAAC,CAAC;YAChG,CAAC,EAAE,SAAS,CAAC,CAAC;QAChB,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,cAAc,EAAE,cAAc,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,EAAE;YACjE,YAAY,CAAC,KAAK,CAAC,CAAC;YACpB,+EAA+E;YAC/E,+CAA+C;QACjD,CAAC,CAAC,CAAC;IACL,CAAC;IAED,cAAc,CAAC,OAAe;QAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IACjC,CAAC;IAEO,WAAW,CAAC,IAAe;QACjC,IAAI,CAAC,OAAO,EAAE,CAAC;QACf,IAAI,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,EAAG,CAAC;YACnC,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,IAAI,EAAE,CAAC;QACT,CAAC;IACH,CAAC;CACF"}
|
|
@@ -0,0 +1,12 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Deep merge utility for configuration objects.
|
|
3
|
+
*
|
|
4
|
+
* Rules:
|
|
5
|
+
* - Plain objects are merged recursively.
|
|
6
|
+
* - For keys in CONCAT_KEYS ('plugins', 'modelGateways'), arrays are concatenated.
|
|
7
|
+
* - All other arrays, primitives, and other types are replaced by the later value.
|
|
8
|
+
* - null/undefined sources are skipped.
|
|
9
|
+
* - The target is NOT mutated; a new object is returned.
|
|
10
|
+
*/
|
|
11
|
+
export declare function deepMerge(target: Record<string, unknown>, ...sources: Record<string, unknown>[]): Record<string, unknown>;
|
|
12
|
+
//# sourceMappingURL=config-merge.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config-merge.d.ts","sourceRoot":"","sources":["../src/config-merge.ts"],"names":[],"mappings":"AAKA;;;;;;;;;GASG;AACH,wBAAgB,SAAS,CACvB,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC/B,GAAG,OAAO,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,GACpC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAmCzB"}
|
|
@@ -0,0 +1,47 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Keys whose arrays should be concatenated rather than replaced during merge.
|
|
3
|
+
*/
|
|
4
|
+
const CONCAT_KEYS = new Set(['plugins', 'modelGateways']);
|
|
5
|
+
/**
|
|
6
|
+
* Deep merge utility for configuration objects.
|
|
7
|
+
*
|
|
8
|
+
* Rules:
|
|
9
|
+
* - Plain objects are merged recursively.
|
|
10
|
+
* - For keys in CONCAT_KEYS ('plugins', 'modelGateways'), arrays are concatenated.
|
|
11
|
+
* - All other arrays, primitives, and other types are replaced by the later value.
|
|
12
|
+
* - null/undefined sources are skipped.
|
|
13
|
+
* - The target is NOT mutated; a new object is returned.
|
|
14
|
+
*/
|
|
15
|
+
export function deepMerge(target, ...sources) {
|
|
16
|
+
const result = { ...target };
|
|
17
|
+
for (const key in result) {
|
|
18
|
+
if (isObject(result[key])) {
|
|
19
|
+
result[key] = { ...result[key] };
|
|
20
|
+
}
|
|
21
|
+
}
|
|
22
|
+
for (const source of sources) {
|
|
23
|
+
if (source == null)
|
|
24
|
+
continue;
|
|
25
|
+
for (const key of Object.keys(source)) {
|
|
26
|
+
const sourceVal = source[key];
|
|
27
|
+
if (sourceVal == null)
|
|
28
|
+
continue;
|
|
29
|
+
if (isObject(sourceVal) && isObject(result[key])) {
|
|
30
|
+
result[key] = deepMerge(result[key], sourceVal);
|
|
31
|
+
}
|
|
32
|
+
else if (Array.isArray(sourceVal) &&
|
|
33
|
+
Array.isArray(result[key]) &&
|
|
34
|
+
CONCAT_KEYS.has(key)) {
|
|
35
|
+
result[key] = [...result[key], ...sourceVal];
|
|
36
|
+
}
|
|
37
|
+
else {
|
|
38
|
+
result[key] = sourceVal;
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
return result;
|
|
43
|
+
}
|
|
44
|
+
function isObject(value) {
|
|
45
|
+
return typeof value === 'object' && value !== null && !Array.isArray(value);
|
|
46
|
+
}
|
|
47
|
+
//# sourceMappingURL=config-merge.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config-merge.js","sourceRoot":"","sources":["../src/config-merge.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,GAAG,IAAI,GAAG,CAAC,CAAC,SAAS,EAAE,eAAe,CAAC,CAAC,CAAC;AAE1D;;;;;;;;;GASG;AACH,MAAM,UAAU,SAAS,CACvB,MAA+B,EAC/B,GAAG,OAAkC;IAErC,MAAM,MAAM,GAA4B,EAAE,GAAG,MAAM,EAAE,CAAC;IAEtD,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;QACzB,IAAI,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,GAAI,MAAM,CAAC,GAAG,CAA6B,EAAE,CAAC;QAChE,CAAC;IACH,CAAC;IAED,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,MAAM,IAAI,IAAI;YAAE,SAAS;QAE7B,KAAK,MAAM,GAAG,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,EAAE,CAAC;YACtC,MAAM,SAAS,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;YAE9B,IAAI,SAAS,IAAI,IAAI;gBAAE,SAAS;YAEhC,IAAI,QAAQ,CAAC,SAAS,CAAC,IAAI,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;gBACjD,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CACrB,MAAM,CAAC,GAAG,CAA4B,EACtC,SAAoC,CACrC,CAAC;YACJ,CAAC;iBAAM,IACL,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC;gBACxB,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBAC1B,WAAW,CAAC,GAAG,CAAC,GAAG,CAAC,EACpB,CAAC;gBACD,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,GAAI,MAAM,CAAC,GAAG,CAAe,EAAE,GAAG,SAAS,CAAC,CAAC;YAC9D,CAAC;iBAAM,CAAC;gBACN,MAAM,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC9E,CAAC"}
|
package/dist/config.d.ts
ADDED
|
@@ -0,0 +1,16 @@
|
|
|
1
|
+
import type { HarnessConfig } from '@primo-ai/sdk';
|
|
2
|
+
export interface ConfigSource {
|
|
3
|
+
global?: string;
|
|
4
|
+
project?: string;
|
|
5
|
+
env?: string;
|
|
6
|
+
session?: Partial<HarnessConfig>;
|
|
7
|
+
}
|
|
8
|
+
export declare class ConfigLoader {
|
|
9
|
+
private fileReader;
|
|
10
|
+
constructor(options?: {
|
|
11
|
+
fileReader?: (path: string) => Promise<string>;
|
|
12
|
+
});
|
|
13
|
+
parseJsonc(content: string): Record<string, unknown>;
|
|
14
|
+
load(sources: ConfigSource): Promise<HarnessConfig>;
|
|
15
|
+
}
|
|
16
|
+
//# sourceMappingURL=config.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.d.ts","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAQnD,MAAM,WAAW,YAAY;IAC3B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,OAAO,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;CAClC;AAMD,qBAAa,YAAY;IACvB,OAAO,CAAC,UAAU,CAAoC;gBAE1C,OAAO,CAAC,EAAE;QAAE,UAAU,CAAC,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,MAAM,CAAC,CAAA;KAAE;IAYxE,UAAU,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAc9C,IAAI,CAAC,OAAO,EAAE,YAAY,GAAG,OAAO,CAAC,aAAa,CAAC;CAmD1D"}
|
package/dist/config.js
ADDED
|
@@ -0,0 +1,144 @@
|
|
|
1
|
+
import { z } from 'zod';
|
|
2
|
+
import { deepMerge } from './config-merge.js';
|
|
3
|
+
// ---------------------------------------------------------------------------
|
|
4
|
+
// ConfigLoader
|
|
5
|
+
// ---------------------------------------------------------------------------
|
|
6
|
+
export class ConfigLoader {
|
|
7
|
+
fileReader;
|
|
8
|
+
constructor(options) {
|
|
9
|
+
this.fileReader =
|
|
10
|
+
options?.fileReader ??
|
|
11
|
+
((path) => import('node:fs/promises').then((fs) => fs.readFile(path, 'utf-8')));
|
|
12
|
+
}
|
|
13
|
+
// -------------------------------------------------------------------------
|
|
14
|
+
// JSONC parser — strips comments and trailing commas without touching
|
|
15
|
+
// string contents.
|
|
16
|
+
// -------------------------------------------------------------------------
|
|
17
|
+
parseJsonc(content) {
|
|
18
|
+
const stripped = stripJsonc(content);
|
|
19
|
+
try {
|
|
20
|
+
return JSON.parse(stripped);
|
|
21
|
+
}
|
|
22
|
+
catch (err) {
|
|
23
|
+
const msg = err instanceof Error ? err.message : String(err);
|
|
24
|
+
throw new Error(`Invalid JSONC: ${msg}`);
|
|
25
|
+
}
|
|
26
|
+
}
|
|
27
|
+
// -------------------------------------------------------------------------
|
|
28
|
+
// Load — merge multiple layers
|
|
29
|
+
// -------------------------------------------------------------------------
|
|
30
|
+
async load(sources) {
|
|
31
|
+
const layers = [];
|
|
32
|
+
// 1. Environment (lowest priority)
|
|
33
|
+
if (sources.env) {
|
|
34
|
+
layers.push(this.parseJsonc(sources.env));
|
|
35
|
+
}
|
|
36
|
+
// 2. Global config file
|
|
37
|
+
if (sources.global) {
|
|
38
|
+
try {
|
|
39
|
+
const content = await this.fileReader(sources.global);
|
|
40
|
+
layers.push(this.parseJsonc(content));
|
|
41
|
+
}
|
|
42
|
+
catch {
|
|
43
|
+
// missing global config is ok
|
|
44
|
+
}
|
|
45
|
+
}
|
|
46
|
+
// 3. Project config file
|
|
47
|
+
if (sources.project) {
|
|
48
|
+
try {
|
|
49
|
+
const content = await this.fileReader(sources.project);
|
|
50
|
+
layers.push(this.parseJsonc(content));
|
|
51
|
+
}
|
|
52
|
+
catch {
|
|
53
|
+
// missing project config is ok
|
|
54
|
+
}
|
|
55
|
+
}
|
|
56
|
+
// 4. Session (highest priority)
|
|
57
|
+
if (sources.session) {
|
|
58
|
+
layers.push(sources.session);
|
|
59
|
+
}
|
|
60
|
+
if (layers.length === 0)
|
|
61
|
+
return {};
|
|
62
|
+
// Merge left to right: later layers win
|
|
63
|
+
const merged = layers.reduce((acc, layer) => deepMerge(acc, layer), {});
|
|
64
|
+
const result = HarnessConfigSchema.safeParse(merged);
|
|
65
|
+
if (!result.success) {
|
|
66
|
+
const issues = result.error.issues
|
|
67
|
+
.map((i) => ` ${i.path.join('.')}: ${i.message}`)
|
|
68
|
+
.join('\n');
|
|
69
|
+
throw new Error(`Invalid config:\n${issues}`);
|
|
70
|
+
}
|
|
71
|
+
return result.data;
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
// ---------------------------------------------------------------------------
|
|
75
|
+
// JSONC stripping — character-by-character to respect string boundaries
|
|
76
|
+
// ---------------------------------------------------------------------------
|
|
77
|
+
function stripJsonc(input) {
|
|
78
|
+
let result = '';
|
|
79
|
+
let i = 0;
|
|
80
|
+
const len = input.length;
|
|
81
|
+
while (i < len) {
|
|
82
|
+
// Inside a string — copy verbatim until closing quote
|
|
83
|
+
if (input[i] === '"') {
|
|
84
|
+
result += input[i];
|
|
85
|
+
i++;
|
|
86
|
+
while (i < len) {
|
|
87
|
+
if (input[i] === '\\' && i + 1 < len) {
|
|
88
|
+
// Escaped character — copy both
|
|
89
|
+
result += input[i] + input[i + 1];
|
|
90
|
+
i += 2;
|
|
91
|
+
}
|
|
92
|
+
else if (input[i] === '"') {
|
|
93
|
+
result += input[i];
|
|
94
|
+
i++;
|
|
95
|
+
break;
|
|
96
|
+
}
|
|
97
|
+
else {
|
|
98
|
+
result += input[i];
|
|
99
|
+
i++;
|
|
100
|
+
}
|
|
101
|
+
}
|
|
102
|
+
continue;
|
|
103
|
+
}
|
|
104
|
+
// Single-line comment
|
|
105
|
+
if (input[i] === '/' && i + 1 < len && input[i + 1] === '/') {
|
|
106
|
+
// Skip until end of line
|
|
107
|
+
i += 2;
|
|
108
|
+
while (i < len && input[i] !== '\n') {
|
|
109
|
+
i++;
|
|
110
|
+
}
|
|
111
|
+
continue;
|
|
112
|
+
}
|
|
113
|
+
// Block comment
|
|
114
|
+
if (input[i] === '/' && i + 1 < len && input[i + 1] === '*') {
|
|
115
|
+
i += 2;
|
|
116
|
+
while (i < len && !(input[i] === '*' && i + 1 < len && input[i + 1] === '/')) {
|
|
117
|
+
i++;
|
|
118
|
+
}
|
|
119
|
+
i += 2; // skip */
|
|
120
|
+
continue;
|
|
121
|
+
}
|
|
122
|
+
result += input[i];
|
|
123
|
+
i++;
|
|
124
|
+
}
|
|
125
|
+
// Strip trailing commas before } or ]
|
|
126
|
+
result = result.replace(/,\s*([}\]])/g, '$1');
|
|
127
|
+
return result;
|
|
128
|
+
}
|
|
129
|
+
// ---------------------------------------------------------------------------
|
|
130
|
+
// Zod schema for HarnessConfig validation
|
|
131
|
+
// ---------------------------------------------------------------------------
|
|
132
|
+
const HarnessConfigSchema = z.object({
|
|
133
|
+
agents: z.record(z.string(), z.unknown()).optional(),
|
|
134
|
+
tools: z.object({ enabled: z.array(z.string()).optional(), disabled: z.array(z.string()).optional() }).optional(),
|
|
135
|
+
plugins: z.array(z.string()).optional(),
|
|
136
|
+
session: z.object({ storage: z.enum(['file', 'memory']).optional(), path: z.string().optional() }).optional(),
|
|
137
|
+
modelProfiles: z.array(z.unknown()).optional(),
|
|
138
|
+
modelGateways: z.array(z.object({
|
|
139
|
+
name: z.string(),
|
|
140
|
+
url: z.string(),
|
|
141
|
+
apiKey: z.string().optional(),
|
|
142
|
+
})).optional(),
|
|
143
|
+
}).passthrough();
|
|
144
|
+
//# sourceMappingURL=config.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"config.js","sourceRoot":"","sources":["../src/config.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAa9C,8EAA8E;AAC9E,eAAe;AACf,8EAA8E;AAE9E,MAAM,OAAO,YAAY;IACf,UAAU,CAAoC;IAEtD,YAAY,OAA4D;QACtE,IAAI,CAAC,UAAU;YACb,OAAO,EAAE,UAAU;gBACnB,CAAC,CAAC,IAAY,EAAE,EAAE,CAChB,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,QAAQ,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,CAAC;IAC3E,CAAC;IAED,4EAA4E;IAC5E,sEAAsE;IACtE,mBAAmB;IACnB,4EAA4E;IAE5E,UAAU,CAAC,OAAe;QACxB,MAAM,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACrC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,KAAK,CAAC,QAAQ,CAA4B,CAAC;QACzD,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,MAAM,IAAI,KAAK,CAAC,kBAAkB,GAAG,EAAE,CAAC,CAAC;QAC3C,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,+BAA+B;IAC/B,4EAA4E;IAE5E,KAAK,CAAC,IAAI,CAAC,OAAqB;QAC9B,MAAM,MAAM,GAA8B,EAAE,CAAC;QAE7C,mCAAmC;QACnC,IAAI,OAAO,CAAC,GAAG,EAAE,CAAC;YAChB,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;QAC5C,CAAC;QAED,wBAAwB;QACxB,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;YACnB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;gBACtD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;YACxC,CAAC;YAAC,MAAM,CAAC;gBACP,8BAA8B;YAChC,CAAC;QACH,CAAC;QAED,yBAAyB;QACzB,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;gBACvD,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC;YACxC,CAAC;YAAC,MAAM,CAAC;gBACP,+BAA+B;YACjC,CAAC;QACH,CAAC;QAED,gCAAgC;QAChC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC,OAAkC,CAAC,CAAC;QAC1D,CAAC;QAED,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,EAAE,CAAC;QAEnC,wCAAwC;QACxC,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAC1B,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,KAAK,CAAC,EACrC,EAA6B,CAC9B,CAAC;QAEF,MAAM,MAAM,GAAG,mBAAmB,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,MAAM;iBAC/B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,OAAO,EAAE,CAAC;iBACjD,IAAI,CAAC,IAAI,CAAC,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,oBAAoB,MAAM,EAAE,CAAC,CAAC;QAChD,CAAC;QAED,OAAO,MAAM,CAAC,IAAqB,CAAC;IACtC,CAAC;CACF;AAED,8EAA8E;AAC9E,wEAAwE;AACxE,8EAA8E;AAE9E,SAAS,UAAU,CAAC,KAAa;IAC/B,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,CAAC,GAAG,CAAC,CAAC;IACV,MAAM,GAAG,GAAG,KAAK,CAAC,MAAM,CAAC;IAEzB,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC;QACf,sDAAsD;QACtD,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;YACrB,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;YACnB,CAAC,EAAE,CAAC;YACJ,OAAO,CAAC,GAAG,GAAG,EAAE,CAAC;gBACf,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,EAAE,CAAC;oBACrC,gCAAgC;oBAChC,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;oBAClC,CAAC,IAAI,CAAC,CAAC;gBACT,CAAC;qBAAM,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;oBAC5B,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;oBACnB,CAAC,EAAE,CAAC;oBACJ,MAAM;gBACR,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;oBACnB,CAAC,EAAE,CAAC;gBACN,CAAC;YACH,CAAC;YACD,SAAS;QACX,CAAC;QAED,sBAAsB;QACtB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;YAC5D,yBAAyB;YACzB,CAAC,IAAI,CAAC,CAAC;YACP,OAAO,CAAC,GAAG,GAAG,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;gBACpC,CAAC,EAAE,CAAC;YACN,CAAC;YACD,SAAS;QACX,CAAC;QAED,gBAAgB;QAChB,IAAI,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;YAC5D,CAAC,IAAI,CAAC,CAAC;YACP,OAAO,CAAC,GAAG,GAAG,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,GAAG,IAAI,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,EAAE,CAAC;gBAC7E,CAAC,EAAE,CAAC;YACN,CAAC;YACD,CAAC,IAAI,CAAC,CAAC,CAAC,UAAU;YAClB,SAAS;QACX,CAAC;QAED,MAAM,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,CAAC,EAAE,CAAC;IACN,CAAC;IAED,sCAAsC;IACtC,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;IAE9C,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,8EAA8E;AAC9E,0CAA0C;AAC1C,8EAA8E;AAE9E,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;IACpD,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE;IACjH,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;IACvC,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,EAAE,EAAE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC7G,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC,QAAQ,EAAE;IAC9C,aAAa,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC;QAC9B,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE;QAChB,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE;QACf,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;KAC9B,CAAC,CAAC,CAAC,QAAQ,EAAE;CACf,CAAC,CAAC,WAAW,EAAE,CAAC"}
|
|
@@ -0,0 +1,29 @@
|
|
|
1
|
+
import type { CompressionStrategy, ContextBudget, Message, ModelProfile, PipelineContext, Processor, TokenCounter } from '@primo-ai/sdk';
|
|
2
|
+
import type { ToolRegistry } from './tool-registry.js';
|
|
3
|
+
export interface ContextBuilderOptions {
|
|
4
|
+
registry: ToolRegistry;
|
|
5
|
+
tokenCounter?: TokenCounter;
|
|
6
|
+
compressionStrategy?: CompressionStrategy;
|
|
7
|
+
profiles?: ModelProfile[];
|
|
8
|
+
budget?: ContextBudget;
|
|
9
|
+
}
|
|
10
|
+
export declare function slidingWindow(messages: Message[], _tc: TokenCounter, _budget: number): Message[];
|
|
11
|
+
export declare class ContextBuilder {
|
|
12
|
+
private readonly registry;
|
|
13
|
+
private readonly tokenCounter;
|
|
14
|
+
private compressionStrategy;
|
|
15
|
+
private profiles;
|
|
16
|
+
private budget;
|
|
17
|
+
constructor(options: ContextBuilderOptions);
|
|
18
|
+
setCompressionStrategy(strategy: CompressionStrategy): void;
|
|
19
|
+
setProfiles(profiles: ModelProfile[]): void;
|
|
20
|
+
setBudget(budget: ContextBudget): void;
|
|
21
|
+
assemble(ctx: PipelineContext): Promise<PipelineContext>;
|
|
22
|
+
createProcessor(): Processor;
|
|
23
|
+
private resolveSystemPrompt;
|
|
24
|
+
private resolveToolDeclarations;
|
|
25
|
+
private trimHistory;
|
|
26
|
+
private countToolDeclarations;
|
|
27
|
+
private matchProfile;
|
|
28
|
+
}
|
|
29
|
+
//# sourceMappingURL=context-builder.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-builder.d.ts","sourceRoot":"","sources":["../src/context-builder.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,mBAAmB,EACnB,aAAa,EACb,OAAO,EACP,YAAY,EACZ,eAAe,EACf,SAAS,EACT,YAAY,EACb,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAGvD,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,YAAY,CAAC;IACvB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,mBAAmB,CAAC,EAAE,mBAAmB,CAAC;IAC1C,QAAQ,CAAC,EAAE,YAAY,EAAE,CAAC;IAC1B,MAAM,CAAC,EAAE,aAAa,CAAC;CACxB;AAID,wBAAgB,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,EAAE,GAAG,EAAE,YAAY,EAAE,OAAO,EAAE,MAAM,GAAG,OAAO,EAAE,CAEhG;AAgDD,qBAAa,cAAc;IACzB,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAe;IACxC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAe;IAC5C,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,QAAQ,CAAiB;IACjC,OAAO,CAAC,MAAM,CAAgB;gBAElB,OAAO,EAAE,qBAAqB;IAQ1C,sBAAsB,CAAC,QAAQ,EAAE,mBAAmB,GAAG,IAAI;IAI3D,WAAW,CAAC,QAAQ,EAAE,YAAY,EAAE,GAAG,IAAI;IAI3C,SAAS,CAAC,MAAM,EAAE,aAAa,GAAG,IAAI;IAIhC,QAAQ,CAAC,GAAG,EAAE,eAAe,GAAG,OAAO,CAAC,eAAe,CAAC;IAW9D,eAAe,IAAI,SAAS;IAO5B,OAAO,CAAC,mBAAmB;IA+B3B,OAAO,CAAC,uBAAuB;YA0BjB,WAAW;IAkCzB,OAAO,CAAC,qBAAqB;IAM7B,OAAO,CAAC,YAAY;CAUrB"}
|
|
@@ -0,0 +1,178 @@
|
|
|
1
|
+
import { TiktokenCounter } from './token-counter.js';
|
|
2
|
+
const DEFAULT_BUDGET = { maxTokens: 128_000 };
|
|
3
|
+
export function slidingWindow(messages, _tc, _budget) {
|
|
4
|
+
return messages.length > 50 ? messages.slice(-50) : messages;
|
|
5
|
+
}
|
|
6
|
+
const RECENT_KEEP = 10;
|
|
7
|
+
const TOOL_PREVIEW_LENGTH = 200;
|
|
8
|
+
function semanticTruncation(messages, tc, budget) {
|
|
9
|
+
if (messages.length === 0)
|
|
10
|
+
return messages;
|
|
11
|
+
const recent = messages.slice(-RECENT_KEEP);
|
|
12
|
+
const recentTokens = tc.countMessages(recent);
|
|
13
|
+
if (recentTokens >= budget) {
|
|
14
|
+
const kept = [];
|
|
15
|
+
for (let i = recent.length - 1; i >= 0; i--) {
|
|
16
|
+
const candidate = [recent[i], ...kept];
|
|
17
|
+
if (tc.countMessages(candidate) > budget)
|
|
18
|
+
break;
|
|
19
|
+
kept.unshift(recent[i]);
|
|
20
|
+
}
|
|
21
|
+
return kept;
|
|
22
|
+
}
|
|
23
|
+
const older = messages.slice(0, -RECENT_KEEP);
|
|
24
|
+
if (older.length === 0)
|
|
25
|
+
return messages;
|
|
26
|
+
const remainingBudget = budget - recentTokens;
|
|
27
|
+
const truncatedOlder = older.map(msg => {
|
|
28
|
+
if (msg.role === 'tool') {
|
|
29
|
+
const content = typeof msg.content === 'string' ? msg.content : String(msg.content);
|
|
30
|
+
if (content.length > TOOL_PREVIEW_LENGTH) {
|
|
31
|
+
return { ...msg, content: content.slice(0, TOOL_PREVIEW_LENGTH) + '... [truncated]' };
|
|
32
|
+
}
|
|
33
|
+
}
|
|
34
|
+
return msg;
|
|
35
|
+
});
|
|
36
|
+
const filled = [];
|
|
37
|
+
let filledTokens = 0;
|
|
38
|
+
for (let i = truncatedOlder.length - 1; i >= 0; i--) {
|
|
39
|
+
const candidateTokens = tc.countMessages([truncatedOlder[i]]);
|
|
40
|
+
if (filledTokens + candidateTokens > remainingBudget)
|
|
41
|
+
break;
|
|
42
|
+
filled.unshift(truncatedOlder[i]);
|
|
43
|
+
filledTokens += candidateTokens;
|
|
44
|
+
}
|
|
45
|
+
return [...filled, ...recent];
|
|
46
|
+
}
|
|
47
|
+
export class ContextBuilder {
|
|
48
|
+
registry;
|
|
49
|
+
tokenCounter;
|
|
50
|
+
compressionStrategy;
|
|
51
|
+
profiles;
|
|
52
|
+
budget;
|
|
53
|
+
constructor(options) {
|
|
54
|
+
this.registry = options.registry;
|
|
55
|
+
this.tokenCounter = options.tokenCounter ?? new TiktokenCounter();
|
|
56
|
+
this.compressionStrategy = options.compressionStrategy ?? semanticTruncation;
|
|
57
|
+
this.profiles = options.profiles ?? [];
|
|
58
|
+
this.budget = options.budget ?? DEFAULT_BUDGET;
|
|
59
|
+
}
|
|
60
|
+
setCompressionStrategy(strategy) {
|
|
61
|
+
this.compressionStrategy = strategy;
|
|
62
|
+
}
|
|
63
|
+
setProfiles(profiles) {
|
|
64
|
+
this.profiles = profiles;
|
|
65
|
+
}
|
|
66
|
+
setBudget(budget) {
|
|
67
|
+
this.budget = budget;
|
|
68
|
+
}
|
|
69
|
+
async assemble(ctx) {
|
|
70
|
+
const model = ctx.agent.config.model;
|
|
71
|
+
const profile = this.matchProfile(model);
|
|
72
|
+
ctx = this.resolveSystemPrompt(ctx, profile);
|
|
73
|
+
ctx = this.resolveToolDeclarations(ctx, profile);
|
|
74
|
+
ctx = await this.trimHistory(ctx, model);
|
|
75
|
+
return ctx;
|
|
76
|
+
}
|
|
77
|
+
createProcessor() {
|
|
78
|
+
return {
|
|
79
|
+
stage: 'buildContext',
|
|
80
|
+
execute: (ctx) => this.assemble(ctx),
|
|
81
|
+
};
|
|
82
|
+
}
|
|
83
|
+
resolveSystemPrompt(ctx, profile) {
|
|
84
|
+
const parts = [];
|
|
85
|
+
if (ctx.agent.config.systemPrompt) {
|
|
86
|
+
parts.push(ctx.agent.config.systemPrompt);
|
|
87
|
+
}
|
|
88
|
+
const extraFragments = profile?.extraPromptFragments?.map((f) => f.content) ?? [];
|
|
89
|
+
const promptFragments = [...ctx.agent.promptFragments, ...extraFragments];
|
|
90
|
+
if (promptFragments.length > 0) {
|
|
91
|
+
parts.push(...promptFragments);
|
|
92
|
+
}
|
|
93
|
+
if (profile?.systemPromptSuffix) {
|
|
94
|
+
parts.push(profile.systemPromptSuffix);
|
|
95
|
+
}
|
|
96
|
+
const systemPrompt = parts.length > 0 ? parts.join('\n\n') : undefined;
|
|
97
|
+
return {
|
|
98
|
+
...ctx,
|
|
99
|
+
agent: {
|
|
100
|
+
...ctx.agent,
|
|
101
|
+
systemPrompt,
|
|
102
|
+
promptFragments,
|
|
103
|
+
_assembledFragmentCount: promptFragments.length,
|
|
104
|
+
},
|
|
105
|
+
};
|
|
106
|
+
}
|
|
107
|
+
resolveToolDeclarations(ctx, profile) {
|
|
108
|
+
const overrides = profile?.toolOverrides ?? {};
|
|
109
|
+
const toolDeclarations = this.registry.getAll().map((t) => ({
|
|
110
|
+
name: t.name,
|
|
111
|
+
description: t.description,
|
|
112
|
+
})).filter((tool) => {
|
|
113
|
+
const override = overrides[tool.name];
|
|
114
|
+
return !override?.exclude;
|
|
115
|
+
}).map((tool) => {
|
|
116
|
+
const override = overrides[tool.name];
|
|
117
|
+
if (override?.description) {
|
|
118
|
+
return { ...tool, description: override.description };
|
|
119
|
+
}
|
|
120
|
+
return tool;
|
|
121
|
+
});
|
|
122
|
+
return {
|
|
123
|
+
...ctx,
|
|
124
|
+
agent: {
|
|
125
|
+
...ctx.agent,
|
|
126
|
+
toolDeclarations,
|
|
127
|
+
providerOptions: ctx.agent.config.providerOptions,
|
|
128
|
+
},
|
|
129
|
+
};
|
|
130
|
+
}
|
|
131
|
+
async trimHistory(ctx, model) {
|
|
132
|
+
const history = ctx.session.messageHistory;
|
|
133
|
+
if (!history || history.length === 0)
|
|
134
|
+
return ctx;
|
|
135
|
+
const systemTokens = ctx.agent.systemPrompt
|
|
136
|
+
? this.tokenCounter.count(ctx.agent.systemPrompt, model)
|
|
137
|
+
: 0;
|
|
138
|
+
const toolTokens = this.countToolDeclarations(ctx.agent.toolDeclarations, model);
|
|
139
|
+
const reservedSystem = this.budget.reservedForSystem ?? 0;
|
|
140
|
+
const reservedTools = this.budget.reservedForTools ?? 0;
|
|
141
|
+
const overhead = Math.max(systemTokens, reservedSystem) + Math.max(toolTokens, reservedTools);
|
|
142
|
+
const historyBudget = this.budget.maxTokens - overhead;
|
|
143
|
+
const historyTokens = this.tokenCounter.countMessages(history, model);
|
|
144
|
+
if (historyTokens <= historyBudget)
|
|
145
|
+
return ctx;
|
|
146
|
+
let compressed = history;
|
|
147
|
+
let compressedTokens = historyTokens;
|
|
148
|
+
const maxPasses = 5;
|
|
149
|
+
for (let pass = 0; pass < maxPasses && compressedTokens > historyBudget; pass++) {
|
|
150
|
+
compressed = await this.compressionStrategy(compressed, this.tokenCounter, historyBudget);
|
|
151
|
+
compressedTokens = this.tokenCounter.countMessages(compressed, model);
|
|
152
|
+
}
|
|
153
|
+
return {
|
|
154
|
+
...ctx,
|
|
155
|
+
session: { ...ctx.session, messageHistory: compressed },
|
|
156
|
+
};
|
|
157
|
+
}
|
|
158
|
+
countToolDeclarations(declarations, model) {
|
|
159
|
+
if (declarations.length === 0)
|
|
160
|
+
return 0;
|
|
161
|
+
const text = declarations.map((d) => `${d.name}: ${d.description}`).join('\n');
|
|
162
|
+
return this.tokenCounter.count(text, model);
|
|
163
|
+
}
|
|
164
|
+
matchProfile(model) {
|
|
165
|
+
for (const profile of this.profiles) {
|
|
166
|
+
if (profile.modelPattern instanceof RegExp) {
|
|
167
|
+
if (profile.modelPattern.test(model))
|
|
168
|
+
return profile;
|
|
169
|
+
}
|
|
170
|
+
else {
|
|
171
|
+
if (model.includes(profile.modelPattern))
|
|
172
|
+
return profile;
|
|
173
|
+
}
|
|
174
|
+
}
|
|
175
|
+
return undefined;
|
|
176
|
+
}
|
|
177
|
+
}
|
|
178
|
+
//# sourceMappingURL=context-builder.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"context-builder.js","sourceRoot":"","sources":["../src/context-builder.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,eAAe,EAAE,MAAM,oBAAoB,CAAC;AAUrD,MAAM,cAAc,GAAkB,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC;AAE7D,MAAM,UAAU,aAAa,CAAC,QAAmB,EAAE,GAAiB,EAAE,OAAe;IACnF,OAAO,QAAQ,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC;AAC/D,CAAC;AAED,MAAM,WAAW,GAAG,EAAE,CAAC;AACvB,MAAM,mBAAmB,GAAG,GAAG,CAAC;AAEhC,SAAS,kBAAkB,CAAC,QAAmB,EAAE,EAAgB,EAAE,MAAc;IAC/E,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IAE3C,MAAM,MAAM,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,WAAW,CAAC,CAAC;IAC5C,MAAM,YAAY,GAAG,EAAE,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;IAE9C,IAAI,YAAY,IAAI,MAAM,EAAE,CAAC;QAC3B,MAAM,IAAI,GAAc,EAAE,CAAC;QAC3B,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,SAAS,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,CAAC;YACvC,IAAI,EAAE,CAAC,aAAa,CAAC,SAAS,CAAC,GAAG,MAAM;gBAAE,MAAM;YAChD,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1B,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,KAAK,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,WAAW,CAAC,CAAC;IAC9C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,QAAQ,CAAC;IAExC,MAAM,eAAe,GAAG,MAAM,GAAG,YAAY,CAAC;IAE9C,MAAM,cAAc,GAAG,KAAK,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;QACrC,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YACxB,MAAM,OAAO,GAAG,OAAO,GAAG,CAAC,OAAO,KAAK,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YACpF,IAAI,OAAO,CAAC,MAAM,GAAG,mBAAmB,EAAE,CAAC;gBACzC,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,mBAAmB,CAAC,GAAG,iBAAiB,EAAE,CAAC;YACxF,CAAC;QACH,CAAC;QACD,OAAO,GAAG,CAAC;IACb,CAAC,CAAC,CAAC;IAEH,MAAM,MAAM,GAAc,EAAE,CAAC;IAC7B,IAAI,YAAY,GAAG,CAAC,CAAC;IACrB,KAAK,IAAI,CAAC,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACpD,MAAM,eAAe,GAAG,EAAE,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QAC9D,IAAI,YAAY,GAAG,eAAe,GAAG,eAAe;YAAE,MAAM;QAC5D,MAAM,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,YAAY,IAAI,eAAe,CAAC;IAClC,CAAC;IAED,OAAO,CAAC,GAAG,MAAM,EAAE,GAAG,MAAM,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,OAAO,cAAc;IACR,QAAQ,CAAe;IACvB,YAAY,CAAe;IACpC,mBAAmB,CAAsB;IACzC,QAAQ,CAAiB;IACzB,MAAM,CAAgB;IAE9B,YAAY,OAA8B;QACxC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;QACjC,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,IAAI,eAAe,EAAE,CAAC;QAClE,IAAI,CAAC,mBAAmB,GAAG,OAAO,CAAC,mBAAmB,IAAI,kBAAkB,CAAC;QAC7E,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,cAAc,CAAC;IACjD,CAAC;IAED,sBAAsB,CAAC,QAA6B;QAClD,IAAI,CAAC,mBAAmB,GAAG,QAAQ,CAAC;IACtC,CAAC;IAED,WAAW,CAAC,QAAwB;QAClC,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;IAC3B,CAAC;IAED,SAAS,CAAC,MAAqB;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,QAAQ,CAAC,GAAoB;QACjC,MAAM,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;QAEzC,GAAG,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QAC7C,GAAG,GAAG,IAAI,CAAC,uBAAuB,CAAC,GAAG,EAAE,OAAO,CAAC,CAAC;QACjD,GAAG,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;QAEzC,OAAO,GAAG,CAAC;IACb,CAAC;IAED,eAAe;QACb,OAAO;YACL,KAAK,EAAE,cAAc;YACrB,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC;SACrC,CAAC;IACJ,CAAC;IAEO,mBAAmB,CAAC,GAAoB,EAAE,OAAsB;QACtE,MAAM,KAAK,GAAa,EAAE,CAAC;QAE3B,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,YAAY,EAAE,CAAC;YAClC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,YAAsB,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,cAAc,GAAG,OAAO,EAAE,oBAAoB,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;QAClF,MAAM,eAAe,GAAG,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,cAAc,CAAC,CAAC;QAE1E,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC/B,KAAK,CAAC,IAAI,CAAC,GAAG,eAAe,CAAC,CAAC;QACjC,CAAC;QAED,IAAI,OAAO,EAAE,kBAAkB,EAAE,CAAC;YAChC,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;QACzC,CAAC;QAED,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEvE,OAAO;YACL,GAAG,GAAG;YACN,KAAK,EAAE;gBACL,GAAG,GAAG,CAAC,KAAK;gBACZ,YAAY;gBACZ,eAAe;gBACf,uBAAuB,EAAE,eAAe,CAAC,MAAM;aACU;SAC5D,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,GAAoB,EAAE,OAAsB;QAC1E,MAAM,SAAS,GAAG,OAAO,EAAE,aAAa,IAAI,EAAE,CAAC;QAC/C,MAAM,gBAAgB,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC1D,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;SAC3B,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YAClB,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC;QAC5B,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACd,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACtC,IAAI,QAAQ,EAAE,WAAW,EAAE,CAAC;gBAC1B,OAAO,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC;YACxD,CAAC;YACD,OAAO,IAAI,CAAC;QACd,CAAC,CAAC,CAAC;QAEH,OAAO;YACL,GAAG,GAAG;YACN,KAAK,EAAE;gBACL,GAAG,GAAG,CAAC,KAAK;gBACZ,gBAAgB;gBAChB,eAAe,EAAE,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,eAAe;aAClD;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,GAAoB,EAAE,KAAc;QAC5D,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,cAAc,CAAC;QAC3C,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QAEjD,MAAM,YAAY,GAAG,GAAG,CAAC,KAAK,CAAC,YAAY;YACzC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC;YACxD,CAAC,CAAC,CAAC,CAAC;QAEN,MAAM,UAAU,GAAG,IAAI,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;QAEjF,MAAM,cAAc,GAAG,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,CAAC,CAAC;QAC1D,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,IAAI,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,EAAE,cAAc,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;QAE9F,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,QAAQ,CAAC;QAEvD,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAEtE,IAAI,aAAa,IAAI,aAAa;YAAE,OAAO,GAAG,CAAC;QAE/C,IAAI,UAAU,GAAG,OAAO,CAAC;QACzB,IAAI,gBAAgB,GAAG,aAAa,CAAC;QACrC,MAAM,SAAS,GAAG,CAAC,CAAC;QACpB,KAAK,IAAI,IAAI,GAAG,CAAC,EAAE,IAAI,GAAG,SAAS,IAAI,gBAAgB,GAAG,aAAa,EAAE,IAAI,EAAE,EAAE,CAAC;YAChF,UAAU,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,EAAE,aAAa,CAAC,CAAC;YAC1F,gBAAgB,GAAG,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;QACxE,CAAC;QAED,OAAO;YACL,GAAG,GAAG;YACN,OAAO,EAAE,EAAE,GAAG,GAAG,CAAC,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE;SACxD,CAAC;IACJ,CAAC;IAEO,qBAAqB,CAAC,YAA0D,EAAE,KAAc;QACtG,IAAI,YAAY,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QACxC,MAAM,IAAI,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC/E,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAC9C,CAAC;IAEO,YAAY,CAAC,KAAa;QAChC,KAAK,MAAM,OAAO,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YACpC,IAAI,OAAO,CAAC,YAAY,YAAY,MAAM,EAAE,CAAC;gBAC3C,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC;oBAAE,OAAO,OAAO,CAAC;YACvD,CAAC;iBAAM,CAAC;gBACN,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC;oBAAE,OAAO,OAAO,CAAC;YAC3D,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;CACF"}
|
|
@@ -0,0 +1,7 @@
|
|
|
1
|
+
import type { Dynamic, ResolveContext } from '@primo-ai/sdk';
|
|
2
|
+
/**
|
|
3
|
+
* Resolve a Dynamic<T> value: if it's a function, call it with the context;
|
|
4
|
+
* otherwise return the static value as-is.
|
|
5
|
+
*/
|
|
6
|
+
export declare function resolveDynamic<T>(value: Dynamic<T>, ctx: ResolveContext): Promise<T>;
|
|
7
|
+
//# sourceMappingURL=dynamic-resolver.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dynamic-resolver.d.ts","sourceRoot":"","sources":["../src/dynamic-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE7D;;;GAGG;AACH,wBAAsB,cAAc,CAAC,CAAC,EACpC,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EACjB,GAAG,EAAE,cAAc,GAClB,OAAO,CAAC,CAAC,CAAC,CAKZ"}
|
|
@@ -0,0 +1,11 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Resolve a Dynamic<T> value: if it's a function, call it with the context;
|
|
3
|
+
* otherwise return the static value as-is.
|
|
4
|
+
*/
|
|
5
|
+
export async function resolveDynamic(value, ctx) {
|
|
6
|
+
if (typeof value === 'function') {
|
|
7
|
+
return value(ctx);
|
|
8
|
+
}
|
|
9
|
+
return value;
|
|
10
|
+
}
|
|
11
|
+
//# sourceMappingURL=dynamic-resolver.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"dynamic-resolver.js","sourceRoot":"","sources":["../src/dynamic-resolver.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CAClC,KAAiB,EACjB,GAAmB;IAEnB,IAAI,OAAO,KAAK,KAAK,UAAU,EAAE,CAAC;QAChC,OAAQ,KAAiD,CAAC,GAAG,CAAC,CAAC;IACjE,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC"}
|
package/dist/errors.d.ts
ADDED
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
export interface AgentErrorOptions {
|
|
2
|
+
code: string;
|
|
3
|
+
recoverable?: boolean;
|
|
4
|
+
retryCount?: number;
|
|
5
|
+
maxRetries?: number;
|
|
6
|
+
cause?: Error;
|
|
7
|
+
}
|
|
8
|
+
export declare class AgentForgeError extends Error {
|
|
9
|
+
readonly code: string;
|
|
10
|
+
readonly recoverable: boolean;
|
|
11
|
+
readonly retryCount?: number;
|
|
12
|
+
readonly maxRetries?: number;
|
|
13
|
+
constructor(message: string, options: AgentErrorOptions);
|
|
14
|
+
}
|
|
15
|
+
export declare class RecoverableError extends AgentForgeError {
|
|
16
|
+
constructor(message: string, options: Omit<AgentErrorOptions, 'recoverable'>);
|
|
17
|
+
}
|
|
18
|
+
export declare class FatalError extends AgentForgeError {
|
|
19
|
+
constructor(message: string, options: Omit<AgentErrorOptions, 'recoverable'>);
|
|
20
|
+
}
|
|
21
|
+
export declare class AuthError extends FatalError {
|
|
22
|
+
constructor(message: string, cause?: Error);
|
|
23
|
+
}
|
|
24
|
+
export declare class ModelNotFoundError extends FatalError {
|
|
25
|
+
constructor(modelString: string, cause?: Error);
|
|
26
|
+
}
|
|
27
|
+
export declare class ToolExecutionError extends RecoverableError {
|
|
28
|
+
constructor(message: string, cause?: Error);
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=errors.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"errors.d.ts","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,iBAAiB;IAChC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,CAAC,EAAE,OAAO,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,UAAU,CAAC,EAAE,MAAM,CAAC;IACpB,KAAK,CAAC,EAAE,KAAK,CAAC;CACf;AAED,qBAAa,eAAgB,SAAQ,KAAK;IACxC,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,QAAQ,CAAC,WAAW,EAAE,OAAO,CAAC;IAC9B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,CAAC;gBAEjB,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,iBAAiB;CAQxD;AAED,qBAAa,gBAAiB,SAAQ,eAAe;gBACvC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,aAAa,CAAC;CAG7E;AAED,qBAAa,UAAW,SAAQ,eAAe;gBACjC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,CAAC,iBAAiB,EAAE,aAAa,CAAC;CAG7E;AAED,qBAAa,SAAU,SAAQ,UAAU;gBAC3B,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK;CAG3C;AAED,qBAAa,kBAAmB,SAAQ,UAAU;gBACpC,WAAW,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK;CAG/C;AAED,qBAAa,kBAAmB,SAAQ,gBAAgB;gBAC1C,OAAO,EAAE,MAAM,EAAE,KAAK,CAAC,EAAE,KAAK;CAG3C"}
|
package/dist/errors.js
ADDED
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
export class AgentForgeError extends Error {
|
|
2
|
+
code;
|
|
3
|
+
recoverable;
|
|
4
|
+
retryCount;
|
|
5
|
+
maxRetries;
|
|
6
|
+
constructor(message, options) {
|
|
7
|
+
super(message, { cause: options.cause });
|
|
8
|
+
this.name = new.target.name;
|
|
9
|
+
this.code = options.code;
|
|
10
|
+
this.recoverable = options.recoverable ?? false;
|
|
11
|
+
this.retryCount = options.retryCount;
|
|
12
|
+
this.maxRetries = options.maxRetries;
|
|
13
|
+
}
|
|
14
|
+
}
|
|
15
|
+
export class RecoverableError extends AgentForgeError {
|
|
16
|
+
constructor(message, options) {
|
|
17
|
+
super(message, { ...options, recoverable: true });
|
|
18
|
+
}
|
|
19
|
+
}
|
|
20
|
+
export class FatalError extends AgentForgeError {
|
|
21
|
+
constructor(message, options) {
|
|
22
|
+
super(message, { ...options, recoverable: false });
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
export class AuthError extends FatalError {
|
|
26
|
+
constructor(message, cause) {
|
|
27
|
+
super(message, { code: 'AUTH', cause });
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
export class ModelNotFoundError extends FatalError {
|
|
31
|
+
constructor(modelString, cause) {
|
|
32
|
+
super(`Model not found: "${modelString}"`, { code: 'MODEL_NOT_FOUND', cause });
|
|
33
|
+
}
|
|
34
|
+
}
|
|
35
|
+
export class ToolExecutionError extends RecoverableError {
|
|
36
|
+
constructor(message, cause) {
|
|
37
|
+
super(message, { code: 'TOOL_ERROR', cause });
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=errors.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"errors.js","sourceRoot":"","sources":["../src/errors.ts"],"names":[],"mappings":"AAQA,MAAM,OAAO,eAAgB,SAAQ,KAAK;IAC/B,IAAI,CAAS;IACb,WAAW,CAAU;IACrB,UAAU,CAAU;IACpB,UAAU,CAAU;IAE7B,YAAY,OAAe,EAAE,OAA0B;QACrD,KAAK,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,KAAK,EAAE,CAAC,CAAC;QACzC,IAAI,CAAC,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC;QAC5B,IAAI,CAAC,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;QACzB,IAAI,CAAC,WAAW,GAAG,OAAO,CAAC,WAAW,IAAI,KAAK,CAAC;QAChD,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACrC,IAAI,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;IACvC,CAAC;CACF;AAED,MAAM,OAAO,gBAAiB,SAAQ,eAAe;IACnD,YAAY,OAAe,EAAE,OAA+C;QAC1E,KAAK,CAAC,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;CACF;AAED,MAAM,OAAO,UAAW,SAAQ,eAAe;IAC7C,YAAY,OAAe,EAAE,OAA+C;QAC1E,KAAK,CAAC,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;IACrD,CAAC;CACF;AAED,MAAM,OAAO,SAAU,SAAQ,UAAU;IACvC,YAAY,OAAe,EAAE,KAAa;QACxC,KAAK,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC,CAAC;IAC1C,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,UAAU;IAChD,YAAY,WAAmB,EAAE,KAAa;QAC5C,KAAK,CAAC,qBAAqB,WAAW,GAAG,EAAE,EAAE,IAAI,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;IACjF,CAAC;CACF;AAED,MAAM,OAAO,kBAAmB,SAAQ,gBAAgB;IACtD,YAAY,OAAe,EAAE,KAAa;QACxC,KAAK,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;IAChD,CAAC;CACF"}
|
|
@@ -0,0 +1,8 @@
|
|
|
1
|
+
export declare class EventBus {
|
|
2
|
+
private onError?;
|
|
3
|
+
private handlers;
|
|
4
|
+
constructor(onError?: ((error: unknown, eventType: string) => void) | undefined);
|
|
5
|
+
emit(eventType: string, data?: unknown): void;
|
|
6
|
+
subscribe(eventType: string, handler: (data?: unknown) => void): () => void;
|
|
7
|
+
}
|
|
8
|
+
//# sourceMappingURL=event-bus.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-bus.d.ts","sourceRoot":"","sources":["../src/event-bus.ts"],"names":[],"mappings":"AAAA,qBAAa,QAAQ;IAGP,OAAO,CAAC,OAAO,CAAC;IAF5B,OAAO,CAAC,QAAQ,CAAoD;gBAEhD,OAAO,CAAC,GAAE,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,IAAI,aAAA;IAQzE,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,IAAI;IAc7C,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,GAAG,MAAM,IAAI;CAS5E"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
export class EventBus {
|
|
2
|
+
onError;
|
|
3
|
+
handlers = new Map();
|
|
4
|
+
constructor(onError) {
|
|
5
|
+
this.onError = onError;
|
|
6
|
+
if (!this.onError) {
|
|
7
|
+
this.onError = (error, eventType) => {
|
|
8
|
+
console.error(`[EventBus] Unhandled error in "${eventType}" handler:`, error);
|
|
9
|
+
};
|
|
10
|
+
}
|
|
11
|
+
}
|
|
12
|
+
emit(eventType, data) {
|
|
13
|
+
const set = this.handlers.get(eventType);
|
|
14
|
+
if (set) {
|
|
15
|
+
for (const handler of set) {
|
|
16
|
+
try {
|
|
17
|
+
handler(data);
|
|
18
|
+
}
|
|
19
|
+
catch (err) {
|
|
20
|
+
// isolate handler errors — one failing handler must not prevent others from running
|
|
21
|
+
this.onError?.(err, eventType);
|
|
22
|
+
}
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
subscribe(eventType, handler) {
|
|
27
|
+
let set = this.handlers.get(eventType);
|
|
28
|
+
if (!set) {
|
|
29
|
+
set = new Set();
|
|
30
|
+
this.handlers.set(eventType, set);
|
|
31
|
+
}
|
|
32
|
+
set.add(handler);
|
|
33
|
+
return () => set.delete(handler);
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
//# sourceMappingURL=event-bus.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-bus.js","sourceRoot":"","sources":["../src/event-bus.ts"],"names":[],"mappings":"AAAA,MAAM,OAAO,QAAQ;IAGC;IAFZ,QAAQ,GAAG,IAAI,GAAG,EAAyC,CAAC;IAEpE,YAAoB,OAAqD;QAArD,YAAO,GAAP,OAAO,CAA8C;QACvE,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAClB,IAAI,CAAC,OAAO,GAAG,CAAC,KAAc,EAAE,SAAiB,EAAE,EAAE;gBACnD,OAAO,CAAC,KAAK,CAAC,kCAAkC,SAAS,YAAY,EAAE,KAAK,CAAC,CAAC;YAChF,CAAC,CAAC;QACJ,CAAC;IACH,CAAC;IAED,IAAI,CAAC,SAAiB,EAAE,IAAc;QACpC,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACzC,IAAI,GAAG,EAAE,CAAC;YACR,KAAK,MAAM,OAAO,IAAI,GAAG,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,OAAO,CAAC,IAAI,CAAC,CAAC;gBAChB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,oFAAoF;oBACpF,IAAI,CAAC,OAAO,EAAE,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;gBACjC,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC;IAED,SAAS,CAAC,SAAiB,EAAE,OAAiC;QAC5D,IAAI,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvC,IAAI,CAAC,GAAG,EAAE,CAAC;YACT,GAAG,GAAG,IAAI,GAAG,EAAE,CAAC;YAChB,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;QACpC,CAAC;QACD,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QACjB,OAAO,GAAG,EAAE,CAAC,GAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;CACF"}
|