@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,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 @@
|
|
|
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 @@
|
|
|
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 @@
|
|
|
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 @@
|
|
|
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 @@
|
|
|
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"}
|