@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,19 @@
|
|
|
1
|
+
import type { ReplayBackend, ReplayOptions, SessionEvent } from '@primo-ai/sdk';
|
|
2
|
+
import { EventBus } from './event-bus.js';
|
|
3
|
+
/** Sentinel key added to replayed event payloads. Reserved namespace: plugins MUST NOT use this key in event payloads. */
|
|
4
|
+
export declare const REPLAY_SENTINEL = "__agentforge_replay";
|
|
5
|
+
export declare class EventSystem {
|
|
6
|
+
private _bus;
|
|
7
|
+
private backend?;
|
|
8
|
+
constructor(onError?: (error: unknown, eventType: string) => void);
|
|
9
|
+
/** Attach a ReplayBackend (typically wraps SessionStorage). */
|
|
10
|
+
setReplayBackend(backend: ReplayBackend): void;
|
|
11
|
+
/** The underlying EventBus — used for wiring to HookManager, ToolRegistry, etc. */
|
|
12
|
+
get bus(): EventBus;
|
|
13
|
+
emit(eventType: string, data?: unknown): void;
|
|
14
|
+
subscribe(eventType: string, handler: (data?: unknown) => void): () => void;
|
|
15
|
+
query(sessionId: string): Promise<SessionEvent[]>;
|
|
16
|
+
replay(sessionId: string, options?: ReplayOptions): Promise<void>;
|
|
17
|
+
private filterEvents;
|
|
18
|
+
}
|
|
19
|
+
//# sourceMappingURL=event-system.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-system.d.ts","sourceRoot":"","sources":["../src/event-system.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAChF,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,0HAA0H;AAC1H,eAAO,MAAM,eAAe,wBAAwB,CAAC;AAErD,qBAAa,WAAW;IACtB,OAAO,CAAC,IAAI,CAAW;IACvB,OAAO,CAAC,OAAO,CAAC,CAAgB;gBAEpB,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,KAAK,IAAI;IAIjE,+DAA+D;IAC/D,gBAAgB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAI9C,mFAAmF;IACnF,IAAI,GAAG,IAAI,QAAQ,CAElB;IAED,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,IAAI;IAI7C,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,GAAG,MAAM,IAAI;IAIrE,KAAK,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;IAKjD,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAavE,OAAO,CAAC,YAAY;CAerB"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { EventBus } from './event-bus.js';
|
|
2
|
+
/** Sentinel key added to replayed event payloads. Reserved namespace: plugins MUST NOT use this key in event payloads. */
|
|
3
|
+
export const REPLAY_SENTINEL = '__agentforge_replay';
|
|
4
|
+
export class EventSystem {
|
|
5
|
+
_bus;
|
|
6
|
+
backend;
|
|
7
|
+
constructor(onError) {
|
|
8
|
+
this._bus = new EventBus(onError);
|
|
9
|
+
}
|
|
10
|
+
/** Attach a ReplayBackend (typically wraps SessionStorage). */
|
|
11
|
+
setReplayBackend(backend) {
|
|
12
|
+
this.backend = backend;
|
|
13
|
+
}
|
|
14
|
+
/** The underlying EventBus — used for wiring to HookManager, ToolRegistry, etc. */
|
|
15
|
+
get bus() {
|
|
16
|
+
return this._bus;
|
|
17
|
+
}
|
|
18
|
+
emit(eventType, data) {
|
|
19
|
+
this._bus.emit(eventType, data);
|
|
20
|
+
}
|
|
21
|
+
subscribe(eventType, handler) {
|
|
22
|
+
return this._bus.subscribe(eventType, handler);
|
|
23
|
+
}
|
|
24
|
+
async query(sessionId) {
|
|
25
|
+
if (!this.backend)
|
|
26
|
+
return [];
|
|
27
|
+
return this.backend.query(sessionId);
|
|
28
|
+
}
|
|
29
|
+
async replay(sessionId, options) {
|
|
30
|
+
const events = await this.query(sessionId);
|
|
31
|
+
if (events.length === 0)
|
|
32
|
+
return;
|
|
33
|
+
const filtered = this.filterEvents(events, options);
|
|
34
|
+
for (const event of filtered) {
|
|
35
|
+
const payload = typeof event.payload === 'object' && event.payload !== null && !Array.isArray(event.payload)
|
|
36
|
+
? { ...event.payload, [REPLAY_SENTINEL]: true }
|
|
37
|
+
: event.payload;
|
|
38
|
+
this._bus.emit(event.type, payload);
|
|
39
|
+
}
|
|
40
|
+
}
|
|
41
|
+
filterEvents(events, options) {
|
|
42
|
+
if (!options)
|
|
43
|
+
return events;
|
|
44
|
+
let result = events;
|
|
45
|
+
if (options.fromSeq !== undefined) {
|
|
46
|
+
result = result.filter(e => e.seq >= options.fromSeq);
|
|
47
|
+
}
|
|
48
|
+
if (options.toSeq !== undefined) {
|
|
49
|
+
result = result.filter(e => e.seq <= options.toSeq);
|
|
50
|
+
}
|
|
51
|
+
if (options.eventTypes && options.eventTypes.length > 0) {
|
|
52
|
+
const typeSet = new Set(options.eventTypes);
|
|
53
|
+
result = result.filter(e => typeSet.has(e.type));
|
|
54
|
+
}
|
|
55
|
+
return result;
|
|
56
|
+
}
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=event-system.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"event-system.js","sourceRoot":"","sources":["../src/event-system.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,0HAA0H;AAC1H,MAAM,CAAC,MAAM,eAAe,GAAG,qBAAqB,CAAC;AAErD,MAAM,OAAO,WAAW;IACd,IAAI,CAAW;IACf,OAAO,CAAiB;IAEhC,YAAY,OAAqD;QAC/D,IAAI,CAAC,IAAI,GAAG,IAAI,QAAQ,CAAC,OAAO,CAAC,CAAC;IACpC,CAAC;IAED,+DAA+D;IAC/D,gBAAgB,CAAC,OAAsB;QACrC,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,mFAAmF;IACnF,IAAI,GAAG;QACL,OAAO,IAAI,CAAC,IAAI,CAAC;IACnB,CAAC;IAED,IAAI,CAAC,SAAiB,EAAE,IAAc;QACpC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,SAAS,CAAC,SAAiB,EAAE,OAAiC;QAC5D,OAAO,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;IACjD,CAAC;IAED,KAAK,CAAC,KAAK,CAAC,SAAiB;QAC3B,IAAI,CAAC,IAAI,CAAC,OAAO;YAAE,OAAO,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,OAAuB;QACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAC3C,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAEhC,MAAM,QAAQ,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;QACpD,KAAK,MAAM,KAAK,IAAI,QAAQ,EAAE,CAAC;YAC7B,MAAM,OAAO,GAAG,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC;gBAC1G,CAAC,CAAC,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,eAAe,CAAC,EAAE,IAAI,EAAE;gBAC/C,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC;YAClB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAEO,YAAY,CAAC,MAAsB,EAAE,OAAuB;QAClE,IAAI,CAAC,OAAO;YAAE,OAAO,MAAM,CAAC;QAC5B,IAAI,MAAM,GAAG,MAAM,CAAC;QACpB,IAAI,OAAO,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAClC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,OAAO,CAAC,OAAQ,CAAC,CAAC;QACzD,CAAC;QACD,IAAI,OAAO,CAAC,KAAK,KAAK,SAAS,EAAE,CAAC;YAChC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,IAAI,OAAO,CAAC,KAAM,CAAC,CAAC;QACvD,CAAC;QACD,IAAI,OAAO,CAAC,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxD,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC5C,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;CACF"}
|
|
@@ -0,0 +1,30 @@
|
|
|
1
|
+
import type { FallbackEntry, TokenUsage } from '@primo-ai/sdk';
|
|
2
|
+
import type { EventBus } from './event-bus.js';
|
|
3
|
+
export interface FallbackInvoker {
|
|
4
|
+
invoke(input: {
|
|
5
|
+
prompt: string;
|
|
6
|
+
tools?: unknown;
|
|
7
|
+
}): Promise<{
|
|
8
|
+
response: string;
|
|
9
|
+
tokenUsage: TokenUsage;
|
|
10
|
+
}>;
|
|
11
|
+
}
|
|
12
|
+
export interface FallbackRunnerOptions {
|
|
13
|
+
entries: FallbackEntry[];
|
|
14
|
+
invokerFactory: (model: string) => FallbackInvoker;
|
|
15
|
+
eventBus?: EventBus;
|
|
16
|
+
}
|
|
17
|
+
export declare class FallbackRunner {
|
|
18
|
+
private sorted;
|
|
19
|
+
private invokerFactory;
|
|
20
|
+
private eventBus?;
|
|
21
|
+
constructor(options: FallbackRunnerOptions);
|
|
22
|
+
run(input: {
|
|
23
|
+
prompt: string;
|
|
24
|
+
tools?: unknown;
|
|
25
|
+
}): Promise<{
|
|
26
|
+
response: string;
|
|
27
|
+
tokenUsage: TokenUsage;
|
|
28
|
+
}>;
|
|
29
|
+
}
|
|
30
|
+
//# sourceMappingURL=fallback-runner.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fallback-runner.d.ts","sourceRoot":"","sources":["../src/fallback-runner.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAC/D,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE/C,MAAM,WAAW,eAAe;IAC9B,MAAM,CAAC,KAAK,EAAE;QACZ,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,OAAO,CAAC;KACjB,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,UAAU,CAAA;KAAE,CAAC,CAAC;CAC3D;AAED,MAAM,WAAW,qBAAqB;IACpC,OAAO,EAAE,aAAa,EAAE,CAAC;IACzB,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,eAAe,CAAC;IACnD,QAAQ,CAAC,EAAE,QAAQ,CAAC;CACrB;AAED,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAkB;IAChC,OAAO,CAAC,cAAc,CAAqC;IAC3D,OAAO,CAAC,QAAQ,CAAC,CAAW;gBAEhB,OAAO,EAAE,qBAAqB;IAOpC,GAAG,CAAC,KAAK,EAAE;QACf,MAAM,EAAE,MAAM,CAAC;QACf,KAAK,CAAC,EAAE,OAAO,CAAC;KACjB,GAAG,OAAO,CAAC;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,UAAU,EAAE,UAAU,CAAA;KAAE,CAAC;CA+B1D"}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
export class FallbackRunner {
|
|
2
|
+
sorted;
|
|
3
|
+
invokerFactory;
|
|
4
|
+
eventBus;
|
|
5
|
+
constructor(options) {
|
|
6
|
+
// Sort by priority ascending (0 = highest priority)
|
|
7
|
+
this.sorted = [...options.entries].sort((a, b) => a.priority - b.priority);
|
|
8
|
+
this.invokerFactory = options.invokerFactory;
|
|
9
|
+
this.eventBus = options.eventBus;
|
|
10
|
+
}
|
|
11
|
+
async run(input) {
|
|
12
|
+
if (this.sorted.length === 0) {
|
|
13
|
+
throw new Error('No fallback entries configured');
|
|
14
|
+
}
|
|
15
|
+
let lastError;
|
|
16
|
+
for (let i = 0; i < this.sorted.length; i++) {
|
|
17
|
+
const entry = this.sorted[i];
|
|
18
|
+
const start = Date.now();
|
|
19
|
+
try {
|
|
20
|
+
const invoker = this.invokerFactory(entry.model);
|
|
21
|
+
return await invoker.invoke(input);
|
|
22
|
+
}
|
|
23
|
+
catch (err) {
|
|
24
|
+
lastError = err instanceof Error ? err : new Error(String(err));
|
|
25
|
+
const latencyMs = Date.now() - start;
|
|
26
|
+
// Emit fallback event if there's a next entry
|
|
27
|
+
if (i < this.sorted.length - 1) {
|
|
28
|
+
this.eventBus?.emit('task:fallback', {
|
|
29
|
+
from: entry.model,
|
|
30
|
+
to: this.sorted[i + 1].model,
|
|
31
|
+
error: lastError,
|
|
32
|
+
latencyMs,
|
|
33
|
+
});
|
|
34
|
+
}
|
|
35
|
+
}
|
|
36
|
+
}
|
|
37
|
+
throw lastError;
|
|
38
|
+
}
|
|
39
|
+
}
|
|
40
|
+
//# sourceMappingURL=fallback-runner.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"fallback-runner.js","sourceRoot":"","sources":["../src/fallback-runner.ts"],"names":[],"mappings":"AAgBA,MAAM,OAAO,cAAc;IACjB,MAAM,CAAkB;IACxB,cAAc,CAAqC;IACnD,QAAQ,CAAY;IAE5B,YAAY,OAA8B;QACxC,oDAAoD;QACpD,IAAI,CAAC,MAAM,GAAG,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC,QAAQ,CAAC,CAAC;QAC3E,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,cAAc,CAAC;QAC7C,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAGT;QACC,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC7B,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;QACpD,CAAC;QAED,IAAI,SAA4B,CAAC;QAEjC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC5C,MAAM,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;YAC7B,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACzB,IAAI,CAAC;gBACH,MAAM,OAAO,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;gBACjD,OAAO,MAAM,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACrC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,SAAS,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;gBAChE,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,KAAK,CAAC;gBAErC,8CAA8C;gBAC9C,IAAI,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBAC/B,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,eAAe,EAAE;wBACnC,IAAI,EAAE,KAAK,CAAC,KAAK;wBACjB,EAAE,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,KAAK;wBAC5B,KAAK,EAAE,SAAS;wBAChB,SAAS;qBACV,CAAC,CAAC;gBACL,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,SAAU,CAAC;IACnB,CAAC;CACF"}
|
|
@@ -0,0 +1,17 @@
|
|
|
1
|
+
import type { LanguageModel } from 'ai';
|
|
2
|
+
import type { ModelGateway } from '@primo-ai/sdk';
|
|
3
|
+
type ProviderFactory = (modelId: string) => LanguageModel;
|
|
4
|
+
export declare class BuiltInGateway implements ModelGateway {
|
|
5
|
+
name: string;
|
|
6
|
+
private customProviders;
|
|
7
|
+
private sdkCache;
|
|
8
|
+
registerProvider(name: string, factory: ProviderFactory): void;
|
|
9
|
+
canResolve(modelString: string): boolean;
|
|
10
|
+
resolve(modelString: string): Promise<unknown>;
|
|
11
|
+
}
|
|
12
|
+
/** Register a custom provider factory on the default gateway instance. */
|
|
13
|
+
export declare function registerProvider(name: string, factory: ProviderFactory): void;
|
|
14
|
+
/** @internal Access the default singleton instance. */
|
|
15
|
+
export declare function getDefaultBuiltInGateway(): BuiltInGateway;
|
|
16
|
+
export {};
|
|
17
|
+
//# sourceMappingURL=builtin-gateway.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"builtin-gateway.d.ts","sourceRoot":"","sources":["../../src/gateways/builtin-gateway.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAGlD,KAAK,eAAe,GAAG,CAAC,OAAO,EAAE,MAAM,KAAK,aAAa,CAAC;AAsB1D,qBAAa,cAAe,YAAW,YAAY;IACjD,IAAI,SAAa;IAEjB,OAAO,CAAC,eAAe,CAAsC;IAC7D,OAAO,CAAC,QAAQ,CAAkC;IAElD,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,GAAG,IAAI;IAI9D,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO;IAKlC,OAAO,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAqBrD;AAKD,0EAA0E;AAC1E,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,eAAe,GAAG,IAAI,CAE7E;AAED,uDAAuD;AACvD,wBAAgB,wBAAwB,IAAI,cAAc,CAEzD"}
|
|
@@ -0,0 +1,58 @@
|
|
|
1
|
+
import { parseModel } from '../parse-model.js';
|
|
2
|
+
const PROVIDER_MAP = {
|
|
3
|
+
openai: () => import('@ai-sdk/openai').then((m) => {
|
|
4
|
+
const sdk = m.createOpenAI();
|
|
5
|
+
return { languageModel: (id) => sdk(id) };
|
|
6
|
+
}),
|
|
7
|
+
anthropic: () => import('@ai-sdk/anthropic').then((m) => {
|
|
8
|
+
const sdk = m.createAnthropic();
|
|
9
|
+
return { languageModel: (id) => sdk(id) };
|
|
10
|
+
}),
|
|
11
|
+
google: () => import('@ai-sdk/google').then((m) => {
|
|
12
|
+
const sdk = m.createGoogleGenerativeAI();
|
|
13
|
+
return { languageModel: (id) => sdk(id) };
|
|
14
|
+
}),
|
|
15
|
+
deepseek: () => import('@ai-sdk/deepseek').then((m) => {
|
|
16
|
+
const sdk = m.createDeepSeek();
|
|
17
|
+
return { languageModel: (id) => sdk(id) };
|
|
18
|
+
}),
|
|
19
|
+
};
|
|
20
|
+
export class BuiltInGateway {
|
|
21
|
+
name = 'builtin';
|
|
22
|
+
customProviders = new Map();
|
|
23
|
+
sdkCache = new Map();
|
|
24
|
+
registerProvider(name, factory) {
|
|
25
|
+
this.customProviders.set(name, factory);
|
|
26
|
+
}
|
|
27
|
+
canResolve(modelString) {
|
|
28
|
+
const { provider } = parseModel(modelString);
|
|
29
|
+
return provider in PROVIDER_MAP || this.customProviders.has(provider);
|
|
30
|
+
}
|
|
31
|
+
async resolve(modelString) {
|
|
32
|
+
const { provider, modelId } = parseModel(modelString);
|
|
33
|
+
const custom = this.customProviders.get(provider);
|
|
34
|
+
if (custom)
|
|
35
|
+
return custom(modelId);
|
|
36
|
+
const loader = PROVIDER_MAP[provider];
|
|
37
|
+
if (!loader) {
|
|
38
|
+
throw new Error(`Unknown provider: "${provider}". Use registerProvider() or install an @ai-sdk/* package.`);
|
|
39
|
+
}
|
|
40
|
+
let instance = this.sdkCache.get(provider);
|
|
41
|
+
if (!instance) {
|
|
42
|
+
instance = await loader();
|
|
43
|
+
this.sdkCache.set(provider, instance);
|
|
44
|
+
}
|
|
45
|
+
return instance.languageModel(modelId);
|
|
46
|
+
}
|
|
47
|
+
}
|
|
48
|
+
// Singleton instance for backward-compatible module-level registerProvider()
|
|
49
|
+
const defaultInstance = new BuiltInGateway();
|
|
50
|
+
/** Register a custom provider factory on the default gateway instance. */
|
|
51
|
+
export function registerProvider(name, factory) {
|
|
52
|
+
defaultInstance.registerProvider(name, factory);
|
|
53
|
+
}
|
|
54
|
+
/** @internal Access the default singleton instance. */
|
|
55
|
+
export function getDefaultBuiltInGateway() {
|
|
56
|
+
return defaultInstance;
|
|
57
|
+
}
|
|
58
|
+
//# sourceMappingURL=builtin-gateway.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"builtin-gateway.js","sourceRoot":"","sources":["../../src/gateways/builtin-gateway.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAK/C,MAAM,YAAY,GAA+C;IAC/D,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;QAChD,MAAM,GAAG,GAAG,CAAC,CAAC,YAAY,EAAE,CAAC;QAC7B,OAAO,EAAE,aAAa,EAAE,CAAC,EAAU,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IACpD,CAAC,CAAC;IACF,SAAS,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;QACtD,MAAM,GAAG,GAAG,CAAC,CAAC,eAAe,EAAE,CAAC;QAChC,OAAO,EAAE,aAAa,EAAE,CAAC,EAAU,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IACpD,CAAC,CAAC;IACF,MAAM,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,gBAAgB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;QAChD,MAAM,GAAG,GAAG,CAAC,CAAC,wBAAwB,EAAE,CAAC;QACzC,OAAO,EAAE,aAAa,EAAE,CAAC,EAAU,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IACpD,CAAC,CAAC;IACF,QAAQ,EAAE,GAAG,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE;QACpD,MAAM,GAAG,GAAG,CAAC,CAAC,cAAc,EAAE,CAAC;QAC/B,OAAO,EAAE,aAAa,EAAE,CAAC,EAAU,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC;IACpD,CAAC,CAAC;CACH,CAAC;AAEF,MAAM,OAAO,cAAc;IACzB,IAAI,GAAG,SAAS,CAAC;IAET,eAAe,GAAG,IAAI,GAAG,EAA2B,CAAC;IACrD,QAAQ,GAAG,IAAI,GAAG,EAAuB,CAAC;IAElD,gBAAgB,CAAC,IAAY,EAAE,OAAwB;QACrD,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC1C,CAAC;IAED,UAAU,CAAC,WAAmB;QAC5B,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7C,OAAO,QAAQ,IAAI,YAAY,IAAI,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACxE,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,WAAmB;QAC/B,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;QAEtD,MAAM,MAAM,GAAG,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAClD,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC,OAAO,CAAC,CAAC;QAEnC,MAAM,MAAM,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CACb,sBAAsB,QAAQ,4DAA4D,CAC3F,CAAC;QACJ,CAAC;QAED,IAAI,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,QAAQ,GAAG,MAAM,MAAM,EAAE,CAAC;YAC1B,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;QACxC,CAAC;QAED,OAAO,QAAQ,CAAC,aAAa,CAAC,OAAO,CAAC,CAAC;IACzC,CAAC;CACF;AAED,6EAA6E;AAC7E,MAAM,eAAe,GAAG,IAAI,cAAc,EAAE,CAAC;AAE7C,0EAA0E;AAC1E,MAAM,UAAU,gBAAgB,CAAC,IAAY,EAAE,OAAwB;IACrE,eAAe,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;AAClD,CAAC;AAED,uDAAuD;AACvD,MAAM,UAAU,wBAAwB;IACtC,OAAO,eAAe,CAAC;AACzB,CAAC"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import type { LanguageModel } from 'ai';
|
|
2
|
+
import type { ModelGateway } from '@primo-ai/sdk';
|
|
3
|
+
export declare class GatewayChain {
|
|
4
|
+
private gateways;
|
|
5
|
+
register(gateway: ModelGateway): void;
|
|
6
|
+
resolve(modelString: string): Promise<LanguageModel>;
|
|
7
|
+
get size(): number;
|
|
8
|
+
}
|
|
9
|
+
//# sourceMappingURL=gateway-chain.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gateway-chain.d.ts","sourceRoot":"","sources":["../../src/gateways/gateway-chain.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAElD,qBAAa,YAAY;IACvB,OAAO,CAAC,QAAQ,CAAsB;IAEtC,QAAQ,CAAC,OAAO,EAAE,YAAY,GAAG,IAAI;IAI/B,OAAO,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC;IAW1D,IAAI,IAAI,IAAI,MAAM,CAEjB;CACF"}
|
|
@@ -0,0 +1,18 @@
|
|
|
1
|
+
export class GatewayChain {
|
|
2
|
+
gateways = [];
|
|
3
|
+
register(gateway) {
|
|
4
|
+
this.gateways.push(gateway);
|
|
5
|
+
}
|
|
6
|
+
async resolve(modelString) {
|
|
7
|
+
for (const gw of this.gateways) {
|
|
8
|
+
if (gw.canResolve(modelString)) {
|
|
9
|
+
return gw.resolve(modelString);
|
|
10
|
+
}
|
|
11
|
+
}
|
|
12
|
+
throw new Error(`Unknown provider for model: "${modelString}". No gateway can resolve it. Use registerProvider() or add a custom gateway.`);
|
|
13
|
+
}
|
|
14
|
+
get size() {
|
|
15
|
+
return this.gateways.length;
|
|
16
|
+
}
|
|
17
|
+
}
|
|
18
|
+
//# sourceMappingURL=gateway-chain.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"gateway-chain.js","sourceRoot":"","sources":["../../src/gateways/gateway-chain.ts"],"names":[],"mappings":"AAGA,MAAM,OAAO,YAAY;IACf,QAAQ,GAAmB,EAAE,CAAC;IAEtC,QAAQ,CAAC,OAAqB;QAC5B,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,WAAmB;QAC/B,KAAK,MAAM,EAAE,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAC/B,IAAI,EAAE,CAAC,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;gBAC/B,OAAO,EAAE,CAAC,OAAO,CAAC,WAAW,CAA2B,CAAC;YAC3D,CAAC;QACH,CAAC;QACD,MAAM,IAAI,KAAK,CACb,gCAAgC,WAAW,+EAA+E,CAC3H,CAAC;IACJ,CAAC;IAED,IAAI,IAAI;QACN,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;IAC9B,CAAC;CACF"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { ModelGateway, GatewayConfig } from '@primo-ai/sdk';
|
|
2
|
+
export declare class OpenAICompatibleGateway implements ModelGateway {
|
|
3
|
+
readonly name: string;
|
|
4
|
+
private readonly url;
|
|
5
|
+
private readonly apiKey?;
|
|
6
|
+
constructor(config: GatewayConfig);
|
|
7
|
+
canResolve(modelString: string): boolean;
|
|
8
|
+
resolve(modelString: string): Promise<unknown>;
|
|
9
|
+
}
|
|
10
|
+
//# sourceMappingURL=openai-compatible-gateway.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-compatible-gateway.d.ts","sourceRoot":"","sources":["../../src/gateways/openai-compatible-gateway.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AAGjE,qBAAa,uBAAwB,YAAW,YAAY;IAC1D,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IACtB,OAAO,CAAC,QAAQ,CAAC,GAAG,CAAS;IAC7B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAS;gBAErB,MAAM,EAAE,aAAa;IAMjC,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO;IAKlC,OAAO,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC;CAUrD"}
|
|
@@ -0,0 +1,26 @@
|
|
|
1
|
+
import { parseModel } from '../parse-model.js';
|
|
2
|
+
export class OpenAICompatibleGateway {
|
|
3
|
+
name;
|
|
4
|
+
url;
|
|
5
|
+
apiKey;
|
|
6
|
+
constructor(config) {
|
|
7
|
+
this.name = config.name;
|
|
8
|
+
this.url = config.url;
|
|
9
|
+
this.apiKey = config.apiKey;
|
|
10
|
+
}
|
|
11
|
+
canResolve(modelString) {
|
|
12
|
+
const { provider } = parseModel(modelString);
|
|
13
|
+
return provider === this.name;
|
|
14
|
+
}
|
|
15
|
+
async resolve(modelString) {
|
|
16
|
+
const { modelId } = parseModel(modelString);
|
|
17
|
+
const { createOpenAICompatible } = await import('@ai-sdk/openai-compatible');
|
|
18
|
+
const sdk = createOpenAICompatible({
|
|
19
|
+
name: this.name,
|
|
20
|
+
baseURL: this.url,
|
|
21
|
+
...(this.apiKey ? { apiKey: this.apiKey } : {}),
|
|
22
|
+
});
|
|
23
|
+
return sdk.chatModel(modelId);
|
|
24
|
+
}
|
|
25
|
+
}
|
|
26
|
+
//# sourceMappingURL=openai-compatible-gateway.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"openai-compatible-gateway.js","sourceRoot":"","sources":["../../src/gateways/openai-compatible-gateway.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/C,MAAM,OAAO,uBAAuB;IACzB,IAAI,CAAS;IACL,GAAG,CAAS;IACZ,MAAM,CAAU;IAEjC,YAAY,MAAqB;QAC/B,IAAI,CAAC,IAAI,GAAG,MAAM,CAAC,IAAI,CAAC;QACxB,IAAI,CAAC,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;QACtB,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;IAC9B,CAAC;IAED,UAAU,CAAC,WAAmB;QAC5B,MAAM,EAAE,QAAQ,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;QAC7C,OAAO,QAAQ,KAAK,IAAI,CAAC,IAAI,CAAC;IAChC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,WAAmB;QAC/B,MAAM,EAAE,OAAO,EAAE,GAAG,UAAU,CAAC,WAAW,CAAC,CAAC;QAC5C,MAAM,EAAE,sBAAsB,EAAE,GAAG,MAAM,MAAM,CAAC,2BAA2B,CAAC,CAAC;QAC7E,MAAM,GAAG,GAAG,sBAAsB,CAAC;YACjC,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI,CAAC,GAAG;YACjB,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChD,CAAC,CAAC;QACH,OAAO,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAChC,CAAC;CACF"}
|
|
@@ -0,0 +1,44 @@
|
|
|
1
|
+
import type { CompressionStrategy, HarnessAPI, Hook, Processor, ResourceDeclaration, StageMutation, StageName, ToolDefinition } from '@primo-ai/sdk';
|
|
2
|
+
import type { PipelineRunner } from './pipeline.js';
|
|
3
|
+
import type { ToolRegistry } from './tool-registry.js';
|
|
4
|
+
import type { ContextBuilder } from './context-builder.js';
|
|
5
|
+
import type { HookManager } from './hook-manager.js';
|
|
6
|
+
import type { EventBus } from './event-bus.js';
|
|
7
|
+
import type { EventSystem } from './event-system.js';
|
|
8
|
+
export interface HarnessDeps {
|
|
9
|
+
runner: PipelineRunner;
|
|
10
|
+
registry: ToolRegistry;
|
|
11
|
+
hookManager: HookManager;
|
|
12
|
+
eventSystem: EventSystem;
|
|
13
|
+
eventBus: EventBus;
|
|
14
|
+
contextBuilder?: ContextBuilder;
|
|
15
|
+
emitEvent: (eventType: string, data?: unknown) => void;
|
|
16
|
+
registerProvider: (name: string, factory: unknown) => void;
|
|
17
|
+
mutateStages?: (mutation: StageMutation) => void;
|
|
18
|
+
}
|
|
19
|
+
export declare class HarnessAPIImpl implements HarnessAPI {
|
|
20
|
+
private deps;
|
|
21
|
+
private commands;
|
|
22
|
+
private resources;
|
|
23
|
+
private unsubFns;
|
|
24
|
+
private frozen;
|
|
25
|
+
constructor(deps: HarnessDeps);
|
|
26
|
+
freeze(): void;
|
|
27
|
+
registerProcessor(stage: StageName, processor: Processor): void;
|
|
28
|
+
registerTool(tool: ToolDefinition): void;
|
|
29
|
+
unregisterTool(name: string): boolean;
|
|
30
|
+
registerCommand(name: string, handler: (args: string) => Promise<void>): void;
|
|
31
|
+
registerHook(hook: Hook): void;
|
|
32
|
+
subscribe(eventType: string, handler: (data?: unknown) => void): () => void;
|
|
33
|
+
registerResource(declaration: ResourceDeclaration): void;
|
|
34
|
+
registerProvider(name: string, factory: unknown): void;
|
|
35
|
+
registerCompressionStrategy(strategy: CompressionStrategy): void;
|
|
36
|
+
emit(eventType: string, data?: unknown): void;
|
|
37
|
+
insertStage(phase: 'preLoop' | 'loop' | 'postLoop', after: StageName, newStage: StageName): void;
|
|
38
|
+
removeStage(phase: 'preLoop' | 'loop' | 'postLoop', stage: StageName): void;
|
|
39
|
+
replaceStages(phase: 'preLoop' | 'loop' | 'postLoop', stages: StageName[]): void;
|
|
40
|
+
getCommands(): Map<string, (args: string) => Promise<void>>;
|
|
41
|
+
getResources(): ResourceDeclaration[];
|
|
42
|
+
getUnsubFns(): Array<() => void>;
|
|
43
|
+
}
|
|
44
|
+
//# sourceMappingURL=harness.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"harness.d.ts","sourceRoot":"","sources":["../src/harness.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,mBAAmB,EACnB,UAAU,EACV,IAAI,EACJ,SAAS,EACT,mBAAmB,EACnB,aAAa,EACb,SAAS,EACT,cAAc,EACf,MAAM,eAAe,CAAC;AACvB,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AACpD,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AACvD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAC3D,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AACrD,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAErD,MAAM,WAAW,WAAW;IAC1B,MAAM,EAAE,cAAc,CAAC;IACvB,QAAQ,EAAE,YAAY,CAAC;IACvB,WAAW,EAAE,WAAW,CAAC;IACzB,WAAW,EAAE,WAAW,CAAC;IACzB,QAAQ,EAAE,QAAQ,CAAC;IACnB,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,SAAS,EAAE,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,CAAC;IACvD,gBAAgB,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,KAAK,IAAI,CAAC;IAC3D,YAAY,CAAC,EAAE,CAAC,QAAQ,EAAE,aAAa,KAAK,IAAI,CAAC;CAClD;AAED,qBAAa,cAAe,YAAW,UAAU;IAMnC,OAAO,CAAC,IAAI;IALxB,OAAO,CAAC,QAAQ,CAAsD;IACtE,OAAO,CAAC,SAAS,CAA6B;IAC9C,OAAO,CAAC,QAAQ,CAAyB;IACzC,OAAO,CAAC,MAAM,CAAS;gBAEH,IAAI,EAAE,WAAW;IAErC,MAAM,IAAI,IAAI;IAEd,iBAAiB,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,GAAG,IAAI;IAI/D,YAAY,CAAC,IAAI,EAAE,cAAc,GAAG,IAAI;IAIxC,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO;IAIrC,eAAe,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,GAAG,IAAI;IAI7E,YAAY,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAI9B,SAAS,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,EAAE,OAAO,KAAK,IAAI,GAAG,MAAM,IAAI;IAM3E,gBAAgB,CAAC,WAAW,EAAE,mBAAmB,GAAG,IAAI;IAIxD,gBAAgB,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,IAAI;IAItD,2BAA2B,CAAC,QAAQ,EAAE,mBAAmB,GAAG,IAAI;IAMhE,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,CAAC,EAAE,OAAO,GAAG,IAAI;IAI7C,WAAW,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM,GAAG,UAAU,EAAE,KAAK,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,GAAG,IAAI;IAKhG,WAAW,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM,GAAG,UAAU,EAAE,KAAK,EAAE,SAAS,GAAG,IAAI;IAK3E,aAAa,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM,GAAG,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,IAAI;IAKhF,WAAW,IAAI,GAAG,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IAI3D,YAAY,IAAI,mBAAmB,EAAE;IAIrC,WAAW,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC;CAGjC"}
|
package/dist/harness.js
ADDED
|
@@ -0,0 +1,70 @@
|
|
|
1
|
+
export class HarnessAPIImpl {
|
|
2
|
+
deps;
|
|
3
|
+
commands = new Map();
|
|
4
|
+
resources = [];
|
|
5
|
+
unsubFns = [];
|
|
6
|
+
frozen = false;
|
|
7
|
+
constructor(deps) {
|
|
8
|
+
this.deps = deps;
|
|
9
|
+
}
|
|
10
|
+
freeze() { this.frozen = true; }
|
|
11
|
+
registerProcessor(stage, processor) {
|
|
12
|
+
this.deps.runner.register(processor);
|
|
13
|
+
}
|
|
14
|
+
registerTool(tool) {
|
|
15
|
+
this.deps.registry.register(tool);
|
|
16
|
+
}
|
|
17
|
+
unregisterTool(name) {
|
|
18
|
+
return this.deps.registry.unregister(name);
|
|
19
|
+
}
|
|
20
|
+
registerCommand(name, handler) {
|
|
21
|
+
this.commands.set(name, handler);
|
|
22
|
+
}
|
|
23
|
+
registerHook(hook) {
|
|
24
|
+
this.deps.hookManager.register(hook);
|
|
25
|
+
}
|
|
26
|
+
subscribe(eventType, handler) {
|
|
27
|
+
const unsub = this.deps.eventBus.subscribe(eventType, handler);
|
|
28
|
+
this.unsubFns.push(unsub);
|
|
29
|
+
return unsub;
|
|
30
|
+
}
|
|
31
|
+
registerResource(declaration) {
|
|
32
|
+
this.resources.push(declaration);
|
|
33
|
+
}
|
|
34
|
+
registerProvider(name, factory) {
|
|
35
|
+
this.deps.registerProvider(name, factory);
|
|
36
|
+
}
|
|
37
|
+
registerCompressionStrategy(strategy) {
|
|
38
|
+
if (this.deps.contextBuilder) {
|
|
39
|
+
this.deps.contextBuilder.setCompressionStrategy(strategy);
|
|
40
|
+
}
|
|
41
|
+
}
|
|
42
|
+
emit(eventType, data) {
|
|
43
|
+
this.deps.emitEvent(eventType, data);
|
|
44
|
+
}
|
|
45
|
+
insertStage(phase, after, newStage) {
|
|
46
|
+
if (this.frozen)
|
|
47
|
+
throw new Error('Cannot mutate stages after agent has started running');
|
|
48
|
+
this.deps.mutateStages?.({ type: 'insert', phase, after, stage: newStage });
|
|
49
|
+
}
|
|
50
|
+
removeStage(phase, stage) {
|
|
51
|
+
if (this.frozen)
|
|
52
|
+
throw new Error('Cannot mutate stages after agent has started running');
|
|
53
|
+
this.deps.mutateStages?.({ type: 'remove', phase, stage });
|
|
54
|
+
}
|
|
55
|
+
replaceStages(phase, stages) {
|
|
56
|
+
if (this.frozen)
|
|
57
|
+
throw new Error('Cannot mutate stages after agent has started running');
|
|
58
|
+
this.deps.mutateStages?.({ type: 'replace', phase, stages });
|
|
59
|
+
}
|
|
60
|
+
getCommands() {
|
|
61
|
+
return this.commands;
|
|
62
|
+
}
|
|
63
|
+
getResources() {
|
|
64
|
+
return this.resources;
|
|
65
|
+
}
|
|
66
|
+
getUnsubFns() {
|
|
67
|
+
return this.unsubFns;
|
|
68
|
+
}
|
|
69
|
+
}
|
|
70
|
+
//# sourceMappingURL=harness.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"harness.js","sourceRoot":"","sources":["../src/harness.ts"],"names":[],"mappings":"AA6BA,MAAM,OAAO,cAAc;IAML;IALZ,QAAQ,GAAG,IAAI,GAAG,EAA2C,CAAC;IAC9D,SAAS,GAA0B,EAAE,CAAC;IACtC,QAAQ,GAAsB,EAAE,CAAC;IACjC,MAAM,GAAG,KAAK,CAAC;IAEvB,YAAoB,IAAiB;QAAjB,SAAI,GAAJ,IAAI,CAAa;IAAG,CAAC;IAEzC,MAAM,KAAW,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC,CAAC,CAAC;IAEtC,iBAAiB,CAAC,KAAgB,EAAE,SAAoB;QACtD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IACvC,CAAC;IAED,YAAY,CAAC,IAAoB;QAC/B,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACpC,CAAC;IAED,cAAc,CAAC,IAAY;QACzB,OAAO,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAC7C,CAAC;IAED,eAAe,CAAC,IAAY,EAAE,OAAwC;QACpE,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;IAED,YAAY,CAAC,IAAU;QACrB,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,SAAS,CAAC,SAAiB,EAAE,OAAiC;QAC5D,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC1B,OAAO,KAAK,CAAC;IACf,CAAC;IAED,gBAAgB,CAAC,WAAgC;QAC/C,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;IACnC,CAAC;IAED,gBAAgB,CAAC,IAAY,EAAE,OAAgB;QAC7C,IAAI,CAAC,IAAI,CAAC,gBAAgB,CAAC,IAAI,EAAE,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,2BAA2B,CAAC,QAA6B;QACvD,IAAI,IAAI,CAAC,IAAI,CAAC,cAAc,EAAE,CAAC;YAC7B,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC;IACH,CAAC;IAED,IAAI,CAAC,SAAiB,EAAE,IAAc;QACpC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IACvC,CAAC;IAED,WAAW,CAAC,KAAsC,EAAE,KAAgB,EAAE,QAAmB;QACvF,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QACzF,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;IAC9E,CAAC;IAED,WAAW,CAAC,KAAsC,EAAE,KAAgB;QAClE,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QACzF,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7D,CAAC;IAED,aAAa,CAAC,KAAsC,EAAE,MAAmB;QACvE,IAAI,IAAI,CAAC,MAAM;YAAE,MAAM,IAAI,KAAK,CAAC,sDAAsD,CAAC,CAAC;QACzF,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,YAAY;QACV,OAAO,IAAI,CAAC,SAAS,CAAC;IACxB,CAAC;IAED,WAAW;QACT,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;CACF"}
|
|
@@ -0,0 +1,22 @@
|
|
|
1
|
+
import type { Hook, HookPoint, HookProfile } from '@primo-ai/sdk';
|
|
2
|
+
import type { EventBus } from './event-bus.js';
|
|
3
|
+
export interface HookManagerOptions {
|
|
4
|
+
profile?: HookProfile;
|
|
5
|
+
disabledHooks?: string[];
|
|
6
|
+
}
|
|
7
|
+
export declare class HookManager {
|
|
8
|
+
private hooks;
|
|
9
|
+
private sortedHooks;
|
|
10
|
+
private disabledPoints;
|
|
11
|
+
private profile;
|
|
12
|
+
private disabledHookNames;
|
|
13
|
+
private eventBus;
|
|
14
|
+
constructor(eventBus: EventBus, options?: HookManagerOptions);
|
|
15
|
+
register(hook: Hook): void;
|
|
16
|
+
invoke(point: HookPoint, input: unknown, output: unknown): Promise<void>;
|
|
17
|
+
setProfile(profile: HookProfile): void;
|
|
18
|
+
disablePoint(point: HookPoint): void;
|
|
19
|
+
private getSortedHooks;
|
|
20
|
+
private bridge;
|
|
21
|
+
}
|
|
22
|
+
//# sourceMappingURL=hook-manager.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hook-manager.d.ts","sourceRoot":"","sources":["../src/hook-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAClE,OAAO,KAAK,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAgB/C,MAAM,WAAW,kBAAkB;IACjC,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;CAC1B;AAED,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAgC;IAC7C,OAAO,CAAC,WAAW,CAAgC;IACnD,OAAO,CAAC,cAAc,CAAwB;IAC9C,OAAO,CAAC,OAAO,CAA2B;IAC1C,OAAO,CAAC,iBAAiB,CAAc;IACvC,OAAO,CAAC,QAAQ,CAAW;gBAEf,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,EAAE,kBAAkB;IAM5D,QAAQ,CAAC,IAAI,EAAE,IAAI,GAAG,IAAI;IAUpB,MAAM,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,EAAE,OAAO,EAAE,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,IAAI,CAAC;IA8B9E,UAAU,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI;IAItC,YAAY,CAAC,KAAK,EAAE,SAAS,GAAG,IAAI;IAIpC,OAAO,CAAC,cAAc;IAetB,OAAO,CAAC,MAAM;CAQf"}
|
|
@@ -0,0 +1,92 @@
|
|
|
1
|
+
const DEFAULT_PRIORITY = 100;
|
|
2
|
+
const HOOK_TO_EVENT = {
|
|
3
|
+
'agent.start': 'agent:start',
|
|
4
|
+
'agent.end': 'agent:end',
|
|
5
|
+
'stage.before': 'stage:before',
|
|
6
|
+
'stage.after': 'stage:after',
|
|
7
|
+
'llm.before': 'llm:before',
|
|
8
|
+
'llm.after': 'llm:after',
|
|
9
|
+
'tool.before': 'tool:before',
|
|
10
|
+
'tool.after': 'tool:after',
|
|
11
|
+
'error': 'error',
|
|
12
|
+
};
|
|
13
|
+
export class HookManager {
|
|
14
|
+
hooks = new Map();
|
|
15
|
+
sortedHooks = new Map();
|
|
16
|
+
disabledPoints = new Set();
|
|
17
|
+
profile = 'standard';
|
|
18
|
+
disabledHookNames;
|
|
19
|
+
eventBus;
|
|
20
|
+
constructor(eventBus, options) {
|
|
21
|
+
this.eventBus = eventBus;
|
|
22
|
+
if (options?.profile)
|
|
23
|
+
this.profile = options.profile;
|
|
24
|
+
this.disabledHookNames = new Set(options?.disabledHooks ?? []);
|
|
25
|
+
}
|
|
26
|
+
register(hook) {
|
|
27
|
+
let list = this.hooks.get(hook.point);
|
|
28
|
+
if (!list) {
|
|
29
|
+
list = [];
|
|
30
|
+
this.hooks.set(hook.point, list);
|
|
31
|
+
}
|
|
32
|
+
list.push(hook);
|
|
33
|
+
this.sortedHooks.delete(hook.point);
|
|
34
|
+
}
|
|
35
|
+
async invoke(point, input, output) {
|
|
36
|
+
if (this.disabledPoints.has(point))
|
|
37
|
+
return;
|
|
38
|
+
const hooks = this.getSortedHooks(point);
|
|
39
|
+
if (hooks.length === 0) {
|
|
40
|
+
this.bridge(point, input);
|
|
41
|
+
return;
|
|
42
|
+
}
|
|
43
|
+
// minimal profile: only run error hooks
|
|
44
|
+
if (this.profile === 'minimal' && point !== 'error') {
|
|
45
|
+
this.bridge(point, input);
|
|
46
|
+
return;
|
|
47
|
+
}
|
|
48
|
+
for (const hook of hooks) {
|
|
49
|
+
if (hook.name && this.disabledHookNames.has(hook.name))
|
|
50
|
+
continue;
|
|
51
|
+
try {
|
|
52
|
+
await hook.handler(input, output);
|
|
53
|
+
}
|
|
54
|
+
catch (err) {
|
|
55
|
+
this.eventBus.emit('hook:error', { point, hookName: hook.name, error: err instanceof Error ? err.message : String(err) });
|
|
56
|
+
if (this.profile === 'strict')
|
|
57
|
+
break; // circuit-break
|
|
58
|
+
// standard: isolate error, continue
|
|
59
|
+
}
|
|
60
|
+
}
|
|
61
|
+
this.bridge(point, input);
|
|
62
|
+
}
|
|
63
|
+
setProfile(profile) {
|
|
64
|
+
this.profile = profile;
|
|
65
|
+
}
|
|
66
|
+
disablePoint(point) {
|
|
67
|
+
this.disabledPoints.add(point);
|
|
68
|
+
}
|
|
69
|
+
getSortedHooks(point) {
|
|
70
|
+
const cached = this.sortedHooks.get(point);
|
|
71
|
+
if (cached)
|
|
72
|
+
return cached;
|
|
73
|
+
const hooks = this.hooks.get(point) ?? [];
|
|
74
|
+
if (hooks.length <= 1) {
|
|
75
|
+
this.sortedHooks.set(point, hooks);
|
|
76
|
+
return hooks;
|
|
77
|
+
}
|
|
78
|
+
const sorted = [...hooks].sort((a, b) => (a.priority ?? DEFAULT_PRIORITY) - (b.priority ?? DEFAULT_PRIORITY));
|
|
79
|
+
this.sortedHooks.set(point, sorted);
|
|
80
|
+
return sorted;
|
|
81
|
+
}
|
|
82
|
+
bridge(point, data) {
|
|
83
|
+
const eventType = HOOK_TO_EVENT[point] ?? point.replace('.', ':');
|
|
84
|
+
try {
|
|
85
|
+
this.eventBus.emit(eventType, data);
|
|
86
|
+
}
|
|
87
|
+
catch {
|
|
88
|
+
// bridge failure is isolated — never affects hook pipeline
|
|
89
|
+
}
|
|
90
|
+
}
|
|
91
|
+
}
|
|
92
|
+
//# sourceMappingURL=hook-manager.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"hook-manager.js","sourceRoot":"","sources":["../src/hook-manager.ts"],"names":[],"mappings":"AAGA,MAAM,gBAAgB,GAAG,GAAG,CAAC;AAE7B,MAAM,aAAa,GAA2B;IAC5C,aAAa,EAAE,aAAa;IAC5B,WAAW,EAAE,WAAW;IACxB,cAAc,EAAE,cAAc;IAC9B,aAAa,EAAE,aAAa;IAC5B,YAAY,EAAE,YAAY;IAC1B,WAAW,EAAE,WAAW;IACxB,aAAa,EAAE,aAAa;IAC5B,YAAY,EAAE,YAAY;IAC1B,OAAO,EAAE,OAAO;CACjB,CAAC;AAOF,MAAM,OAAO,WAAW;IACd,KAAK,GAAG,IAAI,GAAG,EAAqB,CAAC;IACrC,WAAW,GAAG,IAAI,GAAG,EAAqB,CAAC;IAC3C,cAAc,GAAG,IAAI,GAAG,EAAa,CAAC;IACtC,OAAO,GAAgB,UAAU,CAAC;IAClC,iBAAiB,CAAc;IAC/B,QAAQ,CAAW;IAE3B,YAAY,QAAkB,EAAE,OAA4B;QAC1D,IAAI,CAAC,QAAQ,GAAG,QAAQ,CAAC;QACzB,IAAI,OAAO,EAAE,OAAO;YAAE,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QACrD,IAAI,CAAC,iBAAiB,GAAG,IAAI,GAAG,CAAC,OAAO,EAAE,aAAa,IAAI,EAAE,CAAC,CAAC;IACjE,CAAC;IAED,QAAQ,CAAC,IAAU;QACjB,IAAI,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACtC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,EAAE,CAAC;YACV,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACnC,CAAC;QACD,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAChB,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,KAAgB,EAAE,KAAc,EAAE,MAAe;QAC5D,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC;YAAE,OAAO;QAE3C,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,CAAC;QACzC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACvB,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAC1B,OAAO;QACT,CAAC;QAED,wCAAwC;QACxC,IAAI,IAAI,CAAC,OAAO,KAAK,SAAS,IAAI,KAAK,KAAK,OAAO,EAAE,CAAC;YACpD,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YAC1B,OAAO;QACT,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,IAAI,CAAC,IAAI,IAAI,IAAI,CAAC,iBAAiB,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC;gBAAE,SAAS;YAEjE,IAAI,CAAC;gBACH,MAAM,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;YACpC,CAAC;YAAC,OAAO,GAAG,EAAE,CAAC;gBACb,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,CAAC,IAAI,EAAE,KAAK,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBAC1H,IAAI,IAAI,CAAC,OAAO,KAAK,QAAQ;oBAAE,MAAM,CAAC,gBAAgB;gBACtD,oCAAoC;YACtC,CAAC;QACH,CAAC;QAED,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC5B,CAAC;IAED,UAAU,CAAC,OAAoB;QAC7B,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,YAAY,CAAC,KAAgB;QAC3B,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACjC,CAAC;IAEO,cAAc,CAAC,KAAgB;QACrC,MAAM,MAAM,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAC3C,IAAI,MAAM;YAAE,OAAO,MAAM,CAAC;QAE1B,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC;QAC1C,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,EAAE,CAAC;YACtB,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;YACnC,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,MAAM,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,gBAAgB,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,IAAI,gBAAgB,CAAC,CAAC,CAAC;QAC9G,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACpC,OAAO,MAAM,CAAC;IAChB,CAAC;IAEO,MAAM,CAAC,KAAgB,EAAE,IAAa;QAC5C,MAAM,SAAS,GAAG,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAClE,IAAI,CAAC;YACH,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QACtC,CAAC;QAAC,MAAM,CAAC;YACP,2DAA2D;QAC7D,CAAC;IACH,CAAC;CACF"}
|
package/dist/index.d.ts
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
1
|
+
export { PipelineRunner, type PipelineRunnerOptions } from './pipeline.js';
|
|
2
|
+
export { Agent, type AgentDependencies, type AgentRunResult } from './agent.js';
|
|
3
|
+
export { LLMInvoker, type LLMInvokerOptions, type LLMInvokeInput, type LLMInvokeResult, type LLMStreamHandle } from './llm-invoker.js';
|
|
4
|
+
export { resolveModel, registerProvider, parseModel, type ParsedModel } from './model-resolver.js';
|
|
5
|
+
export { streamWithRetry, type RetryOptions } from './retry.js';
|
|
6
|
+
export { PluginManager, type PluginFactory } from './plugin-manager.js';
|
|
7
|
+
export { EventBus } from './event-bus.js';
|
|
8
|
+
export { EventSystem, REPLAY_SENTINEL } from './event-system.js';
|
|
9
|
+
export { StorageReplayBackend } from './storage-replay-backend.js';
|
|
10
|
+
export { HookManager, type HookManagerOptions } from './hook-manager.js';
|
|
11
|
+
export { ToolRegistry, type AiSdkToolSchema, type ToolRegistryOptions } from './tool-registry.js';
|
|
12
|
+
export { FilesystemSessionStorage } from './session-storage.js';
|
|
13
|
+
export { SessionPersistence } from './session-persistence.js';
|
|
14
|
+
export { SessionManagerImpl } from './session-manager.js';
|
|
15
|
+
export { createSubAgentTool } from './sub-agent.js';
|
|
16
|
+
export { deepMerge } from './config-merge.js';
|
|
17
|
+
export { resolveDynamic } from './dynamic-resolver.js';
|
|
18
|
+
export { matchProfile, applyProfile } from './model-profile.js';
|
|
19
|
+
export { ConfigLoader, type ConfigSource } from './config.js';
|
|
20
|
+
export { ConcurrencyController } from './concurrency-controller.js';
|
|
21
|
+
export { FallbackRunner, type FallbackInvoker, type FallbackRunnerOptions } from './fallback-runner.js';
|
|
22
|
+
export { TaskManagerImpl } from './task-manager.js';
|
|
23
|
+
export { StateMachine, type AgentState } from './state-machine.js';
|
|
24
|
+
export { ModelFactory } from './model-factory.js';
|
|
25
|
+
export { LoopOrchestrator, type LoopOptions } from './loop-orchestrator.js';
|
|
26
|
+
export { serialize, deserialize } from './serialize.js';
|
|
27
|
+
export { InMemoryCheckpointStore, JsonlCheckpointStore } from './checkpoint-store.js';
|
|
28
|
+
export { ContextBuilder, type ContextBuilderOptions } from './context-builder.js';
|
|
29
|
+
export { HarnessAPIImpl, type HarnessDeps } from './harness.js';
|
|
30
|
+
export { TiktokenCounter } from './token-counter.js';
|
|
31
|
+
export type { StreamEvent } from '@primo-ai/sdk';
|
|
32
|
+
export { OpenAICompatibleGateway } from './gateways/openai-compatible-gateway.js';
|
|
33
|
+
export { AgentForgeError, RecoverableError, FatalError, AuthError, ModelNotFoundError, ToolExecutionError, type AgentErrorOptions, } from './errors.js';
|
|
34
|
+
//# sourceMappingURL=index.d.ts.map
|