kraken-ai 0.0.6 → 0.0.7

package/dist/index.cjs

@@ -492,14 +492,14 @@ var composeOnLLMCall = (middleware, ctx, coreFn) => {
   }
   return fn;
 };
-var composeOnToolCall = (middleware, ctx, coreFn, toolName, args) => {
+var composeOnToolCall = (middleware, ctx, coreFn, toolName, toolCallId, args) => {
   let fn = coreFn;
   for (let i = middleware.length - 1; i >= 0; i--) {
     const mw = middleware[i];
     if (mw?.onToolCall) {
       const prev = fn;
       const hook = mw.onToolCall;
-      fn = () => hook(ctx, toolName, args, prev);
+      fn = () => hook(ctx, toolName, toolCallId, args, prev);
     }
   }
   return fn;
@@ -511,6 +511,41 @@ var applyTransformTools = (middleware, ctx, tools) => {
   }
   return result;
 };
+var composeOnEvent = (middleware, ctx, sink, warn) => {
+  const hooks = [];
+  for (const mw of middleware) {
+    if (mw.onEvent) hooks.push(mw.onEvent);
+  }
+  if (hooks.length === 0) return sink;
+  return (event) => {
+    const origSeq = event.seq;
+    const origRunId = event.runId;
+    const origAgentName = event.agentName;
+    let next = () => {
+      if (event.seq !== origSeq || event.runId !== origRunId || event.agentName !== origAgentName) {
+        warn?.("onEvent middleware mutated immutable fields (seq/runId/agentName) \u2014 restoring");
+        event.seq = origSeq;
+        event.runId = origRunId;
+        event.agentName = origAgentName;
+      }
+      sink?.(event);
+    };
+    for (let i = hooks.length - 1; i >= 0; i--) {
+      const hook = hooks[i];
+      if (!hook) continue;
+      const prev = next;
+      next = () => {
+        try {
+          hook(ctx, event, prev);
+        } catch (err) {
+          warn?.(`onEvent middleware threw: ${err instanceof Error ? err.message : String(err)}`);
+          prev();
+        }
+      };
+    }
+    next();
+  };
+};
 var createRunner = (config2) => {
   const { agent, model, tools } = config2;
   const toolMap = new Map(tools.map((t) => [t.name, t]));
@@ -532,7 +567,13 @@ var createRunner = (config2) => {
         runId,
         parentRunId: options?.parentRunId
       };
-      const emitter = createEventEmitter(agent.name, runId, options?.onEvent, options?.parentRunId);
+      const composedOnEvent = composeOnEvent(
+        middleware,
+        ctx,
+        options?.onEvent,
+        (msg) => logger.warn(msg)
+      );
+      const emitter = createEventEmitter(agent.name, runId, composedOnEvent, options?.parentRunId);
       const lifecycle = new LifecycleManager((from, to) => {
         emitter.emit({ type: "lifecycle", from, to });
         for (const mw of middleware) mw.onTransition?.(ctx, from, to);
@@ -586,7 +627,19 @@ var createRunner = (config2) => {
             thinkingLevel: config2.thinkingLevel,
             responseSchema: config2.responseSchema
           };
-          const response = await wrappedGenerate(params);
+          let response;
+          try {
+            response = await wrappedGenerate(params);
+          } catch (err) {
+            emitter.emit({
+              type: "llm_call",
+              inputTokens: 0,
+              outputTokens: 0,
+              output: "",
+              error: err instanceof Error ? err.message : String(err)
+            });
+            throw err;
+          }
           usage.inputTokens += response.usage.inputTokens;
           usage.outputTokens += response.usage.outputTokens;
           if (!model.generateStream && response.thoughts) {
@@ -676,6 +729,7 @@ var createRunner = (config2) => {
                 ctx,
                 executeFn,
                 tc.name,
+                tc.id,
                 tc.args
               );
               const startTime = Date.now();
@@ -1014,7 +1068,7 @@ var formatEvent = (e) => {
     case "interrupt":
       return `${tag} interrupt: ${e.toolName} awaiting approval`;
     case "kill":
-      return `${tag} killed${e.reason ? `: ${e.reason}` : ""}`;
+      return `${tag} killed: ${e.reason}`;
     case "error":
       return `${tag} error: ${e.message}`;
     case "notification":

package/dist/index.d.cts

@@ -36,7 +36,7 @@ interface Middleware {
     /** Augment tool definitions visible to the LLM. Called once at run start. */
     transformTools?: (ctx: MiddlewareContext, tools: ToolDefinition[]) => ToolDefinition[];
     /** Wrap tool execution. Called for EVERY tool call including delegation. */
-    onToolCall?: (ctx: MiddlewareContext, toolName: string, args: unknown, next: () => Promise<unknown>) => Promise<unknown>;
+    onToolCall?: (ctx: MiddlewareContext, toolName: string, toolCallId: string, args: unknown, next: () => Promise<unknown>) => Promise<unknown>;
     /** Wrap LLM generation. Pre-LLM and post-LLM gates via next(). */
     onLLMCall?: (ctx: MiddlewareContext, params: GenerateParams, next: () => Promise<GenerateResponse>) => Promise<GenerateResponse>;
     /** Called on every lifecycle transition. */
@@ -196,11 +196,21 @@ interface AgentLLMCallEvent extends AgentEventBase {
         name: string;
         args: Record<string, unknown>;
     }[];
+    /** Set when the LLM call failed. */
+    error?: string;
+    /** Middleware enrichment: correlation ID linking this LLM call to its tool calls. */
+    turnId?: string;
+    /** Middleware enrichment: wall-clock duration in milliseconds. */
+    durationMs?: number;
+    /** Middleware enrichment: number of tool calls in the response. */
+    toolCallCount?: number;
 }
 interface AgentThoughtEvent extends AgentEventBase {
     type: "thought";
     text: string;
     callIndex: number;
+    /** Middleware enrichment: correlation ID of the active LLM call turn. */
+    turnId?: string;
 }
 interface AgentToolCallEvent extends AgentEventBase {
     type: "tool_call";
@@ -211,6 +221,8 @@ interface AgentToolCallEvent extends AgentEventBase {
     result?: unknown;
     error?: string;
     durationMs: number;
+    /** Middleware enrichment: correlation ID linking this tool call to its parent LLM call. */
+    turnId?: string;
 }
 interface AgentDelegationEvent extends AgentEventBase {
     type: "delegation";
@@ -227,7 +239,7 @@ interface AgentInterruptEvent extends AgentEventBase {
 }
 interface AgentKillEvent extends AgentEventBase {
     type: "kill";
-    reason?: string;
+    reason: string;
 }
 interface AgentErrorEvent extends AgentEventBase {
     type: "error";

package/dist/index.d.ts

@@ -36,7 +36,7 @@ interface Middleware {
     /** Augment tool definitions visible to the LLM. Called once at run start. */
     transformTools?: (ctx: MiddlewareContext, tools: ToolDefinition[]) => ToolDefinition[];
     /** Wrap tool execution. Called for EVERY tool call including delegation. */
-    onToolCall?: (ctx: MiddlewareContext, toolName: string, args: unknown, next: () => Promise<unknown>) => Promise<unknown>;
+    onToolCall?: (ctx: MiddlewareContext, toolName: string, toolCallId: string, args: unknown, next: () => Promise<unknown>) => Promise<unknown>;
     /** Wrap LLM generation. Pre-LLM and post-LLM gates via next(). */
     onLLMCall?: (ctx: MiddlewareContext, params: GenerateParams, next: () => Promise<GenerateResponse>) => Promise<GenerateResponse>;
     /** Called on every lifecycle transition. */
@@ -196,11 +196,21 @@ interface AgentLLMCallEvent extends AgentEventBase {
         name: string;
         args: Record<string, unknown>;
     }[];
+    /** Set when the LLM call failed. */
+    error?: string;
+    /** Middleware enrichment: correlation ID linking this LLM call to its tool calls. */
+    turnId?: string;
+    /** Middleware enrichment: wall-clock duration in milliseconds. */
+    durationMs?: number;
+    /** Middleware enrichment: number of tool calls in the response. */
+    toolCallCount?: number;
 }
 interface AgentThoughtEvent extends AgentEventBase {
     type: "thought";
     text: string;
     callIndex: number;
+    /** Middleware enrichment: correlation ID of the active LLM call turn. */
+    turnId?: string;
 }
 interface AgentToolCallEvent extends AgentEventBase {
     type: "tool_call";
@@ -211,6 +221,8 @@ interface AgentToolCallEvent extends AgentEventBase {
     result?: unknown;
     error?: string;
     durationMs: number;
+    /** Middleware enrichment: correlation ID linking this tool call to its parent LLM call. */
+    turnId?: string;
 }
 interface AgentDelegationEvent extends AgentEventBase {
     type: "delegation";
@@ -227,7 +239,7 @@ interface AgentInterruptEvent extends AgentEventBase {
 }
 interface AgentKillEvent extends AgentEventBase {
     type: "kill";
-    reason?: string;
+    reason: string;
 }
 interface AgentErrorEvent extends AgentEventBase {
     type: "error";

package/dist/index.js

@@ -174,14 +174,14 @@ var composeOnLLMCall = (middleware, ctx, coreFn) => {
   }
   return fn;
 };
-var composeOnToolCall = (middleware, ctx, coreFn, toolName, args) => {
+var composeOnToolCall = (middleware, ctx, coreFn, toolName, toolCallId, args) => {
   let fn = coreFn;
   for (let i = middleware.length - 1; i >= 0; i--) {
     const mw = middleware[i];
     if (mw?.onToolCall) {
       const prev = fn;
       const hook = mw.onToolCall;
-      fn = () => hook(ctx, toolName, args, prev);
+      fn = () => hook(ctx, toolName, toolCallId, args, prev);
     }
   }
   return fn;
@@ -193,6 +193,41 @@ var applyTransformTools = (middleware, ctx, tools) => {
   }
   return result;
 };
+var composeOnEvent = (middleware, ctx, sink, warn) => {
+  const hooks = [];
+  for (const mw of middleware) {
+    if (mw.onEvent) hooks.push(mw.onEvent);
+  }
+  if (hooks.length === 0) return sink;
+  return (event) => {
+    const origSeq = event.seq;
+    const origRunId = event.runId;
+    const origAgentName = event.agentName;
+    let next = () => {
+      if (event.seq !== origSeq || event.runId !== origRunId || event.agentName !== origAgentName) {
+        warn?.("onEvent middleware mutated immutable fields (seq/runId/agentName) \u2014 restoring");
+        event.seq = origSeq;
+        event.runId = origRunId;
+        event.agentName = origAgentName;
+      }
+      sink?.(event);
+    };
+    for (let i = hooks.length - 1; i >= 0; i--) {
+      const hook = hooks[i];
+      if (!hook) continue;
+      const prev = next;
+      next = () => {
+        try {
+          hook(ctx, event, prev);
+        } catch (err) {
+          warn?.(`onEvent middleware threw: ${err instanceof Error ? err.message : String(err)}`);
+          prev();
+        }
+      };
+    }
+    next();
+  };
+};
 var createRunner = (config2) => {
   const { agent, model, tools } = config2;
   const toolMap = new Map(tools.map((t) => [t.name, t]));
@@ -214,7 +249,13 @@ var createRunner = (config2) => {
         runId,
         parentRunId: options?.parentRunId
       };
-      const emitter = createEventEmitter(agent.name, runId, options?.onEvent, options?.parentRunId);
+      const composedOnEvent = composeOnEvent(
+        middleware,
+        ctx,
+        options?.onEvent,
+        (msg) => logger.warn(msg)
+      );
+      const emitter = createEventEmitter(agent.name, runId, composedOnEvent, options?.parentRunId);
       const lifecycle = new LifecycleManager((from, to) => {
         emitter.emit({ type: "lifecycle", from, to });
         for (const mw of middleware) mw.onTransition?.(ctx, from, to);
@@ -268,7 +309,19 @@ var createRunner = (config2) => {
             thinkingLevel: config2.thinkingLevel,
             responseSchema: config2.responseSchema
           };
-          const response = await wrappedGenerate(params);
+          let response;
+          try {
+            response = await wrappedGenerate(params);
+          } catch (err) {
+            emitter.emit({
+              type: "llm_call",
+              inputTokens: 0,
+              outputTokens: 0,
+              output: "",
+              error: err instanceof Error ? err.message : String(err)
+            });
+            throw err;
+          }
           usage.inputTokens += response.usage.inputTokens;
           usage.outputTokens += response.usage.outputTokens;
           if (!model.generateStream && response.thoughts) {
@@ -358,6 +411,7 @@ var createRunner = (config2) => {
                 ctx,
                 executeFn,
                 tc.name,
+                tc.id,
                 tc.args
               );
               const startTime = Date.now();
@@ -693,7 +747,7 @@ var formatEvent = (e) => {
     case "interrupt":
       return `${tag} interrupt: ${e.toolName} awaiting approval`;
     case "kill":
-      return `${tag} killed${e.reason ? `: ${e.reason}` : ""}`;
+      return `${tag} killed: ${e.reason}`;
     case "error":
       return `${tag} error: ${e.message}`;
     case "notification":

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "kraken-ai",
-  "version": "0.0.6",
+  "version": "0.0.7",
   "description": "Kraken AI SDK: a very lightweight, fully-typed multi-model AI agent framework",
   "type": "module",
   "main": "./dist/index.cjs",