@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.
Files changed (194) hide show
  1. package/README.md +114 -0
  2. package/dist/agent.d.ts +66 -0
  3. package/dist/agent.d.ts.map +1 -0
  4. package/dist/agent.js +260 -0
  5. package/dist/agent.js.map +1 -0
  6. package/dist/checkpoint-store.d.ts +19 -0
  7. package/dist/checkpoint-store.d.ts.map +1 -0
  8. package/dist/checkpoint-store.js +79 -0
  9. package/dist/checkpoint-store.js.map +1 -0
  10. package/dist/concurrency-controller.d.ts +9 -0
  11. package/dist/concurrency-controller.d.ts.map +1 -0
  12. package/dist/concurrency-controller.js +71 -0
  13. package/dist/concurrency-controller.js.map +1 -0
  14. package/dist/config-merge.d.ts +12 -0
  15. package/dist/config-merge.d.ts.map +1 -0
  16. package/dist/config-merge.js +47 -0
  17. package/dist/config-merge.js.map +1 -0
  18. package/dist/config.d.ts +16 -0
  19. package/dist/config.d.ts.map +1 -0
  20. package/dist/config.js +144 -0
  21. package/dist/config.js.map +1 -0
  22. package/dist/context-builder.d.ts +29 -0
  23. package/dist/context-builder.d.ts.map +1 -0
  24. package/dist/context-builder.js +178 -0
  25. package/dist/context-builder.js.map +1 -0
  26. package/dist/dynamic-resolver.d.ts +7 -0
  27. package/dist/dynamic-resolver.d.ts.map +1 -0
  28. package/dist/dynamic-resolver.js +11 -0
  29. package/dist/dynamic-resolver.js.map +1 -0
  30. package/dist/errors.d.ts +30 -0
  31. package/dist/errors.d.ts.map +1 -0
  32. package/dist/errors.js +40 -0
  33. package/dist/errors.js.map +1 -0
  34. package/dist/event-bus.d.ts +8 -0
  35. package/dist/event-bus.d.ts.map +1 -0
  36. package/dist/event-bus.js +36 -0
  37. package/dist/event-bus.js.map +1 -0
  38. package/dist/event-system.d.ts +19 -0
  39. package/dist/event-system.d.ts.map +1 -0
  40. package/dist/event-system.js +58 -0
  41. package/dist/event-system.js.map +1 -0
  42. package/dist/fallback-runner.d.ts +30 -0
  43. package/dist/fallback-runner.d.ts.map +1 -0
  44. package/dist/fallback-runner.js +40 -0
  45. package/dist/fallback-runner.js.map +1 -0
  46. package/dist/gateways/builtin-gateway.d.ts +17 -0
  47. package/dist/gateways/builtin-gateway.d.ts.map +1 -0
  48. package/dist/gateways/builtin-gateway.js +58 -0
  49. package/dist/gateways/builtin-gateway.js.map +1 -0
  50. package/dist/gateways/gateway-chain.d.ts +9 -0
  51. package/dist/gateways/gateway-chain.d.ts.map +1 -0
  52. package/dist/gateways/gateway-chain.js +18 -0
  53. package/dist/gateways/gateway-chain.js.map +1 -0
  54. package/dist/gateways/openai-compatible-gateway.d.ts +10 -0
  55. package/dist/gateways/openai-compatible-gateway.d.ts.map +1 -0
  56. package/dist/gateways/openai-compatible-gateway.js +26 -0
  57. package/dist/gateways/openai-compatible-gateway.js.map +1 -0
  58. package/dist/harness.d.ts +44 -0
  59. package/dist/harness.d.ts.map +1 -0
  60. package/dist/harness.js +70 -0
  61. package/dist/harness.js.map +1 -0
  62. package/dist/hook-manager.d.ts +22 -0
  63. package/dist/hook-manager.d.ts.map +1 -0
  64. package/dist/hook-manager.js +92 -0
  65. package/dist/hook-manager.js.map +1 -0
  66. package/dist/index.d.ts +34 -0
  67. package/dist/index.d.ts.map +1 -0
  68. package/dist/index.js +34 -0
  69. package/dist/index.js.map +1 -0
  70. package/dist/llm-invoker.d.ts +34 -0
  71. package/dist/llm-invoker.d.ts.map +1 -0
  72. package/dist/llm-invoker.js +124 -0
  73. package/dist/llm-invoker.js.map +1 -0
  74. package/dist/loop-orchestrator.d.ts +48 -0
  75. package/dist/loop-orchestrator.d.ts.map +1 -0
  76. package/dist/loop-orchestrator.js +247 -0
  77. package/dist/loop-orchestrator.js.map +1 -0
  78. package/dist/model-factory.d.ts +13 -0
  79. package/dist/model-factory.d.ts.map +1 -0
  80. package/dist/model-factory.js +40 -0
  81. package/dist/model-factory.js.map +1 -0
  82. package/dist/model-profile.d.ts +17 -0
  83. package/dist/model-profile.d.ts.map +1 -0
  84. package/dist/model-profile.js +57 -0
  85. package/dist/model-profile.js.map +1 -0
  86. package/dist/model-resolver.d.ts +16 -0
  87. package/dist/model-resolver.d.ts.map +1 -0
  88. package/dist/model-resolver.js +28 -0
  89. package/dist/model-resolver.js.map +1 -0
  90. package/dist/parse-model.d.ts +6 -0
  91. package/dist/parse-model.d.ts.map +1 -0
  92. package/dist/parse-model.js +11 -0
  93. package/dist/parse-model.js.map +1 -0
  94. package/dist/pipeline.d.ts +28 -0
  95. package/dist/pipeline.d.ts.map +1 -0
  96. package/dist/pipeline.js +286 -0
  97. package/dist/pipeline.js.map +1 -0
  98. package/dist/plugin-manager.d.ts +42 -0
  99. package/dist/plugin-manager.d.ts.map +1 -0
  100. package/dist/plugin-manager.js +148 -0
  101. package/dist/plugin-manager.js.map +1 -0
  102. package/dist/processors/build-context.d.ts +3 -0
  103. package/dist/processors/build-context.d.ts.map +1 -0
  104. package/dist/processors/build-context.js +5 -0
  105. package/dist/processors/build-context.js.map +1 -0
  106. package/dist/processors/evaluate-iteration.d.ts +13 -0
  107. package/dist/processors/evaluate-iteration.d.ts.map +1 -0
  108. package/dist/processors/evaluate-iteration.js +196 -0
  109. package/dist/processors/evaluate-iteration.js.map +1 -0
  110. package/dist/processors/execute-tools.d.ts +4 -0
  111. package/dist/processors/execute-tools.d.ts.map +1 -0
  112. package/dist/processors/execute-tools.js +69 -0
  113. package/dist/processors/execute-tools.js.map +1 -0
  114. package/dist/processors/gate-tool.d.ts +3 -0
  115. package/dist/processors/gate-tool.d.ts.map +1 -0
  116. package/dist/processors/gate-tool.js +6 -0
  117. package/dist/processors/gate-tool.js.map +1 -0
  118. package/dist/processors/index.d.ts +10 -0
  119. package/dist/processors/index.d.ts.map +1 -0
  120. package/dist/processors/index.js +10 -0
  121. package/dist/processors/index.js.map +1 -0
  122. package/dist/processors/invoke-llm.d.ts +13 -0
  123. package/dist/processors/invoke-llm.d.ts.map +1 -0
  124. package/dist/processors/invoke-llm.js +138 -0
  125. package/dist/processors/invoke-llm.js.map +1 -0
  126. package/dist/processors/prepare-step.d.ts +3 -0
  127. package/dist/processors/prepare-step.d.ts.map +1 -0
  128. package/dist/processors/prepare-step.js +5 -0
  129. package/dist/processors/prepare-step.js.map +1 -0
  130. package/dist/processors/process-input.d.ts +3 -0
  131. package/dist/processors/process-input.d.ts.map +1 -0
  132. package/dist/processors/process-input.js +20 -0
  133. package/dist/processors/process-input.js.map +1 -0
  134. package/dist/processors/process-output.d.ts +3 -0
  135. package/dist/processors/process-output.d.ts.map +1 -0
  136. package/dist/processors/process-output.js +5 -0
  137. package/dist/processors/process-output.js.map +1 -0
  138. package/dist/processors/process-step-output.d.ts +3 -0
  139. package/dist/processors/process-step-output.d.ts.map +1 -0
  140. package/dist/processors/process-step-output.js +23 -0
  141. package/dist/processors/process-step-output.js.map +1 -0
  142. package/dist/processors/provider-history-compat.d.ts +7 -0
  143. package/dist/processors/provider-history-compat.d.ts.map +1 -0
  144. package/dist/processors/provider-history-compat.js +166 -0
  145. package/dist/processors/provider-history-compat.js.map +1 -0
  146. package/dist/provider-capabilities.d.ts +4 -0
  147. package/dist/provider-capabilities.d.ts.map +1 -0
  148. package/dist/provider-capabilities.js +47 -0
  149. package/dist/provider-capabilities.js.map +1 -0
  150. package/dist/retry.d.ts +6 -0
  151. package/dist/retry.d.ts.map +1 -0
  152. package/dist/retry.js +26 -0
  153. package/dist/retry.js.map +1 -0
  154. package/dist/serialize.d.ts +12 -0
  155. package/dist/serialize.d.ts.map +1 -0
  156. package/dist/serialize.js +22 -0
  157. package/dist/serialize.js.map +1 -0
  158. package/dist/session-manager.d.ts +17 -0
  159. package/dist/session-manager.d.ts.map +1 -0
  160. package/dist/session-manager.js +143 -0
  161. package/dist/session-manager.js.map +1 -0
  162. package/dist/session-persistence.d.ts +13 -0
  163. package/dist/session-persistence.d.ts.map +1 -0
  164. package/dist/session-persistence.js +63 -0
  165. package/dist/session-persistence.js.map +1 -0
  166. package/dist/session-storage.d.ts +19 -0
  167. package/dist/session-storage.d.ts.map +1 -0
  168. package/dist/session-storage.js +103 -0
  169. package/dist/session-storage.js.map +1 -0
  170. package/dist/state-machine.d.ts +13 -0
  171. package/dist/state-machine.d.ts.map +1 -0
  172. package/dist/state-machine.js +49 -0
  173. package/dist/state-machine.js.map +1 -0
  174. package/dist/storage-replay-backend.d.ts +7 -0
  175. package/dist/storage-replay-backend.d.ts.map +1 -0
  176. package/dist/storage-replay-backend.js +14 -0
  177. package/dist/storage-replay-backend.js.map +1 -0
  178. package/dist/sub-agent.d.ts +13 -0
  179. package/dist/sub-agent.d.ts.map +1 -0
  180. package/dist/sub-agent.js +87 -0
  181. package/dist/sub-agent.js.map +1 -0
  182. package/dist/task-manager.d.ts +22 -0
  183. package/dist/task-manager.d.ts.map +1 -0
  184. package/dist/task-manager.js +151 -0
  185. package/dist/task-manager.js.map +1 -0
  186. package/dist/token-counter.d.ts +9 -0
  187. package/dist/token-counter.d.ts.map +1 -0
  188. package/dist/token-counter.js +74 -0
  189. package/dist/token-counter.js.map +1 -0
  190. package/dist/tool-registry.d.ts +31 -0
  191. package/dist/tool-registry.d.ts.map +1 -0
  192. package/dist/tool-registry.js +147 -0
  193. package/dist/tool-registry.js.map +1 -0
  194. package/package.json +37 -0
@@ -0,0 +1,148 @@
1
+ import { resolve, relative, isAbsolute } from 'node:path';
2
+ import { EventSystem } from './event-system.js';
3
+ import { HookManager } from './hook-manager.js';
4
+ import { HarnessAPIImpl } from './harness.js';
5
+ function validatePluginPath(filePath) {
6
+ if (/^(@[a-zA-Z0-9_-]+\/)?[a-zA-Z0-9_-]+$/.test(filePath)) {
7
+ return { valid: true };
8
+ }
9
+ const root = resolve(process.cwd());
10
+ const resolved = resolve(root, filePath);
11
+ const rel = relative(root, resolved);
12
+ if (rel.startsWith('..') || isAbsolute(rel)) {
13
+ return { valid: false, reason: `Plugin path "${filePath}" resolves outside project root` };
14
+ }
15
+ return { valid: true };
16
+ }
17
+ export class PluginManager {
18
+ runner;
19
+ registry;
20
+ contextBuilder;
21
+ _eventSystem = new EventSystem();
22
+ hookManager;
23
+ _harnessInstances = [];
24
+ resourceInstances = new Map();
25
+ errors = [];
26
+ stageMutator;
27
+ get eventBus() {
28
+ return this._eventSystem.bus;
29
+ }
30
+ get eventSystem() {
31
+ return this._eventSystem;
32
+ }
33
+ constructor(runner, registry, contextBuilder) {
34
+ this.runner = runner;
35
+ this.registry = registry;
36
+ this.contextBuilder = contextBuilder;
37
+ this.hookManager = new HookManager(this.eventBus);
38
+ registry.setHookManager(this.hookManager);
39
+ }
40
+ async loadPlugin(filePath) {
41
+ const validation = validatePluginPath(filePath);
42
+ if (!validation.valid) {
43
+ this.errors.push({ source: filePath, error: new Error(validation.reason) });
44
+ return;
45
+ }
46
+ try {
47
+ const module = await import(filePath);
48
+ const factory = module.default ?? module;
49
+ if (typeof factory !== 'function') {
50
+ throw new Error(`Plugin at "${filePath}" does not export a factory function`);
51
+ }
52
+ this.initializePlugin(factory);
53
+ }
54
+ catch (err) {
55
+ this.errors.push({
56
+ source: filePath,
57
+ error: err instanceof Error ? err : new Error(String(err)),
58
+ });
59
+ }
60
+ }
61
+ initializePlugin(factory) {
62
+ const api = this.createHarnessAPI();
63
+ factory(api);
64
+ }
65
+ async loadPluginsFromConfig(config) {
66
+ const plugins = config.plugins ?? [];
67
+ for (const { path } of plugins) {
68
+ await this.loadPlugin(path);
69
+ }
70
+ }
71
+ getCommand(name) {
72
+ for (const h of this._harnessInstances) {
73
+ const cmd = h.getCommands().get(name);
74
+ if (cmd)
75
+ return cmd;
76
+ }
77
+ return undefined;
78
+ }
79
+ getErrors() {
80
+ return this.errors;
81
+ }
82
+ setStageMutator(mutator) {
83
+ this.stageMutator = mutator;
84
+ }
85
+ freezeHarnessInstances() {
86
+ for (const h of this._harnessInstances)
87
+ h.freeze();
88
+ }
89
+ emitEvent(eventType, ...args) {
90
+ this.eventBus.emit(eventType, args[0]);
91
+ }
92
+ async initializeAll() {
93
+ const allResources = this._harnessInstances.flatMap(h => h.getResources());
94
+ for (const resource of allResources) {
95
+ try {
96
+ const instance = await resource.start();
97
+ this.resourceInstances.set(resource.id, instance);
98
+ }
99
+ catch (err) {
100
+ this.errors.push({
101
+ source: `resource:${resource.id}`,
102
+ error: err instanceof Error ? err : new Error(String(err)),
103
+ });
104
+ }
105
+ }
106
+ }
107
+ _shutdown = false;
108
+ async shutdown() {
109
+ if (this._shutdown)
110
+ return;
111
+ this._shutdown = true;
112
+ const allResources = this._harnessInstances.flatMap(h => h.getResources());
113
+ for (const resource of [...allResources].reverse()) {
114
+ const instance = this.resourceInstances.get(resource.id);
115
+ try {
116
+ await resource.stop(instance);
117
+ }
118
+ catch (err) {
119
+ this.errors.push({
120
+ source: `resource:${resource.id}`,
121
+ error: err instanceof Error ? err : new Error(String(err)),
122
+ });
123
+ }
124
+ }
125
+ this.resourceInstances.clear();
126
+ for (const h of this._harnessInstances) {
127
+ for (const unsub of h.getUnsubFns())
128
+ unsub();
129
+ }
130
+ this._harnessInstances = [];
131
+ }
132
+ createHarnessAPI() {
133
+ const impl = new HarnessAPIImpl({
134
+ runner: this.runner,
135
+ registry: this.registry,
136
+ hookManager: this.hookManager,
137
+ eventSystem: this._eventSystem,
138
+ eventBus: this.eventBus,
139
+ contextBuilder: this.contextBuilder,
140
+ emitEvent: (eventType, data) => this.emitEvent(eventType, data),
141
+ registerProvider: (_name, _factory) => { },
142
+ mutateStages: (mutation) => this.stageMutator?.(mutation),
143
+ });
144
+ this._harnessInstances.push(impl);
145
+ return impl;
146
+ }
147
+ }
148
+ //# sourceMappingURL=plugin-manager.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"plugin-manager.js","sourceRoot":"","sources":["../src/plugin-manager.ts"],"names":[],"mappings":"AASA,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAE1D,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,cAAc,EAAE,MAAM,cAAc,CAAC;AAI9C,SAAS,kBAAkB,CAAC,QAAgB;IAC1C,IAAI,sCAAsC,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;QAC1D,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC;IACD,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE,CAAC,CAAC;IACpC,MAAM,QAAQ,GAAG,OAAO,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACzC,MAAM,GAAG,GAAG,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACrC,IAAI,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QAC5C,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,gBAAgB,QAAQ,iCAAiC,EAAE,CAAC;IAC7F,CAAC;IACD,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC;AACzB,CAAC;AAED,MAAM,OAAO,aAAa;IAChB,MAAM,CAAiB;IACvB,QAAQ,CAAe;IACvB,cAAc,CAAkB;IAChC,YAAY,GAAG,IAAI,WAAW,EAAE,CAAC;IAChC,WAAW,CAAc;IAC1B,iBAAiB,GAAqB,EAAE,CAAC;IACzC,iBAAiB,GAAG,IAAI,GAAG,EAAmB,CAAC;IAC/C,MAAM,GAA4C,EAAE,CAAC;IACrD,YAAY,CAAqC;IAEzD,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;IAC/B,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,YAAY,CAAC;IAC3B,CAAC;IAED,YAAY,MAAsB,EAAE,QAAsB,EAAE,cAA+B;QACzF,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,CAAC,cAAc,GAAG,cAAc,CAAC;QACrC,IAAI,CAAC,WAAW,GAAG,IAAI,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAClD,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,UAAU,CAAC,QAAgB;QAC/B,MAAM,UAAU,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAChD,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;YACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,QAAQ,EAAE,KAAK,EAAE,IAAI,KAAK,CAAC,UAAU,CAAC,MAAO,CAAC,EAAE,CAAC,CAAC;YAC7E,OAAO;QACT,CAAC;QACD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,CAAC;YACtC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC;YACzC,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;gBAClC,MAAM,IAAI,KAAK,CAAC,cAAc,QAAQ,sCAAsC,CAAC,CAAC;YAChF,CAAC;YACD,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;gBACf,MAAM,EAAE,QAAQ;gBAChB,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;aAC3D,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,gBAAgB,CAAC,OAAsB;QACrC,MAAM,GAAG,GAAG,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACpC,OAAO,CAAC,GAAG,CAAC,CAAC;IACf,CAAC;IAED,KAAK,CAAC,qBAAqB,CAAC,MAA6C;QACvE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,IAAI,EAAE,CAAC;QACrC,KAAK,MAAM,EAAE,IAAI,EAAE,IAAI,OAAO,EAAE,CAAC;YAC/B,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;QAC9B,CAAC;IACH,CAAC;IAED,UAAU,CAAC,IAAY;QACrB,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvC,MAAM,GAAG,GAAG,CAAC,CAAC,WAAW,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACtC,IAAI,GAAG;gBAAE,OAAO,GAAG,CAAC;QACtB,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,SAAS;QACP,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,eAAe,CAAC,OAA0C;QACxD,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;IAC9B,CAAC;IAED,sBAAsB;QACpB,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,iBAAiB;YAAE,CAAC,CAAC,MAAM,EAAE,CAAC;IACrD,CAAC;IAED,SAAS,CAAC,SAAoB,EAAE,GAAG,IAAe;QAChD,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC;IACzC,CAAC;IAED,KAAK,CAAC,aAAa;QACjB,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC;QAC3E,KAAK,MAAM,QAAQ,IAAI,YAAY,EAAE,CAAC;YACpC,IAAI,CAAC;gBACH,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,KAAK,EAAE,CAAC;gBACxC,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,QAAQ,CAAC,CAAC;YACpD,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBACf,MAAM,EAAE,YAAY,QAAQ,CAAC,EAAE,EAAE;oBACjC,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBAC3D,CAAC,CAAC;YACL,CAAC;QACH,CAAC;IACH,CAAC;IAEO,SAAS,GAAG,KAAK,CAAC;IAE1B,KAAK,CAAC,QAAQ;QACZ,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3B,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QAEtB,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,YAAY,EAAE,CAAC,CAAC;QAC3E,KAAK,MAAM,QAAQ,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,MAAM,QAAQ,GAAG,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YACzD,IAAI,CAAC;gBACH,MAAM,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAChC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC;oBACf,MAAM,EAAE,YAAY,QAAQ,CAAC,EAAE,EAAE;oBACjC,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;iBAC3D,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QACD,IAAI,CAAC,iBAAiB,CAAC,KAAK,EAAE,CAAC;QAE/B,KAAK,MAAM,CAAC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACvC,KAAK,MAAM,KAAK,IAAI,CAAC,CAAC,WAAW,EAAE;gBAAE,KAAK,EAAE,CAAC;QAC/C,CAAC;QACD,IAAI,CAAC,iBAAiB,GAAG,EAAE,CAAC;IAC9B,CAAC;IAEO,gBAAgB;QACtB,MAAM,IAAI,GAAG,IAAI,cAAc,CAAC;YAC9B,MAAM,EAAE,IAAI,CAAC,MAAM;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,WAAW,EAAE,IAAI,CAAC,WAAW;YAC7B,WAAW,EAAE,IAAI,CAAC,YAAY;YAC9B,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,SAAS,EAAE,CAAC,SAAS,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC;YAC/D,gBAAgB,EAAE,CAAC,KAAK,EAAE,QAAQ,EAAE,EAAE,GAAE,CAAC;YACzC,YAAY,EAAE,CAAC,QAAQ,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,QAAQ,CAAC;SAC1D,CAAC,CAAC;QACH,IAAI,CAAC,iBAAiB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAClC,OAAO,IAAI,CAAC;IACd,CAAC;CACF"}
@@ -0,0 +1,3 @@
1
+ import type { Processor } from '@primo-ai/sdk';
2
+ export declare const buildContextExtensionPoint: Processor;
3
+ //# sourceMappingURL=build-context.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"build-context.d.ts","sourceRoot":"","sources":["../../src/processors/build-context.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C,eAAO,MAAM,0BAA0B,EAAE,SAGxC,CAAC"}
@@ -0,0 +1,5 @@
1
+ export const buildContextExtensionPoint = {
2
+ stage: 'buildContext',
3
+ execute: async (ctx) => ctx,
4
+ };
5
+ //# sourceMappingURL=build-context.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"build-context.js","sourceRoot":"","sources":["../../src/processors/build-context.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,0BAA0B,GAAc;IACnD,KAAK,EAAE,cAAc;IACrB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG;CAC5B,CAAC"}
@@ -0,0 +1,13 @@
1
+ import type { Processor } from '@primo-ai/sdk';
2
+ import type { EventBus } from '../event-bus.js';
3
+ export interface EvaluateIterationDeps {
4
+ eventBus?: EventBus;
5
+ /** Maximum total tokens before stopping the loop. Defaults to 100_000. */
6
+ maxTotalTokens?: number;
7
+ }
8
+ export declare function createEvaluateIterationProcessor(deps?: EvaluateIterationDeps): Processor;
9
+ /**
10
+ * @deprecated Use `createEvaluateIterationProcessor({ eventBus })` for full functionality.
11
+ */
12
+ export declare const evaluateIterationProcessor: Processor;
13
+ //# sourceMappingURL=evaluate-iteration.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"evaluate-iteration.d.ts","sourceRoot":"","sources":["../../src/processors/evaluate-iteration.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAgD,MAAM,eAAe,CAAC;AAC7F,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAEhD,MAAM,WAAW,qBAAqB;IACpC,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,0EAA0E;IAC1E,cAAc,CAAC,EAAE,MAAM,CAAC;CACzB;AAqBD,wBAAgB,gCAAgC,CAAC,IAAI,CAAC,EAAE,qBAAqB,GAAG,SAAS,CAuMxF;AAED;;GAEG;AACH,eAAO,MAAM,0BAA0B,EAAE,SAA8C,CAAC"}
@@ -0,0 +1,196 @@
1
+ function collectCalledToolNames(history, pendingToolCalls) {
2
+ const called = new Set();
3
+ for (const msg of history) {
4
+ if (msg.role === 'assistant' && 'toolCalls' in msg && msg.toolCalls) {
5
+ for (const tc of msg.toolCalls) {
6
+ called.add(tc.name);
7
+ }
8
+ }
9
+ }
10
+ if (pendingToolCalls) {
11
+ for (const tc of pendingToolCalls) {
12
+ called.add(tc.name);
13
+ }
14
+ }
15
+ return called;
16
+ }
17
+ const REQUIRED_TOOLS_MAX_RETRIES = 3;
18
+ export function createEvaluateIterationProcessor(deps) {
19
+ const eventBus = deps?.eventBus;
20
+ const maxTotalTokens = deps?.maxTotalTokens ?? 100_000;
21
+ let warnedUnknownTools = false;
22
+ const failCounts = {};
23
+ return {
24
+ stage: 'evaluateIteration',
25
+ execute: async (ctx) => {
26
+ const prevTotal = ctx.session.totalTokenUsage ?? { input: 0, output: 0 };
27
+ const iterUsage = ctx.iteration.tokenUsage ?? { input: 0, output: 0 };
28
+ const totalTokenUsage = {
29
+ input: prevTotal.input + iterUsage.input,
30
+ output: prevTotal.output + iterUsage.output,
31
+ };
32
+ const totalTokens = totalTokenUsage.input + totalTokenUsage.output;
33
+ const requiredTools = ctx.agent.config.requiredTools;
34
+ if (totalTokens > maxTotalTokens) {
35
+ ctx.iteration.span?.setAttribute('token.overflow', true);
36
+ ctx.iteration.span?.setAttribute('token.total', totalTokens);
37
+ // Emit unsatisfied if required tools are not all called
38
+ if (requiredTools && requiredTools.length > 0) {
39
+ const calledTools = collectCalledToolNames(ctx.session.messageHistory ?? [], ctx.iteration.pendingToolCalls);
40
+ const uncalled = requiredTools.filter(name => !calledTools.has(name));
41
+ if (uncalled.length > 0) {
42
+ eventBus?.emit('required_tools:unsatisfied', {
43
+ uncalled,
44
+ reason: 'token_overflow',
45
+ step: ctx.iteration.step,
46
+ sessionId: ctx.request.sessionId,
47
+ });
48
+ }
49
+ }
50
+ return {
51
+ ...ctx,
52
+ iteration: {
53
+ ...ctx.iteration,
54
+ loopDirective: { action: 'stop' },
55
+ },
56
+ session: {
57
+ ...ctx.session,
58
+ totalTokenUsage,
59
+ },
60
+ };
61
+ }
62
+ if (requiredTools && requiredTools.length > 0) {
63
+ if (!warnedUnknownTools) {
64
+ const registeredNames = new Set(ctx.agent.toolDeclarations.map(t => t.name));
65
+ const unknown = requiredTools.filter(name => !registeredNames.has(name));
66
+ if (unknown.length > 0) {
67
+ eventBus?.emit('required_tools:unknown', {
68
+ unknown,
69
+ registered: [...registeredNames],
70
+ step: ctx.iteration.step,
71
+ sessionId: ctx.request.sessionId,
72
+ });
73
+ ctx.iteration.span?.setAttribute('required_tools.unknown', unknown.join(','));
74
+ }
75
+ warnedUnknownTools = true;
76
+ }
77
+ const calledTools = collectCalledToolNames(ctx.session.messageHistory ?? [], ctx.iteration.pendingToolCalls);
78
+ const uncalled = requiredTools.filter(name => !calledTools.has(name));
79
+ if (uncalled.length > 0) {
80
+ for (const name of uncalled) {
81
+ failCounts[name] = (failCounts[name] ?? 0) + 1;
82
+ }
83
+ for (const name of requiredTools) {
84
+ if (calledTools.has(name))
85
+ failCounts[name] = 0;
86
+ }
87
+ const exhausted = uncalled.filter(name => (failCounts[name] ?? 0) >= REQUIRED_TOOLS_MAX_RETRIES);
88
+ ctx.iteration.span?.setAttribute('required_tools.incomplete', true);
89
+ ctx.iteration.span?.setAttribute('required_tools.uncalled', uncalled.join(','));
90
+ if (exhausted.length > 0) {
91
+ const exhaustedMsg = `Required tools exhausted after ${REQUIRED_TOOLS_MAX_RETRIES} retries: ${exhausted.join(', ')}. Halting loop.`;
92
+ eventBus?.emit('required_tools:exhausted', {
93
+ exhausted,
94
+ failCounts: { ...failCounts },
95
+ step: ctx.iteration.step,
96
+ sessionId: ctx.request.sessionId,
97
+ });
98
+ ctx.iteration.span?.setAttribute('required_tools.exhausted', exhausted.join(','));
99
+ const policy = ctx.agent.config.requiredToolPolicy ?? 'advise';
100
+ if (policy === 'enforce') {
101
+ // Inject synthetic tool calls for exhausted uncalled tools and continue the loop
102
+ const syntheticCalls = exhausted.map((name, i) => ({
103
+ id: `required-${i}`,
104
+ name,
105
+ args: {},
106
+ }));
107
+ eventBus?.emit('required_tools:enforced', {
108
+ tools: exhausted,
109
+ syntheticCalls,
110
+ step: ctx.iteration.step,
111
+ sessionId: ctx.request.sessionId,
112
+ });
113
+ ctx.iteration.span?.setAttribute('required_tools.enforced', exhausted.join(','));
114
+ return {
115
+ ...ctx,
116
+ iteration: {
117
+ ...ctx.iteration,
118
+ loopDirective: { action: 'continue' },
119
+ pendingToolCalls: syntheticCalls,
120
+ },
121
+ session: {
122
+ ...ctx.session,
123
+ totalTokenUsage,
124
+ },
125
+ };
126
+ }
127
+ // Default 'advise' behavior: stop with error message
128
+ return {
129
+ ...ctx,
130
+ agent: {
131
+ ...ctx.agent,
132
+ promptFragments: [
133
+ ...ctx.agent.promptFragments,
134
+ `[system] ${exhaustedMsg}`,
135
+ ],
136
+ },
137
+ iteration: {
138
+ ...ctx.iteration,
139
+ loopDirective: { action: 'stop' },
140
+ response: exhaustedMsg,
141
+ },
142
+ session: {
143
+ ...ctx.session,
144
+ totalTokenUsage,
145
+ },
146
+ };
147
+ }
148
+ eventBus?.emit('required_tools:incomplete', {
149
+ uncalled,
150
+ called: [...calledTools],
151
+ step: ctx.iteration.step,
152
+ sessionId: ctx.request.sessionId,
153
+ });
154
+ return {
155
+ ...ctx,
156
+ agent: {
157
+ ...ctx.agent,
158
+ promptFragments: [
159
+ ...ctx.agent.promptFragments,
160
+ `[system] Required tools not yet called: ${uncalled.join(', ')}. Please call them before finishing.`,
161
+ ],
162
+ },
163
+ iteration: {
164
+ ...ctx.iteration,
165
+ loopDirective: { action: 'continue' },
166
+ },
167
+ session: {
168
+ ...ctx.session,
169
+ totalTokenUsage,
170
+ },
171
+ };
172
+ }
173
+ }
174
+ const hasToolResults = (ctx.iteration.toolResults?.length ?? 0) > 0;
175
+ const directive = hasToolResults
176
+ ? { action: 'continue' }
177
+ : { action: 'stop' };
178
+ return {
179
+ ...ctx,
180
+ iteration: {
181
+ ...ctx.iteration,
182
+ loopDirective: directive,
183
+ },
184
+ session: {
185
+ ...ctx.session,
186
+ totalTokenUsage,
187
+ },
188
+ };
189
+ },
190
+ };
191
+ }
192
+ /**
193
+ * @deprecated Use `createEvaluateIterationProcessor({ eventBus })` for full functionality.
194
+ */
195
+ export const evaluateIterationProcessor = createEvaluateIterationProcessor();
196
+ //# sourceMappingURL=evaluate-iteration.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"evaluate-iteration.js","sourceRoot":"","sources":["../../src/processors/evaluate-iteration.ts"],"names":[],"mappings":"AASA,SAAS,sBAAsB,CAAC,OAAkB,EAAE,gBAA6B;IAC/E,MAAM,MAAM,GAAG,IAAI,GAAG,EAAU,CAAC;IACjC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC1B,IAAI,GAAG,CAAC,IAAI,KAAK,WAAW,IAAI,WAAW,IAAI,GAAG,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;YACpE,KAAK,MAAM,EAAE,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;gBAC/B,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;YACtB,CAAC;QACH,CAAC;IACH,CAAC;IACD,IAAI,gBAAgB,EAAE,CAAC;QACrB,KAAK,MAAM,EAAE,IAAI,gBAAgB,EAAE,CAAC;YAClC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,MAAM,0BAA0B,GAAG,CAAC,CAAC;AAErC,MAAM,UAAU,gCAAgC,CAAC,IAA4B;IAC3E,MAAM,QAAQ,GAAG,IAAI,EAAE,QAAQ,CAAC;IAChC,MAAM,cAAc,GAAG,IAAI,EAAE,cAAc,IAAI,OAAO,CAAC;IACvD,IAAI,kBAAkB,GAAG,KAAK,CAAC;IAC/B,MAAM,UAAU,GAA2B,EAAE,CAAC;IAE9C,OAAO;QACL,KAAK,EAAE,mBAAmB;QAC1B,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACrB,MAAM,SAAS,GAAG,GAAG,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YACzE,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC,UAAU,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC;YACtE,MAAM,eAAe,GAAe;gBAClC,KAAK,EAAE,SAAS,CAAC,KAAK,GAAG,SAAS,CAAC,KAAK;gBACxC,MAAM,EAAE,SAAS,CAAC,MAAM,GAAG,SAAS,CAAC,MAAM;aAC5C,CAAC;YAEF,MAAM,WAAW,GAAG,eAAe,CAAC,KAAK,GAAG,eAAe,CAAC,MAAM,CAAC;YAEnE,MAAM,aAAa,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,aAAa,CAAC;YAErD,IAAI,WAAW,GAAG,cAAc,EAAE,CAAC;gBACjC,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,gBAAgB,EAAE,IAAI,CAAC,CAAC;gBACzD,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,aAAa,EAAE,WAAW,CAAC,CAAC;gBAE7D,wDAAwD;gBACxD,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC9C,MAAM,WAAW,GAAG,sBAAsB,CACxC,GAAG,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,EAChC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAC/B,CAAC;oBACF,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;oBACtE,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACxB,QAAQ,EAAE,IAAI,CAAC,4BAA4B,EAAE;4BAC3C,QAAQ;4BACR,MAAM,EAAE,gBAAgB;4BACxB,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,IAAI;4BACxB,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS;yBACjC,CAAC,CAAC;oBACL,CAAC;gBACH,CAAC;gBAED,OAAO;oBACL,GAAG,GAAG;oBACN,SAAS,EAAE;wBACT,GAAG,GAAG,CAAC,SAAS;wBAChB,aAAa,EAAE,EAAE,MAAM,EAAE,MAAM,EAAmB;qBACnD;oBACD,OAAO,EAAE;wBACP,GAAG,GAAG,CAAC,OAAO;wBACd,eAAe;qBAChB;iBACF,CAAC;YACJ,CAAC;YAED,IAAI,aAAa,IAAI,aAAa,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBAC9C,IAAI,CAAC,kBAAkB,EAAE,CAAC;oBACxB,MAAM,eAAe,GAAG,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,CAAC,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;oBAC7E,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;oBACzE,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACvB,QAAQ,EAAE,IAAI,CAAC,wBAAwB,EAAE;4BACvC,OAAO;4BACP,UAAU,EAAE,CAAC,GAAG,eAAe,CAAC;4BAChC,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,IAAI;4BACxB,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS;yBACjC,CAAC,CAAC;wBACH,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,wBAAwB,EAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;oBAChF,CAAC;oBACD,kBAAkB,GAAG,IAAI,CAAC;gBAC5B,CAAC;gBAED,MAAM,WAAW,GAAG,sBAAsB,CACxC,GAAG,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,EAChC,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAC/B,CAAC;gBACF,MAAM,QAAQ,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;gBAEtE,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACxB,KAAK,MAAM,IAAI,IAAI,QAAQ,EAAE,CAAC;wBAC5B,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;oBACjD,CAAC;oBACD,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;wBACjC,IAAI,WAAW,CAAC,GAAG,CAAC,IAAI,CAAC;4BAAE,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBAClD,CAAC;oBAED,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,0BAA0B,CAAC,CAAC;oBAEjG,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,2BAA2B,EAAE,IAAI,CAAC,CAAC;oBACpE,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,yBAAyB,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;oBAEhF,IAAI,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACzB,MAAM,YAAY,GAAG,kCAAkC,0BAA0B,aAAa,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,iBAAiB,CAAC;wBACpI,QAAQ,EAAE,IAAI,CAAC,0BAA0B,EAAE;4BACzC,SAAS;4BACT,UAAU,EAAE,EAAE,GAAG,UAAU,EAAE;4BAC7B,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,IAAI;4BACxB,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS;yBACjC,CAAC,CAAC;wBACH,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,0BAA0B,EAAE,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;wBAElF,MAAM,MAAM,GAAG,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,kBAAkB,IAAI,QAAQ,CAAC;wBAE/D,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;4BACzB,iFAAiF;4BACjF,MAAM,cAAc,GAAe,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gCAC7D,EAAE,EAAE,YAAY,CAAC,EAAE;gCACnB,IAAI;gCACJ,IAAI,EAAE,EAAE;6BACT,CAAC,CAAC,CAAC;4BAEJ,QAAQ,EAAE,IAAI,CAAC,yBAAyB,EAAE;gCACxC,KAAK,EAAE,SAAS;gCAChB,cAAc;gCACd,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,IAAI;gCACxB,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS;6BACjC,CAAC,CAAC;4BACH,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,YAAY,CAAC,yBAAyB,EAAE,SAAS,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC;4BAEjF,OAAO;gCACL,GAAG,GAAG;gCACN,SAAS,EAAE;oCACT,GAAG,GAAG,CAAC,SAAS;oCAChB,aAAa,EAAE,EAAE,MAAM,EAAE,UAAU,EAAmB;oCACtD,gBAAgB,EAAE,cAAc;iCACjC;gCACD,OAAO,EAAE;oCACP,GAAG,GAAG,CAAC,OAAO;oCACd,eAAe;iCAChB;6BACF,CAAC;wBACJ,CAAC;wBAED,qDAAqD;wBACrD,OAAO;4BACL,GAAG,GAAG;4BACN,KAAK,EAAE;gCACL,GAAG,GAAG,CAAC,KAAK;gCACZ,eAAe,EAAE;oCACf,GAAG,GAAG,CAAC,KAAK,CAAC,eAAe;oCAC5B,YAAY,YAAY,EAAE;iCAC3B;6BACF;4BACD,SAAS,EAAE;gCACT,GAAG,GAAG,CAAC,SAAS;gCAChB,aAAa,EAAE,EAAE,MAAM,EAAE,MAAM,EAAmB;gCAClD,QAAQ,EAAE,YAAY;6BACvB;4BACD,OAAO,EAAE;gCACP,GAAG,GAAG,CAAC,OAAO;gCACd,eAAe;6BAChB;yBACF,CAAC;oBACJ,CAAC;oBAED,QAAQ,EAAE,IAAI,CAAC,2BAA2B,EAAE;wBAC1C,QAAQ;wBACR,MAAM,EAAE,CAAC,GAAG,WAAW,CAAC;wBACxB,IAAI,EAAE,GAAG,CAAC,SAAS,CAAC,IAAI;wBACxB,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS;qBACjC,CAAC,CAAC;oBAEH,OAAO;wBACL,GAAG,GAAG;wBACN,KAAK,EAAE;4BACL,GAAG,GAAG,CAAC,KAAK;4BACZ,eAAe,EAAE;gCACf,GAAG,GAAG,CAAC,KAAK,CAAC,eAAe;gCAC5B,2CAA2C,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,sCAAsC;6BACrG;yBACF;wBACD,SAAS,EAAE;4BACT,GAAG,GAAG,CAAC,SAAS;4BAChB,aAAa,EAAE,EAAE,MAAM,EAAE,UAAU,EAAmB;yBACvD;wBACD,OAAO,EAAE;4BACP,GAAG,GAAG,CAAC,OAAO;4BACd,eAAe;yBAChB;qBACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,MAAM,cAAc,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,WAAW,EAAE,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;YACpE,MAAM,SAAS,GAAkB,cAAc;gBAC7C,CAAC,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE;gBACxB,CAAC,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC;YAEvB,OAAO;gBACL,GAAG,GAAG;gBACN,SAAS,EAAE;oBACT,GAAG,GAAG,CAAC,SAAS;oBAChB,aAAa,EAAE,SAAS;iBACzB;gBACD,OAAO,EAAE;oBACP,GAAG,GAAG,CAAC,OAAO;oBACd,eAAe;iBAChB;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,0BAA0B,GAAc,gCAAgC,EAAE,CAAC"}
@@ -0,0 +1,4 @@
1
+ import type { Processor } from '@primo-ai/sdk';
2
+ import type { ToolRegistry } from '../tool-registry.js';
3
+ export declare function createExecuteToolsProcessor(registry: ToolRegistry): Processor;
4
+ //# sourceMappingURL=execute-tools.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"execute-tools.d.ts","sourceRoot":"","sources":["../../src/processors/execute-tools.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAuB,MAAM,eAAe,CAAC;AAEpE,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAExD,wBAAgB,2BAA2B,CAAC,QAAQ,EAAE,YAAY,GAAG,SAAS,CAkE7E"}
@@ -0,0 +1,69 @@
1
+ import { SpanAttributeKeys, SpanType } from '@primo-ai/sdk';
2
+ export function createExecuteToolsProcessor(registry) {
3
+ return {
4
+ stage: 'executeTools',
5
+ execute: async (ctx) => {
6
+ const toolCalls = ctx.iteration.pendingToolCalls;
7
+ if (!toolCalls || toolCalls.length === 0) {
8
+ return ctx;
9
+ }
10
+ const toolResults = [];
11
+ for (const tc of toolCalls) {
12
+ const toolSpan = ctx.iteration.span?.startChild(SpanType.TOOL_EXECUTE);
13
+ toolSpan?.setAttribute(SpanAttributeKeys.TOOL_NAME, tc.name);
14
+ try {
15
+ const result = await registry.executeTool(tc.name, tc.args, {
16
+ toolCallId: tc.id,
17
+ span: toolSpan ?? ctx.iteration.span,
18
+ sessionId: ctx.request.sessionId,
19
+ });
20
+ const outputSize = typeof result.output === 'string'
21
+ ? result.output.length
22
+ : JSON.stringify(result.output ?? '').length;
23
+ toolSpan?.setAttribute(SpanAttributeKeys.TOOL_RESULT_SIZE, outputSize);
24
+ toolResults.push({ ...result, toolCallId: tc.id });
25
+ }
26
+ finally {
27
+ toolSpan?.end();
28
+ }
29
+ }
30
+ const toolMessages = toolResults.map((tr) => {
31
+ let content = tr.error ?? (typeof tr.output === 'string' ? tr.output : JSON.stringify(tr.output));
32
+ if (tr.validationError && !tr.error) {
33
+ content = `[Warning: ${tr.validationError}]\n${content}`;
34
+ }
35
+ const msg = {
36
+ role: 'tool',
37
+ content,
38
+ toolCallId: tr.toolCallId,
39
+ toolName: tr.name,
40
+ result: tr.output,
41
+ error: tr.error,
42
+ };
43
+ const ext = msg;
44
+ if (tr.mutated)
45
+ ext.mutated = true;
46
+ if (tr.truncated)
47
+ ext.truncated = true;
48
+ if (tr.validationError)
49
+ ext.validationError = tr.validationError;
50
+ return msg;
51
+ });
52
+ const history = [...(ctx.session.messageHistory ?? [])];
53
+ history.push(...toolMessages);
54
+ return {
55
+ ...ctx,
56
+ iteration: {
57
+ ...ctx.iteration,
58
+ pendingToolCalls: undefined,
59
+ toolResults,
60
+ },
61
+ session: {
62
+ ...ctx.session,
63
+ messageHistory: history,
64
+ },
65
+ };
66
+ },
67
+ };
68
+ }
69
+ //# sourceMappingURL=execute-tools.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"execute-tools.js","sourceRoot":"","sources":["../../src/processors/execute-tools.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,iBAAiB,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AAG5D,MAAM,UAAU,2BAA2B,CAAC,QAAsB;IAChE,OAAO;QACL,KAAK,EAAE,cAAc;QACrB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;YACrB,MAAM,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC;YACjD,IAAI,CAAC,SAAS,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzC,OAAO,GAAG,CAAC;YACb,CAAC;YAED,MAAM,WAAW,GAAiB,EAAE,CAAC;YACrC,KAAK,MAAM,EAAE,IAAI,SAAS,EAAE,CAAC;gBAC3B,MAAM,QAAQ,GAAG,GAAG,CAAC,SAAS,CAAC,IAAI,EAAE,UAAU,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;gBACvE,QAAQ,EAAE,YAAY,CAAC,iBAAiB,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;gBAC7D,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE;wBAC1D,UAAU,EAAE,EAAE,CAAC,EAAE;wBACjB,IAAI,EAAE,QAAQ,IAAI,GAAG,CAAC,SAAS,CAAC,IAAI;wBACpC,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,SAAS;qBACjC,CAAC,CAAC;oBACH,MAAM,UAAU,GAAG,OAAO,MAAM,CAAC,MAAM,KAAK,QAAQ;wBAClD,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,MAAM;wBACtB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;oBAC/C,QAAQ,EAAE,YAAY,CAAC,iBAAiB,CAAC,gBAAgB,EAAE,UAAU,CAAC,CAAC;oBACvE,WAAW,CAAC,IAAI,CAAC,EAAE,GAAG,MAAM,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;gBACrD,CAAC;wBAAS,CAAC;oBACT,QAAQ,EAAE,GAAG,EAAE,CAAC;gBAClB,CAAC;YACH,CAAC;YAED,MAAM,YAAY,GAAc,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE;gBACrD,IAAI,OAAO,GAAG,EAAE,CAAC,KAAK,IAAI,CAAC,OAAO,EAAE,CAAC,MAAM,KAAK,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,CAAC,MAAM,CAAC,CAAC,CAAC;gBAClG,IAAI,EAAE,CAAC,eAAe,IAAI,CAAC,EAAE,CAAC,KAAK,EAAE,CAAC;oBACpC,OAAO,GAAG,aAAa,EAAE,CAAC,eAAe,MAAM,OAAO,EAAE,CAAC;gBAC3D,CAAC;gBACD,MAAM,GAAG,GAAY;oBACnB,IAAI,EAAE,MAAe;oBACrB,OAAO;oBACP,UAAU,EAAE,EAAE,CAAC,UAAU;oBACzB,QAAQ,EAAE,EAAE,CAAC,IAAI;oBACjB,MAAM,EAAE,EAAE,CAAC,MAAM;oBACjB,KAAK,EAAE,EAAE,CAAC,KAAK;iBAChB,CAAC;gBACF,MAAM,GAAG,GAAG,GAAuF,CAAC;gBACpG,IAAI,EAAE,CAAC,OAAO;oBAAE,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC;gBACnC,IAAI,EAAE,CAAC,SAAS;oBAAE,GAAG,CAAC,SAAS,GAAG,IAAI,CAAC;gBACvC,IAAI,EAAE,CAAC,eAAe;oBAAE,GAAG,CAAC,eAAe,GAAG,EAAE,CAAC,eAAe,CAAC;gBACjE,OAAO,GAAG,CAAC;YACb,CAAC,CAAC,CAAC;YAEH,MAAM,OAAO,GAAc,CAAC,GAAG,CAAC,GAAG,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC,CAAC;YACnE,OAAO,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,CAAC;YAE9B,OAAO;gBACL,GAAG,GAAG;gBACN,SAAS,EAAE;oBACT,GAAG,GAAG,CAAC,SAAS;oBAChB,gBAAgB,EAAE,SAAS;oBAC3B,WAAW;iBACZ;gBACD,OAAO,EAAE;oBACP,GAAG,GAAG,CAAC,OAAO;oBACd,cAAc,EAAE,OAAO;iBACxB;aACF,CAAC;QACJ,CAAC;KACF,CAAC;AACJ,CAAC"}
@@ -0,0 +1,3 @@
1
+ import type { Processor } from '@primo-ai/sdk';
2
+ export declare const gateToolExtensionPoint: Processor;
3
+ //# sourceMappingURL=gate-tool.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"gate-tool.d.ts","sourceRoot":"","sources":["../../src/processors/gate-tool.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,eAAe,CAAC;AAE/C,eAAO,MAAM,sBAAsB,EAAE,SAIpC,CAAC"}
@@ -0,0 +1,6 @@
1
+ export const gateToolExtensionPoint = {
2
+ stage: 'gateTool',
3
+ execute: async (ctx) => ctx,
4
+ isNoOp: true,
5
+ };
6
+ //# sourceMappingURL=gate-tool.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"gate-tool.js","sourceRoot":"","sources":["../../src/processors/gate-tool.ts"],"names":[],"mappings":"AAEA,MAAM,CAAC,MAAM,sBAAsB,GAAc;IAC/C,KAAK,EAAE,UAAU;IACjB,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,GAAG;IAC3B,MAAM,EAAE,IAAI;CACb,CAAC"}
@@ -0,0 +1,10 @@
1
+ export { processInputProcessor } from './process-input.js';
2
+ export { buildContextExtensionPoint } from './build-context.js';
3
+ export { prepareStepExtensionPoint } from './prepare-step.js';
4
+ export { createInvokeLLMProcessor, type InvokeLLMDeps } from './invoke-llm.js';
5
+ export { processStepOutputProcessor } from './process-step-output.js';
6
+ export { gateToolExtensionPoint } from './gate-tool.js';
7
+ export { createExecuteToolsProcessor } from './execute-tools.js';
8
+ export { createEvaluateIterationProcessor, evaluateIterationProcessor } from './evaluate-iteration.js';
9
+ export { processOutputProcessor } from './process-output.js';
10
+ //# sourceMappingURL=index.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/processors/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,KAAK,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAC/E,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,2BAA2B,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,gCAAgC,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AACvG,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC"}
@@ -0,0 +1,10 @@
1
+ export { processInputProcessor } from './process-input.js';
2
+ export { buildContextExtensionPoint } from './build-context.js';
3
+ export { prepareStepExtensionPoint } from './prepare-step.js';
4
+ export { createInvokeLLMProcessor } from './invoke-llm.js';
5
+ export { processStepOutputProcessor } from './process-step-output.js';
6
+ export { gateToolExtensionPoint } from './gate-tool.js';
7
+ export { createExecuteToolsProcessor } from './execute-tools.js';
8
+ export { createEvaluateIterationProcessor, evaluateIterationProcessor } from './evaluate-iteration.js';
9
+ export { processOutputProcessor } from './process-output.js';
10
+ //# sourceMappingURL=index.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/processors/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAC3D,OAAO,EAAE,0BAA0B,EAAE,MAAM,oBAAoB,CAAC;AAChE,OAAO,EAAE,yBAAyB,EAAE,MAAM,mBAAmB,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAsB,MAAM,iBAAiB,CAAC;AAC/E,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,sBAAsB,EAAE,MAAM,gBAAgB,CAAC;AACxD,OAAO,EAAE,2BAA2B,EAAE,MAAM,oBAAoB,CAAC;AACjE,OAAO,EAAE,gCAAgC,EAAE,0BAA0B,EAAE,MAAM,yBAAyB,CAAC;AACvG,OAAO,EAAE,sBAAsB,EAAE,MAAM,qBAAqB,CAAC"}
@@ -0,0 +1,13 @@
1
+ import type { Processor } from '@primo-ai/sdk';
2
+ import type { LLMInvoker } from '../llm-invoker.js';
3
+ import type { ToolRegistry } from '../tool-registry.js';
4
+ import type { HookManager } from '../hook-manager.js';
5
+ export declare function validateLlmHookOutput(hookMessages: unknown, original: unknown[]): unknown[];
6
+ export interface InvokeLLMDeps {
7
+ getLLM: (systemPrompt?: string) => Promise<LLMInvoker>;
8
+ registry: ToolRegistry;
9
+ hookManager: HookManager;
10
+ modelString: string;
11
+ }
12
+ export declare function createInvokeLLMProcessor(deps: InvokeLLMDeps): Processor;
13
+ //# sourceMappingURL=invoke-llm.d.ts.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"invoke-llm.d.ts","sourceRoot":"","sources":["../../src/processors/invoke-llm.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAA4B,SAAS,EAAE,MAAM,eAAe,CAAC;AACzE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,oBAAoB,CAAC;AActD,wBAAgB,qBAAqB,CAAC,YAAY,EAAE,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,OAAO,EAAE,CAG3F;AAED,MAAM,WAAW,aAAa;IAC5B,MAAM,EAAE,CAAC,YAAY,CAAC,EAAE,MAAM,KAAK,OAAO,CAAC,UAAU,CAAC,CAAC;IACvD,QAAQ,EAAE,YAAY,CAAC;IACvB,WAAW,EAAE,WAAW,CAAC;IACzB,WAAW,EAAE,MAAM,CAAC;CACrB;AAiFD,wBAAgB,wBAAwB,CAAC,IAAI,EAAE,aAAa,GAAG,SAAS,CA4DvE"}