@primo-ai/core 0.1.0

package/dist/model-profile.js

@@ -0,0 +1,57 @@
+/**
+ * Find the first ModelProfile whose modelPattern matches the given model string.
+ * String patterns use simple `includes`; RegExp patterns use `test()`.
+ */
+export function matchProfile(model, profiles) {
+    for (const profile of profiles) {
+        if (profile.modelPattern instanceof RegExp) {
+            if (profile.modelPattern.test(model))
+                return profile;
+        }
+        else {
+            if (model.includes(profile.modelPattern))
+                return profile;
+        }
+    }
+    return undefined;
+}
+/**
+ * Apply a ModelProfile to a PipelineContext, returning a new context
+ * (original is not mutated).
+ *
+ * - extraPromptFragments are converted from PromptFragment[] to strings
+ *   by extracting `.content`, then appended to promptFragments.
+ * - systemPromptSuffix is appended as a plain string after fragments.
+ * - toolOverrides.exclude removes matching tools; description overrides update them.
+ */
+export function applyProfile(ctx, profile) {
+    // Build new promptFragments: existing + extra (converted) + suffix
+    const newFragments = profile.extraPromptFragments?.map((f) => f.content) ?? [];
+    const promptFragments = [...ctx.agent.promptFragments, ...newFragments];
+    if (profile.systemPromptSuffix) {
+        promptFragments.push(profile.systemPromptSuffix);
+    }
+    // Apply tool overrides
+    const overrides = profile.toolOverrides ?? {};
+    const toolDeclarations = ctx.agent.toolDeclarations
+        .filter((tool) => {
+        const override = overrides[tool.name];
+        return !override?.exclude;
+    })
+        .map((tool) => {
+        const override = overrides[tool.name];
+        if (override?.description) {
+            return { ...tool, description: override.description };
+        }
+        return tool;
+    });
+    return {
+        ...ctx,
+        agent: {
+            ...ctx.agent,
+            promptFragments,
+            toolDeclarations,
+        },
+    };
+}
+//# sourceMappingURL=model-profile.js.map

package/dist/model-profile.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-profile.js","sourceRoot":"","sources":["../src/model-profile.ts"],"names":[],"mappings":"AAEA;;;GAGG;AACH,MAAM,UAAU,YAAY,CAC1B,KAAa,EACb,QAAwB;IAExB,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,IAAI,OAAO,CAAC,YAAY,YAAY,MAAM,EAAE,CAAC;YAC3C,IAAI,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC;gBAAE,OAAO,OAAO,CAAC;QACvD,CAAC;aAAM,CAAC;YACN,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,YAAY,CAAC;gBAAE,OAAO,OAAO,CAAC;QAC3D,CAAC;IACH,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,YAAY,CAC1B,GAAoB,EACpB,OAAqB;IAErB,mEAAmE;IACnE,MAAM,YAAY,GAAG,OAAO,CAAC,oBAAoB,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IAC/E,MAAM,eAAe,GAAG,CAAC,GAAG,GAAG,CAAC,KAAK,CAAC,eAAe,EAAE,GAAG,YAAY,CAAC,CAAC;IAExE,IAAI,OAAO,CAAC,kBAAkB,EAAE,CAAC;QAC/B,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC;IACnD,CAAC;IAED,uBAAuB;IACvB,MAAM,SAAS,GAAG,OAAO,CAAC,aAAa,IAAI,EAAE,CAAC;IAC9C,MAAM,gBAAgB,GAAG,GAAG,CAAC,KAAK,CAAC,gBAAgB;SAChD,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;QACf,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,OAAO,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC5B,CAAC,CAAC;SACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;QACZ,MAAM,QAAQ,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACtC,IAAI,QAAQ,EAAE,WAAW,EAAE,CAAC;YAC1B,OAAO,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,CAAC;QACxD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC,CAAC,CAAC;IAEL,OAAO;QACL,GAAG,GAAG;QACN,KAAK,EAAE;YACL,GAAG,GAAG,CAAC,KAAK;YACZ,eAAe;YACf,gBAAgB;SACjB;KACF,CAAC;AACJ,CAAC"}

package/dist/model-resolver.d.ts

@@ -0,0 +1,16 @@
+import type { LanguageModel } from 'ai';
+import type { GatewayConfig } from '@primo-ai/sdk';
+import { GatewayChain } from './gateways/gateway-chain.js';
+import { BuiltInGateway } from './gateways/builtin-gateway.js';
+import { OpenAICompatibleGateway } from './gateways/openai-compatible-gateway.js';
+export { registerProvider } from './gateways/builtin-gateway.js';
+export { GatewayChain, BuiltInGateway, OpenAICompatibleGateway };
+export { parseModel, type ParsedModel } from './parse-model.js';
+/**
+ * Resolve a model string to a LanguageModel using the default gateway chain.
+ * @deprecated Use `ModelFactory` instead for injectable, instance-based resolution.
+ */
+export declare function resolveModel(modelString: string): Promise<LanguageModel>;
+/** Create a gateway chain with custom gateways prepended before BuiltInGateway. */
+export declare function createChain(configs?: GatewayConfig[]): GatewayChain;
+//# sourceMappingURL=model-resolver.d.ts.map

package/dist/model-resolver.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.d.ts","sourceRoot":"","sources":["../src/model-resolver.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACxC,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,cAAc,EAA4B,MAAM,+BAA+B,CAAC;AACzF,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAElF,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,uBAAuB,EAAE,CAAC;AACjE,OAAO,EAAE,UAAU,EAAE,KAAK,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAMhE;;;GAGG;AACH,wBAAsB,YAAY,CAAC,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,CAAC,CAE9E;AAED,mFAAmF;AACnF,wBAAgB,WAAW,CAAC,OAAO,CAAC,EAAE,aAAa,EAAE,GAAG,YAAY,CASnE"}

package/dist/model-resolver.js

@@ -0,0 +1,28 @@
+import { GatewayChain } from './gateways/gateway-chain.js';
+import { BuiltInGateway, getDefaultBuiltInGateway } from './gateways/builtin-gateway.js';
+import { OpenAICompatibleGateway } from './gateways/openai-compatible-gateway.js';
+export { registerProvider } from './gateways/builtin-gateway.js';
+export { GatewayChain, BuiltInGateway, OpenAICompatibleGateway };
+export { parseModel } from './parse-model.js';
+// Global default chain: uses the singleton BuiltInGateway so registerProvider() works.
+const defaultChain = new GatewayChain();
+defaultChain.register(getDefaultBuiltInGateway());
+/**
+ * Resolve a model string to a LanguageModel using the default gateway chain.
+ * @deprecated Use `ModelFactory` instead for injectable, instance-based resolution.
+ */
+export async function resolveModel(modelString) {
+    return defaultChain.resolve(modelString);
+}
+/** Create a gateway chain with custom gateways prepended before BuiltInGateway. */
+export function createChain(configs) {
+    const chain = new GatewayChain();
+    if (configs) {
+        for (const cfg of configs) {
+            chain.register(new OpenAICompatibleGateway(cfg));
+        }
+    }
+    chain.register(getDefaultBuiltInGateway());
+    return chain;
+}
+//# sourceMappingURL=model-resolver.js.map

package/dist/model-resolver.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"model-resolver.js","sourceRoot":"","sources":["../src/model-resolver.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,YAAY,EAAE,MAAM,6BAA6B,CAAC;AAC3D,OAAO,EAAE,cAAc,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AACzF,OAAO,EAAE,uBAAuB,EAAE,MAAM,yCAAyC,CAAC;AAElF,OAAO,EAAE,gBAAgB,EAAE,MAAM,+BAA+B,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,cAAc,EAAE,uBAAuB,EAAE,CAAC;AACjE,OAAO,EAAE,UAAU,EAAoB,MAAM,kBAAkB,CAAC;AAEhE,uFAAuF;AACvF,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC;AACxC,YAAY,CAAC,QAAQ,CAAC,wBAAwB,EAAE,CAAC,CAAC;AAElD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,YAAY,CAAC,WAAmB;IACpD,OAAO,YAAY,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;AAC3C,CAAC;AAED,mFAAmF;AACnF,MAAM,UAAU,WAAW,CAAC,OAAyB;IACnD,MAAM,KAAK,GAAG,IAAI,YAAY,EAAE,CAAC;IACjC,IAAI,OAAO,EAAE,CAAC;QACZ,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;YAC1B,KAAK,CAAC,QAAQ,CAAC,IAAI,uBAAuB,CAAC,GAAG,CAAC,CAAC,CAAC;QACnD,CAAC;IACH,CAAC;IACD,KAAK,CAAC,QAAQ,CAAC,wBAAwB,EAAE,CAAC,CAAC;IAC3C,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/parse-model.d.ts

@@ -0,0 +1,6 @@
+export interface ParsedModel {
+    provider: string;
+    modelId: string;
+}
+export declare function parseModel(modelString: string): ParsedModel;
+//# sourceMappingURL=parse-model.d.ts.map

package/dist/parse-model.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-model.d.ts","sourceRoot":"","sources":["../src/parse-model.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,WAAW;IAC1B,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;CACjB;AAED,wBAAgB,UAAU,CAAC,WAAW,EAAE,MAAM,GAAG,WAAW,CAW3D"}

package/dist/parse-model.js

@@ -0,0 +1,11 @@
+export function parseModel(modelString) {
+    const idx = modelString.indexOf('/');
+    if (idx < 1 || idx === modelString.length - 1) {
+        throw new Error(`Invalid model string: "${modelString}". Expected format: "provider/model-name"`);
+    }
+    return {
+        provider: modelString.slice(0, idx),
+        modelId: modelString.slice(idx + 1),
+    };
+}
+//# sourceMappingURL=parse-model.js.map

package/dist/parse-model.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"parse-model.js","sourceRoot":"","sources":["../src/parse-model.ts"],"names":[],"mappings":"AAKA,MAAM,UAAU,UAAU,CAAC,WAAmB;IAC5C,MAAM,GAAG,GAAG,WAAW,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACrC,IAAI,GAAG,GAAG,CAAC,IAAI,GAAG,KAAK,WAAW,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC9C,MAAM,IAAI,KAAK,CACb,0BAA0B,WAAW,2CAA2C,CACjF,CAAC;IACJ,CAAC;IACD,OAAO;QACL,QAAQ,EAAE,WAAW,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;QACnC,OAAO,EAAE,WAAW,CAAC,KAAK,CAAC,GAAG,GAAG,CAAC,CAAC;KACpC,CAAC;AACJ,CAAC"}

package/dist/pipeline.d.ts

@@ -0,0 +1,28 @@
+import type { AbortSignal, PipelineContext, Processor, StageName, StreamEvent, SuspensionSignal, Tracer } from '@primo-ai/sdk';
+import type { HookManager } from './hook-manager.js';
+export type RunResult = PipelineContext | AbortSignal | SuspensionSignal;
+export interface PipelineRunnerOptions {
+    tracer?: Tracer;
+    hookManager?: HookManager;
+}
+export declare class PipelineRunner {
+    private processors;
+    private tracer;
+    private hookManager?;
+    constructor(options?: PipelineRunnerOptions);
+    register(processor: Processor): void;
+    unregister(stage: StageName): void;
+    replace(stage: StageName, processor: Processor): void;
+    setHookManager(hookManager: HookManager): void;
+    run(context: PipelineContext, stages: StageName[], options?: {
+        signal?: globalThis.AbortSignal;
+    }): Promise<RunResult>;
+    stream(context: PipelineContext, stages: StageName[], options?: {
+        signal?: globalThis.AbortSignal;
+    }): AsyncGenerator<StreamEvent>;
+    private executeStage;
+    private isAbort;
+    private isSuspend;
+    private consumeStream;
+}
+//# sourceMappingURL=pipeline.d.ts.map

package/dist/pipeline.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.d.ts","sourceRoot":"","sources":["../src/pipeline.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,eAAe,EAEf,SAAS,EAGT,SAAS,EACT,WAAW,EACX,gBAAgB,EAEhB,MAAM,EAEP,MAAM,eAAe,CAAC;AAGvB,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAyBrD,MAAM,MAAM,SAAS,GAAG,eAAe,GAAG,WAAW,GAAG,gBAAgB,CAAC;AAEzE,MAAM,WAAW,qBAAqB;IACpC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,WAAW,CAAC,EAAE,WAAW,CAAC;CAC3B;AA0DD,qBAAa,cAAc;IACzB,OAAO,CAAC,UAAU,CAAmB;IACrC,OAAO,CAAC,MAAM,CAAS;IACvB,OAAO,CAAC,WAAW,CAAC,CAAc;gBAEtB,OAAO,CAAC,EAAE,qBAAqB;IAK3C,QAAQ,CAAC,SAAS,EAAE,SAAS,GAAG,IAAI;IAIpC,UAAU,CAAC,KAAK,EAAE,SAAS,GAAG,IAAI;IAIlC,OAAO,CAAC,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,GAAG,IAAI;IAKrD,cAAc,CAAC,WAAW,EAAE,WAAW,GAAG,IAAI;IAIxC,GAAG,CAAC,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,CAAC;IA6CpH,MAAM,CAAC,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,OAAO,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,CAAA;KAAE,GAAG,cAAc,CAAC,WAAW,CAAC;YAgG1H,YAAY;IAmC1B,OAAO,CAAC,OAAO;IAIf,OAAO,CAAC,SAAS;YAIH,aAAa;CAwB5B"}

package/dist/pipeline.js

@@ -0,0 +1,286 @@
+import { SpanType } from '@primo-ai/sdk';
+import { NoOpTracer } from '@primo-ai/observability';
+import { extractTokenUsage } from './llm-invoker.js';
+/**
+ * Recursively freezes an object and all nested plain objects / arrays.
+ * Uses a WeakSet to guard against circular references.
+ * Skips null, non-objects, and already-frozen objects.
+ */
+function deepFreeze(obj, seen = new WeakSet()) {
+    if (obj === null || typeof obj !== 'object')
+        return obj;
+    if (Object.isFrozen(obj))
+        return obj;
+    if (seen.has(obj))
+        return obj;
+    seen.add(obj);
+    Object.freeze(obj);
+    for (const value of Object.values(obj)) {
+        if (value !== null && typeof value === 'object' && (Array.isArray(value) || Object.getPrototypeOf(value) === Object.prototype)) {
+            deepFreeze(value, seen);
+        }
+    }
+    return obj;
+}
+async function parseFullStream(fullStream, callbacks) {
+    const chunks = [];
+    const toolCalls = [];
+    const reasoningParts = [];
+    let usage;
+    for await (const event of fullStream) {
+        if (event.type === 'text-delta') {
+            chunks.push(event.text);
+            callbacks?.onTextDelta?.(event.text);
+        }
+        else if (event.type === 'tool-call') {
+            const tc = {
+                id: (event.toolCallId ?? event.id ?? ''),
+                name: (event.toolName ?? event.name ?? ''),
+                args: (event.args ?? event.input ?? {}),
+            };
+            toolCalls.push(tc);
+            callbacks?.onToolCall?.(tc);
+        }
+        else if (event.type === 'reasoning') {
+            reasoningParts.push((event.textDelta ?? event.text ?? ''));
+        }
+        else if (event.type === 'finish-step') {
+            usage = extractTokenUsage(event.usage);
+        }
+        else if (event.type === 'error') {
+            throw event.error;
+        }
+    }
+    return { chunks, toolCalls, reasoningParts, usage };
+}
+async function resolveReasoningContent(reasoningParts, reasoningPromise) {
+    let reasoningContent = reasoningParts.length > 0 ? reasoningParts.join('') : undefined;
+    if (!reasoningContent && reasoningPromise) {
+        try {
+            reasoningContent = await reasoningPromise ?? undefined;
+        }
+        catch { /* ignore */ }
+    }
+    return reasoningContent;
+}
+export class PipelineRunner {
+    processors = [];
+    tracer;
+    hookManager;
+    constructor(options) {
+        this.tracer = options?.tracer ?? new NoOpTracer();
+        this.hookManager = options?.hookManager;
+    }
+    register(processor) {
+        this.processors.push(processor);
+    }
+    unregister(stage) {
+        this.processors = this.processors.filter((p) => p.stage !== stage);
+    }
+    replace(stage, processor) {
+        this.processors = this.processors.filter((p) => p.stage !== stage);
+        this.processors.push(processor);
+    }
+    setHookManager(hookManager) {
+        this.hookManager = hookManager;
+    }
+    async run(context, stages, options) {
+        const rootSpan = this.tracer.startSpan(SpanType.AGENT_RUN);
+        const signal = options?.signal;
+        let ctx = context;
+        try {
+            for (const stage of stages) {
+                if (signal?.aborted)
+                    throw new DOMException('Pipeline aborted', 'AbortError');
+                const stageSpan = rootSpan.startChild(stage);
+                try {
+                    const stageResult = await this.executeStage(ctx, stage, stageSpan);
+                    if (this.isAbort(stageResult) || this.isSuspend(stageResult)) {
+                        stageSpan.end();
+                        rootSpan.end();
+                        return stageResult;
+                    }
+                    ctx = stageResult;
+                    ctx = await this.consumeStream(ctx);
+                    // Fire llm.after after stream is consumed (response is now available)
+                    if (stage === 'invokeLLM' && this.hookManager && ctx.iteration._modelString) {
+                        await this.hookManager.invoke('llm.after', { model: ctx.iteration._modelString }, { response: ctx.iteration.response });
+                    }
+                    // Auto-enrich span with token usage after stream consumption
+                    if (stage === 'invokeLLM' && ctx.iteration.tokenUsage) {
+                        stageSpan.setAttribute('tokens.input', ctx.iteration.tokenUsage.input);
+                        stageSpan.setAttribute('tokens.output', ctx.iteration.tokenUsage.output);
+                    }
+                }
+                catch (error) {
+                    if (this.hookManager) {
+                        try {
+                            await this.hookManager.invoke('error', { error, stage }, {});
+                        }
+                        catch { /* hook error must not mask original */ }
+                    }
+                    throw error;
+                }
+                finally {
+                    stageSpan.end();
+                }
+            }
+        }
+        finally {
+            rootSpan.end();
+        }
+        return ctx;
+    }
+    async *stream(context, stages, options) {
+        const rootSpan = this.tracer.startSpan(SpanType.AGENT_RUN);
+        const signal = options?.signal;
+        let ctx = context;
+        try {
+            for (const stage of stages) {
+                if (signal?.aborted)
+                    throw new DOMException('Pipeline aborted', 'AbortError');
+                yield { type: 'stage_start', stage };
+                const stageSpan = rootSpan.startChild(stage);
+                try {
+                    const stageResult = await this.executeStage(ctx, stage, stageSpan);
+                    if (this.isAbort(stageResult)) {
+                        stageSpan.end();
+                        rootSpan.end();
+                        yield { type: 'abort', reason: stageResult.reason, ...(stageResult.retryFrom ? { retryFrom: stageResult.retryFrom } : {}) };
+                        return;
+                    }
+                    if (this.isSuspend(stageResult)) {
+                        stageSpan.end();
+                        rootSpan.end();
+                        yield { type: 'suspended', suspensionId: stageResult.suspensionId, reason: stageResult.reason, checkpoint: stageResult.checkpoint };
+                        return;
+                    }
+                    ctx = stageResult;
+                    const fullStream = ctx.iteration.fullStream;
+                    if (fullStream) {
+                        const chunks = [];
+                        const toolCalls = [];
+                        const reasoningParts = [];
+                        let usage;
+                        for await (const event of fullStream) {
+                            if (event.type === 'text-delta') {
+                                chunks.push(event.text);
+                                yield { type: 'text_delta', text: event.text };
+                            }
+                            else if (event.type === 'tool-call') {
+                                const tc = {
+                                    id: (event.toolCallId ?? event.id ?? ''),
+                                    name: (event.toolName ?? event.name ?? ''),
+                                    args: (event.args ?? event.input ?? {}),
+                                };
+                                toolCalls.push(tc);
+                                yield { type: 'tool_call', name: tc.name, args: tc.args };
+                            }
+                            else if (event.type === 'reasoning') {
+                                reasoningParts.push((event.textDelta ?? event.text ?? ''));
+                            }
+                            else if (event.type === 'finish-step') {
+                                usage = extractTokenUsage(event.usage);
+                            }
+                            else if (event.type === 'error') {
+                                throw event.error;
+                            }
+                        }
+                        const reasoningContent = await resolveReasoningContent(reasoningParts, ctx.iteration.reasoningPromise);
+                        const pendingUsage = ctx.iteration.usagePromise
+                            ? await ctx.iteration.usagePromise
+                            : undefined;
+                        ctx = deepFreeze({
+                            ...ctx,
+                            iteration: {
+                                ...ctx.iteration,
+                                response: chunks.join('') || ctx.iteration.response || '',
+                                pendingToolCalls: toolCalls.length > 0 ? toolCalls : undefined,
+                                reasoningContent,
+                                tokenUsage: usage ?? pendingUsage,
+                                fullStream: undefined,
+                                usagePromise: undefined,
+                                reasoningPromise: undefined,
+                            },
+                        });
+                    }
+                    // Fire llm.after after stream is consumed (response is now available)
+                    if (stage === 'invokeLLM' && this.hookManager && ctx.iteration._modelString) {
+                        await this.hookManager.invoke('llm.after', { model: ctx.iteration._modelString }, { response: ctx.iteration.response });
+                    }
+                    // Auto-enrich span with token usage after stream consumption
+                    if (stage === 'invokeLLM' && ctx.iteration.tokenUsage) {
+                        stageSpan.setAttribute('tokens.input', ctx.iteration.tokenUsage.input);
+                        stageSpan.setAttribute('tokens.output', ctx.iteration.tokenUsage.output);
+                    }
+                }
+                finally {
+                    stageSpan.end();
+                }
+                yield { type: 'stage_complete', stage };
+            }
+        }
+        finally {
+            rootSpan.end();
+        }
+        yield { type: 'complete', context: ctx };
+    }
+    async executeStage(ctx, stage, stageSpan) {
+        const stageProcessors = this.processors.filter((p) => p.stage === stage);
+        if (stageProcessors.length === 0)
+            return ctx;
+        if (stageProcessors.every((p) => p.isNoOp))
+            return ctx;
+        // stage.before hook
+        if (this.hookManager) {
+            await this.hookManager.invoke('stage.before', { stage, context: ctx }, {});
+        }
+        let currentCtx = ctx;
+        for (const processor of stageProcessors) {
+            const ctxWithSpan = deepFreeze({
+                ...currentCtx,
+                iteration: { ...currentCtx.iteration, span: stageSpan },
+            });
+            const result = await processor.execute(ctxWithSpan);
+            if ('type' in result && (result.type === 'abort' || result.type === 'suspend')) {
+                return result;
+            }
+            currentCtx = deepFreeze({ ...result });
+        }
+        // stage.after hook
+        if (this.hookManager) {
+            await this.hookManager.invoke('stage.after', { stage, context: currentCtx }, {});
+        }
+        return currentCtx;
+    }
+    isAbort(result) {
+        return 'type' in result && result.type === 'abort';
+    }
+    isSuspend(result) {
+        return 'type' in result && result.type === 'suspend';
+    }
+    async consumeStream(ctx) {
+        const fullStream = ctx.iteration.fullStream;
+        if (!fullStream)
+            return ctx;
+        const result = await parseFullStream(fullStream);
+        const reasoningContent = await resolveReasoningContent(result.reasoningParts, ctx.iteration.reasoningPromise);
+        const pendingUsage = ctx.iteration.usagePromise
+            ? await ctx.iteration.usagePromise
+            : undefined;
+        return deepFreeze({
+            ...ctx,
+            iteration: {
+                ...ctx.iteration,
+                response: result.chunks.join('') || ctx.iteration.response,
+                pendingToolCalls: result.toolCalls.length > 0 ? result.toolCalls : undefined,
+                reasoningContent,
+                tokenUsage: result.usage ?? pendingUsage,
+                fullStream: undefined,
+                usagePromise: undefined,
+                reasoningPromise: undefined,
+            },
+        });
+    }
+}
+//# sourceMappingURL=pipeline.js.map

package/dist/pipeline.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pipeline.js","sourceRoot":"","sources":["../src/pipeline.ts"],"names":[],"mappings":"AAcA,OAAO,EAAE,QAAQ,EAAE,MAAM,eAAe,CAAC;AACzC,OAAO,EAAE,UAAU,EAAE,MAAM,yBAAyB,CAAC;AAErD,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAErD;;;;GAIG;AACH,SAAS,UAAU,CAAI,GAAM,EAAE,OAAwB,IAAI,OAAO,EAAE;IAClE,IAAI,GAAG,KAAK,IAAI,IAAI,OAAO,GAAG,KAAK,QAAQ;QAAE,OAAO,GAAkB,CAAC;IACvE,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC;QAAE,OAAO,GAAkB,CAAC;IACpD,IAAI,IAAI,CAAC,GAAG,CAAC,GAAa,CAAC;QAAE,OAAO,GAAkB,CAAC;IAEvD,IAAI,CAAC,GAAG,CAAC,GAAa,CAAC,CAAC;IACxB,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAEnB,KAAK,MAAM,KAAK,IAAI,MAAM,CAAC,MAAM,CAAC,GAA8B,CAAC,EAAE,CAAC;QAClE,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,MAAM,CAAC,SAAS,CAAC,EAAE,CAAC;YAC/H,UAAU,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QAC1B,CAAC;IACH,CAAC;IAED,OAAO,GAAkB,CAAC;AAC5B,CAAC;AAqBD,KAAK,UAAU,eAAe,CAC5B,UAAmE,EACnE,SAA+B;IAE/B,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,SAAS,GAAe,EAAE,CAAC;IACjC,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,IAAI,KAA6B,CAAC;IAElC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QACrC,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAc,CAAC,CAAC;YAClC,SAAS,EAAE,WAAW,EAAE,CAAC,KAAK,CAAC,IAAc,CAAC,CAAC;QACjD,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtC,MAAM,EAAE,GAAa;gBACnB,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,EAAE,IAAI,EAAE,CAAW;gBAClD,IAAI,EAAE,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,IAAI,IAAI,EAAE,CAAW;gBACpD,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,IAAI,EAAE,CAA4B;aACnE,CAAC;YACF,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACnB,SAAS,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC;QAC9B,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YACtC,cAAc,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,IAAI,IAAI,EAAE,CAAW,CAAC,CAAC;QACvE,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;YACxC,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACzC,CAAC;aAAM,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAClC,MAAM,KAAK,CAAC,KAAK,CAAC;QACpB,CAAC;IACH,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,cAAc,EAAE,KAAK,EAAE,CAAC;AACtD,CAAC;AAED,KAAK,UAAU,uBAAuB,CACpC,cAAwB,EACxB,gBAAyD;IAEzD,IAAI,gBAAgB,GAAG,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;IACvF,IAAI,CAAC,gBAAgB,IAAI,gBAAgB,EAAE,CAAC;QAC1C,IAAI,CAAC;YAAC,gBAAgB,GAAG,MAAM,gBAAgB,IAAI,SAAS,CAAC;QAAC,CAAC;QAAC,MAAM,CAAC,CAAC,YAAY,CAAC,CAAC;IACxF,CAAC;IACD,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAED,MAAM,OAAO,cAAc;IACjB,UAAU,GAAgB,EAAE,CAAC;IAC7B,MAAM,CAAS;IACf,WAAW,CAAe;IAElC,YAAY,OAA+B;QACzC,IAAI,CAAC,MAAM,GAAG,OAAO,EAAE,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;QAClD,IAAI,CAAC,WAAW,GAAG,OAAO,EAAE,WAAW,CAAC;IAC1C,CAAC;IAED,QAAQ,CAAC,SAAoB;QAC3B,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,UAAU,CAAC,KAAgB;QACzB,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IACrE,CAAC;IAED,OAAO,CAAC,KAAgB,EAAE,SAAoB;QAC5C,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;QACnE,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAED,cAAc,CAAC,WAAwB;QACrC,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,OAAwB,EAAE,MAAmB,EAAE,OAA6C;QACpG,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAC/B,IAAI,GAAG,GAAG,OAAO,CAAC;QAElB,IAAI,CAAC;YACH,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,IAAI,MAAM,EAAE,OAAO;oBAAE,MAAM,IAAI,YAAY,CAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC;gBAC9E,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC7C,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;oBACnE,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC;wBAC7D,SAAS,CAAC,GAAG,EAAE,CAAC;wBAChB,QAAQ,CAAC,GAAG,EAAE,CAAC;wBACf,OAAO,WAAW,CAAC;oBACrB,CAAC;oBACD,GAAG,GAAG,WAAW,CAAC;oBAClB,GAAG,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;oBAEpC,sEAAsE;oBACtE,IAAI,KAAK,KAAK,WAAW,IAAI,IAAI,CAAC,WAAW,IAAK,GAAG,CAAC,SAAkD,CAAC,YAAY,EAAE,CAAC;wBACtH,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,KAAK,EAAG,GAAG,CAAC,SAAkD,CAAC,YAAY,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;oBACpK,CAAC;oBAED,6DAA6D;oBAC7D,IAAI,KAAK,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;wBACtD,SAAS,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;wBACvE,SAAS,CAAC,YAAY,CAAC,eAAe,EAAE,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;oBAC3E,CAAC;gBACH,CAAC;gBAAC,OAAO,KAAK,EAAE,CAAC;oBACf,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;wBACrB,IAAI,CAAC;4BAAC,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;wBAAC,CAAC;wBAAC,MAAM,CAAC,CAAC,uCAAuC,CAAC,CAAC;oBACzH,CAAC;oBACD,MAAM,KAAK,CAAC;gBACd,CAAC;wBAAS,CAAC;oBACT,SAAS,CAAC,GAAG,EAAE,CAAC;gBAClB,CAAC;YACH,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,QAAQ,CAAC,GAAG,EAAE,CAAC;QACjB,CAAC;QAED,OAAO,GAAG,CAAC;IACb,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,OAAwB,EAAE,MAAmB,EAAE,OAA6C;QACxG,MAAM,QAAQ,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,MAAM,GAAG,OAAO,EAAE,MAAM,CAAC;QAC/B,IAAI,GAAG,GAAG,OAAO,CAAC;QAElB,IAAI,CAAC;YACH,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;gBAC3B,IAAI,MAAM,EAAE,OAAO;oBAAE,MAAM,IAAI,YAAY,CAAC,kBAAkB,EAAE,YAAY,CAAC,CAAC;gBAC9E,MAAM,EAAE,IAAI,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC;gBACrC,MAAM,SAAS,GAAG,QAAQ,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;gBAC7C,IAAI,CAAC;oBACH,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;oBACnE,IAAI,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC;wBAC9B,SAAS,CAAC,GAAG,EAAE,CAAC;wBAChB,QAAQ,CAAC,GAAG,EAAE,CAAC;wBACf,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,GAAG,CAAC,WAAW,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,SAAS,EAAE,WAAW,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;wBAC5H,OAAO;oBACT,CAAC;oBACD,IAAI,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,EAAE,CAAC;wBAChC,SAAS,CAAC,GAAG,EAAE,CAAC;wBAChB,QAAQ,CAAC,GAAG,EAAE,CAAC;wBACf,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,YAAY,EAAE,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,WAAW,CAAC,MAAM,EAAE,UAAU,EAAE,WAAW,CAAC,UAAU,EAAE,CAAC;wBACpI,OAAO;oBACT,CAAC;oBACD,GAAG,GAAG,WAAW,CAAC;oBAElB,MAAM,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC;oBAC5C,IAAI,UAAU,EAAE,CAAC;wBACf,MAAM,MAAM,GAAa,EAAE,CAAC;wBAC5B,MAAM,SAAS,GAAe,EAAE,CAAC;wBACjC,MAAM,cAAc,GAAa,EAAE,CAAC;wBACpC,IAAI,KAA6B,CAAC;wBAElC,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,UAAqE,EAAE,CAAC;4BAChG,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gCAChC,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAc,CAAC,CAAC;gCAClC,MAAM,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,IAAc,EAAE,CAAC;4BAC3D,CAAC;iCAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gCACtC,MAAM,EAAE,GAAa;oCACnB,EAAE,EAAE,CAAC,KAAK,CAAC,UAAU,IAAI,KAAK,CAAC,EAAE,IAAI,EAAE,CAAW;oCAClD,IAAI,EAAE,CAAC,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,IAAI,IAAI,EAAE,CAAW;oCACpD,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,IAAI,KAAK,CAAC,KAAK,IAAI,EAAE,CAA4B;iCACnE,CAAC;gCACF,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gCACnB,MAAM,EAAE,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;4BAC5D,CAAC;iCAAM,IAAI,KAAK,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;gCACtC,cAAc,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,IAAI,IAAI,EAAE,CAAW,CAAC,CAAC;4BACvE,CAAC;iCAAM,IAAI,KAAK,CAAC,IAAI,KAAK,aAAa,EAAE,CAAC;gCACxC,KAAK,GAAG,iBAAiB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;4BACzC,CAAC;iCAAM,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gCAClC,MAAM,KAAK,CAAC,KAAK,CAAC;4BACpB,CAAC;wBACH,CAAC;wBAED,MAAM,gBAAgB,GAAG,MAAM,uBAAuB,CAAC,cAAc,EAAE,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;wBACvG,MAAM,YAAY,GAAG,GAAG,CAAC,SAAS,CAAC,YAAY;4BAC7C,CAAC,CAAC,MAAM,GAAG,CAAC,SAAS,CAAC,YAAY;4BAClC,CAAC,CAAC,SAAS,CAAC;wBAEd,GAAG,GAAG,UAAU,CAAC;4BACf,GAAG,GAAG;4BACN,SAAS,EAAE;gCACT,GAAG,GAAG,CAAC,SAAS;gCAChB,QAAQ,EAAE,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,SAAS,CAAC,QAAQ,IAAI,EAAE;gCACzD,gBAAgB,EAAE,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gCAC9D,gBAAgB;gCAChB,UAAU,EAAE,KAAK,IAAI,YAAY;gCACjC,UAAU,EAAE,SAAS;gCACrB,YAAY,EAAE,SAAS;gCACvB,gBAAgB,EAAE,SAAS;6BAC5B;yBACF,CAAC,CAAC;oBACL,CAAC;oBAED,sEAAsE;oBACtE,IAAI,KAAK,KAAK,WAAW,IAAI,IAAI,CAAC,WAAW,IAAK,GAAG,CAAC,SAAkD,CAAC,YAAY,EAAE,CAAC;wBACtH,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,KAAK,EAAG,GAAG,CAAC,SAAkD,CAAC,YAAY,EAAE,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,SAAS,CAAC,QAAQ,EAAE,CAAC,CAAC;oBACpK,CAAC;oBAED,6DAA6D;oBAC7D,IAAI,KAAK,KAAK,WAAW,IAAI,GAAG,CAAC,SAAS,CAAC,UAAU,EAAE,CAAC;wBACtD,SAAS,CAAC,YAAY,CAAC,cAAc,EAAE,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;wBACvE,SAAS,CAAC,YAAY,CAAC,eAAe,EAAE,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;oBAC3E,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,SAAS,CAAC,GAAG,EAAE,CAAC;gBAClB,CAAC;gBACD,MAAM,EAAE,IAAI,EAAE,gBAAgB,EAAE,KAAK,EAAE,CAAC;YAC1C,CAAC;QACH,CAAC;gBAAS,CAAC;YACT,QAAQ,CAAC,GAAG,EAAE,CAAC;QACjB,CAAC;QAED,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC;IAC3C,CAAC;IAEO,KAAK,CAAC,YAAY,CACxB,GAAoB,EACpB,KAAgB,EAChB,SAAe;QAEf,MAAM,eAAe,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;QACzE,IAAI,eAAe,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,GAAG,CAAC;QAC7C,IAAI,eAAe,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC;YAAE,OAAO,GAAG,CAAC;QAEvD,oBAAoB;QACpB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,cAAc,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;QAC7E,CAAC;QAED,IAAI,UAAU,GAAG,GAAG,CAAC;QACrB,KAAK,MAAM,SAAS,IAAI,eAAe,EAAE,CAAC;YACxC,MAAM,WAAW,GAAG,UAAU,CAAC;gBAC7B,GAAG,UAAU;gBACb,SAAS,EAAE,EAAE,GAAG,UAAU,CAAC,SAAS,EAAE,IAAI,EAAE,SAAS,EAAE;aACxD,CAAC,CAAC;YACH,MAAM,MAAM,GAAoB,MAAM,SAAS,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;YACrE,IAAI,MAAM,IAAI,MAAM,IAAI,CAAC,MAAM,CAAC,IAAI,KAAK,OAAO,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC,EAAE,CAAC;gBAC/E,OAAO,MAAM,CAAC;YAChB,CAAC;YACD,UAAU,GAAG,UAAU,CAAC,EAAE,GAAI,MAA0B,EAAE,CAAC,CAAC;QAC9D,CAAC;QAED,mBAAmB;QACnB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACrB,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,CAAC;QACnF,CAAC;QAED,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,OAAO,CAAC,MAAiB;QAC/B,OAAO,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,OAAO,CAAC;IACrD,CAAC;IAEO,SAAS,CAAC,MAAiB;QACjC,OAAO,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,KAAK,SAAS,CAAC;IACvD,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,GAAoB;QAC9C,MAAM,UAAU,GAAG,GAAG,CAAC,SAAS,CAAC,UAAU,CAAC;QAC5C,IAAI,CAAC,UAAU;YAAE,OAAO,GAAG,CAAC;QAE5B,MAAM,MAAM,GAAG,MAAM,eAAe,CAAC,UAAqE,CAAC,CAAC;QAC5G,MAAM,gBAAgB,GAAG,MAAM,uBAAuB,CAAC,MAAM,CAAC,cAAc,EAAE,GAAG,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC;QAC9G,MAAM,YAAY,GAAG,GAAG,CAAC,SAAS,CAAC,YAAY;YAC7C,CAAC,CAAC,MAAM,GAAG,CAAC,SAAS,CAAC,YAAY;YAClC,CAAC,CAAC,SAAS,CAAC;QAEd,OAAO,UAAU,CAAC;YAChB,GAAG,GAAG;YACN,SAAS,EAAE;gBACT,GAAG,GAAG,CAAC,SAAS;gBAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,GAAG,CAAC,SAAS,CAAC,QAAQ;gBAC1D,gBAAgB,EAAE,MAAM,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS;gBAC5E,gBAAgB;gBAChB,UAAU,EAAE,MAAM,CAAC,KAAK,IAAI,YAAY;gBACxC,UAAU,EAAE,SAAS;gBACrB,YAAY,EAAE,SAAS;gBACvB,gBAAgB,EAAE,SAAS;aAC5B;SACF,CAAC,CAAC;IACL,CAAC;CACF"}

package/dist/plugin-manager.d.ts

@@ -0,0 +1,42 @@
+import type { EventType, HarnessAPI, PluginRegistration, StageMutation } from '@primo-ai/sdk';
+import type { PipelineRunner } from './pipeline.js';
+import type { ToolRegistry } from './tool-registry.js';
+import type { ContextBuilder } from './context-builder.js';
+import { EventBus } from './event-bus.js';
+import { EventSystem } from './event-system.js';
+import { HookManager } from './hook-manager.js';
+export type PluginFactory = (api: HarnessAPI) => PluginRegistration | void;
+export declare class PluginManager {
+    private runner;
+    private registry;
+    private contextBuilder?;
+    private _eventSystem;
+    readonly hookManager: HookManager;
+    private _harnessInstances;
+    private resourceInstances;
+    private errors;
+    private stageMutator?;
+    get eventBus(): EventBus;
+    get eventSystem(): EventSystem;
+    constructor(runner: PipelineRunner, registry: ToolRegistry, contextBuilder?: ContextBuilder);
+    loadPlugin(filePath: string): Promise<void>;
+    initializePlugin(factory: PluginFactory): void;
+    loadPluginsFromConfig(config: {
+        plugins?: Array<{
+            path: string;
+        }>;
+    }): Promise<void>;
+    getCommand(name: string): ((args: string) => Promise<void>) | undefined;
+    getErrors(): Array<{
+        source: string;
+        error: Error;
+    }>;
+    setStageMutator(mutator: (mutation: StageMutation) => void): void;
+    freezeHarnessInstances(): void;
+    emitEvent(eventType: EventType, ...args: unknown[]): void;
+    initializeAll(): Promise<void>;
+    private _shutdown;
+    shutdown(): Promise<void>;
+    private createHarnessAPI;
+}
+//# sourceMappingURL=plugin-manager.d.ts.map

package/dist/plugin-manager.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"plugin-manager.d.ts","sourceRoot":"","sources":["../src/plugin-manager.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,SAAS,EACT,UAAU,EACV,kBAAkB,EAClB,aAAa,EACd,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;AAE3D,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAChD,OAAO,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAGhD,MAAM,MAAM,aAAa,GAAG,CAAC,GAAG,EAAE,UAAU,KAAK,kBAAkB,GAAG,IAAI,CAAC;AAe3E,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,YAAY,CAAqB;IACzC,QAAQ,CAAC,WAAW,EAAE,WAAW,CAAC;IAClC,OAAO,CAAC,iBAAiB,CAAwB;IACjD,OAAO,CAAC,iBAAiB,CAA8B;IACvD,OAAO,CAAC,MAAM,CAA+C;IAC7D,OAAO,CAAC,YAAY,CAAC,CAAoC;IAEzD,IAAI,QAAQ,IAAI,QAAQ,CAEvB;IAED,IAAI,WAAW,IAAI,WAAW,CAE7B;gBAEW,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,YAAY,EAAE,cAAc,CAAC,EAAE,cAAc;IAQrF,UAAU,CAAC,QAAQ,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAqBjD,gBAAgB,CAAC,OAAO,EAAE,aAAa,GAAG,IAAI;IAKxC,qBAAqB,CAAC,MAAM,EAAE;QAAE,OAAO,CAAC,EAAE,KAAK,CAAC;YAAE,IAAI,EAAE,MAAM,CAAA;SAAE,CAAC,CAAA;KAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IAOzF,UAAU,CAAC,IAAI,EAAE,MAAM,GAAG,CAAC,CAAC,IAAI,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC,GAAG,SAAS;IAQvE,SAAS,IAAI,KAAK,CAAC;QAAE,MAAM,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,KAAK,CAAA;KAAE,CAAC;IAIpD,eAAe,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,aAAa,KAAK,IAAI,GAAG,IAAI;IAIjE,sBAAsB,IAAI,IAAI;IAI9B,SAAS,CAAC,SAAS,EAAE,SAAS,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,GAAG,IAAI;IAInD,aAAa,IAAI,OAAO,CAAC,IAAI,CAAC;IAepC,OAAO,CAAC,SAAS,CAAS;IAEpB,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAwB/B,OAAO,CAAC,gBAAgB;CAezB"}