@oni.bot/core 0.6.1

package/dist/events/types.d.ts

@@ -0,0 +1,87 @@
+import type { TokenUsage } from "../models/types.js";
+export interface AgentStartEvent {
+    type: "agent.start";
+    agent: string;
+    timestamp: number;
+    step: number;
+}
+export interface AgentEndEvent {
+    type: "agent.end";
+    agent: string;
+    timestamp: number;
+    step: number;
+    duration: number;
+    usage?: TokenUsage;
+}
+export interface LLMRequestEvent {
+    type: "llm.request";
+    agent: string;
+    model: string;
+    timestamp: number;
+    messageCount: number;
+    hasTools: boolean;
+}
+export interface LLMResponseEvent {
+    type: "llm.response";
+    agent: string;
+    model: string;
+    timestamp: number;
+    duration: number;
+    usage: TokenUsage;
+    stopReason: string;
+    hasToolCalls: boolean;
+}
+export interface ToolCallEvent {
+    type: "tool.call";
+    agent: string;
+    tool: string;
+    timestamp: number;
+    input: Record<string, unknown>;
+}
+export interface ToolResultEvent {
+    type: "tool.result";
+    agent: string;
+    tool: string;
+    timestamp: number;
+    duration: number;
+    status: "success" | "error";
+    output?: unknown;
+    error?: string;
+}
+export interface HandoffEvent {
+    type: "handoff";
+    from: string;
+    to: string;
+    timestamp: number;
+    message?: string;
+}
+export interface FilterBlockedEvent {
+    type: "filter.blocked";
+    filter: string;
+    agent: string;
+    direction: "input" | "output";
+    reason: string;
+    timestamp: number;
+}
+export interface BudgetWarningEvent {
+    type: "budget.warning";
+    agent: string;
+    percentUsed: number;
+    metric: string;
+    timestamp: number;
+}
+export interface ErrorEvent {
+    type: "error";
+    agent?: string;
+    error: Error;
+    timestamp: number;
+}
+export type LifecycleEvent = AgentStartEvent | AgentEndEvent | LLMRequestEvent | LLMResponseEvent | ToolCallEvent | ToolResultEvent | HandoffEvent | FilterBlockedEvent | BudgetWarningEvent | ErrorEvent;
+export type EventType = LifecycleEvent["type"];
+export type EventHandler<T extends LifecycleEvent = LifecycleEvent> = (event: T) => void;
+export type EventListeners = {
+    [K in EventType]?: EventHandler<Extract<LifecycleEvent, {
+        type: K;
+    }>>;
+};
+//# sourceMappingURL=types.d.ts.map

package/dist/events/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/events/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AAErD,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,aAAa,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;CACd;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,CAAC,EAAE,UAAU,CAAC;CACpB;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,aAAa,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,YAAY,EAAE,MAAM,CAAC;IACrB,QAAQ,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,WAAW,gBAAgB;IAC/B,IAAI,EAAE,cAAc,CAAC;IACrB,KAAK,EAAE,MAAM,CAAC;IACd,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,KAAK,EAAE,UAAU,CAAC;IAClB,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,OAAO,CAAC;CACvB;AAED,MAAM,WAAW,aAAa;IAC5B,IAAI,EAAE,WAAW,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAChC;AAED,MAAM,WAAW,eAAe;IAC9B,IAAI,EAAE,aAAa,CAAC;IACpB,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,SAAS,GAAG,OAAO,CAAC;IAC5B,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED,MAAM,WAAW,YAAY;IAC3B,IAAI,EAAE,SAAS,CAAC;IAChB,IAAI,EAAE,MAAM,CAAC;IACb,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,EAAE,MAAM,CAAC;IAClB,OAAO,CAAC,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,gBAAgB,CAAC;IACvB,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,OAAO,GAAG,QAAQ,CAAC;IAC9B,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,kBAAkB;IACjC,IAAI,EAAE,gBAAgB,CAAC;IACvB,KAAK,EAAE,MAAM,CAAC;IACd,WAAW,EAAE,MAAM,CAAC;IACpB,MAAM,EAAE,MAAM,CAAC;IACf,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,UAAU;IACzB,IAAI,EAAE,OAAO,CAAC;IACd,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,KAAK,EAAE,KAAK,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,MAAM,cAAc,GACtB,eAAe,GACf,aAAa,GACb,eAAe,GACf,gBAAgB,GAChB,aAAa,GACb,eAAe,GACf,YAAY,GACZ,kBAAkB,GAClB,kBAAkB,GAClB,UAAU,CAAC;AAEf,MAAM,MAAM,SAAS,GAAG,cAAc,CAAC,MAAM,CAAC,CAAC;AAE/C,MAAM,MAAM,YAAY,CAAC,CAAC,SAAS,cAAc,GAAG,cAAc,IAAI,CAAC,KAAK,EAAE,CAAC,KAAK,IAAI,CAAC;AAEzF,MAAM,MAAM,cAAc,GAAG;KAC1B,CAAC,IAAI,SAAS,CAAC,CAAC,EAAE,YAAY,CAAC,OAAO,CAAC,cAAc,EAAE;QAAE,IAAI,EAAE,CAAC,CAAA;KAAE,CAAC,CAAC;CACtE,CAAC"}

package/dist/events/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/events/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../src/events/types.ts"],"names":[],"mappings":""}

package/dist/functional.d.ts

@@ -0,0 +1,36 @@
+import type { ChannelSchema, ONIConfig, ONICheckpointer } from "./types.js";
+import type { ONISkeletonV3 } from "./graph.js";
+export interface TaskOptions {
+    /** Retry policy */
+    maxAttempts?: number;
+    initialDelay?: number;
+}
+export interface TaskDef<TInput, TOutput> {
+    name: string;
+    invoke: (input: TInput, config?: ONIConfig) => Promise<TOutput>;
+}
+export declare function task<TInput, TOutput>(name: string, fn: (input: TInput, config?: ONIConfig) => Promise<TOutput> | TOutput, opts?: TaskOptions): TaskDef<TInput, TOutput>;
+export interface EntrypointOptions<S extends Record<string, unknown>> {
+    channels: ChannelSchema<S>;
+    checkpointer?: ONICheckpointer<S>;
+    /** Node name (default: "entrypoint") */
+    name?: string;
+}
+/**
+ * Create a compiled skeleton from a single async function.
+ * The function receives full state and returns a partial update.
+ *
+ * @example
+ * const app = entrypoint(
+ *   { channels: { query: lastValue(() => ""), answer: lastValue(() => "") } },
+ *   async (state) => {
+ *     const answer = await myLLM(state.query);
+ *     return { answer };
+ *   }
+ * );
+ */
+export declare function entrypoint<S extends Record<string, unknown>>(opts: EntrypointOptions<S>, fn: (state: S, config?: ONIConfig) => Promise<Partial<S>> | Partial<S>): ONISkeletonV3<S>;
+export declare function pipe<S extends Record<string, unknown>>(opts: EntrypointOptions<S>, ...tasks: Array<(state: S, config?: ONIConfig) => Promise<Partial<S>> | Partial<S>>): ONISkeletonV3<S>;
+export declare function branch<S extends Record<string, unknown>>(condition: (state: S) => string, branches: Record<string, (state: S, config?: ONIConfig) => Promise<Partial<S>> | Partial<S>>, opts: EntrypointOptions<S>): ONISkeletonV3<S>;
+export { interrupt, getUserInput, getUserApproval, getUserSelection } from "./hitl/interrupt.js";
+//# sourceMappingURL=functional.d.ts.map

package/dist/functional.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"functional.d.ts","sourceRoot":"","sources":["../src/functional.ts"],"names":[],"mappings":"AAwBA,OAAO,KAAK,EAAE,aAAa,EAAE,SAAS,EAAe,eAAe,EAAE,MAAM,YAAY,CAAC;AACzF,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAQhD,MAAM,WAAW,WAAW;IAC1B,mBAAmB;IACnB,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,YAAY,CAAC,EAAE,MAAM,CAAC;CACvB;AAED,MAAM,WAAW,OAAO,CAAC,MAAM,EAAE,OAAO;IACtC,IAAI,EAAI,MAAM,CAAC;IACf,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC;CACjE;AAED,wBAAgB,IAAI,CAAC,MAAM,EAAE,OAAO,EAClC,IAAI,EAAI,MAAM,EACd,EAAE,EAAM,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,GAAG,OAAO,EACzE,IAAI,CAAC,EAAG,WAAW,GAClB,OAAO,CAAC,MAAM,EAAE,OAAO,CAAC,CAuB1B;AAMD,MAAM,WAAW,iBAAiB,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAClE,QAAQ,EAAO,aAAa,CAAC,CAAC,CAAC,CAAC;IAChC,YAAY,CAAC,EAAE,eAAe,CAAC,CAAC,CAAC,CAAC;IAClC,wCAAwC;IACxC,IAAI,CAAC,EAAU,MAAM,CAAC;CACvB;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,UAAU,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAC1D,IAAI,EAAE,iBAAiB,CAAC,CAAC,CAAC,EAC1B,EAAE,EAAI,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,GACvE,aAAa,CAAC,CAAC,CAAC,CASlB;AAMD,wBAAgB,IAAI,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACpD,IAAI,EAAG,iBAAiB,CAAC,CAAC,CAAC,EAC3B,GAAG,KAAK,EAAE,KAAK,CAAC,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,GAClF,aAAa,CAAC,CAAC,CAAC,CAgBlB;AAMD,wBAAgB,MAAM,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EACtD,SAAS,EAAE,CAAC,KAAK,EAAE,CAAC,KAAK,MAAM,EAC/B,QAAQ,EAAG,MAAM,CAAC,MAAM,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC,EAC7F,IAAI,EAAO,iBAAiB,CAAC,CAAC,CAAC,GAC9B,aAAa,CAAC,CAAC,CAAC,CAclB;AAGD,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/functional.js

@@ -0,0 +1,103 @@
+// ============================================================
+// @oni.bot/core — Functional API
+// ============================================================
+// Alternative to the builder pattern. Compose agent skeletons
+// using decorated functions instead of StateGraph.addNode().
+//
+// @example
+// const search = task("search", async (query: string) => {
+//   return { results: await searchWeb(query) };
+// });
+//
+// const agent = entrypoint(
+//   { channels: myChannels, checkpointer },
+//   async (state: MyState) => {
+//     const results = await search(state.query);
+//     return { ...results };
+//   }
+// );
+//
+// const result = await agent.invoke({ query: "hello" });
+// ============================================================
+import { StateGraph } from "./graph.js";
+import { START, END } from "./types.js";
+export function task(name, fn, opts) {
+    return {
+        name,
+        async invoke(input, config) {
+            if (opts?.maxAttempts && opts.maxAttempts > 1) {
+                let last;
+                let delay = opts.initialDelay ?? 500;
+                for (let i = 0; i < opts.maxAttempts; i++) {
+                    try {
+                        return await fn(input, config);
+                    }
+                    catch (err) {
+                        last = err instanceof Error ? err : new Error(String(err));
+                        if (i < opts.maxAttempts - 1) {
+                            await new Promise((r) => setTimeout(r, delay));
+                            delay *= 2;
+                        }
+                    }
+                }
+                throw last;
+            }
+            return fn(input, config);
+        },
+    };
+}
+/**
+ * Create a compiled skeleton from a single async function.
+ * The function receives full state and returns a partial update.
+ *
+ * @example
+ * const app = entrypoint(
+ *   { channels: { query: lastValue(() => ""), answer: lastValue(() => "") } },
+ *   async (state) => {
+ *     const answer = await myLLM(state.query);
+ *     return { answer };
+ *   }
+ * );
+ */
+export function entrypoint(opts, fn) {
+    const nodeName = opts.name ?? "entrypoint";
+    const graph = new StateGraph({ channels: opts.channels });
+    graph.addNode(nodeName, fn);
+    graph.addEdge(START, nodeName);
+    graph.addEdge(nodeName, END);
+    return graph.compile({ checkpointer: opts.checkpointer });
+}
+// ----------------------------------------------------------------
+// pipe() — compose multiple tasks into a linear pipeline
+// ----------------------------------------------------------------
+export function pipe(opts, ...tasks) {
+    const graph = new StateGraph({ channels: opts.channels });
+    const names = tasks.map((_, i) => `step_${i}`);
+    for (let i = 0; i < tasks.length; i++) {
+        graph.addNode(names[i], tasks[i]);
+    }
+    graph.addEdge(START, names[0]);
+    for (let i = 0; i < names.length - 1; i++) {
+        graph.addEdge(names[i], names[i + 1]);
+    }
+    graph.addEdge(names[names.length - 1], END);
+    return graph.compile({ checkpointer: opts.checkpointer });
+}
+// ----------------------------------------------------------------
+// branch() — conditional routing shorthand
+// ----------------------------------------------------------------
+export function branch(condition, branches, opts) {
+    const graph = new StateGraph({ channels: opts.channels });
+    for (const [name, fn] of Object.entries(branches)) {
+        graph.addNode(name, fn);
+        graph.addEdge(name, END);
+    }
+    // Router node
+    graph.addNode("__router__", async (state) => state);
+    graph.addEdge(START, "__router__");
+    graph.addConditionalEdges("__router__", condition);
+    return graph.compile({ checkpointer: opts.checkpointer });
+}
+// Re-export interrupt for functional API users
+export { interrupt, getUserInput, getUserApproval, getUserSelection } from "./hitl/interrupt.js";
+//# sourceMappingURL=functional.js.map

package/dist/functional.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"functional.js","sourceRoot":"","sources":["../src/functional.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,iCAAiC;AACjC,+DAA+D;AAC/D,8DAA8D;AAC9D,6DAA6D;AAC7D,EAAE;AACF,WAAW;AACX,2DAA2D;AAC3D,gDAAgD;AAChD,MAAM;AACN,EAAE;AACF,4BAA4B;AAC5B,4CAA4C;AAC5C,gCAAgC;AAChC,iDAAiD;AACjD,6BAA6B;AAC7B,MAAM;AACN,KAAK;AACL,EAAE;AACF,yDAAyD;AACzD,+DAA+D;AAE/D,OAAO,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AACxC,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,YAAY,CAAC;AAqBxC,MAAM,UAAU,IAAI,CAClB,IAAc,EACd,EAAyE,EACzE,IAAmB;IAEnB,OAAO;QACL,IAAI;QACJ,KAAK,CAAC,MAAM,CAAC,KAAa,EAAE,MAAkB;YAC5C,IAAI,IAAI,EAAE,WAAW,IAAI,IAAI,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;gBAC9C,IAAI,IAAuB,CAAC;gBAC5B,IAAI,KAAK,GAAG,IAAI,CAAC,YAAY,IAAI,GAAG,CAAC;gBACrC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;oBAC1C,IAAI,CAAC;wBACH,OAAO,MAAM,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;oBACjC,CAAC;oBAAC,OAAO,GAAG,EAAE,CAAC;wBACb,IAAI,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;wBAC3D,IAAI,CAAC,GAAG,IAAI,CAAC,WAAW,GAAG,CAAC,EAAE,CAAC;4BAC7B,MAAM,IAAI,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,UAAU,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;4BAC/C,KAAK,IAAI,CAAC,CAAC;wBACb,CAAC;oBACH,CAAC;gBACH,CAAC;gBACD,MAAM,IAAI,CAAC;YACb,CAAC;YACD,OAAO,EAAE,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QAC3B,CAAC;KACF,CAAC;AACJ,CAAC;AAaD;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,UAAU,CACxB,IAA0B,EAC1B,EAAwE;IAExE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,IAAI,YAAY,CAAC;IAE3C,MAAM,KAAK,GAAG,IAAI,UAAU,CAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAC7D,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IAC5B,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC/B,KAAK,CAAC,OAAO,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;IAE7B,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;AAC5D,CAAC;AAED,mEAAmE;AACnE,yDAAyD;AACzD,mEAAmE;AAEnE,MAAM,UAAU,IAAI,CAClB,IAA2B,EAC3B,GAAG,KAAgF;IAEnF,MAAM,KAAK,GAAG,IAAI,UAAU,CAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAE7D,MAAM,KAAK,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;IAE/C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACtC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC,CAAE,CAAC,CAAC;IAChC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC1C,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAE,EAAE,KAAK,CAAC,CAAC,GAAG,CAAC,CAAE,CAAC,CAAC;IAC1C,CAAC;IACD,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAE,EAAE,GAAG,CAAC,CAAC;IAE7C,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;AAC5D,CAAC;AAED,mEAAmE;AACnE,2CAA2C;AAC3C,mEAAmE;AAEnE,MAAM,UAAU,MAAM,CACpB,SAA+B,EAC/B,QAA6F,EAC7F,IAA+B;IAE/B,MAAM,KAAK,GAAG,IAAI,UAAU,CAAI,EAAE,QAAQ,EAAE,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC;IAE7D,KAAK,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAC;QAClD,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACxB,KAAK,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC;IAC3B,CAAC;IAED,cAAc;IACd,KAAK,CAAC,OAAO,CAAC,YAAY,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,CAAC;IACpD,KAAK,CAAC,OAAO,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC;IACnC,KAAK,CAAC,mBAAmB,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IAEnD,OAAO,KAAK,CAAC,OAAO,CAAC,EAAE,YAAY,EAAE,IAAI,CAAC,YAAY,EAAE,CAAC,CAAC;AAC5D,CAAC;AAED,+CAA+C;AAC/C,OAAO,EAAE,SAAS,EAAE,YAAY,EAAE,eAAe,EAAE,gBAAgB,EAAE,MAAM,qBAAqB,CAAC"}

package/dist/graph.d.ts

@@ -0,0 +1,74 @@
+import { type NodeName, type NodeFn, type ChannelSchema, type ONIConfig, type ONISkeleton, type ONICheckpointer, type RetryPolicy, type CachePolicy } from "./types.js";
+import type { AgentNode } from "./agents/types.js";
+import type { HITLSession } from "./hitl/index.js";
+import type { BaseStore } from "./store/index.js";
+import type { GuardrailsConfig } from "./guardrails/types.js";
+import type { EventListeners } from "./events/types.js";
+export interface ONISkeletonV3<S> extends ONISkeleton<S> {
+    /**
+     * Resume a paused execution by providing a response to an interrupt.
+     * The resumeId comes from the HITLInterruptException.interrupt.resumeId.
+     */
+    resume(config: {
+        threadId: string;
+        resumeId: string;
+    }, value: unknown): Promise<S>;
+    /** Get all pending interrupt sessions for a thread */
+    getPendingInterrupts(config: {
+        threadId: string;
+    }): HITLSession<S>[];
+}
+export declare class StateGraph<S extends Record<string, unknown>> {
+    private nodes;
+    private edges;
+    private readonly channels;
+    constructor(schema: {
+        channels: ChannelSchema<S>;
+    });
+    addNode(name: string, fn: NodeFn<S>, opts?: {
+        retry?: RetryPolicy;
+        cache?: boolean | CachePolicy;
+    }): this;
+    addSubgraph(name: string, subgraph: ONISkeleton<S>, opts?: {
+        retry?: RetryPolicy;
+    }): this;
+    addAgent(agentNode: AgentNode<any>): this;
+    addEdge(from: NodeName, to: NodeName | NodeName[]): this;
+    addConditionalEdges(from: NodeName, condition: (state: S, config?: ONIConfig) => NodeName | NodeName[], pathMap?: Record<string, NodeName>): this;
+    private validate;
+    compile(options?: CompileOptions<S>): ONISkeletonV3<S>;
+    toMermaid(): string;
+}
+export interface CompileOptions<S> {
+    checkpointer?: ONICheckpointer<S>;
+    interruptBefore?: string[];
+    interruptAfter?: string[];
+    store?: BaseStore;
+    guardrails?: GuardrailsConfig;
+    listeners?: EventListeners;
+}
+export interface ONIMessage {
+    role: "user" | "assistant" | "system" | "tool";
+    content: string;
+    name?: string;
+    tool_call_id?: string;
+    tool_calls?: ONIToolCall[];
+}
+export interface ONIToolCall {
+    id: string;
+    name: string;
+    args: Record<string, unknown>;
+}
+export type MessageState = {
+    messages: ONIMessage[];
+};
+export declare class MessageGraph extends StateGraph<MessageState> {
+    constructor();
+}
+declare module "./graph.js" {
+    interface StateGraph<S> {
+        getGraph(): import("./inspect.js").GraphDescriptor;
+        toMermaidDetailed(): string;
+    }
+}
+//# sourceMappingURL=graph.d.ts.map

package/dist/graph.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph.d.ts","sourceRoot":"","sources":["../src/graph.ts"],"names":[],"mappings":"AAKA,OAAO,EAEL,KAAK,QAAQ,EAAE,KAAK,MAAM,EAC1B,KAAK,aAAa,EAAE,KAAK,SAAS,EAAE,KAAK,WAAW,EACpD,KAAK,eAAe,EAAE,KAAK,WAAW,EAAE,KAAK,WAAW,EAGzD,MAAM,YAAY,CAAC;AAEpB,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAGnD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,KAAK,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAC9D,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAMxD,MAAM,WAAW,aAAa,CAAC,CAAC,CAAE,SAAQ,WAAW,CAAC,CAAC,CAAC;IACtD;;;OAGG;IACH,MAAM,CACJ,MAAM,EAAI;QAAE,QAAQ,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,CAAA;KAAE,EAChD,KAAK,EAAK,OAAO,GAChB,OAAO,CAAC,CAAC,CAAC,CAAC;IAEd,sDAAsD;IACtD,oBAAoB,CAAC,MAAM,EAAE;QAAE,QAAQ,EAAE,MAAM,CAAA;KAAE,GAAG,WAAW,CAAC,CAAC,CAAC,EAAE,CAAC;CACtE;AAMD,qBAAa,UAAU,CAAC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IACvD,OAAO,CAAC,KAAK,CAA0C;IACvD,OAAO,CAAC,KAAK,CAAoB;IACjC,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAmB;gBAEhC,MAAM,EAAE;QAAE,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC,CAAA;KAAE;IAIlD,OAAO,CAAC,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,WAAW,CAAC;QAAC,KAAK,CAAC,EAAE,OAAO,GAAG,WAAW,CAAA;KAAE,GAAG,IAAI;IASzG,WAAW,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE;QAAE,KAAK,CAAC,EAAE,WAAW,CAAA;KAAE,GAAG,IAAI;IAOzF,QAAQ,CAAC,SAAS,EAAE,SAAS,CAAC,GAAG,CAAC,GAAG,IAAI;IAIzC,OAAO,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,EAAE,QAAQ,GAAG,QAAQ,EAAE,GAAG,IAAI;IAUxD,mBAAmB,CACjB,IAAI,EAAE,QAAQ,EACd,SAAS,EAAE,CAAC,KAAK,EAAE,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,KAAK,QAAQ,GAAG,QAAQ,EAAE,EAClE,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,QAAQ,CAAC,GACjC,IAAI;IAKP,OAAO,CAAC,QAAQ;IAUhB,OAAO,CAAC,OAAO,GAAE,cAAc,CAAC,CAAC,CAAM,GAAG,aAAa,CAAC,CAAC,CAAC;IA+C1D,SAAS,IAAI,MAAM;CAepB;AAED,MAAM,WAAW,cAAc,CAAC,CAAC;IAC/B,YAAY,CAAC,EAAK,eAAe,CAAC,CAAC,CAAC,CAAC;IACrC,eAAe,CAAC,EAAE,MAAM,EAAE,CAAC;IAC3B,cAAc,CAAC,EAAG,MAAM,EAAE,CAAC;IAC3B,KAAK,CAAC,EAAY,SAAS,CAAC;IAC5B,UAAU,CAAC,EAAO,gBAAgB,CAAC;IACnC,SAAS,CAAC,EAAQ,cAAc,CAAC;CAClC;AAMD,MAAM,WAAW,UAAU;IACzB,IAAI,EAAW,MAAM,GAAG,WAAW,GAAG,QAAQ,GAAG,MAAM,CAAC;IACxD,OAAO,EAAQ,MAAM,CAAC;IACtB,IAAI,CAAC,EAAU,MAAM,CAAC;IACtB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAI,WAAW,EAAE,CAAC;CAC9B;AAED,MAAM,WAAW,WAAW;IAC1B,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACzD;AAED,MAAM,MAAM,YAAY,GAAG;IAAE,QAAQ,EAAE,UAAU,EAAE,CAAA;CAAE,CAAC;AAEtD,qBAAa,YAAa,SAAQ,UAAU,CAAC,YAAY,CAAC;;CAIzD;AAQD,OAAO,QAAQ,YAAY,CAAC;IAC1B,UAAU,UAAU,CAAC,CAAC;QACpB,QAAQ,IAAI,OAAO,cAAc,EAAE,eAAe,CAAC;QACnD,iBAAiB,IAAI,MAAM,CAAC;KAC7B;CACF"}

package/dist/graph.js

@@ -0,0 +1,119 @@
+// ============================================================
+// @oni.bot/core — StateGraph v3
+// New: HITL resume(), getPendingInterrupts(), token streaming
+// ============================================================
+import { START, END, appendList, } from "./types.js";
+import { InvalidSkeletonError, EdgeConflictError, NodeNotFoundError } from "./errors.js";
+import { ONIPregelRunner } from "./pregel.js";
+// ----------------------------------------------------------------
+// StateGraph
+// ----------------------------------------------------------------
+export class StateGraph {
+    nodes = new Map();
+    edges = [];
+    channels;
+    constructor(schema) {
+        this.channels = schema.channels;
+    }
+    addNode(name, fn, opts) {
+        if (name === START || name === END)
+            throw new InvalidSkeletonError(`"${name}" is reserved.`);
+        if (this.nodes.has(name))
+            throw new InvalidSkeletonError(`Node "${name}" already registered.`);
+        this.nodes.set(name, { name, fn, retry: opts?.retry, cache: opts?.cache });
+        return this;
+    }
+    addSubgraph(name, subgraph, opts) {
+        if (this.nodes.has(name))
+            throw new InvalidSkeletonError(`Node "${name}" already registered.`);
+        const fn = async (state, config) => subgraph.invoke(state, config);
+        this.nodes.set(name, { name, fn, retry: opts?.retry, subgraph });
+        return this;
+    }
+    addAgent(agentNode) {
+        return this.addNode(agentNode.name, agentNode._nodeFn);
+    }
+    addEdge(from, to) {
+        const targets = Array.isArray(to) ? to : [to];
+        for (const t of targets) {
+            const dup = this.edges.find((e) => e.type === "static" && e.from === from && e.to === t);
+            if (dup)
+                throw new EdgeConflictError(from, t);
+            this.edges.push({ type: "static", from, to: t });
+        }
+        return this;
+    }
+    addConditionalEdges(from, condition, pathMap) {
+        this.edges.push({ type: "conditional", from, condition, pathMap });
+        return this;
+    }
+    validate() {
+        if (!this.edges.some((e) => e.from === START))
+            throw new InvalidSkeletonError("Skeleton must have at least one edge from START.");
+        for (const edge of this.edges) {
+            if (edge.type !== "static")
+                continue;
+            if (edge.to !== END && !this.nodes.has(edge.to))
+                throw new NodeNotFoundError(edge.to);
+            if (edge.from !== START && !this.nodes.has(edge.from))
+                throw new NodeNotFoundError(edge.from);
+        }
+    }
+    compile(options = {}) {
+        this.validate();
+        const runner = new ONIPregelRunner(this.nodes, this.edges, this.channels, { interruptBefore: options.interruptBefore, interruptAfter: options.interruptAfter }, options.checkpointer ?? null, options.store ?? null, options.guardrails, options.listeners);
+        const skeleton = {
+            invoke: (input, config) => runner.invoke(input, config),
+            stream: (input, config) => runner.stream(input, config),
+            batch: (inputs, config) => runner.batch(inputs, config),
+            getState: (cfg) => runner.getState(cfg.threadId),
+            updateState: (cfg, update) => runner.updateState(cfg.threadId, update),
+            getStateAt: (cfg) => runner.getStateAt(cfg.threadId, cfg.step),
+            getHistory: (cfg) => runner.getHistory(cfg.threadId),
+            forkFrom: (cfg) => runner.forkFrom(cfg.threadId, cfg.step, cfg.newThreadId),
+            // ---- HITL ----
+            async resume(cfg, value) {
+                // Inject the resume value into the config so the node receives it
+                return runner.invoke({}, {
+                    threadId: cfg.threadId,
+                    __resumeValues: { [runner.getPendingInterrupts(cfg.threadId)[0]?.node ?? ""]: value },
+                });
+            },
+            getPendingInterrupts(cfg) {
+                return runner.getPendingInterrupts(cfg.threadId);
+            },
+        };
+        // Attach runner as internal property for subgraph Command.PARENT support
+        skeleton._runner = runner;
+        return skeleton;
+    }
+    toMermaid() {
+        const lines = ["graph TD"];
+        const lbl = (n) => n === START ? "__start__" : n === END ? "__end__" : n;
+        for (const edge of this.edges) {
+            lines.push(edge.type === "static"
+                ? `    ${lbl(edge.from)} --> ${lbl(edge.to)}`
+                : `    ${lbl(edge.from)} -->|?| conditional_${lbl(edge.from)}`);
+        }
+        lines.push(`    style __start__ fill:#7c3aed,color:#fff`);
+        lines.push(`    style __end__   fill:#1e1e2e,color:#fff`);
+        return lines.join("\n");
+    }
+}
+export class MessageGraph extends StateGraph {
+    constructor() {
+        super({ channels: { messages: appendList(() => []) } });
+    }
+}
+// ----------------------------------------------------------------
+// Re-export getGraph helper — called via skeleton.getGraph()
+// ----------------------------------------------------------------
+import { buildGraphDescriptor, toMermaidDetailed } from "./inspect.js";
+StateGraph.prototype.getGraph = function () {
+    return buildGraphDescriptor(this.nodes, this.edges);
+};
+StateGraph.prototype.toMermaidDetailed = function () {
+    const desc = buildGraphDescriptor(this.nodes, this.edges);
+    return toMermaidDetailed(desc);
+};
+//# sourceMappingURL=graph.js.map

package/dist/graph.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"graph.js","sourceRoot":"","sources":["../src/graph.ts"],"names":[],"mappings":"AAAA,+DAA+D;AAC/D,gCAAgC;AAChC,8DAA8D;AAC9D,+DAA+D;AAE/D,OAAO,EACL,KAAK,EAAE,GAAG,EAKV,UAAU,GACX,MAAM,YAAY,CAAC;AACpB,OAAO,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAEzF,OAAO,EAAE,eAAe,EAAE,MAAM,aAAa,CAAC;AAyB9C,mEAAmE;AACnE,aAAa;AACb,mEAAmE;AAEnE,MAAM,OAAO,UAAU;IACb,KAAK,GAAK,IAAI,GAAG,EAA6B,CAAC;IAC/C,KAAK,GAAiB,EAAE,CAAC;IAChB,QAAQ,CAAmB;IAE5C,YAAY,MAAsC;QAChD,IAAI,CAAC,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;IAClC,CAAC;IAED,OAAO,CAAC,IAAY,EAAE,EAAa,EAAE,IAA6D;QAChG,IAAI,IAAI,KAAK,KAAK,IAAI,IAAI,KAAK,GAAG;YAChC,MAAM,IAAI,oBAAoB,CAAC,IAAI,IAAI,gBAAgB,CAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;YACtB,MAAM,IAAI,oBAAoB,CAAC,SAAS,IAAI,uBAAuB,CAAC,CAAC;QACvE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC3E,OAAO,IAAI,CAAC;IACd,CAAC;IAED,WAAW,CAAC,IAAY,EAAE,QAAwB,EAAE,IAA8B;QAChF,IAAI,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,MAAM,IAAI,oBAAoB,CAAC,SAAS,IAAI,uBAAuB,CAAC,CAAC;QAC/F,MAAM,EAAE,GAAc,KAAK,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAwB,CAAC;QACrG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QACjE,OAAO,IAAI,CAAC;IACd,CAAC;IAED,QAAQ,CAAC,SAAyB;QAChC,OAAO,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,OAAO,CAAC,CAAC;IACzD,CAAC;IAED,OAAO,CAAC,IAAc,EAAE,EAAyB;QAC/C,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;QAC9C,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,QAAQ,IAAI,CAAC,CAAC,IAAI,KAAK,IAAI,IAAI,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,CAAC;YACzF,IAAI,GAAG;gBAAE,MAAM,IAAI,iBAAiB,CAAC,IAAc,EAAE,CAAW,CAAC,CAAC;YAClE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC;QACnD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,mBAAmB,CACjB,IAAc,EACd,SAAkE,EAClE,OAAkC;QAElC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,CAAC,CAAC;QACnE,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,QAAQ;QACd,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC;YAC3C,MAAM,IAAI,oBAAoB,CAAC,kDAAkD,CAAC,CAAC;QACrF,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,IAAI,KAAK,QAAQ;gBAAE,SAAS;YACrC,IAAI,IAAI,CAAC,EAAE,KAAO,GAAG,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAc,CAAC;gBAAE,MAAM,IAAI,iBAAiB,CAAC,IAAI,CAAC,EAAc,CAAC,CAAC;YAChH,IAAI,IAAI,CAAC,IAAI,KAAK,KAAK,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,IAAc,CAAC;gBAAE,MAAM,IAAI,iBAAiB,CAAC,IAAI,CAAC,IAAc,CAAC,CAAC;QACpH,CAAC;IACH,CAAC;IAED,OAAO,CAAC,UAA6B,EAAE;QACrC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAEhB,MAAM,MAAM,GAAG,IAAI,eAAe,CAChC,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,KAAK,EACV,IAAI,CAAC,QAAQ,EACb,EAAE,eAAe,EAAE,OAAO,CAAC,eAAe,EAAE,cAAc,EAAE,OAAO,CAAC,cAAc,EAAE,EACpF,OAAO,CAAC,YAAY,IAAI,IAAI,EAC5B,OAAO,CAAC,KAAK,IAAI,IAAI,EACrB,OAAO,CAAC,UAAU,EAClB,OAAO,CAAC,SAAS,CAClB,CAAC;QAEF,MAAM,QAAQ,GAAqB;YACjC,MAAM,EAAO,CAAC,KAAK,EAAE,MAAM,EAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;YAC7D,MAAM,EAAO,CAAC,KAAK,EAAE,MAAM,EAAG,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,CAAC;YAC7D,KAAK,EAAQ,CAAC,MAAM,EAAE,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,MAAM,CAAC;YAC7D,QAAQ,EAAK,CAAC,GAAG,EAAa,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC;YAC9D,WAAW,EAAE,CAAC,GAAG,EAAE,MAAM,EAAK,EAAE,CAAC,MAAM,CAAC,WAAW,CAAC,GAAG,CAAC,QAAQ,EAAE,MAAM,CAAC;YACzE,UAAU,EAAG,CAAC,GAAG,EAAa,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,CAAC;YAC1E,UAAU,EAAG,CAAC,GAAG,EAAa,EAAE,CAAC,MAAM,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC;YAChE,QAAQ,EAAK,CAAC,GAAG,EAAa,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,WAAW,CAAC;YAEzF,iBAAiB;YAEjB,KAAK,CAAC,MAAM,CAAC,GAAG,EAAE,KAAK;gBACrB,kEAAkE;gBAClE,OAAO,MAAM,CAAC,MAAM,CAClB,EAAE,EACF;oBACE,QAAQ,EAAE,GAAG,CAAC,QAAQ;oBACtB,cAAc,EAAE,EAAE,CAAC,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,EAAE,KAAK,EAAE;iBAC3B,CAC7D,CAAC;YACJ,CAAC;YAED,oBAAoB,CAAC,GAAG;gBACtB,OAAO,MAAM,CAAC,oBAAoB,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACnD,CAAC;SACF,CAAC;QAEF,yEAAyE;QACxE,QAAgB,CAAC,OAAO,GAAG,MAAM,CAAC;QACnC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,SAAS;QACP,MAAM,KAAK,GAAa,CAAC,UAAU,CAAC,CAAC;QACrC,MAAM,GAAG,GAAG,CAAC,CAAW,EAAE,EAAE,CAC1B,CAAC,KAAK,KAAK,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAE,CAAY,CAAC;QACpE,KAAK,MAAM,IAAI,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YAC9B,KAAK,CAAC,IAAI,CACR,IAAI,CAAC,IAAI,KAAK,QAAQ;gBACpB,CAAC,CAAC,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE;gBAC7C,CAAC,CAAC,OAAO,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,uBAAuB,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CACjE,CAAC;QACJ,CAAC;QACD,KAAK,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC1D,KAAK,CAAC,IAAI,CAAC,6CAA6C,CAAC,CAAC;QAC1D,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC1B,CAAC;CACF;AA6BD,MAAM,OAAO,YAAa,SAAQ,UAAwB;IACxD;QACE,KAAK,CAAC,EAAE,QAAQ,EAAE,EAAE,QAAQ,EAAE,UAAU,CAAa,GAAG,EAAE,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC;IACtE,CAAC;CACF;AAED,mEAAmE;AACnE,6DAA6D;AAC7D,mEAAmE;AACnE,OAAO,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAUvE,UAAU,CAAC,SAAS,CAAC,QAAQ,GAAG;IAC9B,OAAO,oBAAoB,CAAE,IAAY,CAAC,KAAK,EAAG,IAAY,CAAC,KAAK,CAAC,CAAC;AACxE,CAAC,CAAC;AAEF,UAAU,CAAC,SAAS,CAAC,iBAAiB,GAAG;IACvC,MAAM,IAAI,GAAG,oBAAoB,CAAE,IAAY,CAAC,KAAK,EAAG,IAAY,CAAC,KAAK,CAAC,CAAC;IAC5E,OAAO,iBAAiB,CAAC,IAAI,CAAC,CAAC;AACjC,CAAC,CAAC"}

package/dist/guardrails/audit.d.ts

@@ -0,0 +1,13 @@
+import type { AuditEntry } from "./types.js";
+export declare class AuditLog {
+    private logs;
+    record(threadId: string, entry: AuditEntry): void;
+    getLog(threadId: string): AuditEntry[];
+    getByAgent(threadId: string, agent: string): AuditEntry[];
+    getByAction(threadId: string, action: AuditEntry["action"]): AuditEntry[];
+    clear(threadId: string): void;
+    clearAll(): void;
+    toJSON(threadId: string): string;
+    fromJSON(threadId: string, json: string): void;
+}
+//# sourceMappingURL=audit.d.ts.map

package/dist/guardrails/audit.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"audit.d.ts","sourceRoot":"","sources":["../../src/guardrails/audit.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE7C,qBAAa,QAAQ;IACnB,OAAO,CAAC,IAAI,CAAwC;IAEpD,MAAM,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,IAAI;IAMjD,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,UAAU,EAAE;IAItC,UAAU,CAAC,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,UAAU,EAAE;IAIzD,WAAW,CAAC,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,UAAU,CAAC,QAAQ,CAAC,GAAG,UAAU,EAAE;IAIzE,KAAK,CAAC,QAAQ,EAAE,MAAM,GAAG,IAAI;IAI7B,QAAQ,IAAI,IAAI;IAIhB,MAAM,CAAC,QAAQ,EAAE,MAAM,GAAG,MAAM;IAIhC,QAAQ,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,IAAI;CAI/C"}

package/dist/guardrails/audit.js

@@ -0,0 +1,31 @@
+export class AuditLog {
+    logs = new Map();
+    record(threadId, entry) {
+        const existing = this.logs.get(threadId) ?? [];
+        existing.push(entry);
+        this.logs.set(threadId, existing);
+    }
+    getLog(threadId) {
+        return this.logs.get(threadId) ?? [];
+    }
+    getByAgent(threadId, agent) {
+        return this.getLog(threadId).filter(e => e.agent === agent);
+    }
+    getByAction(threadId, action) {
+        return this.getLog(threadId).filter(e => e.action === action);
+    }
+    clear(threadId) {
+        this.logs.delete(threadId);
+    }
+    clearAll() {
+        this.logs.clear();
+    }
+    toJSON(threadId) {
+        return JSON.stringify(this.getLog(threadId));
+    }
+    fromJSON(threadId, json) {
+        const entries = JSON.parse(json);
+        this.logs.set(threadId, entries);
+    }
+}
+//# sourceMappingURL=audit.js.map

package/dist/guardrails/audit.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"audit.js","sourceRoot":"","sources":["../../src/guardrails/audit.ts"],"names":[],"mappings":"AAEA,MAAM,OAAO,QAAQ;IACX,IAAI,GAA8B,IAAI,GAAG,EAAE,CAAC;IAEpD,MAAM,CAAC,QAAgB,EAAE,KAAiB;QACxC,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC/C,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACrB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,QAAQ,CAAC,CAAC;IACpC,CAAC;IAED,MAAM,CAAC,QAAgB;QACrB,OAAO,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IACvC,CAAC;IAED,UAAU,CAAC,QAAgB,EAAE,KAAa;QACxC,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,KAAK,KAAK,CAAC,CAAC;IAC9D,CAAC;IAED,WAAW,CAAC,QAAgB,EAAE,MAA4B;QACxD,OAAO,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,MAAM,KAAK,MAAM,CAAC,CAAC;IAChE,CAAC;IAED,KAAK,CAAC,QAAgB;QACpB,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;IAC7B,CAAC;IAED,QAAQ;QACN,IAAI,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;IAED,MAAM,CAAC,QAAgB;QACrB,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,CAAC,QAAgB,EAAE,IAAY;QACrC,MAAM,OAAO,GAAiB,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QAC/C,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnC,CAAC;CACF"}

package/dist/guardrails/budget.d.ts

@@ -0,0 +1,30 @@
+import type { TokenUsage } from "../models/types.js";
+import type { BudgetConfig, AuditEntry } from "./types.js";
+export declare class BudgetExceededError extends Error {
+    readonly kind: "agent" | "run" | "cost";
+    readonly current: number;
+    readonly limit: number;
+    constructor(kind: "agent" | "run" | "cost", current: number, limit: number);
+}
+/** Per-million-token pricing: [input, output] */
+export interface ModelPricing {
+    input: number;
+    output: number;
+}
+/** Default pricing per million tokens for common models */
+export declare const DEFAULT_PRICING: Record<string, ModelPricing>;
+export declare class BudgetTracker {
+    private agentTokens;
+    private totalInput;
+    private totalOutput;
+    private totalCost;
+    private pricing;
+    private config;
+    constructor(config: BudgetConfig, customPricing?: Record<string, ModelPricing>);
+    record(agentName: string, modelId: string, usage: TokenUsage): AuditEntry[];
+    getAgentTokens(name: string): number;
+    getTotalTokens(): number;
+    getTotalCost(): number;
+    isOverBudget(): boolean;
+}
+//# sourceMappingURL=budget.d.ts.map

package/dist/guardrails/budget.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"budget.d.ts","sourceRoot":"","sources":["../../src/guardrails/budget.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,oBAAoB,CAAC;AACrD,OAAO,KAAK,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE3D,qBAAa,mBAAoB,SAAQ,KAAK;aAE1B,IAAI,EAAE,OAAO,GAAG,KAAK,GAAG,MAAM;aAC9B,OAAO,EAAE,MAAM;aACf,KAAK,EAAE,MAAM;gBAFb,IAAI,EAAE,OAAO,GAAG,KAAK,GAAG,MAAM,EAC9B,OAAO,EAAE,MAAM,EACf,KAAK,EAAE,MAAM;CAShC;AAED,iDAAiD;AACjD,MAAM,WAAW,YAAY;IAC3B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;CAChB;AAED,2DAA2D;AAC3D,eAAO,MAAM,eAAe,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAMxD,CAAC;AAEF,qBAAa,aAAa;IACxB,OAAO,CAAC,WAAW,CAA6D;IAChF,OAAO,CAAC,UAAU,CAAK;IACvB,OAAO,CAAC,WAAW,CAAK;IACxB,OAAO,CAAC,SAAS,CAAK;IACtB,OAAO,CAAC,OAAO,CAA+B;IAC9C,OAAO,CAAC,MAAM,CAAe;gBAEjB,MAAM,EAAE,YAAY,EAAE,aAAa,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,YAAY,CAAC;IAK9E,MAAM,CAAC,SAAS,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,GAAG,UAAU,EAAE;IAmG3E,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAKpC,cAAc,IAAI,MAAM;IAIxB,YAAY,IAAI,MAAM;IAItB,YAAY,IAAI,OAAO;CAkBxB"}