kernl 0.12.3 → 0.12.6

package/dist/thread/thread.js

@@ -1,6 +1,7 @@
 import assert from "assert";
 import * as z from "zod";
 import { Context } from "../context.js";
+import { span, event, } from "../tracing/index.js";
 import { logger } from "../lib/logger.js";
 import { FAILED, RUNNING, STOPPED, message, } from "@kernl-sdk/protocol";
 import { randomID, filter } from "@kernl-sdk/shared/lib";
@@ -64,8 +65,9 @@ export class Thread {
     persisted; /* indicates thread was hydrated from storage */
     history;
     tickres; /* final result from terminal tick */
-    abort;
+    _abort;
     storage;
+    _span; /* tracing span for current execution */
     constructor(options) {
         this.tid = options.tid ?? `tid_${randomID()}`;
         this.namespace = options.namespace ?? "kernl";
@@ -85,6 +87,7 @@ export class Thread {
         this.cpbuf = [];
         this.persisted = options.persisted ?? false;
         this.history = options.history ?? [];
+        this._abort = options.abort;
         // seek to latest seq (not persisted)
         if (this.history.length > 0) {
             this._seq = Math.max(...this.history.map((e) => e.seq));
@@ -113,20 +116,35 @@ export class Thread {
      *   - Exactly one thread.stop (with result on success, error on failure)
      */
     async *stream() {
-        if (this.state === RUNNING && this.abort) {
+        if (this.state === RUNNING) {
             throw new Error("thread already running");
         }
         this.state = RUNNING;
-        this.abort = new AbortController();
         this.tickres = undefined; // reset for this run
         await this.checkpoint(); /* c1: persist RUNNING state + initial input */
+        // create thread span (root span for this execution)
+        this._span = span({
+            kind: "thread",
+            threadId: this.tid,
+            agentId: this.agent.id,
+            namespace: this.namespace,
+            context: this.context.context,
+        }, null);
+        this._span.enter();
         this.emit("thread.start");
         yield { kind: "stream.start" }; // always yield start immediately
         try {
             yield* this._execute();
+            this._span.record({ state: "stopped", result: this.tickres });
             this.emit("thread.stop", { state: STOPPED, result: this.tickres });
         }
         catch (err) {
+            this._span.error(err instanceof Error ? err : new Error(String(err)));
+            event({
+                kind: "thread.error",
+                message: err instanceof Error ? err.message : String(err),
+                stack: err instanceof Error ? err.stack : undefined,
+            });
             this.emit("thread.stop", {
                 state: STOPPED,
                 error: err instanceof Error ? err.message : String(err),
@@ -135,7 +153,9 @@ export class Thread {
         }
         finally {
             this.state = STOPPED;
-            this.abort = undefined;
+            this._span.close();
+            // (TODO): questionable whether this should be undefined. perhaps a single thread should exit + resume..
+            this._span = undefined;
             await this.checkpoint(); /* c4: final checkpoint - persist STOPPED state */
         }
     }
@@ -148,7 +168,7 @@ export class Thread {
     async *_execute() {
         for (;;) {
             let err = undefined;
-            if (this.abort?.signal.aborted) {
+            if (this._abort?.aborted) {
                 return;
             }
             const events = [];
@@ -214,17 +234,36 @@ export class Thread {
         // (TODO): run input guardrails on first tick (if this._tick === 1)
         // (TODO): compaction if necessary
         const req = await this.prepareModelRequest(this.history);
+        const s = span({
+            kind: "model.call",
+            provider: this.model.provider,
+            modelId: this.model.modelId,
+            request: {
+                input: req.input,
+                settings: req.settings,
+                responseType: req.responseType,
+                tools: req.tools,
+            },
+        }, this._span.id);
+        s.enter();
         this.emit("model.call.start", { settings: req.settings ?? {} });
+        // tracing / observability
+        const content = [];
         let usage;
-        let finishReason = { unified: "other", raw: undefined };
+        let finishReason = {
+            unified: "other",
+            raw: undefined,
+        };
         try {
             if (this.model.stream) {
-                for await (const event of this.model.stream(req)) {
-                    if (event.kind === "finish") {
-                        usage = event.usage;
-                        finishReason = event.finishReason;
+                for await (const e of this.model.stream(req)) {
+                    if (e.kind === "finish") {
+                        usage = e.usage;
+                        finishReason = e.finishReason;
                     }
-                    yield event;
+                    if (notDelta(e))
+                        content.push(e);
+                    yield e;
                 }
             }
             else {
@@ -232,19 +271,31 @@ export class Thread {
                 const res = await this.model.generate(req);
                 usage = res.usage;
                 finishReason = res.finishReason;
-                for (const event of res.content) {
-                    yield event;
+                for (const e of res.content) {
+                    content.push(e);
+                    yield e;
                 }
             }
+            s.record({
+                response: {
+                    content,
+                    finishReason,
+                    usage,
+                },
+            });
             this.emit("model.call.end", { finishReason, usage });
         }
         catch (error) {
+            s.error(error instanceof Error ? error : new Error(String(error)));
             this.emit("model.call.end", { finishReason: "error" });
             yield {
                 kind: "error",
                 error: error instanceof Error ? error : new Error(String(error)),
             };
         }
+        finally {
+            s.close();
+        }
     }
     /**
      * Persist current thread state to storage.
@@ -309,12 +360,12 @@ export class Thread {
         return events;
     }
     /**
-     * Cancel the running thread
+     * Abort the running thread.
      *
-     * TODO: Emit thread.stop when cancelled (neither result nor error set)
+     * @throws {Error} Not implemented - use AbortSignal via options instead
      */
-    cancel() {
-        this.abort?.abort();
+    abort() {
+        throw new Error("Not implemented: use AbortSignal via ThreadExecuteOptions");
     }
     /**
      * Emit an agent event with common fields auto-filled.
@@ -384,6 +435,14 @@ export class Thread {
     async executeTools(calls) {
         return await Promise.all(calls.map(async (call) => {
             const parsedArgs = JSON.parse(call.arguments || "{}");
+            // create tool.call span
+            const s = span({
+                kind: "tool.call",
+                toolId: call.toolId,
+                callId: call.callId,
+                args: parsedArgs,
+            }, this._span.id);
+            s.enter();
             this.emit("tool.call.start", {
                 toolId: call.toolId,
                 callId: call.callId,
@@ -402,13 +461,16 @@ export class Thread {
                 ctx.agent = this.agent;
                 ctx.approve(call.callId); // mark this call as approved
                 const res = await tool.invoke(ctx, call.arguments, call.callId);
+                s.record({
+                    state: res.state,
+                    result: res.result,
+                    error: res.error,
+                });
                 this.emit("tool.call.end", {
                     toolId: call.toolId,
                     callId: call.callId,
                     state: res.state,
-                    result: typeof res.result === "string"
-                        ? res.result
-                        : JSON.stringify(res.result),
+                    result: res.result,
                     error: res.error,
                 });
                 return {
@@ -421,11 +483,14 @@ export class Thread {
                 };
             }
             catch (error) {
+                const errMsg = error instanceof Error ? error.message : String(error);
+                s.error(error instanceof Error ? error : new Error(errMsg));
+                s.record({ state: "failed", error: errMsg });
                 this.emit("tool.call.end", {
                     toolId: call.toolId,
                     callId: call.callId,
                     state: FAILED,
-                    error: error instanceof Error ? error.message : String(error),
+                    error: errMsg,
                 });
                 return {
                     kind: "tool.result",
@@ -433,9 +498,12 @@ export class Thread {
                     toolId: call.toolId,
                     state: FAILED,
                     result: undefined,
-                    error: error instanceof Error ? error.message : String(error),
+                    error: errMsg,
                 };
             }
+            finally {
+                s.close();
+            }
         }));
     }
     /**
@@ -479,6 +547,7 @@ export class Thread {
             settings,
             tools,
             responseType,
+            abort: this._abort,
         };
     }
 }

package/dist/thread/types.d.ts

@@ -134,6 +134,11 @@ export interface ThreadOptions<TContext = unknown, TOutput extends AgentOutputTy
      * hydrating from a store. Callers creating new threads should omit it.
      */
     persisted?: boolean;
+    /**
+     * Abort signal for cancelling thread execution.
+     * When aborted, the thread will stop at the next safe point.
+     */
+    abort?: AbortSignal;
 }
 /**
  * Options passed to agent.execute() and agent.stream().

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

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/thread/types.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,aAAa,EACb,iBAAiB,EACjB,OAAO,EACP,OAAO,EACP,aAAa,EACb,eAAe,EACf,MAAM,EACN,IAAI,EAEJ,cAAc,EACd,YAAY,EACZ,cAAc,EACd,mBAAmB,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,UAAU,EACV,WAAW,EACX,UAAU,EACV,UAAU,EACV,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAC9B,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAEhC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAE7C;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG,iBAAiB,GAAG,eAAe,CAAC;AAEpE;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,aAAa,uFAOhB,CAAC;AAEX;;GAEG;AACH,MAAM,MAAM,WAAW,GACnB,OAAO,OAAO,GACd,OAAO,OAAO,GACd,OAAO,aAAa,GACpB,OAAO,eAAe,GACtB,OAAO,MAAM,GACb,OAAO,IAAI,CAAC;AAEhB;;;GAGG;AACH,eAAO,MAAM,iBAAiB,sBAAsB,CAAC;AAErD;;;;GAIG;AACH,MAAM,WAAW,OAAO,CACtB,QAAQ,GAAG,OAAO,EAClB,OAAO,SAAS,eAAe,GAAG,MAAM;IAExC,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChC,KAAK,EAAE,aAAa,CAAC;IAErB,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC3B,KAAK,EAAE,iBAAiB,EAAE,CAAoC;IAC9D,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAsD;IAGjF,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,WAAW,CAA+B;IACjD,SAAS,EAAE,MAAM,CAAC;IAGlB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC1C;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,WAAW,EAAE,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAEjD;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,GAAG,EAAE,MAAM,CAAC;IACZ,SAAS,EAAE,IAAI,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,eAAe;IACxD,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;CAEzB;AAED;;;;;GAKG;AACH,MAAM,MAAM,WAAW,GACnB,CAAC,iBAAiB,GAAG,eAAe,CAAC,GACrC,iBAAiB,CAAC;AAEtB;;GAEG;AACH,MAAM,MAAM,gBAAgB,GACxB,cAAc,GACd,mBAAmB,GACnB,mBAAmB,CAAC;AAExB;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAC1B,cAAc,GACd,YAAY,GACZ,mBAAmB,GACnB,iBAAiB,GACjB,mBAAmB,GACnB,iBAAiB,GACjB,UAAU,GACV,WAAW,GACX,UAAU,GACV,UAAU,GACV,QAAQ,CAAC;AAEb;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,gBAAgB,GAAG,kBAAkB,CAAC;AAEhE;;;;;GAKG;AACH,MAAM,MAAM,iBAAiB,GAAG,WAAW,GAAG,WAAW,CAAC;AAE1D;;GAEG;AACH,MAAM,WAAW,mBAAmB,CAAC,SAAS,GAAG,OAAO;IACtD;;OAEG;IACH,QAAQ,EAAE,SAAS,CAAC;IACpB;;OAEG;IACH,KAAK,EAAE,GAAG,CAAC;CACZ;AAED;;GAEG;AACH,MAAM,WAAW,aAAa,CAC5B,QAAQ,GAAG,OAAO,EAClB,OAAO,SAAS,eAAe,GAAG,MAAM;IAExC,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAC5B,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC;IACxB,OAAO,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC5B,KAAK,CAAC,EAAE,aAAa,CAAC;IACtB,IAAI,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC1C;;;;OAIG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB,CAAC,QAAQ;IAC5C,OAAO,CAAC,EAAE,QAAQ,CAAC;IACnB,KAAK,CAAC,EAAE,aAAa,CAAC;IACtB,IAAI,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,SAAS,EAAE,QAAQ,EAAE,CAAC;CAEvB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC5B;;OAEG;IACH,gBAAgB,EAAE,QAAQ,EAAE,CAAC;CAC9B"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/thread/types.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,QAAQ,EACR,aAAa,EACb,iBAAiB,EACjB,OAAO,EACP,OAAO,EACP,aAAa,EACb,eAAe,EACf,MAAM,EACN,IAAI,EAEJ,cAAc,EACd,YAAY,EACZ,cAAc,EACd,mBAAmB,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,mBAAmB,EACnB,iBAAiB,EACjB,mBAAmB,EACnB,UAAU,EACV,WAAW,EACX,UAAU,EACV,UAAU,EACV,QAAQ,EACT,MAAM,qBAAqB,CAAC;AAE7B,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC;AAC9B,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAEhC,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,eAAe,CAAC;AACrD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,WAAW,CAAC;AAE7C;;GAEG;AACH,MAAM,MAAM,iBAAiB,GAAG,iBAAiB,GAAG,eAAe,CAAC;AAEpE;;GAEG;AACH,MAAM,MAAM,UAAU,GAAG,MAAM,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,aAAa,uFAOhB,CAAC;AAEX;;GAEG;AACH,MAAM,MAAM,WAAW,GACnB,OAAO,OAAO,GACd,OAAO,OAAO,GACd,OAAO,aAAa,GACpB,OAAO,eAAe,GACtB,OAAO,MAAM,GACb,OAAO,IAAI,CAAC;AAEhB;;;GAGG;AACH,eAAO,MAAM,iBAAiB,sBAAsB,CAAC;AAErD;;;;GAIG;AACH,MAAM,WAAW,OAAO,CACtB,QAAQ,GAAG,OAAO,EAClB,OAAO,SAAS,eAAe,GAAG,MAAM;IAExC,GAAG,EAAE,MAAM,CAAC;IACZ,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChC,KAAK,EAAE,aAAa,CAAC;IAErB,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC3B,KAAK,EAAE,iBAAiB,EAAE,CAAoC;IAC9D,OAAO,EAAE,WAAW,EAAE,CAAC;IACvB,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAsD;IAGjF,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,WAAW,CAA+B;IACjD,SAAS,EAAE,MAAM,CAAC;IAGlB,SAAS,EAAE,IAAI,CAAC;IAChB,SAAS,EAAE,IAAI,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC1C;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,MAAM,CAAC,EAAE,WAAW,EAAE,CAAC;CACxB;AAED;;GAEG;AACH,MAAM,MAAM,gBAAgB,GAAG,iBAAiB,CAAC;AAEjD;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,EAAE,EAAE,MAAM,CAAC;IACX,GAAG,EAAE,MAAM,CAAC;IACZ,GAAG,EAAE,MAAM,CAAC;IACZ,SAAS,EAAE,IAAI,CAAC;IAChB,QAAQ,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CACnC;AAED;;GAEG;AACH,MAAM,WAAW,iBAAkB,SAAQ,eAAe;IACxD,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;CAEzB;AAED;;;;;GAKG;AACH,MAAM,MAAM,WAAW,GACnB,CAAC,iBAAiB,GAAG,eAAe,CAAC,GACrC,iBAAiB,CAAC;AAEtB;;GAEG;AACH,MAAM,MAAM,gBAAgB,GACxB,cAAc,GACd,mBAAmB,GACnB,mBAAmB,CAAC;AAExB;;GAEG;AACH,MAAM,MAAM,kBAAkB,GAC1B,cAAc,GACd,YAAY,GACZ,mBAAmB,GACnB,iBAAiB,GACjB,mBAAmB,GACnB,iBAAiB,GACjB,UAAU,GACV,WAAW,GACX,UAAU,GACV,UAAU,GACV,QAAQ,CAAC;AAEb;;GAEG;AACH,MAAM,MAAM,WAAW,GAAG,gBAAgB,GAAG,kBAAkB,CAAC;AAEhE;;;;;GAKG;AACH,MAAM,MAAM,iBAAiB,GAAG,WAAW,GAAG,WAAW,CAAC;AAE1D;;GAEG;AACH,MAAM,WAAW,mBAAmB,CAAC,SAAS,GAAG,OAAO;IACtD;;OAEG;IACH,QAAQ,EAAE,SAAS,CAAC;IACpB;;OAEG;IACH,KAAK,EAAE,GAAG,CAAC;CACZ;AAED;;GAEG;AACH,MAAM,WAAW,aAAa,CAC5B,QAAQ,GAAG,OAAO,EAClB,OAAO,SAAS,eAAe,GAAG,MAAM;IAExC,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IAChC,KAAK,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAC5B,OAAO,CAAC,EAAE,WAAW,EAAE,CAAC;IACxB,OAAO,CAAC,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;IAC5B,KAAK,CAAC,EAAE,aAAa,CAAC;IACtB,IAAI,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,GAAG,IAAI,CAAC;IAC7B,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,GAAG,CAAC,EAAE,MAAM,CAAC;IACb,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,OAAO,CAAC,EAAE,WAAW,CAAC;IACtB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,SAAS,CAAC,EAAE,IAAI,CAAC;IACjB,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC1C;;;;OAIG;IACH,SAAS,CAAC,EAAE,OAAO,CAAC;IACpB;;;OAGG;IACH,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB,CAAC,QAAQ;IAC5C,OAAO,CAAC,EAAE,QAAQ,CAAC;IACnB,KAAK,CAAC,EAAE,aAAa,CAAC;IACtB,IAAI,CAAC,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC;IACtB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,KAAK,CAAC,EAAE,WAAW,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,SAAS;IACxB,SAAS,EAAE,QAAQ,EAAE,CAAC;CAEvB;AAED;;GAEG;AACH,MAAM,WAAW,oBAAoB;IACnC;;OAEG;IACH,OAAO,EAAE,gBAAgB,EAAE,CAAC;IAC5B;;OAEG;IACH,gBAAgB,EAAE,QAAQ,EAAE,CAAC;CAC9B"}

package/dist/tool/tool.d.ts

@@ -1,6 +1,6 @@
 import { Context, UnknownContext } from "../context.js";
 import { type LanguageModelTool } from "@kernl-sdk/protocol";
-import type { ToolConfig, ToolApprovalFunction, ToolEnabledFunction, ToolErrorFunction, ToolInputParameters, ToolResult } from "./types.js";
+import type { ToolConfig, ToolApprovalFunction, ToolEnabledFunction, ToolErrorFunction, ToolExecuteFunction, ToolInputParameters, ToolResult } from "./types.js";
 /**
  * Exposes a function to the agent as a tool to be called
  *
@@ -42,7 +42,7 @@ export declare class FunctionTool<TContext = UnknownContext, TParameters extends
     readonly description: string;
     readonly parameters?: TParameters;
     readonly mode: "blocking" | "async";
-    private execute;
+    execute: ToolExecuteFunction<TContext, TParameters, TResult>;
     errorfn: ToolErrorFunction | null;
     requiresApproval: ToolApprovalFunction<TParameters>;
     isEnabled: ToolEnabledFunction<TContext>;

package/dist/tool/tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/tool/tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAMpD,OAAO,EAIL,KAAK,iBAAiB,EACvB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EACV,UAAU,EACV,oBAAoB,EACpB,mBAAmB,EAEnB,iBAAiB,EAGjB,mBAAmB,EACnB,UAAU,EACX,MAAM,SAAS,CAAC;AAEjB;;;;;GAKG;AACH,wBAAgB,IAAI,CAClB,QAAQ,GAAG,cAAc,EACzB,WAAW,SAAS,mBAAmB,GAAG,SAAS,EACnD,OAAO,GAAG,MAAM,EAEhB,MAAM,EAAE,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,GACjD,YAAY,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,CAE9C;AAED;;GAEG;AACH,8BAAsB,QAAQ,CAAC,QAAQ,GAAG,cAAc;IACtD,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,aAAa,CAAC;IACnD,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAEhC;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAE3C;;OAEG;IACH,QAAQ,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAErD;;OAEG;IACH,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;IAEhE;;OAEG;IACH,QAAQ,CAAC,SAAS,IAAI,iBAAiB;CACxC;AAED;;GAEG;AACH,qBAAa,YAAY,CACvB,QAAQ,GAAG,cAAc,EACzB,WAAW,SAAS,mBAAmB,GAAG,SAAS,EACnD,OAAO,GAAG,OAAO,CACjB,SAAQ,QAAQ,CAAC,QAAQ,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAG,UAAU,CAAU;IACpC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC;IACpC,OAAO,CAAC,OAAO,CAAsD;IAErE,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAClC,gBAAgB,EAAE,oBAAoB,CAAC,WAAW,CAAC,CAAC;IACpD,SAAS,EAAE,mBAAmB,CAAC,QAAQ,CAAC,CAAC;gBAE7B,MAAM,EAAE,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC;IAqC9D;;;;OAIG;IACG,MAAM,CACV,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,IAAI,EAAE,MAAM,EACZ,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAgB/B;;OAEG;YACW,OAAO;IAuCrB;;OAEG;IACH,SAAS,IAAI,iBAAiB;CAU/B;AAED;;GAEG;AACH,qBAAa,UAAW,SAAQ,QAAQ;IACtC,QAAQ,CAAC,IAAI,EAAG,aAAa,CAAU;IACvC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAE5C;;OAEG;IACH,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAA4B;IAE7D;;OAEG;IACH,gBAAgB,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAqB;gBAEpD,MAAM,EAAE;QAClB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACpC;IAOD;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;IAInC;;OAEG;IACH,SAAS,IAAI,iBAAiB;CAQ/B"}
+{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../src/tool/tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAMpD,OAAO,EAIL,KAAK,iBAAiB,EACvB,MAAM,qBAAqB,CAAC;AAG7B,OAAO,KAAK,EACV,UAAU,EACV,oBAAoB,EACpB,mBAAmB,EAEnB,iBAAiB,EAEjB,mBAAmB,EACnB,mBAAmB,EACnB,UAAU,EACX,MAAM,SAAS,CAAC;AAEjB;;;;;GAKG;AACH,wBAAgB,IAAI,CAClB,QAAQ,GAAG,cAAc,EACzB,WAAW,SAAS,mBAAmB,GAAG,SAAS,EACnD,OAAO,GAAG,MAAM,EAEhB,MAAM,EAAE,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,GACjD,YAAY,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,CAE9C;AAED;;GAEG;AACH,8BAAsB,QAAQ,CAAC,QAAQ,GAAG,cAAc;IACtD,QAAQ,CAAC,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,aAAa,CAAC;IACnD,QAAQ,CAAC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAEhC;;OAEG;IACH,QAAQ,CAAC,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAE3C;;OAEG;IACH,QAAQ,CAAC,gBAAgB,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC;IAErD;;OAEG;IACH,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,OAAO,CAAC;IAEhE;;OAEG;IACH,QAAQ,CAAC,SAAS,IAAI,iBAAiB;CACxC;AAED;;GAEG;AACH,qBAAa,YAAY,CACvB,QAAQ,GAAG,cAAc,EACzB,WAAW,SAAS,mBAAmB,GAAG,SAAS,EACnD,OAAO,GAAG,OAAO,CACjB,SAAQ,QAAQ,CAAC,QAAQ,CAAC;IAC1B,QAAQ,CAAC,IAAI,EAAG,UAAU,CAAU;IACpC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,MAAM,CAAC;IAC7B,QAAQ,CAAC,UAAU,CAAC,EAAE,WAAW,CAAC;IAClC,QAAQ,CAAC,IAAI,EAAE,UAAU,GAAG,OAAO,CAAC;IACpC,OAAO,EAAE,mBAAmB,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IAE7D,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAClC,gBAAgB,EAAE,oBAAoB,CAAC,WAAW,CAAC,CAAC;IACpD,SAAS,EAAE,mBAAmB,CAAC,QAAQ,CAAC,CAAC;gBAE7B,MAAM,EAAE,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC;IAqC9D;;;;OAIG;IACG,MAAM,CACV,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,EAC1B,IAAI,EAAE,MAAM,EACZ,MAAM,CAAC,EAAE,MAAM,GACd,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAgB/B;;OAEG;YACW,OAAO;IAuCrB;;OAEG;IACH,SAAS,IAAI,iBAAiB;CAU/B;AAED;;GAEG;AACH,qBAAa,UAAW,SAAQ,QAAQ;IACtC,QAAQ,CAAC,IAAI,EAAG,aAAa,CAAU;IACvC,QAAQ,CAAC,EAAE,EAAE,MAAM,CAAC;IACpB,QAAQ,CAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IACvB,QAAQ,CAAC,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;IAE5C;;OAEG;IACH,OAAO,EAAE,iBAAiB,GAAG,IAAI,CAA4B;IAE7D;;OAEG;IACH,gBAAgB,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAqB;gBAEpD,MAAM,EAAE;QAClB,EAAE,EAAE,MAAM,CAAC;QACX,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,YAAY,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC;KACpC;IAOD;;OAEG;IACG,SAAS,IAAI,OAAO,CAAC,OAAO,CAAC;IAInC;;OAEG;IACH,SAAS,IAAI,iBAAiB;CAQ/B"}

package/dist/tracing/__tests__/composite.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=composite.test.d.ts.map

package/dist/tracing/__tests__/composite.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"composite.test.d.ts","sourceRoot":"","sources":["../../../src/tracing/__tests__/composite.test.ts"],"names":[],"mappings":""}

package/dist/tracing/__tests__/composite.test.js

@@ -0,0 +1,146 @@
+import { describe, it, expect, beforeEach } from "vitest";
+import { CompositeSubscriber } from "../subscribers/composite.js";
+import { TestSubscriber } from "./helpers.js";
+describe("CompositeSubscriber", () => {
+    let sub1;
+    let sub2;
+    let composite;
+    beforeEach(() => {
+        sub1 = new TestSubscriber();
+        sub2 = new TestSubscriber();
+        composite = new CompositeSubscriber([sub1, sub2]);
+    });
+    describe("enabled", () => {
+        it("should return true if any subscriber is enabled", () => {
+            sub1.enabledKinds = new Set(["thread"]);
+            sub2.enabledKinds = new Set(["model.call"]);
+            expect(composite.enabled({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" })).toBe(true);
+            expect(composite.enabled({ kind: "model.call", provider: "test", modelId: "m1" })).toBe(true);
+        });
+        it("should return false if no subscriber is enabled", () => {
+            sub1.enabledKinds = new Set(["thread"]);
+            sub2.enabledKinds = new Set(["thread"]);
+            expect(composite.enabled({ kind: "model.call", provider: "test", modelId: "m1" })).toBe(false);
+        });
+        it("should return true if all subscribers are enabled (default)", () => {
+            expect(composite.enabled({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" })).toBe(true);
+        });
+    });
+    describe("span", () => {
+        it("should create span in all enabled subscribers", () => {
+            const data = { kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" };
+            composite.span(data, null);
+            expect(sub1.spans.size).toBe(1);
+            expect(sub2.spans.size).toBe(1);
+            const [, s1] = [...sub1.spans.entries()][0];
+            const [, s2] = [...sub2.spans.entries()][0];
+            expect(s1.data).toEqual(data);
+            expect(s2.data).toEqual(data);
+        });
+        it("should return composite span ID", () => {
+            const id = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            expect(id).toMatch(/^composite_\d+$/);
+        });
+        it("should skip disabled subscribers", () => {
+            sub1.enabledKinds = new Set(["thread"]);
+            sub2.enabledKinds = new Set(["model.call"]); // not enabled for thread
+            composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            expect(sub1.spans.size).toBe(1);
+            expect(sub2.spans.size).toBe(0);
+        });
+        it("should map parent span IDs correctly", () => {
+            // Create parent
+            const parentId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            // Create child with parent
+            composite.span({ kind: "model.call", provider: "test", modelId: "m1" }, parentId);
+            // Each subscriber should have correct parent mapping
+            const threadSpan1 = sub1.spansOfKind("thread")[0];
+            const modelSpan1 = sub1.spansOfKind("model.call")[0];
+            expect(modelSpan1.parent).toBe(threadSpan1.id);
+            const threadSpan2 = sub2.spansOfKind("thread")[0];
+            const modelSpan2 = sub2.spansOfKind("model.call")[0];
+            expect(modelSpan2.parent).toBe(threadSpan2.id);
+        });
+    });
+    describe("enter / exit", () => {
+        it("should dispatch to all subscribers", () => {
+            const spanId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            composite.enter(spanId);
+            expect(sub1.entered.size).toBe(1);
+            expect(sub2.entered.size).toBe(1);
+            composite.exit(spanId);
+            expect(sub1.exited.size).toBe(1);
+            expect(sub2.exited.size).toBe(1);
+        });
+        it("should skip disabled subscribers", () => {
+            sub2.enabledKinds = new Set(["model.call"]); // not enabled for thread
+            const spanId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            composite.enter(spanId);
+            expect(sub1.entered.size).toBe(1);
+            expect(sub2.entered.size).toBe(0);
+        });
+    });
+    describe("record", () => {
+        it("should dispatch to all subscribers", () => {
+            const spanId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            composite.record(spanId, { state: "running" });
+            const sub1SpanId = [...sub1.spans.keys()][0];
+            const sub2SpanId = [...sub2.spans.keys()][0];
+            expect(sub1.getRecorded(sub1SpanId)).toHaveLength(1);
+            expect(sub2.getRecorded(sub2SpanId)).toHaveLength(1);
+        });
+    });
+    describe("error", () => {
+        it("should dispatch to all subscribers", () => {
+            const spanId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            const err = new Error("test");
+            composite.error(spanId, err);
+            const sub1SpanId = [...sub1.spans.keys()][0];
+            const sub2SpanId = [...sub2.spans.keys()][0];
+            expect(sub1.errors.get(sub1SpanId)).toHaveLength(1);
+            expect(sub2.errors.get(sub2SpanId)).toHaveLength(1);
+        });
+    });
+    describe("close", () => {
+        it("should dispatch to all subscribers and clean up", () => {
+            const spanId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            composite.close(spanId);
+            expect(sub1.closed.size).toBe(1);
+            expect(sub2.closed.size).toBe(1);
+        });
+    });
+    describe("event", () => {
+        it("should dispatch to all subscribers", () => {
+            composite.event({ kind: "thread.error", message: "test" }, null);
+            expect(sub1.events).toHaveLength(1);
+            expect(sub2.events).toHaveLength(1);
+        });
+        it("should map parent span IDs correctly", () => {
+            const spanId = composite.span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            composite.event({ kind: "thread.error", message: "test" }, spanId);
+            const sub1SpanId = [...sub1.spans.keys()][0];
+            const sub2SpanId = [...sub2.spans.keys()][0];
+            expect(sub1.events[0].parent).toBe(sub1SpanId);
+            expect(sub2.events[0].parent).toBe(sub2SpanId);
+        });
+        it("should handle null parent", () => {
+            composite.event({ kind: "thread.error", message: "test" }, null);
+            expect(sub1.events[0].parent).toBeNull();
+            expect(sub2.events[0].parent).toBeNull();
+        });
+    });
+    describe("flush", () => {
+        it("should flush all subscribers", async () => {
+            await composite.flush();
+            expect(sub1.calls.some((c) => c.method === "flush")).toBe(true);
+            expect(sub2.calls.some((c) => c.method === "flush")).toBe(true);
+        });
+    });
+    describe("shutdown", () => {
+        it("should shutdown all subscribers", async () => {
+            await composite.shutdown(5000);
+            expect(sub1.calls.some((c) => c.method === "shutdown")).toBe(true);
+            expect(sub2.calls.some((c) => c.method === "shutdown")).toBe(true);
+        });
+    });
+});

package/dist/tracing/__tests__/dispatch.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=dispatch.test.d.ts.map

package/dist/tracing/__tests__/dispatch.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"dispatch.test.d.ts","sourceRoot":"","sources":["../../../src/tracing/__tests__/dispatch.test.ts"],"names":[],"mappings":""}

package/dist/tracing/__tests__/dispatch.test.js

@@ -0,0 +1,160 @@
+import { describe, it, expect, beforeEach, afterEach } from "vitest";
+import { span, event, run, current, setSubscriber, clearSubscriber, getSubscriber, } from "../dispatch.js";
+import { NoopSpan, SpanImpl } from "../span.js";
+import { TestSubscriber } from "./helpers.js";
+describe("dispatch", () => {
+    let subscriber;
+    beforeEach(() => {
+        subscriber = new TestSubscriber();
+    });
+    afterEach(() => {
+        clearSubscriber();
+    });
+    describe("setSubscriber / clearSubscriber / getSubscriber", () => {
+        it("should set and get subscriber", () => {
+            expect(getSubscriber()).toBeNull();
+            setSubscriber(subscriber);
+            expect(getSubscriber()).toBe(subscriber);
+        });
+        it("should throw if subscriber already set", () => {
+            setSubscriber(subscriber);
+            expect(() => setSubscriber(new TestSubscriber())).toThrow("Global subscriber already set");
+        });
+        it("should allow re-setting after clear", () => {
+            setSubscriber(subscriber);
+            clearSubscriber();
+            expect(getSubscriber()).toBeNull();
+            const newSubscriber = new TestSubscriber();
+            setSubscriber(newSubscriber);
+            expect(getSubscriber()).toBe(newSubscriber);
+        });
+    });
+    describe("span", () => {
+        it("should return NoopSpan when no subscriber is set", () => {
+            const s = span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" });
+            expect(s).toBeInstanceOf(NoopSpan);
+            expect(s.noop()).toBe(true);
+        });
+        it("should return NoopSpan when subscriber.enabled returns false", () => {
+            subscriber.enabledKinds = new Set(["model.call"]); // only model.call enabled
+            setSubscriber(subscriber);
+            const s = span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" });
+            expect(s).toBeInstanceOf(NoopSpan);
+            expect(subscriber.spans.size).toBe(0);
+        });
+        it("should return SpanImpl when subscriber is set and enabled", () => {
+            setSubscriber(subscriber);
+            const s = span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" });
+            expect(s).toBeInstanceOf(SpanImpl);
+            expect(s.noop()).toBe(false);
+            expect(s.id).toBeDefined();
+        });
+        it("should pass span data to subscriber", () => {
+            setSubscriber(subscriber);
+            const data = { kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" };
+            span(data);
+            expect(subscriber.spans.size).toBe(1);
+            const [, captured] = [...subscriber.spans.entries()][0];
+            expect(captured.data).toEqual(data);
+        });
+        it("should use null parent when parent=null", () => {
+            setSubscriber(subscriber);
+            span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            const [, captured] = [...subscriber.spans.entries()][0];
+            expect(captured.parent).toBeNull();
+        });
+        it("should use explicit parent when provided", () => {
+            setSubscriber(subscriber);
+            span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, "parent_123");
+            const [, captured] = [...subscriber.spans.entries()][0];
+            expect(captured.parent).toBe("parent_123");
+        });
+        it("should resolve parent from context when parent='current' (default)", () => {
+            setSubscriber(subscriber);
+            // Create parent span
+            const parentSpan = span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" }, null);
+            // Create child span within run() context
+            run(parentSpan.id, () => {
+                span({ kind: "model.call", provider: "test", modelId: "m1" });
+            });
+            const modelSpans = subscriber.spansOfKind("model.call");
+            expect(modelSpans).toHaveLength(1);
+            expect(modelSpans[0].parent).toBe(parentSpan.id);
+        });
+        it("should use null parent when parent='current' but no context", () => {
+            setSubscriber(subscriber);
+            // No run() context
+            span({ kind: "thread", threadId: "t1", agentId: "a1", namespace: "ns" });
+            const [, captured] = [...subscriber.spans.entries()][0];
+            expect(captured.parent).toBeNull();
+        });
+    });
+    describe("run / current", () => {
+        it("should return null when no context", () => {
+            expect(current()).toBeNull();
+        });
+        it("should return spanId within run context", () => {
+            const result = run("span_123", () => {
+                return current();
+            });
+            expect(result).toBe("span_123");
+        });
+        it("should return to null after run completes", () => {
+            run("span_123", () => {
+                expect(current()).toBe("span_123");
+            });
+            expect(current()).toBeNull();
+        });
+        it("should support nested run contexts", () => {
+            const results = [];
+            run("outer", () => {
+                results.push(current());
+                run("inner", () => {
+                    results.push(current());
+                });
+                results.push(current());
+            });
+            expect(results).toEqual(["outer", "inner", "outer"]);
+        });
+        it("should handle null spanId in run", () => {
+            run("span_123", () => {
+                run(null, () => {
+                    expect(current()).toBeNull();
+                });
+            });
+        });
+    });
+    describe("event", () => {
+        it("should do nothing when no subscriber is set", () => {
+            // Should not throw
+            event({ kind: "thread.error", message: "test" });
+        });
+        it("should emit event to subscriber", () => {
+            setSubscriber(subscriber);
+            event({ kind: "thread.error", message: "test error", stack: "stack" });
+            expect(subscriber.events).toHaveLength(1);
+            expect(subscriber.events[0].data).toEqual({
+                kind: "thread.error",
+                message: "test error",
+                stack: "stack",
+            });
+        });
+        it("should use null parent when parent=null", () => {
+            setSubscriber(subscriber);
+            event({ kind: "thread.error", message: "test" }, null);
+            expect(subscriber.events[0].parent).toBeNull();
+        });
+        it("should use explicit parent when provided", () => {
+            setSubscriber(subscriber);
+            event({ kind: "thread.error", message: "test" }, "parent_123");
+            expect(subscriber.events[0].parent).toBe("parent_123");
+        });
+        it("should resolve parent from context when parent='current' (default)", () => {
+            setSubscriber(subscriber);
+            run("span_123", () => {
+                event({ kind: "thread.error", message: "test" });
+            });
+            expect(subscriber.events[0].parent).toBe("span_123");
+        });
+    });
+});