@primo-ai/core 0.1.0

package/README.md

@@ -0,0 +1,114 @@
+# @primo-ai/core
+
+Core agent loop, processor pipeline, and orchestration layer.
+
+## Overview
+
+This package provides the main building blocks of the AgentForge framework:
+
+- **Agent** -- Top-level facade for creating, configuring, and running agents
+- **PipelineRunner** -- Executes the linear pipeline of processors stage by stage
+- **LoopOrchestrator** -- Manages the agentic loop (repeating stages until stop)
+- **LLMInvoker** -- Wraps Vercel AI SDK `streamText()` for single-step LLM calls
+- **ToolRegistry** -- Registers tools and converts them to AI SDK schemas
+- **EventBus / EventSystem** -- Pub/sub event dispatch with replay support
+- **HookManager** -- Fixed-point interception (before/after hooks)
+- **StateMachine** -- Agent lifecycle states (pending/running/completed/paused/cancelled/error)
+- **ModelFactory** -- Canonical model resolution with pluggable gateways
+- **ContextBuilder** -- Assembles PipelineContext from config, tools, and prompt fragments
+
+## Quick Example
+
+```typescript
+import { Agent, registerProvider } from '@primo-ai/core';
+import { createOpenAICompatible } from '@ai-sdk/openai-compatible';
+
+registerProvider('deepseek', (modelId) => {
+  const sdk = createOpenAICompatible({
+    baseURL: 'https://api.deepseek.com',
+    apiKey: process.env.DEEPSEEK_API_KEY!,
+  });
+  return sdk.languageModel(modelId);
+});
+
+const agent = new Agent({
+  model: 'deepseek/deepseek-v4-flash',
+  systemPrompt: 'You are a helpful assistant.',
+  maxIterations: 5,
+});
+
+// Run (returns full result)
+const result = await agent.run('Hello');
+console.log(result.response, result.tokenUsage, result.sessionId);
+
+// Stream (yields text chunks)
+for await (const chunk of agent.stream('Hello')) {
+  process.stdout.write(chunk);
+}
+```
+
+## Agent API
+
+### `new Agent(config, deps?)`
+
+| Parameter | Type | Description |
+|-----------|------|-------------|
+| `config.model` | `string` | Model string in `provider/modelId` format |
+| `config.systemPrompt` | `Dynamic<string>` | System prompt (static or per-request function) |
+| `config.maxIterations` | `Dynamic<number>` | Max agentic loop iterations (default: 10) |
+| `config.tools` | `Tool[]` | Tools available to the agent |
+| `config.providerOptions` | `Record<string, Record<string, unknown>>` | Per-provider options passed to streamText() |
+
+### `agent.run(input, signal?)` -> `Promise<AgentRunResult>`
+
+Runs the agent pipeline to completion. Returns `{ response, tokenUsage, sessionId, compatRetries }`.
+
+### `agent.stream(input, signal?)` -> `AsyncGenerator<string>`
+
+Streams text chunks as the agent generates them.
+
+### `agent.streamEvents(input, signal?)` -> `AsyncGenerator<StreamEvent>`
+
+Streams structured events (text_delta, tool_call, tool_result, stage_start, etc.).
+
+### `agent.resume(sessionId, signal?)` -> `Promise<AgentRunResult>`
+
+Resumes a suspended agent from a checkpoint.
+
+### `agent.use(factory)`
+
+Registers a plugin or processor. Accepts a `PluginFactory` function or a `Processor` instance.
+
+## Key Exports
+
+| Export | Description |
+|--------|-------------|
+| `PipelineRunner` | Executes pipeline stages sequentially |
+| `LoopOrchestrator` | Manages agentic loop with abort/retry/suspend |
+| `LLMInvoker` | Single-step LLM invocation via AI SDK |
+| `ToolRegistry` | Tool registration and schema conversion |
+| `EventBus` | Lightweight pub/sub event bus |
+| `EventSystem` | Event dispatch with replay backend |
+| `HookManager` | Before/after hook invocation |
+| `StateMachine` | Agent lifecycle state transitions |
+| `ModelFactory` | Pluggable model resolution chain |
+| `ContextBuilder` | PipelineContext assembly |
+| `PluginManager` | Plugin lifecycle (initialize, shutdown) |
+| `SessionManagerImpl` | Session CRUD with suspend/resume |
+| `FilesystemSessionStorage` | JSONL file-based session storage |
+| `createSubAgentTool` | Creates a tool that delegates to a sub-agent |
+| `ConcurrencyController` | Limits parallel task execution |
+| `TaskManagerImpl` | Async sub-agent task management |
+| `ConfigLoader` | Multi-layer JSONC config loading |
+| `InMemoryCheckpointStore` | In-memory checkpoint storage |
+| `JsonlCheckpointStore` | JSONL file-based checkpoint storage |
+| `serialize` / `deserialize` | PipelineContext serialization for checkpoints |
+| `registerProvider` | Register a model provider factory |
+| `AgentForgeError` | Base error class with domain hierarchy |
+
+## Dependencies
+
+- `@primo-ai/sdk` -- type definitions
+- `@primo-ai/observability` -- tracing and metrics
+- `@primo-ai/tools` -- built-in tools
+- `ai` + provider SDKs -- Vercel AI SDK core

package/dist/agent.d.ts

@@ -0,0 +1,66 @@
+import type { AgentConfig, CheckpointStore, PipelineStageConfig, Processor, SessionManager, Tracer } from '@primo-ai/sdk';
+import { PipelineRunner } from './pipeline.js';
+import { serialize } from './serialize.js';
+import { ToolRegistry } from './tool-registry.js';
+import { PluginManager, type PluginFactory } from './plugin-manager.js';
+import { ModelFactory } from './model-factory.js';
+import { ContextBuilder } from './context-builder.js';
+export interface AgentDependencies {
+    runner?: PipelineRunner;
+    registry?: ToolRegistry;
+    pluginManager?: PluginManager;
+    tracer?: Tracer;
+    modelFactory?: ModelFactory;
+    sessionManager?: SessionManager;
+    checkpointStore?: CheckpointStore<ReturnType<typeof serialize>>;
+    /** When provided, creates a JsonlCheckpointStore for crash-safe checkpoint persistence */
+    checkpointDir?: string;
+    /** When true, saves a checkpoint after each completed iteration */
+    autoCheckpoint?: boolean;
+    contextBuilder?: ContextBuilder;
+    /** Override default pipeline stage order. */
+    stageConfig?: PipelineStageConfig;
+}
+export interface AgentRunResult {
+    response: string;
+    tokenUsage: import('@primo-ai/sdk').TokenUsage;
+    sessionId: string;
+    /** Number of compat retries that occurred during the agentic loop. */
+    compatRetries: number;
+}
+export declare class Agent {
+    private config;
+    private runner;
+    private registry;
+    private _pluginManager;
+    private _tracer?;
+    private _autoCheckpoint;
+    private _model;
+    private modelFactory;
+    private orchestrator;
+    private sessionManager?;
+    private contextBuilder;
+    constructor(config: AgentConfig, deps?: AgentDependencies);
+    use(factory: Processor | PluginFactory): void;
+    teardown(): Promise<void>;
+    get pipelineRunner(): PipelineRunner;
+    get toolRegistry(): ToolRegistry;
+    get pluginManager(): PluginManager;
+    get eventBus(): import('./event-bus.js').EventBus;
+    get eventSystem(): import('./event-system.js').EventSystem;
+    get state(): import('./state-machine.js').AgentState;
+    get _contextBuilder(): ContextBuilder;
+    run(input: string, signal?: globalThis.AbortSignal): Promise<AgentRunResult>;
+    resume(sessionId: string, signal?: globalThis.AbortSignal): Promise<AgentRunResult>;
+    stream(input: string, signal?: globalThis.AbortSignal): AsyncGenerator<string>;
+    streamEvents(input: string, signal?: globalThis.AbortSignal): AsyncGenerator<import('@primo-ai/sdk').StreamEvent>;
+    /** Clear the cached model so the next run re-resolves from the factory. */
+    invalidateModel(): void;
+    /** Auto-invalidate cached model when the error indicates auth failure or model-not-found. */
+    private autoInvalidateModel;
+    private getLLM;
+    private createContext;
+    private registerTools;
+    private registerBuiltinProcessors;
+}
+//# sourceMappingURL=agent.d.ts.map

package/dist/agent.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.d.ts","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EACV,WAAW,EACX,eAAe,EAEf,mBAAmB,EACnB,SAAS,EACT,cAAc,EAEd,MAAM,EACP,MAAM,eAAe,CAAC;AACvB,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAC/C,OAAO,EAAE,SAAS,EAAE,MAAM,gBAAgB,CAAC;AAC3C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,KAAK,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAExE,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAgBlD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,MAAM,WAAW,iBAAiB;IAChC,MAAM,CAAC,EAAE,cAAc,CAAC;IACxB,QAAQ,CAAC,EAAE,YAAY,CAAC;IACxB,aAAa,CAAC,EAAE,aAAa,CAAC;IAC9B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,YAAY,CAAC,EAAE,YAAY,CAAC;IAC5B,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,eAAe,CAAC,EAAE,eAAe,CAAC,UAAU,CAAC,OAAO,SAAS,CAAC,CAAC,CAAC;IAChE,0FAA0F;IAC1F,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,mEAAmE;IACnE,cAAc,CAAC,EAAE,OAAO,CAAC;IACzB,cAAc,CAAC,EAAE,cAAc,CAAC;IAChC,6CAA6C;IAC7C,WAAW,CAAC,EAAE,mBAAmB,CAAC;CACnC;AAED,MAAM,WAAW,cAAc;IAC7B,QAAQ,EAAE,MAAM,CAAC;IACjB,UAAU,EAAE,OAAO,eAAe,EAAE,UAAU,CAAC;IAC/C,SAAS,EAAE,MAAM,CAAC;IAClB,sEAAsE;IACtE,aAAa,EAAE,MAAM,CAAC;CACvB;AAED,qBAAa,KAAK;IAChB,OAAO,CAAC,MAAM,CAAc;IAC5B,OAAO,CAAC,MAAM,CAAiB;IAC/B,OAAO,CAAC,QAAQ,CAAe;IAC/B,OAAO,CAAC,cAAc,CAAgB;IACtC,OAAO,CAAC,OAAO,CAAC,CAAS;IACzB,OAAO,CAAC,eAAe,CAAU;IACjC,OAAO,CAAC,MAAM,CAA2C;IACzD,OAAO,CAAC,YAAY,CAAe;IACnC,OAAO,CAAC,YAAY,CAAmB;IACvC,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,cAAc,CAAiB;gBAE3B,MAAM,EAAE,WAAW,EAAE,IAAI,CAAC,EAAE,iBAAiB;IAqBzD,GAAG,CAAC,OAAO,EAAE,SAAS,GAAG,aAAa,GAAG,IAAI;IAQvC,QAAQ,IAAI,OAAO,CAAC,IAAI,CAAC;IAI/B,IAAI,cAAc,IAAI,cAAc,CAEnC;IAED,IAAI,YAAY,IAAI,YAAY,CAE/B;IAED,IAAI,aAAa,IAAI,aAAa,CAEjC;IAED,IAAI,QAAQ,IAAI,OAAO,gBAAgB,EAAE,QAAQ,CAEhD;IAED,IAAI,WAAW,IAAI,OAAO,mBAAmB,EAAE,WAAW,CAEzD;IAED,IAAI,KAAK,IAAI,OAAO,oBAAoB,EAAE,UAAU,CAEnD;IAED,IAAI,eAAe,IAAI,cAAc,CAEpC;IAEK,GAAG,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC;IAmC5E,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,OAAO,CAAC,cAAc,CAAC;IA6BlF,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,cAAc,CAAC,MAAM,CAAC;IAuB9E,YAAY,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,UAAU,CAAC,WAAW,GAAG,cAAc,CAAC,OAAO,eAAe,EAAE,WAAW,CAAC;IAsBxH,2EAA2E;IAC3E,eAAe,IAAI,IAAI;IAIvB,6FAA6F;IAC7F,OAAO,CAAC,mBAAmB;YAMb,MAAM;YAYN,aAAa;IAc3B,OAAO,CAAC,aAAa;IAUrB,OAAO,CAAC,yBAAyB;CAgBlC"}

package/dist/agent.js

@@ -0,0 +1,260 @@
+import { PipelineRunner } from './pipeline.js';
+import { ToolRegistry } from './tool-registry.js';
+import { PluginManager } from './plugin-manager.js';
+import { LLMInvoker } from './llm-invoker.js';
+import { ModelFactory } from './model-factory.js';
+import { getDefaultBuiltInGateway } from './gateways/builtin-gateway.js';
+import { AuthError, ModelNotFoundError } from './errors.js';
+import { LoopOrchestrator } from './loop-orchestrator.js';
+import { JsonlCheckpointStore } from './checkpoint-store.js';
+import { echoTool } from '@primo-ai/tools';
+import { processInputProcessor, prepareStepExtensionPoint, createInvokeLLMProcessor, processStepOutputProcessor, gateToolExtensionPoint, createExecuteToolsProcessor, createEvaluateIterationProcessor, processOutputProcessor, } from './processors/index.js';
+import { ContextBuilder } from './context-builder.js';
+export class Agent {
+    config;
+    runner;
+    registry;
+    _pluginManager;
+    _tracer;
+    _autoCheckpoint;
+    _model = null;
+    modelFactory;
+    orchestrator;
+    sessionManager;
+    contextBuilder;
+    constructor(config, deps) {
+        this.config = config;
+        this.modelFactory = deps?.modelFactory ?? new ModelFactory();
+        this.modelFactory.registerGateway(getDefaultBuiltInGateway());
+        this._tracer = deps?.tracer;
+        this.runner = deps?.runner ?? new PipelineRunner({ tracer: deps?.tracer });
+        this.registry = deps?.registry ?? new ToolRegistry();
+        this.contextBuilder = deps?.contextBuilder ?? new ContextBuilder({ registry: this.registry });
+        this._pluginManager = deps?.pluginManager ?? new PluginManager(this.runner, this.registry, this.contextBuilder);
+        this.registry.setHookManager(this._pluginManager.hookManager);
+        this.registry.setEventBus(this._pluginManager.eventBus);
+        this.runner.setHookManager(this._pluginManager.hookManager);
+        const store = deps?.checkpointStore ?? (deps?.checkpointDir ? new JsonlCheckpointStore(deps.checkpointDir) : undefined);
+        this.orchestrator = new LoopOrchestrator(this.runner, this._pluginManager.hookManager, store, this._pluginManager.eventBus, deps?.stageConfig);
+        this.sessionManager = deps?.sessionManager;
+        this._autoCheckpoint = deps?.autoCheckpoint ?? false;
+        this.registerTools();
+        this.registerBuiltinProcessors();
+        this._pluginManager.setStageMutator((m) => this.orchestrator.applyMutation(m));
+    }
+    use(factory) {
+        if (typeof factory === 'function') {
+            this._pluginManager.initializePlugin(factory);
+        }
+        else {
+            this.runner.register(factory);
+        }
+    }
+    async teardown() {
+        await this._pluginManager.shutdown();
+    }
+    get pipelineRunner() {
+        return this.runner;
+    }
+    get toolRegistry() {
+        return this.registry;
+    }
+    get pluginManager() {
+        return this._pluginManager;
+    }
+    get eventBus() {
+        return this._pluginManager.eventBus;
+    }
+    get eventSystem() {
+        return this._pluginManager.eventSystem;
+    }
+    get state() {
+        return this.orchestrator.state;
+    }
+    get _contextBuilder() {
+        return this.contextBuilder;
+    }
+    async run(input, signal) {
+        if (signal?.aborted)
+            throw new DOMException('Agent run aborted', 'AbortError');
+        this._pluginManager.freezeHarnessInstances();
+        const context = await this.createContext(input);
+        const hm = this._pluginManager.hookManager;
+        // agent.start hook
+        await hm.invoke('agent.start', { sessionId: context.request.sessionId, request: context.request, agentConfig: this.config }, {});
+        const maxIter = typeof this.config.maxIterations === 'number' ? this.config.maxIterations : 10;
+        try {
+            const { context: finalCtx, compatRetries } = await this.orchestrator.runLoop(context, {
+                maxIterations: maxIter,
+                signal,
+                modelString: this.config.model,
+                sessionId: context.request.sessionId,
+                autoCheckpoint: this._autoCheckpoint,
+            });
+            return {
+                response: finalCtx.iteration.response ?? '',
+                tokenUsage: finalCtx.session.totalTokenUsage ?? { input: 0, output: 0 },
+                sessionId: context.request.sessionId,
+                compatRetries,
+            };
+        }
+        catch (error) {
+            this.autoInvalidateModel(error);
+            throw error;
+        }
+        finally {
+            // agent.end hook — always fires, even on error; suppress hook errors to preserve original
+            try {
+                await hm.invoke('agent.end', { sessionId: context.request.sessionId }, {});
+            }
+            catch { /* hook error must not mask original */ }
+        }
+    }
+    async resume(sessionId, signal) {
+        if (signal?.aborted)
+            throw new DOMException('Agent resume aborted', 'AbortError');
+        const hm = this._pluginManager.hookManager;
+        const maxIter = typeof this.config.maxIterations === 'number' ? this.config.maxIterations : 10;
+        try {
+            const { context: finalCtx, compatRetries } = await this.orchestrator.resumeLoop(sessionId, {
+                maxIterations: maxIter,
+                signal,
+                modelString: this.config.model,
+                sessionId,
+                autoCheckpoint: this._autoCheckpoint,
+            });
+            return {
+                response: finalCtx.iteration.response ?? '',
+                tokenUsage: finalCtx.session.totalTokenUsage ?? { input: 0, output: 0 },
+                sessionId,
+                compatRetries,
+            };
+        }
+        catch (error) {
+            this.autoInvalidateModel(error);
+            throw error;
+        }
+        finally {
+            try {
+                await hm.invoke('agent.end', { sessionId }, {});
+            }
+            catch { /* hook error must not mask original */ }
+        }
+    }
+    async *stream(input, signal) {
+        if (signal?.aborted)
+            throw new DOMException('Agent stream aborted', 'AbortError');
+        const context = await this.createContext(input);
+        const hm = this._pluginManager.hookManager;
+        // agent.start hook
+        await hm.invoke('agent.start', { sessionId: context.request.sessionId, request: context.request, agentConfig: this.config }, {});
+        const maxIter = typeof this.config.maxIterations === 'number' ? this.config.maxIterations : 10;
+        try {
+            yield* this.orchestrator.streamLoop(context, {
+                maxIterations: maxIter,
+                signal,
+                modelString: this.config.model,
+                sessionId: context.request.sessionId,
+                autoCheckpoint: this._autoCheckpoint,
+            });
+        }
+        finally {
+            try {
+                await hm.invoke('agent.end', { sessionId: context.request.sessionId }, {});
+            }
+            catch { /* hook error must not mask original */ }
+        }
+    }
+    async *streamEvents(input, signal) {
+        if (signal?.aborted)
+            throw new DOMException('Agent stream aborted', 'AbortError');
+        const context = await this.createContext(input);
+        const hm = this._pluginManager.hookManager;
+        await hm.invoke('agent.start', { sessionId: context.request.sessionId, request: context.request, agentConfig: this.config }, {});
+        const maxIter = typeof this.config.maxIterations === 'number' ? this.config.maxIterations : 10;
+        try {
+            yield* this.orchestrator.streamEvents(context, {
+                maxIterations: maxIter,
+                signal,
+                modelString: this.config.model,
+                sessionId: context.request.sessionId,
+                autoCheckpoint: this._autoCheckpoint,
+            });
+        }
+        finally {
+            try {
+                await hm.invoke('agent.end', { sessionId: context.request.sessionId }, {});
+            }
+            catch { /* hook error must not mask original */ }
+        }
+    }
+    /** Clear the cached model so the next run re-resolves from the factory. */
+    invalidateModel() {
+        this._model = null;
+    }
+    /** Auto-invalidate cached model when the error indicates auth failure or model-not-found. */
+    autoInvalidateModel(error) {
+        if (isAuthOrNotFoundError(error)) {
+            this._model = null;
+        }
+    }
+    async getLLM(systemPrompt) {
+        if (!this._model) {
+            this._model = await this.modelFactory.resolve(this.config.model);
+        }
+        return new LLMInvoker({
+            model: this._model,
+            system: systemPrompt,
+            retryOptions: { maxRetries: 3, baseDelay: 1000 },
+            tracer: this._tracer,
+        });
+    }
+    async createContext(input) {
+        let sessionId = crypto.randomUUID();
+        if (this.sessionManager) {
+            const record = await this.sessionManager.start(input);
+            sessionId = record.sessionId;
+        }
+        return {
+            request: { input, sessionId },
+            agent: { config: { ...this.config }, promptFragments: [], toolDeclarations: [] },
+            iteration: { step: 0 },
+            session: { custom: {} },
+        };
+    }
+    registerTools() {
+        const userToolNames = new Set((this.config.tools ?? []).map(t => t.name));
+        if (!userToolNames.has(echoTool.name)) {
+            this.registry.register(echoTool);
+        }
+        for (const tool of this.config.tools ?? []) {
+            this.registry.register(tool);
+        }
+    }
+    registerBuiltinProcessors() {
+        this.runner.register(processInputProcessor);
+        this.runner.register(this.contextBuilder.createProcessor());
+        this.runner.register(prepareStepExtensionPoint);
+        this.runner.register(createInvokeLLMProcessor({
+            getLLM: (systemPrompt) => this.getLLM(systemPrompt),
+            registry: this.registry,
+            hookManager: this._pluginManager.hookManager,
+            modelString: this.config.model,
+        }));
+        this.runner.register(processStepOutputProcessor);
+        this.runner.register(gateToolExtensionPoint);
+        this.runner.register(createExecuteToolsProcessor(this.registry));
+        this.runner.register(createEvaluateIterationProcessor({ eventBus: this._pluginManager.eventBus }));
+        this.runner.register(processOutputProcessor);
+    }
+}
+function isAuthOrNotFoundError(error) {
+    if (error instanceof AuthError || error instanceof ModelNotFoundError)
+        return true;
+    if (!(error instanceof Error))
+        return false;
+    const err = error;
+    const code = err.statusCode ?? err.status;
+    return code === 401 || code === 403 || code === 404;
+}
+//# sourceMappingURL=agent.js.map

package/dist/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../src/agent.ts"],"names":[],"mappings":"AAUA,OAAO,EAAE,cAAc,EAAE,MAAM,eAAe,CAAC;AAE/C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,aAAa,EAAsB,MAAM,qBAAqB,CAAC;AACxE,OAAO,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AAC9C,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAkB,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AACzF,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AAC5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,EAAE,QAAQ,EAAE,MAAM,iBAAiB,CAAC;AAC3C,OAAO,EACL,qBAAqB,EACrB,yBAAyB,EACzB,wBAAwB,EACxB,0BAA0B,EAC1B,sBAAsB,EACtB,2BAA2B,EAC3B,gCAAgC,EAChC,sBAAsB,GACvB,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AA2BtD,MAAM,OAAO,KAAK;IACR,MAAM,CAAc;IACpB,MAAM,CAAiB;IACvB,QAAQ,CAAe;IACvB,cAAc,CAAgB;IAC9B,OAAO,CAAU;IACjB,eAAe,CAAU;IACzB,MAAM,GAAsC,IAAI,CAAC;IACjD,YAAY,CAAe;IAC3B,YAAY,CAAmB;IAC/B,cAAc,CAAkB;IAChC,cAAc,CAAiB;IAEvC,YAAY,MAAmB,EAAE,IAAwB;QACvD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,YAAY,GAAG,IAAI,EAAE,YAAY,IAAI,IAAI,YAAY,EAAE,CAAC;QAC7D,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,wBAAwB,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,OAAO,GAAG,IAAI,EAAE,MAAM,CAAC;QAC5B,IAAI,CAAC,MAAM,GAAG,IAAI,EAAE,MAAM,IAAI,IAAI,cAAc,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;QAC3E,IAAI,CAAC,QAAQ,GAAG,IAAI,EAAE,QAAQ,IAAI,IAAI,YAAY,EAAE,CAAC;QACrD,IAAI,CAAC,cAAc,GAAG,IAAI,EAAE,cAAc,IAAI,IAAI,cAAc,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;QAC9F,IAAI,CAAC,cAAc,GAAG,IAAI,EAAE,aAAa,IAAI,IAAI,aAAa,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,CAAC;QAChH,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QAC9D,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACxD,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC,CAAC;QAC5D,MAAM,KAAK,GAAG,IAAI,EAAE,eAAe,IAAI,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC,CAAC,IAAI,oBAAoB,CAA+B,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACtJ,IAAI,CAAC,YAAY,GAAG,IAAI,gBAAgB,CAAC,IAAI,CAAC,MAAM,EAAE,IAAI,CAAC,cAAc,CAAC,WAAW,EAAE,KAAK,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QAC/I,IAAI,CAAC,cAAc,GAAG,IAAI,EAAE,cAAc,CAAC;QAC3C,IAAI,CAAC,eAAe,GAAG,IAAI,EAAE,cAAc,IAAI,KAAK,CAAC;QACrD,IAAI,CAAC,aAAa,EAAE,CAAC;QACrB,IAAI,CAAC,yBAAyB,EAAE,CAAC;QACjC,IAAI,CAAC,cAAc,CAAC,eAAe,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,YAAY,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,CAAC;IACjF,CAAC;IAED,GAAG,CAAC,OAAkC;QACpC,IAAI,OAAO,OAAO,KAAK,UAAU,EAAE,CAAC;YAClC,IAAI,CAAC,cAAc,CAAC,gBAAgB,CAAC,OAAwB,CAAC,CAAC;QACjE,CAAC;aAAM,CAAC;YACN,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,OAAoB,CAAC,CAAC;QAC7C,CAAC;IACH,CAAC;IAED,KAAK,CAAC,QAAQ;QACZ,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC;IACvC,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,MAAM,CAAC;IACrB,CAAC;IAED,IAAI,YAAY;QACd,OAAO,IAAI,CAAC,QAAQ,CAAC;IACvB,CAAC;IAED,IAAI,aAAa;QACf,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED,IAAI,QAAQ;QACV,OAAO,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC;IACtC,CAAC;IAED,IAAI,WAAW;QACb,OAAO,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;IACzC,CAAC;IAED,IAAI,KAAK;QACP,OAAO,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC;IACjC,CAAC;IAED,IAAI,eAAe;QACjB,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAED,KAAK,CAAC,GAAG,CAAC,KAAa,EAAE,MAA+B;QACtD,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,YAAY,CAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;QAC/E,IAAI,CAAC,cAAc,CAAC,sBAAsB,EAAE,CAAC;QAE7C,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;QAE3C,mBAAmB;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEjI,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,OAAO,EAAE;gBACpF,aAAa,EAAE,OAAO;gBACtB,MAAM;gBACN,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBAC9B,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS;gBACpC,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,CAAC;YAEH,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,QAAkB,IAAI,EAAE;gBACrD,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;gBACvE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS;gBACpC,aAAa;aACd,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,0FAA0F;YAC1F,IAAI,CAAC;gBAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,uCAAuC,CAAC,CAAC;QACvI,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB,EAAE,MAA+B;QAC7D,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;QAElF,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;QAC3C,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;QAE/F,IAAI,CAAC;YACH,MAAM,EAAE,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAE,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,SAAS,EAAE;gBACzF,aAAa,EAAE,OAAO;gBACtB,MAAM;gBACN,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBAC9B,SAAS;gBACT,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,CAAC;YAEH,OAAO;gBACL,QAAQ,EAAE,QAAQ,CAAC,SAAS,CAAC,QAAkB,IAAI,EAAE;gBACrD,UAAU,EAAE,QAAQ,CAAC,OAAO,CAAC,eAAe,IAAI,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE;gBACvE,SAAS;gBACT,aAAa;aACd,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAChC,MAAM,KAAK,CAAC;QACd,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,uCAAuC,CAAC,CAAC;QAC5G,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CAAC,KAAa,EAAE,MAA+B;QAC1D,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;QAElF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;QAE3C,mBAAmB;QACnB,MAAM,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEjI,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,IAAI,CAAC;YACH,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,OAAO,EAAE;gBAC3C,aAAa,EAAE,OAAO;gBACtB,MAAM;gBACN,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBAC9B,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS;gBACpC,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,uCAAuC,CAAC,CAAC;QACvI,CAAC;IACH,CAAC;IAED,KAAK,CAAC,CAAC,YAAY,CAAC,KAAa,EAAE,MAA+B;QAChE,IAAI,MAAM,EAAE,OAAO;YAAE,MAAM,IAAI,YAAY,CAAC,sBAAsB,EAAE,YAAY,CAAC,CAAC;QAElF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAChD,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,WAAW,CAAC;QAE3C,MAAM,EAAE,CAAC,MAAM,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,OAAO,EAAE,WAAW,EAAE,IAAI,CAAC,MAAM,EAAE,EAAE,EAAE,CAAC,CAAC;QAEjI,MAAM,OAAO,GAAG,OAAO,IAAI,CAAC,MAAM,CAAC,aAAa,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,CAAC,CAAC,EAAE,CAAC;QAC/F,IAAI,CAAC;YACH,KAAK,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,YAAY,CAAC,OAAO,EAAE;gBAC7C,aAAa,EAAE,OAAO;gBACtB,MAAM;gBACN,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;gBAC9B,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS;gBACpC,cAAc,EAAE,IAAI,CAAC,eAAe;aACrC,CAAC,CAAC;QACL,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC;gBAAC,MAAM,EAAE,CAAC,MAAM,CAAC,WAAW,EAAE,EAAE,SAAS,EAAE,OAAO,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;YAAC,CAAC;YAAC,MAAM,CAAC,CAAC,uCAAuC,CAAC,CAAC;QACvI,CAAC;IACH,CAAC;IAED,2EAA2E;IAC3E,eAAe;QACb,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;IACrB,CAAC;IAED,6FAA6F;IACrF,mBAAmB,CAAC,KAAc;QACxC,IAAI,qBAAqB,CAAC,KAAK,CAAC,EAAE,CAAC;YACjC,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACrB,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,MAAM,CAAC,YAAqB;QACxC,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC;YACjB,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnE,CAAC;QACD,OAAO,IAAI,UAAU,CAAC;YACpB,KAAK,EAAE,IAAI,CAAC,MAAM;YAClB,MAAM,EAAE,YAAY;YACpB,YAAY,EAAE,EAAE,UAAU,EAAE,CAAC,EAAE,SAAS,EAAE,IAAI,EAAE;YAChD,MAAM,EAAE,IAAI,CAAC,OAAO;SACrB,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,KAAa;QACvC,IAAI,SAAS,GAAW,MAAM,CAAC,UAAU,EAAE,CAAC;QAC5C,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACtD,SAAS,GAAG,MAAM,CAAC,SAAS,CAAC;QAC/B,CAAC;QACD,OAAO;YACL,OAAO,EAAE,EAAE,KAAK,EAAE,SAAS,EAAE;YAC7B,KAAK,EAAE,EAAE,MAAM,EAAE,EAAE,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,eAAe,EAAE,EAAE,EAAE,gBAAgB,EAAE,EAAE,EAAE;YAChF,SAAS,EAAE,EAAE,IAAI,EAAE,CAAC,EAAE;YACtB,OAAO,EAAE,EAAE,MAAM,EAAE,EAAE,EAAE;SACxB,CAAC;IACJ,CAAC;IAEO,aAAa;QACnB,MAAM,aAAa,GAAG,IAAI,GAAG,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QAC1E,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACtC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,QAAgB,CAAC,CAAC;QAC3C,CAAC;QACD,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,EAAE,CAAC;YAC3C,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,IAAY,CAAC,CAAC;QACvC,CAAC;IACH,CAAC;IAEO,yBAAyB;QAC/B,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,eAAe,EAAE,CAAC,CAAC;QAC5D,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,yBAAyB,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,wBAAwB,CAAC;YAC5C,MAAM,EAAE,CAAC,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;YACnD,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,WAAW,EAAE,IAAI,CAAC,cAAc,CAAC,WAAW;YAC5C,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,KAAK;SAC/B,CAAC,CAAC,CAAC;QACJ,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,0BAA0B,CAAC,CAAC;QACjD,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,2BAA2B,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,gCAAgC,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC;QACnG,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,sBAAsB,CAAC,CAAC;IAC/C,CAAC;CACF;AAED,SAAS,qBAAqB,CAAC,KAAc;IAC3C,IAAI,KAAK,YAAY,SAAS,IAAI,KAAK,YAAY,kBAAkB;QAAE,OAAO,IAAI,CAAC;IACnF,IAAI,CAAC,CAAC,KAAK,YAAY,KAAK,CAAC;QAAE,OAAO,KAAK,CAAC;IAC5C,MAAM,GAAG,GAAG,KAAyD,CAAC;IACtE,MAAM,IAAI,GAAG,GAAG,CAAC,UAAU,IAAI,GAAG,CAAC,MAAM,CAAC;IAC1C,OAAO,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,CAAC;AACtD,CAAC"}

package/dist/checkpoint-store.d.ts

@@ -0,0 +1,19 @@
+import type { CheckpointStore } from '@primo-ai/sdk';
+export declare class InMemoryCheckpointStore<T = unknown> implements CheckpointStore<T> {
+    private store;
+    save(sessionId: string, data: T): Promise<void>;
+    load(sessionId: string): Promise<T | undefined>;
+    delete(sessionId: string): Promise<void>;
+    list(): Promise<string[]>;
+}
+export declare class JsonlCheckpointStore<T = unknown> implements CheckpointStore<T> {
+    private dir;
+    constructor(dir: string);
+    private validateSessionId;
+    private path;
+    save(sessionId: string, data: T): Promise<void>;
+    load(sessionId: string): Promise<T | undefined>;
+    delete(sessionId: string): Promise<void>;
+    list(): Promise<string[]>;
+}
+//# sourceMappingURL=checkpoint-store.d.ts.map

package/dist/checkpoint-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint-store.d.ts","sourceRoot":"","sources":["../src/checkpoint-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAQrD,qBAAa,uBAAuB,CAAC,CAAC,GAAG,OAAO,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAC7E,OAAO,CAAC,KAAK,CAAwB;IAE/B,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAI/C,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAI/C,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIxC,IAAI,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAGhC;AAED,qBAAa,oBAAoB,CAAC,CAAC,GAAG,OAAO,CAAE,YAAW,eAAe,CAAC,CAAC,CAAC;IAC9D,OAAO,CAAC,GAAG;gBAAH,GAAG,EAAE,MAAM;IAE/B,OAAO,CAAC,iBAAiB;IAMzB,OAAO,CAAC,IAAI;IAIN,IAAI,CAAC,SAAS,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,GAAG,OAAO,CAAC,IAAI,CAAC;IAS/C,IAAI,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,CAAC,GAAG,SAAS,CAAC;IAa/C,MAAM,CAAC,SAAS,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IASxC,IAAI,IAAI,OAAO,CAAC,MAAM,EAAE,CAAC;CAShC"}

package/dist/checkpoint-store.js

@@ -0,0 +1,79 @@
+import { readFile, writeFile, mkdir, rm, readdir, rename } from 'node:fs/promises';
+import { join } from 'node:path';
+function hasCode(err) {
+    return typeof err === 'object' && err !== null && 'code' in err;
+}
+export class InMemoryCheckpointStore {
+    store = new Map();
+    async save(sessionId, data) {
+        this.store.set(sessionId, data);
+    }
+    async load(sessionId) {
+        return this.store.get(sessionId);
+    }
+    async delete(sessionId) {
+        this.store.delete(sessionId);
+    }
+    async list() {
+        return [...this.store.keys()];
+    }
+}
+export class JsonlCheckpointStore {
+    dir;
+    constructor(dir) {
+        this.dir = dir;
+    }
+    validateSessionId(sessionId) {
+        if (sessionId.includes('/') || sessionId.includes('\\') || sessionId.includes('..')) {
+            throw new Error(`Invalid sessionId: ${sessionId}`);
+        }
+    }
+    path(sessionId) {
+        return join(this.dir, `${sessionId}.jsonl`);
+    }
+    async save(sessionId, data) {
+        this.validateSessionId(sessionId);
+        await mkdir(this.dir, { recursive: true });
+        const target = this.path(sessionId);
+        const tmp = `${target}.tmp`;
+        await writeFile(tmp, JSON.stringify(data) + '\n', 'utf-8');
+        await rename(tmp, target);
+    }
+    async load(sessionId) {
+        this.validateSessionId(sessionId);
+        try {
+            const content = await readFile(this.path(sessionId), 'utf-8');
+            const line = content.trim();
+            if (!line)
+                return undefined;
+            return JSON.parse(line);
+        }
+        catch (err) {
+            if (hasCode(err) && err.code === 'ENOENT')
+                return undefined;
+            throw err;
+        }
+    }
+    async delete(sessionId) {
+        this.validateSessionId(sessionId);
+        try {
+            await rm(this.path(sessionId));
+        }
+        catch (err) {
+            if (!hasCode(err) || err.code !== 'ENOENT')
+                throw err;
+        }
+    }
+    async list() {
+        try {
+            const files = await readdir(this.dir);
+            return files.filter(f => f.endsWith('.jsonl')).map(f => f.slice(0, -6));
+        }
+        catch (err) {
+            if (hasCode(err) && err.code === 'ENOENT')
+                return [];
+            throw err;
+        }
+    }
+}
+//# sourceMappingURL=checkpoint-store.js.map

package/dist/checkpoint-store.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"checkpoint-store.js","sourceRoot":"","sources":["../src/checkpoint-store.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,EAAE,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,kBAAkB,CAAC;AACnF,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,SAAS,OAAO,CAAC,GAAY;IAC3B,OAAO,OAAO,GAAG,KAAK,QAAQ,IAAI,GAAG,KAAK,IAAI,IAAI,MAAM,IAAI,GAAG,CAAC;AAClE,CAAC;AAED,MAAM,OAAO,uBAAuB;IAC1B,KAAK,GAAG,IAAI,GAAG,EAAa,CAAC;IAErC,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,IAAO;QACnC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAClC,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB;QAC1B,OAAO,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IACnC,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;IAC/B,CAAC;IAED,KAAK,CAAC,IAAI;QACR,OAAO,CAAC,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;IAChC,CAAC;CACF;AAED,MAAM,OAAO,oBAAoB;IACX;IAApB,YAAoB,GAAW;QAAX,QAAG,GAAH,GAAG,CAAQ;IAAG,CAAC;IAE3B,iBAAiB,CAAC,SAAiB;QACzC,IAAI,SAAS,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,SAAS,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;YACpF,MAAM,IAAI,KAAK,CAAC,sBAAsB,SAAS,EAAE,CAAC,CAAC;QACrD,CAAC;IACH,CAAC;IAEO,IAAI,CAAC,SAAiB;QAC5B,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,QAAQ,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB,EAAE,IAAO;QACnC,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClC,MAAM,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAC3C,MAAM,MAAM,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QACpC,MAAM,GAAG,GAAG,GAAG,MAAM,MAAM,CAAC;QAC5B,MAAM,SAAS,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,OAAO,CAAC,CAAC;QAC3D,MAAM,MAAM,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;IAC5B,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,SAAiB;QAC1B,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,EAAE,OAAO,CAAC,CAAC;YAC9D,MAAM,IAAI,GAAG,OAAO,CAAC,IAAI,EAAE,CAAC;YAC5B,IAAI,CAAC,IAAI;gBAAE,OAAO,SAAS,CAAC;YAC5B,OAAO,IAAI,CAAC,KAAK,CAAC,IAAI,CAAM,CAAC;QAC/B,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,SAAS,CAAC;YAC5D,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;IAED,KAAK,CAAC,MAAM,CAAC,SAAiB;QAC5B,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC;YACH,MAAM,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,MAAM,GAAG,CAAC;QACxD,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI;QACR,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACtC,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC;QAC1E,CAAC;QAAC,OAAO,GAAY,EAAE,CAAC;YACtB,IAAI,OAAO,CAAC,GAAG,CAAC,IAAI,GAAG,CAAC,IAAI,KAAK,QAAQ;gBAAE,OAAO,EAAE,CAAC;YACrD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC;CACF"}

package/dist/concurrency-controller.d.ts

@@ -0,0 +1,9 @@
+import type { ConcurrencySlot } from '@primo-ai/sdk';
+export declare class ConcurrencyController {
+    private slots;
+    constructor(slotDefs: ConcurrencySlot[]);
+    acquire(slotKey: string, timeoutMs?: number): Promise<() => void>;
+    getActiveCount(slotKey: string): number;
+    private releaseSlot;
+}
+//# sourceMappingURL=concurrency-controller.d.ts.map

package/dist/concurrency-controller.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"concurrency-controller.d.ts","sourceRoot":"","sources":["../src/concurrency-controller.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AAQrD,qBAAa,qBAAqB;IAChC,OAAO,CAAC,KAAK,CAAgC;gBAEjC,QAAQ,EAAE,eAAe,EAAE;IAMvC,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,CAAC;IAsDjE,cAAc,CAAC,OAAO,EAAE,MAAM,GAAG,MAAM;IAKvC,OAAO,CAAC,WAAW;CAQpB"}

package/dist/concurrency-controller.js

@@ -0,0 +1,71 @@
+export class ConcurrencyController {
+    slots = new Map();
+    constructor(slotDefs) {
+        for (const { key, maxConcurrent } of slotDefs) {
+            this.slots.set(key, { current: 0, max: maxConcurrent, waiters: [] });
+        }
+    }
+    acquire(slotKey, timeoutMs) {
+        const slot = this.slots.get(slotKey);
+        if (!slot) {
+            return Promise.reject(new Error(`Unknown concurrency slot: ${slotKey}`));
+        }
+        if (slot.current < slot.max) {
+            slot.current++;
+            let released = false;
+            return Promise.resolve(() => {
+                if (released)
+                    return;
+                released = true;
+                this.releaseSlot(slot);
+            });
+        }
+        let waiterRef = null;
+        const acquirePromise = new Promise((resolve) => {
+            const waiter = () => {
+                let released = false;
+                resolve(() => {
+                    if (released)
+                        return;
+                    released = true;
+                    this.releaseSlot(slot);
+                });
+            };
+            waiterRef = waiter;
+            slot.waiters.push(waiter);
+        });
+        if (timeoutMs == null) {
+            return acquirePromise;
+        }
+        let timer;
+        const timeoutPromise = new Promise((_, reject) => {
+            timer = setTimeout(() => {
+                // Remove our waiter from the queue so it doesn't leak
+                if (waiterRef) {
+                    const idx = slot.waiters.indexOf(waiterRef);
+                    if (idx !== -1)
+                        slot.waiters.splice(idx, 1);
+                }
+                reject(new Error(`Concurrency acquire timed out for slot "${slotKey}" after ${timeoutMs}ms`));
+            }, timeoutMs);
+        });
+        return Promise.race([acquirePromise, timeoutPromise]).finally(() => {
+            clearTimeout(timer);
+            // If acquire won the race, the waiter was already consumed — nothing to clean.
+            // If timeout won, we already removed it above.
+        });
+    }
+    getActiveCount(slotKey) {
+        const slot = this.slots.get(slotKey);
+        return slot ? slot.current : 0;
+    }
+    releaseSlot(slot) {
+        slot.current--;
+        if (slot.waiters.length > 0) {
+            const next = slot.waiters.shift();
+            slot.current++;
+            next();
+        }
+    }
+}
+//# sourceMappingURL=concurrency-controller.js.map