lightrace 0.1.6 → 0.1.8

package/dist/integrations/llamaindex.js

@@ -0,0 +1,222 @@
+/**
+ * LlamaIndex integration for Lightrace.
+ *
+ * Provides a handler that hooks into LlamaIndex's callback system to capture
+ * LLM calls, tool invocations, retrieval, and query events as OTel spans.
+ *
+ * @example
+ * ```ts
+ * import { LightraceLlamaIndexHandler } from "lightrace/integrations/llamaindex";
+ *
+ * const handler = new LightraceLlamaIndexHandler({ userId: "user-123" });
+ * // Register with LlamaIndex Settings or CallbackManager
+ * ```
+ */
+import { generateId, jsonSerializable } from "../utils.js";
+import { TracingMixin } from "./_base.js";
+/** Observation type mapping for LlamaIndex event types. */
+const EVENT_TYPE_MAP = {
+    llm: "generation",
+    embedding: "span",
+    retrieve: "span",
+    query: "span",
+    function_call: "tool",
+    agent_step: "span",
+    chunking: "span",
+    reranking: "span",
+    synthesize: "span",
+    tree: "span",
+    sub_question: "span",
+    templating: "span",
+};
+export class LightraceLlamaIndexHandler extends TracingMixin {
+    traceRunId = null;
+    constructor(opts) {
+        super(opts);
+    }
+    /** Called when a LlamaIndex trace begins. */
+    startTrace(traceId) {
+        try {
+            const runId = generateId();
+            this.traceRunId = runId;
+            this.registerRun(runId, undefined, {
+                type: "span",
+                name: traceId ?? "llamaindex-trace",
+                startTime: new Date(),
+                input: traceId ? { trace_id: traceId } : undefined,
+            });
+        }
+        catch (e) {
+            console.warn("[lightrace] Error in startTrace:", e);
+        }
+    }
+    /** Called when a LlamaIndex trace ends. */
+    endTrace() {
+        try {
+            if (this.traceRunId) {
+                this.endRun(this.traceRunId, null);
+                this.traceRunId = null;
+            }
+        }
+        catch (e) {
+            console.warn("[lightrace] Error in endTrace:", e);
+        }
+    }
+    /** Called when a LlamaIndex event starts. */
+    onEventStart(eventType, payload, eventId, parentId) {
+        try {
+            const obsType = EVENT_TYPE_MAP[eventType.toLowerCase()] ?? "span";
+            const runId = eventId ?? generateId();
+            const parentRunId = parentId && this.runs.has(parentId) ? parentId : (this.traceRunId ?? undefined);
+            let input = undefined;
+            let model;
+            let modelParams;
+            if (payload) {
+                input = extractInput(eventType, payload);
+                if (eventType.toLowerCase() === "llm") {
+                    model = extractModel(payload);
+                    modelParams = extractModelParams(payload);
+                }
+            }
+            this.registerRun(runId, parentRunId, {
+                type: obsType,
+                name: eventType,
+                startTime: new Date(),
+                input,
+                model,
+                modelParameters: modelParams,
+            });
+            return runId;
+        }
+        catch (e) {
+            console.warn("[lightrace] Error in onEventStart:", e);
+            return eventId ?? generateId();
+        }
+    }
+    /** Called when a LlamaIndex event ends. */
+    onEventEnd(eventType, payload, eventId) {
+        try {
+            if (!eventId)
+                return;
+            let output = undefined;
+            let extra;
+            if (payload) {
+                output = extractOutput(eventType, payload);
+                if (eventType.toLowerCase() === "llm") {
+                    extra = extractLlmUsage(payload) ?? undefined;
+                }
+            }
+            this.endRun(eventId, output, "DEFAULT", null, extra);
+        }
+        catch (e) {
+            console.warn("[lightrace] Error in onEventEnd:", e);
+        }
+    }
+}
+// -- Extraction helpers -----------------------------------------------------
+function extractInput(eventType, payload) {
+    const type = eventType.toLowerCase();
+    if (type === "llm") {
+        if (payload.messages)
+            return jsonSerializable(payload.messages);
+        if (payload.template)
+            return String(payload.template);
+        return payload.prompt ?? payload.query_str ?? null;
+    }
+    if (type === "retrieve" || type === "query") {
+        return payload.query_str ?? null;
+    }
+    if (type === "function_call") {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const tool = payload.tool;
+        return {
+            tool: tool?.name ?? payload.function_call ?? "unknown",
+            arguments: jsonSerializable(payload.function_call_args ?? {}),
+        };
+    }
+    if (type === "embedding") {
+        const chunks = payload.chunks;
+        if (chunks)
+            return { num_chunks: chunks.length };
+    }
+    return jsonSerializable(payload);
+}
+function extractOutput(eventType, payload) {
+    const type = eventType.toLowerCase();
+    if (type === "llm") {
+        if (payload.response)
+            return jsonSerializable(payload.response);
+        if (payload.completion)
+            return String(payload.completion);
+        return payload.formatted_prompt ?? null;
+    }
+    if (type === "retrieve") {
+        const nodes = payload.nodes;
+        if (nodes) {
+            return nodes.slice(0, 10).map((n) => {
+                // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                const node = n;
+                return {
+                    text: String(node?.text ?? node).slice(0, 500),
+                    score: node?.score ?? null,
+                };
+            });
+        }
+    }
+    if (type === "query") {
+        if (payload.response)
+            return String(payload.response);
+    }
+    if (type === "function_call") {
+        return payload.function_call_response ?? null;
+    }
+    if (type === "embedding") {
+        const embeddings = payload.embeddings;
+        if (embeddings)
+            return { num_embeddings: embeddings.length };
+    }
+    return jsonSerializable(payload);
+}
+function extractModel(payload) {
+    const serialized = payload.serialized;
+    if (serialized) {
+        const model = serialized.model ?? serialized.model_name;
+        if (model)
+            return String(model);
+    }
+    return undefined;
+}
+function extractModelParams(payload) {
+    const serialized = payload.serialized;
+    if (!serialized)
+        return undefined;
+    const params = {};
+    for (const key of ["temperature", "max_tokens", "top_p"]) {
+        if (serialized[key] !== undefined)
+            params[key] = serialized[key];
+    }
+    return Object.keys(params).length > 0 ? params : undefined;
+}
+function extractLlmUsage(payload) {
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const response = payload.response;
+    if (!response)
+        return null;
+    const usage = response?.raw?.usage;
+    if (usage) {
+        const raw = {};
+        if (usage.prompt_tokens !== undefined)
+            raw.promptTokens = usage.prompt_tokens;
+        if (usage.completion_tokens !== undefined)
+            raw.completionTokens = usage.completion_tokens;
+        if (usage.total_tokens !== undefined)
+            raw.totalTokens = usage.total_tokens;
+        if (usage.input_tokens !== undefined)
+            raw.promptTokens = usage.input_tokens;
+        if (usage.output_tokens !== undefined)
+            raw.completionTokens = usage.output_tokens;
+        return Object.keys(raw).length > 0 ? raw : null;
+    }
+    return null;
+}
+//# sourceMappingURL=llamaindex.js.map

package/dist/integrations/llamaindex.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"llamaindex.js","sourceRoot":"","sources":["../../src/integrations/llamaindex.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;GAaG;AACH,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,YAAY,EAA4C,MAAM,YAAY,CAAC;AAEpF,2DAA2D;AAC3D,MAAM,cAAc,GAAmD;IACrE,GAAG,EAAE,YAAY;IACjB,SAAS,EAAE,MAAM;IACjB,QAAQ,EAAE,MAAM;IAChB,KAAK,EAAE,MAAM;IACb,aAAa,EAAE,MAAM;IACrB,UAAU,EAAE,MAAM;IAClB,QAAQ,EAAE,MAAM;IAChB,SAAS,EAAE,MAAM;IACjB,UAAU,EAAE,MAAM;IAClB,IAAI,EAAE,MAAM;IACZ,YAAY,EAAE,MAAM;IACpB,UAAU,EAAE,MAAM;CACnB,CAAC;AAEF,MAAM,OAAO,0BAA2B,SAAQ,YAAY;IAClD,UAAU,GAAkB,IAAI,CAAC;IAEzC,YAAY,IAA0B;QACpC,KAAK,CAAC,IAAI,CAAC,CAAC;IACd,CAAC;IAED,6CAA6C;IAC7C,UAAU,CAAC,OAAgB;QACzB,IAAI,CAAC;YACH,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;YAC3B,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;YACxB,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE;gBACjC,IAAI,EAAE,MAAM;gBACZ,IAAI,EAAE,OAAO,IAAI,kBAAkB;gBACnC,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,KAAK,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,SAAS;aACnD,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;IAED,2CAA2C;IAC3C,QAAQ;QACN,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;gBACpB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,CAAC;gBACnC,IAAI,CAAC,UAAU,GAAG,IAAI,CAAC;YACzB,CAAC;QACH,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,gCAAgC,EAAE,CAAC,CAAC,CAAC;QACpD,CAAC;IACH,CAAC;IAED,6CAA6C;IAC7C,YAAY,CACV,SAAiB,EACjB,OAAwC,EACxC,OAAgB,EAChB,QAAiB;QAEjB,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,CAAC,WAAW,EAAE,CAAC,IAAI,MAAM,CAAC;YAClE,MAAM,KAAK,GAAG,OAAO,IAAI,UAAU,EAAE,CAAC;YACtC,MAAM,WAAW,GACf,QAAQ,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,UAAU,IAAI,SAAS,CAAC,CAAC;YAElF,IAAI,KAAK,GAAY,SAAS,CAAC;YAC/B,IAAI,KAAyB,CAAC;YAC9B,IAAI,WAAgD,CAAC;YAErD,IAAI,OAAO,EAAE,CAAC;gBACZ,KAAK,GAAG,YAAY,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBACzC,IAAI,SAAS,CAAC,WAAW,EAAE,KAAK,KAAK,EAAE,CAAC;oBACtC,KAAK,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;oBAC9B,WAAW,GAAG,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBAC5C,CAAC;YACH,CAAC;YAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,WAAW,EAAE;gBACnC,IAAI,EAAE,OAAO;gBACb,IAAI,EAAE,SAAS;gBACf,SAAS,EAAE,IAAI,IAAI,EAAE;gBACrB,KAAK;gBACL,KAAK;gBACL,eAAe,EAAE,WAAW;aAC7B,CAAC,CAAC;YAEH,OAAO,KAAK,CAAC;QACf,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,oCAAoC,EAAE,CAAC,CAAC,CAAC;YACtD,OAAO,OAAO,IAAI,UAAU,EAAE,CAAC;QACjC,CAAC;IACH,CAAC;IAED,2CAA2C;IAC3C,UAAU,CAAC,SAAiB,EAAE,OAAwC,EAAE,OAAgB;QACtF,IAAI,CAAC;YACH,IAAI,CAAC,OAAO;gBAAE,OAAO;YAErB,IAAI,MAAM,GAAY,SAAS,CAAC;YAChC,IAAI,KAA0C,CAAC;YAE/C,IAAI,OAAO,EAAE,CAAC;gBACZ,MAAM,GAAG,aAAa,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;gBAC3C,IAAI,SAAS,CAAC,WAAW,EAAE,KAAK,KAAK,EAAE,CAAC;oBACtC,KAAK,GAAG,eAAe,CAAC,OAAO,CAAC,IAAI,SAAS,CAAC;gBAChD,CAAC;YACH,CAAC;YAED,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACX,OAAO,CAAC,IAAI,CAAC,kCAAkC,EAAE,CAAC,CAAC,CAAC;QACtD,CAAC;IACH,CAAC;CACF;AAED,8EAA8E;AAE9E,SAAS,YAAY,CAAC,SAAiB,EAAE,OAAgC;IACvE,MAAM,IAAI,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IAErC,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;QACnB,IAAI,OAAO,CAAC,QAAQ;YAAE,OAAO,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAChE,IAAI,OAAO,CAAC,QAAQ;YAAE,OAAO,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACtD,OAAO,OAAO,CAAC,MAAM,IAAI,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;IACrD,CAAC;IAED,IAAI,IAAI,KAAK,UAAU,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QAC5C,OAAO,OAAO,CAAC,SAAS,IAAI,IAAI,CAAC;IACnC,CAAC;IAED,IAAI,IAAI,KAAK,eAAe,EAAE,CAAC;QAC7B,8DAA8D;QAC9D,MAAM,IAAI,GAAG,OAAO,CAAC,IAAW,CAAC;QACjC,OAAO;YACL,IAAI,EAAE,IAAI,EAAE,IAAI,IAAI,OAAO,CAAC,aAAa,IAAI,SAAS;YACtD,SAAS,EAAE,gBAAgB,CAAC,OAAO,CAAC,kBAAkB,IAAI,EAAE,CAAC;SAC9D,CAAC;IACJ,CAAC;IAED,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,OAAO,CAAC,MAAmB,CAAC;QAC3C,IAAI,MAAM;YAAE,OAAO,EAAE,UAAU,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC;IACnD,CAAC;IAED,OAAO,gBAAgB,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,aAAa,CAAC,SAAiB,EAAE,OAAgC;IACxE,MAAM,IAAI,GAAG,SAAS,CAAC,WAAW,EAAE,CAAC;IAErC,IAAI,IAAI,KAAK,KAAK,EAAE,CAAC;QACnB,IAAI,OAAO,CAAC,QAAQ;YAAE,OAAO,gBAAgB,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAChE,IAAI,OAAO,CAAC,UAAU;YAAE,OAAO,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;QAC1D,OAAO,OAAO,CAAC,gBAAgB,IAAI,IAAI,CAAC;IAC1C,CAAC;IAED,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;QACxB,MAAM,KAAK,GAAG,OAAO,CAAC,KAAkB,CAAC;QACzC,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAU,EAAE,EAAE;gBAC3C,8DAA8D;gBAC9D,MAAM,IAAI,GAAG,CAAQ,CAAC;gBACtB,OAAO;oBACL,IAAI,EAAE,MAAM,CAAC,IAAI,EAAE,IAAI,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC;oBAC9C,KAAK,EAAE,IAAI,EAAE,KAAK,IAAI,IAAI;iBAC3B,CAAC;YACJ,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,IAAI,IAAI,KAAK,OAAO,EAAE,CAAC;QACrB,IAAI,OAAO,CAAC,QAAQ;YAAE,OAAO,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACxD,CAAC;IAED,IAAI,IAAI,KAAK,eAAe,EAAE,CAAC;QAC7B,OAAO,OAAO,CAAC,sBAAsB,IAAI,IAAI,CAAC;IAChD,CAAC;IAED,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACzB,MAAM,UAAU,GAAG,OAAO,CAAC,UAAuB,CAAC;QACnD,IAAI,UAAU;YAAE,OAAO,EAAE,cAAc,EAAE,UAAU,CAAC,MAAM,EAAE,CAAC;IAC/D,CAAC;IAED,OAAO,gBAAgB,CAAC,OAAO,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,YAAY,CAAC,OAAgC;IACpD,MAAM,UAAU,GAAG,OAAO,CAAC,UAAiD,CAAC;IAC7E,IAAI,UAAU,EAAE,CAAC;QACf,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,IAAI,UAAU,CAAC,UAAU,CAAC;QACxD,IAAI,KAAK;YAAE,OAAO,MAAM,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,kBAAkB,CAAC,OAAgC;IAC1D,MAAM,UAAU,GAAG,OAAO,CAAC,UAAiD,CAAC;IAC7E,IAAI,CAAC,UAAU;QAAE,OAAO,SAAS,CAAC;IAElC,MAAM,MAAM,GAA4B,EAAE,CAAC;IAC3C,KAAK,MAAM,GAAG,IAAI,CAAC,aAAa,EAAE,YAAY,EAAE,OAAO,CAAC,EAAE,CAAC;QACzD,IAAI,UAAU,CAAC,GAAG,CAAC,KAAK,SAAS;YAAE,MAAM,CAAC,GAAG,CAAC,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC;IACnE,CAAC;IACD,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC;AAC7D,CAAC;AAED,SAAS,eAAe,CAAC,OAAgC;IACvD,8DAA8D;IAC9D,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAe,CAAC;IACzC,IAAI,CAAC,QAAQ;QAAE,OAAO,IAAI,CAAC;IAE3B,MAAM,KAAK,GAAG,QAAQ,EAAE,GAAG,EAAE,KAAK,CAAC;IACnC,IAAI,KAAK,EAAE,CAAC;QACV,MAAM,GAAG,GAA4B,EAAE,CAAC;QACxC,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS;YAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,aAAa,CAAC;QAC9E,IAAI,KAAK,CAAC,iBAAiB,KAAK,SAAS;YAAE,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC,iBAAiB,CAAC;QAC1F,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;YAAE,GAAG,CAAC,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;QAC3E,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;YAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;QAC5E,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS;YAAE,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC,aAAa,CAAC;QAClF,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;IAClD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC"}

package/dist/integrations/openai.d.ts

@@ -0,0 +1,17 @@
+import { TracingMixin, type TracingMixinOptions } from "./_base.js";
+export declare class LightraceOpenAIInstrumentor extends TracingMixin {
+    private patchedTargets;
+    constructor(opts?: TracingMixinOptions);
+    /**
+     * Patch an OpenAI client instance so that `chat.completions.create` and
+     * `responses.create` calls are automatically traced.
+     */
+    instrument(client: unknown): void;
+    /** Restore the original methods. */
+    uninstrument(client: unknown): void;
+    /** Start a trace for a chat completions call. */
+    private startChatTrace;
+    /** Start a trace for a Responses API call. */
+    private startResponsesTrace;
+}
+//# sourceMappingURL=openai.d.ts.map

package/dist/integrations/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/integrations/openai.ts"],"names":[],"mappings":"AAkBA,OAAO,EAAE,YAAY,EAAkB,KAAK,mBAAmB,EAAE,MAAM,YAAY,CAAC;AAKpF,qBAAa,2BAA4B,SAAQ,YAAY;IAC3D,OAAO,CAAC,cAAc,CAQlB;gBAEQ,IAAI,CAAC,EAAE,mBAAmB;IAItC;;;OAGG;IACH,UAAU,CAAC,MAAM,EAAE,OAAO,GAAG,IAAI;IAuFjC,oCAAoC;IACpC,YAAY,CAAC,MAAM,EAAE,OAAO,GAAG,IAAI;IAcnC,iDAAiD;IACjD,OAAO,CAAC,cAAc;IA4CtB,8CAA8C;IAC9C,OAAO,CAAC,mBAAmB;CAsC5B"}

package/dist/integrations/openai.js

@@ -0,0 +1,259 @@
+/**
+ * OpenAI SDK integration for Lightrace.
+ *
+ * Wraps `chat.completions.create` and `responses.create` on OpenAI client
+ * instances to automatically capture traces via OpenTelemetry spans.
+ *
+ * @example
+ * ```ts
+ * import { LightraceOpenAIInstrumentor } from "lightrace/integrations/openai";
+ *
+ * const instrumentor = new LightraceOpenAIInstrumentor();
+ * instrumentor.instrument(client);
+ *
+ * const response = await client.chat.completions.create({ ... });
+ * console.log(instrumentor.lastTraceId);
+ * ```
+ */
+import { generateId, jsonSerializable } from "../utils.js";
+import { TracingMixin } from "./_base.js";
+export class LightraceOpenAIInstrumentor extends TracingMixin {
+    patchedTargets = new Map();
+    constructor(opts) {
+        super(opts);
+    }
+    /**
+     * Patch an OpenAI client instance so that `chat.completions.create` and
+     * `responses.create` calls are automatically traced.
+     */
+    instrument(client) {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const c = client;
+        if (this.patchedTargets.has(client))
+            return;
+        const entry = {};
+        const self = this;
+        // Patch chat.completions.create
+        const completions = c?.chat?.completions;
+        if (completions?.create) {
+            const originalChatCreate = completions.create.bind(completions);
+            entry.completions = completions;
+            entry.originalChatCreate = originalChatCreate;
+            completions.create = async function (...args) {
+                const kwargs = args.length === 1 && typeof args[0] === "object" && args[0] !== null
+                    ? args[0]
+                    : {};
+                const { runId, stream } = self.startChatTrace(kwargs);
+                try {
+                    const result = await originalChatCreate(...args);
+                    if (!stream) {
+                        const output = extractChatOutput(result);
+                        const usage = extractUsage(result);
+                        self.endRun(runId, output, "DEFAULT", null, usage ?? undefined);
+                    }
+                    else {
+                        self.endRun(runId, { streaming: true });
+                    }
+                    return result;
+                }
+                catch (err) {
+                    self.endRun(runId, null, "ERROR", err instanceof Error ? err.message : String(err));
+                    throw err;
+                }
+            };
+        }
+        // Patch responses.create (Responses API)
+        const responses = c?.responses;
+        if (responses?.create) {
+            const originalResponsesCreate = responses.create.bind(responses);
+            entry.responses = responses;
+            entry.originalResponsesCreate = originalResponsesCreate;
+            responses.create = async function (...args) {
+                const kwargs = args.length === 1 && typeof args[0] === "object" && args[0] !== null
+                    ? args[0]
+                    : {};
+                const { runId, stream } = self.startResponsesTrace(kwargs);
+                try {
+                    const result = await originalResponsesCreate(...args);
+                    if (!stream) {
+                        const output = extractResponsesOutput(result);
+                        const usage = extractUsage(result);
+                        self.endRun(runId, output, "DEFAULT", null, usage ?? undefined);
+                    }
+                    else {
+                        self.endRun(runId, { streaming: true });
+                    }
+                    return result;
+                }
+                catch (err) {
+                    self.endRun(runId, null, "ERROR", err instanceof Error ? err.message : String(err));
+                    throw err;
+                }
+            };
+        }
+        if (!entry.completions && !entry.responses)
+            return;
+        this.patchedTargets.set(client, entry);
+    }
+    /** Restore the original methods. */
+    uninstrument(client) {
+        const entry = this.patchedTargets.get(client);
+        if (!entry)
+            return;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        if (entry.completions && entry.originalChatCreate) {
+            entry.completions.create = entry.originalChatCreate;
+        }
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        if (entry.responses && entry.originalResponsesCreate) {
+            entry.responses.create = entry.originalResponsesCreate;
+        }
+        this.patchedTargets.delete(client);
+    }
+    /** Start a trace for a chat completions call. */
+    startChatTrace(kwargs) {
+        const runId = generateId();
+        const model = kwargs.model;
+        const stream = Boolean(kwargs.stream);
+        const inputData = {
+            messages: jsonSerializable(kwargs.messages ?? []),
+        };
+        if (model)
+            inputData.model = model;
+        if (kwargs.tools)
+            inputData.tools = jsonSerializable(kwargs.tools);
+        if (kwargs.tool_choice)
+            inputData.tool_choice = jsonSerializable(kwargs.tool_choice);
+        if (kwargs.response_format)
+            inputData.response_format = jsonSerializable(kwargs.response_format);
+        const modelParams = {};
+        for (const key of [
+            "temperature",
+            "max_tokens",
+            "max_completion_tokens",
+            "top_p",
+            "frequency_penalty",
+            "presence_penalty",
+            "stop",
+            "seed",
+            "n",
+        ]) {
+            if (kwargs[key] !== undefined)
+                modelParams[key] = kwargs[key];
+        }
+        this.registerRun(runId, undefined, {
+            type: "generation",
+            name: model ?? "openai",
+            startTime: new Date(),
+            input: inputData,
+            model,
+            modelParameters: Object.keys(modelParams).length > 0 ? modelParams : undefined,
+        });
+        return { runId, stream };
+    }
+    /** Start a trace for a Responses API call. */
+    startResponsesTrace(kwargs) {
+        const runId = generateId();
+        const model = kwargs.model;
+        const stream = Boolean(kwargs.stream);
+        const inputData = {
+            input: jsonSerializable(kwargs.input ?? []),
+        };
+        if (model)
+            inputData.model = model;
+        if (kwargs.tools)
+            inputData.tools = jsonSerializable(kwargs.tools);
+        if (kwargs.instructions)
+            inputData.instructions = kwargs.instructions;
+        const modelParams = {};
+        for (const key of [
+            "temperature",
+            "max_output_tokens",
+            "top_p",
+            "frequency_penalty",
+            "presence_penalty",
+            "seed",
+        ]) {
+            if (kwargs[key] !== undefined)
+                modelParams[key] = kwargs[key];
+        }
+        this.registerRun(runId, undefined, {
+            type: "generation",
+            name: model ?? "openai-responses",
+            startTime: new Date(),
+            input: inputData,
+            model,
+            modelParameters: Object.keys(modelParams).length > 0 ? modelParams : undefined,
+        });
+        return { runId, stream };
+    }
+}
+// -- Helpers ----------------------------------------------------------------
+function extractChatOutput(response) {
+    if (!response || typeof response !== "object")
+        return response;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const r = response;
+    const choices = r.choices;
+    if (!Array.isArray(choices) || choices.length === 0)
+        return jsonSerializable(response);
+    const message = choices[0]?.message;
+    if (!message)
+        return jsonSerializable(choices[0]);
+    const output = {
+        role: message.role ?? "assistant",
+        content: message.content ?? null,
+    };
+    if (message.tool_calls?.length) {
+        output.tool_calls = message.tool_calls.map((tc) => ({
+            id: tc.id ?? "",
+            type: tc.type ?? "function",
+            function: {
+                name: tc.function?.name ?? "",
+                arguments: tc.function?.arguments ?? "",
+            },
+        }));
+    }
+    const finishReason = choices[0]?.finish_reason;
+    if (finishReason)
+        output.finish_reason = finishReason;
+    return output;
+}
+function extractResponsesOutput(response) {
+    if (!response || typeof response !== "object")
+        return response;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const r = response;
+    if (r.output !== undefined)
+        return jsonSerializable(r.output);
+    return jsonSerializable(response);
+}
+function extractUsage(response) {
+    if (!response || typeof response !== "object")
+        return null;
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const usage = response.usage;
+    if (!usage)
+        return null;
+    const raw = {};
+    // Chat Completions API
+    if (usage.prompt_tokens !== undefined)
+        raw.promptTokens = usage.prompt_tokens;
+    if (usage.completion_tokens !== undefined)
+        raw.completionTokens = usage.completion_tokens;
+    if (usage.total_tokens !== undefined)
+        raw.totalTokens = usage.total_tokens;
+    // Responses API
+    if (usage.input_tokens !== undefined)
+        raw.promptTokens = usage.input_tokens;
+    if (usage.output_tokens !== undefined)
+        raw.completionTokens = usage.output_tokens;
+    // Auto-total
+    if (raw.totalTokens === undefined &&
+        raw.promptTokens !== undefined &&
+        raw.completionTokens !== undefined) {
+        raw.totalTokens = raw.promptTokens + raw.completionTokens;
+    }
+    return Object.keys(raw).length > 0 ? raw : null;
+}
+//# sourceMappingURL=openai.js.map

package/dist/integrations/openai.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.js","sourceRoot":"","sources":["../../src/integrations/openai.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AACH,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,YAAY,EAA4C,MAAM,YAAY,CAAC;AAKpF,MAAM,OAAO,2BAA4B,SAAQ,YAAY;IACnD,cAAc,GAAG,IAAI,GAAG,EAQ7B,CAAC;IAEJ,YAAY,IAA0B;QACpC,KAAK,CAAC,IAAI,CAAC,CAAC;IACd,CAAC;IAED;;;OAGG;IACH,UAAU,CAAC,MAAe;QACxB,8DAA8D;QAC9D,MAAM,CAAC,GAAG,MAAa,CAAC;QACxB,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC;YAAE,OAAO;QAE5C,MAAM,KAAK,GAKP,EAAE,CAAC;QAEP,MAAM,IAAI,GAAG,IAAI,CAAC;QAElB,gCAAgC;QAChC,MAAM,WAAW,GAAG,CAAC,EAAE,IAAI,EAAE,WAAW,CAAC;QACzC,IAAI,WAAW,EAAE,MAAM,EAAE,CAAC;YACxB,MAAM,kBAAkB,GAAG,WAAW,CAAC,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAChE,KAAK,CAAC,WAAW,GAAG,WAAW,CAAC;YAChC,KAAK,CAAC,kBAAkB,GAAG,kBAAkB,CAAC;YAE9C,WAAW,CAAC,MAAM,GAAG,KAAK,WAAW,GAAG,IAAe;gBACrD,MAAM,MAAM,GACV,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;oBAClE,CAAC,CAAE,IAAI,CAAC,CAAC,CAA6B;oBACtC,CAAC,CAAC,EAAE,CAAC;gBAET,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;gBAEtD,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,GAAG,IAAI,CAAC,CAAC;oBAEjD,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,MAAM,MAAM,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;wBACzC,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;wBACnC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,IAAI,SAAS,CAAC,CAAC;oBAClE,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC1C,CAAC;oBAED,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBACpF,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,yCAAyC;QACzC,MAAM,SAAS,GAAG,CAAC,EAAE,SAAS,CAAC;QAC/B,IAAI,SAAS,EAAE,MAAM,EAAE,CAAC;YACtB,MAAM,uBAAuB,GAAG,SAAS,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACjE,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;YAC5B,KAAK,CAAC,uBAAuB,GAAG,uBAAuB,CAAC;YAExD,SAAS,CAAC,MAAM,GAAG,KAAK,WAAW,GAAG,IAAe;gBACnD,MAAM,MAAM,GACV,IAAI,CAAC,MAAM,KAAK,CAAC,IAAI,OAAO,IAAI,CAAC,CAAC,CAAC,KAAK,QAAQ,IAAI,IAAI,CAAC,CAAC,CAAC,KAAK,IAAI;oBAClE,CAAC,CAAE,IAAI,CAAC,CAAC,CAA6B;oBACtC,CAAC,CAAC,EAAE,CAAC;gBAET,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,CAAC;gBAE3D,IAAI,CAAC;oBACH,MAAM,MAAM,GAAG,MAAM,uBAAuB,CAAC,GAAG,IAAI,CAAC,CAAC;oBAEtD,IAAI,CAAC,MAAM,EAAE,CAAC;wBACZ,MAAM,MAAM,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;wBAC9C,MAAM,KAAK,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;wBACnC,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,IAAI,SAAS,CAAC,CAAC;oBAClE,CAAC;yBAAM,CAAC;wBACN,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;oBAC1C,CAAC;oBAED,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,EAAE,OAAO,EAAE,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;oBACpF,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,KAAK,CAAC,WAAW,IAAI,CAAC,KAAK,CAAC,SAAS;YAAE,OAAO;QAEnD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,oCAAoC;IACpC,YAAY,CAAC,MAAe;QAC1B,MAAM,KAAK,GAAG,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAC9C,IAAI,CAAC,KAAK;YAAE,OAAO;QACnB,8DAA8D;QAC9D,IAAI,KAAK,CAAC,WAAW,IAAI,KAAK,CAAC,kBAAkB,EAAE,CAAC;YACjD,KAAK,CAAC,WAAmB,CAAC,MAAM,GAAG,KAAK,CAAC,kBAAkB,CAAC;QAC/D,CAAC;QACD,8DAA8D;QAC9D,IAAI,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,uBAAuB,EAAE,CAAC;YACpD,KAAK,CAAC,SAAiB,CAAC,MAAM,GAAG,KAAK,CAAC,uBAAuB,CAAC;QAClE,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,iDAAiD;IACzC,cAAc,CAAC,MAA+B;QAIpD,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,KAA2B,CAAC;QACjD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEtC,MAAM,SAAS,GAA4B;YACzC,QAAQ,EAAE,gBAAgB,CAAC,MAAM,CAAC,QAAQ,IAAI,EAAE,CAAC;SAClD,CAAC;QACF,IAAI,KAAK;YAAE,SAAS,CAAC,KAAK,GAAG,KAAK,CAAC;QACnC,IAAI,MAAM,CAAC,KAAK;YAAE,SAAS,CAAC,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,MAAM,CAAC,WAAW;YAAE,SAAS,CAAC,WAAW,GAAG,gBAAgB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;QACrF,IAAI,MAAM,CAAC,eAAe;YACxB,SAAS,CAAC,eAAe,GAAG,gBAAgB,CAAC,MAAM,CAAC,eAAe,CAAC,CAAC;QAEvE,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,KAAK,MAAM,GAAG,IAAI;YAChB,aAAa;YACb,YAAY;YACZ,uBAAuB;YACvB,OAAO;YACP,mBAAmB;YACnB,kBAAkB;YAClB,MAAM;YACN,MAAM;YACN,GAAG;SACJ,EAAE,CAAC;YACF,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS;gBAAE,WAAW,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,KAAK,IAAI,QAAQ;YACvB,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,KAAK,EAAE,SAAS;YAChB,KAAK;YACL,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAC/E,CAAC,CAAC;QAEH,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;IAC3B,CAAC;IAED,8CAA8C;IACtC,mBAAmB,CAAC,MAA+B;QAIzD,MAAM,KAAK,GAAG,UAAU,EAAE,CAAC;QAC3B,MAAM,KAAK,GAAG,MAAM,CAAC,KAA2B,CAAC;QACjD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEtC,MAAM,SAAS,GAA4B;YACzC,KAAK,EAAE,gBAAgB,CAAC,MAAM,CAAC,KAAK,IAAI,EAAE,CAAC;SAC5C,CAAC;QACF,IAAI,KAAK;YAAE,SAAS,CAAC,KAAK,GAAG,KAAK,CAAC;QACnC,IAAI,MAAM,CAAC,KAAK;YAAE,SAAS,CAAC,KAAK,GAAG,gBAAgB,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QACnE,IAAI,MAAM,CAAC,YAAY;YAAE,SAAS,CAAC,YAAY,GAAG,MAAM,CAAC,YAAY,CAAC;QAEtE,MAAM,WAAW,GAA4B,EAAE,CAAC;QAChD,KAAK,MAAM,GAAG,IAAI;YAChB,aAAa;YACb,mBAAmB;YACnB,OAAO;YACP,mBAAmB;YACnB,kBAAkB;YAClB,MAAM;SACP,EAAE,CAAC;YACF,IAAI,MAAM,CAAC,GAAG,CAAC,KAAK,SAAS;gBAAE,WAAW,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAChE,CAAC;QAED,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,SAAS,EAAE;YACjC,IAAI,EAAE,YAAY;YAClB,IAAI,EAAE,KAAK,IAAI,kBAAkB;YACjC,SAAS,EAAE,IAAI,IAAI,EAAE;YACrB,KAAK,EAAE,SAAS;YAChB,KAAK;YACL,eAAe,EAAE,MAAM,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS;SAC/E,CAAC,CAAC;QAEH,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC;IAC3B,CAAC;CACF;AAED,8EAA8E;AAE9E,SAAS,iBAAiB,CAAC,QAAiB;IAC1C,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC/D,8DAA8D;IAC9D,MAAM,CAAC,GAAG,QAAe,CAAC;IAC1B,MAAM,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;IAC1B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IAEvF,MAAM,OAAO,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,CAAC;IACpC,IAAI,CAAC,OAAO;QAAE,OAAO,gBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC;IAElD,MAAM,MAAM,GAA4B;QACtC,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,WAAW;QACjC,OAAO,EAAE,OAAO,CAAC,OAAO,IAAI,IAAI;KACjC,CAAC;IAEF,IAAI,OAAO,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;QAC/B,MAAM,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAA2B,EAAE,EAAE,CAAC,CAAC;YAC3E,EAAE,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE;YACf,IAAI,EAAE,EAAE,CAAC,IAAI,IAAI,UAAU;YAC3B,QAAQ,EAAE;gBACR,IAAI,EAAG,EAAE,CAAC,QAAoC,EAAE,IAAI,IAAI,EAAE;gBAC1D,SAAS,EAAG,EAAE,CAAC,QAAoC,EAAE,SAAS,IAAI,EAAE;aACrE;SACF,CAAC,CAAC,CAAC;IACN,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,CAAC,CAAC,EAAE,aAAa,CAAC;IAC/C,IAAI,YAAY;QAAE,MAAM,CAAC,aAAa,GAAG,YAAY,CAAC;IAEtD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,sBAAsB,CAAC,QAAiB;IAC/C,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,QAAQ,CAAC;IAC/D,8DAA8D;IAC9D,MAAM,CAAC,GAAG,QAAe,CAAC;IAC1B,IAAI,CAAC,CAAC,MAAM,KAAK,SAAS;QAAE,OAAO,gBAAgB,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;IAC9D,OAAO,gBAAgB,CAAC,QAAQ,CAAC,CAAC;AACpC,CAAC;AAED,SAAS,YAAY,CAAC,QAAiB;IACrC,IAAI,CAAC,QAAQ,IAAI,OAAO,QAAQ,KAAK,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC3D,8DAA8D;IAC9D,MAAM,KAAK,GAAI,QAAgB,CAAC,KAAK,CAAC;IACtC,IAAI,CAAC,KAAK;QAAE,OAAO,IAAI,CAAC;IAExB,MAAM,GAAG,GAA4B,EAAE,CAAC;IACxC,uBAAuB;IACvB,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,aAAa,CAAC;IAC9E,IAAI,KAAK,CAAC,iBAAiB,KAAK,SAAS;QAAE,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC,iBAAiB,CAAC;IAC1F,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;QAAE,GAAG,CAAC,WAAW,GAAG,KAAK,CAAC,YAAY,CAAC;IAC3E,gBAAgB;IAChB,IAAI,KAAK,CAAC,YAAY,KAAK,SAAS;QAAE,GAAG,CAAC,YAAY,GAAG,KAAK,CAAC,YAAY,CAAC;IAC5E,IAAI,KAAK,CAAC,aAAa,KAAK,SAAS;QAAE,GAAG,CAAC,gBAAgB,GAAG,KAAK,CAAC,aAAa,CAAC;IAClF,aAAa;IACb,IACE,GAAG,CAAC,WAAW,KAAK,SAAS;QAC7B,GAAG,CAAC,YAAY,KAAK,SAAS;QAC9B,GAAG,CAAC,gBAAgB,KAAK,SAAS,EAClC,CAAC;QACD,GAAG,CAAC,WAAW,GAAI,GAAG,CAAC,YAAuB,GAAI,GAAG,CAAC,gBAA2B,CAAC;IACpF,CAAC;IAED,OAAO,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,CAAC;AAClD,CAAC"}

package/dist/trace.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"trace.d.ts","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,KAAK,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAalE,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,IAAI,GAAG,IAAI,CAE7E;AAED,wBAAgB,gBAAgB,IAAI,qBAAqB,GAAG,IAAI,CAE/D;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE;IAAE,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAE1F;AAED,6DAA6D;AAC7D,wBAAgB,gBAAgB,IAAI;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,GAAG,SAAS,CAQhG;AAED,wBAAgB,gBAAgB,IAAI,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAEjE;AAiDD;;;;;;GAMG;AACH,wBAAgB,KAAK,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,EACrD,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,CAAC,GAAG,YAAY,EAC7B,OAAO,CAAC,EAAE,CAAC,GACV,CAAC,CAkIH"}
+{"version":3,"file":"trace.d.ts","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAMA,OAAO,KAAK,EAAE,qBAAqB,EAAE,MAAM,oBAAoB,CAAC;AAEhE,OAAO,KAAK,EAAE,YAAY,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAalE,wBAAgB,gBAAgB,CAAC,QAAQ,EAAE,qBAAqB,GAAG,IAAI,GAAG,IAAI,CAE7E;AAED,wBAAgB,gBAAgB,IAAI,qBAAqB,GAAG,IAAI,CAE/D;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE;IAAE,MAAM,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GAAG,IAAI,CAE1F;AAED,6DAA6D;AAC7D,wBAAgB,gBAAgB,IAAI;IAAE,OAAO,EAAE,MAAM,CAAC;IAAC,aAAa,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,GAAG,SAAS,CAQhG;AAED,wBAAgB,gBAAgB,IAAI,GAAG,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAEjE;AAiDD;;;;;;GAMG;AACH,wBAAgB,KAAK,CAAC,CAAC,SAAS,CAAC,GAAG,IAAI,EAAE,GAAG,EAAE,KAAK,GAAG,EACrD,IAAI,EAAE,MAAM,EACZ,WAAW,EAAE,CAAC,GAAG,YAAY,EAC7B,OAAO,CAAC,EAAE,CAAC,GACV,CAAC,CAmIH"}

package/dist/trace.js

@@ -93,12 +93,13 @@ export function trace(name, fnOrOptions, maybeFn) {
     // Resolve userId/sessionId: per-trace option > client default
     const userId = options.userId ?? _clientDefaults.userId;
     const sessionId = options.sessionId ?? _clientDefaults.sessionId;
-    // Register tool for remote invocation
+    // Register tool for local execution + dev server invocation
     if (obsType === "tool" && invoke) {
         const inputSchema = options.inputSchema ? zodToJsonSchema(options.inputSchema) : null;
         _toolRegistry.set(obsName, {
             fn: fn,
             inputSchema,
+            description: null,
         });
     }
     const wrapped = ((...args) => {

package/dist/trace.js.map

@@ -1 +1 @@
-{"version":3,"file":"trace.js","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAW,KAAK,IAAI,SAAS,EAAa,MAAM,oBAAoB,CAAC;AAE5E,OAAO,KAAK,KAAK,MAAM,oBAAoB,CAAC;AAE5C,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AACnD,OAAO,EAAc,gBAAgB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE3E,sDAAsD;AACtD,IAAI,aAAa,GAAiC,IAAI,CAAC;AAEvD,uEAAuE;AACvE,IAAI,eAAe,GAA4C,EAAE,CAAC;AAElE,gDAAgD;AAChD,MAAM,aAAa,GAAG,IAAI,GAAG,EAA6B,CAAC;AAE3D,MAAM,UAAU,gBAAgB,CAAC,QAAsC;IACrE,aAAa,GAAG,QAAQ,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,QAAiD;IAClF,eAAe,GAAG,QAAQ,CAAC;AAC7B,CAAC;AAED,6DAA6D;AAC7D,MAAM,UAAU,gBAAgB;IAC9B,MAAM,IAAI,GAAG,SAAS,CAAC,aAAa,EAAE,CAAC;IACvC,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAC/B,OAAO;QACL,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,aAAa,EAAE,GAAG,CAAC,MAAM;KAC1B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,2EAA2E;AAC3E,SAAS,iBAAiB,CACxB,IAAU,EACV,IAiBC;IAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACnE,IAAI,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1F,IAAI,IAAI,CAAC,MAAM;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACrE,IAAI,IAAI,CAAC,SAAS;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAChF,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,YAAY,CACf,KAAK,CAAC,gBAAgB,EACtB,qBAAqB,CAAC,IAAI,CAAC,OAAQ,CAAC,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,CAC/D,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACvE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACzE,IAAI,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAChG,IAAI,IAAI,CAAC,KAAK;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,IAAI,CAAC,aAAa;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,0BAA0B,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAChG,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,KAAK,CACnB,IAAY,EACZ,WAA6B,EAC7B,OAAW;IAEX,IAAI,OAAqB,CAAC;IAC1B,IAAI,EAAK,CAAC;IAEV,IAAI,OAAO,WAAW,KAAK,UAAU,EAAE,CAAC;QACtC,OAAO,GAAG,EAAE,CAAC;QACb,EAAE,GAAG,WAAW,CAAC;IACnB,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,WAAW,CAAC;QACtB,EAAE,GAAG,OAAQ,CAAC;IAChB,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;IACrC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC;IACpC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;IAChD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,eAAe;IACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC;IAEpC,8DAA8D;IAC9D,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC;IACxD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,eAAe,CAAC,SAAS,CAAC;IAEjE,sCAAsC;IACtC,IAAI,OAAO,KAAK,MAAM,IAAI,MAAM,EAAE,CAAC;QACjC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACtF,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE;YACzB,EAAE,EAAE,EAAqC;YACzC,WAAW;SACZ,CAAC,CAAC;IACL,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,aAAa,EAAE,MAAM,CAAC;QACrC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QAEhC,MAAM,MAAM,GAAG,OAAO,KAAK,IAAI,CAAC;QAEhC,OAAO,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YAC9C,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;oBAC3E,MAAM,MAAM,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;oBAE3B,yBAAyB;oBACzB,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC9E,OAAQ,MAA2B,CAAC,IAAI,CACtC,CAAC,QAAQ,EAAE,EAAE;4BACX,iBAAiB,CAAC,IAAI,EAAE;gCACtB,MAAM;gCACN,OAAO;gCACP,OAAO;gCACP,KAAK,EAAE,aAAa;gCACpB,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC;gCAClC,KAAK;gCACL,QAAQ,EAAE,cAAc;gCACxB,KAAK,EAAE,SAAS;gCAChB,MAAM;gCACN,SAAS;gCACT,KAAK;6BACN,CAAC,CAAC;4BACH,IAAI,CAAC,GAAG,EAAE,CAAC;4BACX,OAAO,QAAQ,CAAC;wBAClB,CAAC,EACD,CAAC,GAAU,EAAE,EAAE;4BACb,iBAAiB,CAAC,IAAI,EAAE;gCACtB,MAAM;gCACN,OAAO;gCACP,OAAO;gCACP,KAAK,EAAE,aAAa;gCACpB,MAAM,EAAE,IAAI;gCACZ,KAAK;gCACL,QAAQ,EAAE,cAAc;gCACxB,KAAK,EAAE,OAAO;gCACd,aAAa,EAAE,GAAG,CAAC,OAAO;gCAC1B,MAAM;gCACN,SAAS;gCACT,KAAK;6BACN,CAAC,CAAC;4BACH,IAAI,CAAC,GAAG,EAAE,CAAC;4BACX,MAAM,GAAG,CAAC;wBACZ,CAAC,CACF,CAAC;oBACJ,CAAC;oBAED,cAAc;oBACd,iBAAiB,CAAC,IAAI,EAAE;wBACtB,MAAM;wBACN,OAAO;wBACP,OAAO;wBACP,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC;wBAChC,KAAK;wBACL,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE,SAAS;wBAChB,MAAM;wBACN,SAAS;wBACT,KAAK;qBACN,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBACjE,iBAAiB,CAAC,IAAI,EAAE;wBACtB,MAAM;wBACN,OAAO;wBACP,OAAO;wBACP,KAAK,EAAE,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;wBAC3D,MAAM,EAAE,IAAI;wBACZ,KAAK;wBACL,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE,OAAO;wBACd,aAAa,EAAE,OAAO;wBACtB,MAAM;wBACN,SAAS;wBACT,KAAK;qBACN,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;YAEF,OAAO,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC,CAAiB,CAAC;IAEnB,yBAAyB;IACzB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;IAEnE,OAAO,OAAO,CAAC;AACjB,CAAC"}
+{"version":3,"file":"trace.js","sourceRoot":"","sources":["../src/trace.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AACH,OAAO,EAAW,KAAK,IAAI,SAAS,EAAa,MAAM,oBAAoB,CAAC;AAE5E,OAAO,KAAK,KAAK,MAAM,oBAAoB,CAAC;AAE5C,OAAO,EAAE,qBAAqB,EAAE,MAAM,YAAY,CAAC;AACnD,OAAO,EAAc,gBAAgB,EAAE,eAAe,EAAE,MAAM,YAAY,CAAC;AAE3E,sDAAsD;AACtD,IAAI,aAAa,GAAiC,IAAI,CAAC;AAEvD,uEAAuE;AACvE,IAAI,eAAe,GAA4C,EAAE,CAAC;AAElE,gDAAgD;AAChD,MAAM,aAAa,GAAG,IAAI,GAAG,EAA6B,CAAC;AAE3D,MAAM,UAAU,gBAAgB,CAAC,QAAsC;IACrE,aAAa,GAAG,QAAQ,CAAC;AAC3B,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,MAAM,UAAU,kBAAkB,CAAC,QAAiD;IAClF,eAAe,GAAG,QAAQ,CAAC;AAC7B,CAAC;AAED,6DAA6D;AAC7D,MAAM,UAAU,gBAAgB;IAC9B,MAAM,IAAI,GAAG,SAAS,CAAC,aAAa,EAAE,CAAC;IACvC,IAAI,CAAC,IAAI;QAAE,OAAO,SAAS,CAAC;IAC5B,MAAM,GAAG,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IAC/B,OAAO;QACL,OAAO,EAAE,GAAG,CAAC,OAAO;QACpB,aAAa,EAAE,GAAG,CAAC,MAAM;KAC1B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB;IAC9B,OAAO,aAAa,CAAC;AACvB,CAAC;AAED,2EAA2E;AAC3E,SAAS,iBAAiB,CACxB,IAAU,EACV,IAiBC;IAED,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;QAChB,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QACzC,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,WAAW,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACjE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,YAAY,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACnE,IAAI,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,cAAc,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC1F,IAAI,IAAI,CAAC,MAAM;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QACrE,IAAI,IAAI,CAAC,SAAS;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,gBAAgB,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAChF,CAAC;SAAM,CAAC;QACN,IAAI,CAAC,YAAY,CACf,KAAK,CAAC,gBAAgB,EACtB,qBAAqB,CAAC,IAAI,CAAC,OAAQ,CAAC,IAAI,IAAI,CAAC,OAAO,IAAI,MAAM,CAC/D,CAAC;QACF,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACvE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,kBAAkB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC;QACzE,IAAI,IAAI,CAAC,QAAQ;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,oBAAoB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;QAChG,IAAI,IAAI,CAAC,KAAK;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACvE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,iBAAiB,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,IAAI,CAAC,aAAa;YAAE,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,0BAA0B,EAAE,IAAI,CAAC,aAAa,CAAC,CAAC;QAChG,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;YACf,IAAI,CAAC,YAAY,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QACjF,CAAC;IACH,CAAC;AACH,CAAC;AAED;;;;;;GAMG;AACH,MAAM,UAAU,KAAK,CACnB,IAAY,EACZ,WAA6B,EAC7B,OAAW;IAEX,IAAI,OAAqB,CAAC;IAC1B,IAAI,EAAK,CAAC;IAEV,IAAI,OAAO,WAAW,KAAK,UAAU,EAAE,CAAC;QACtC,OAAO,GAAG,EAAE,CAAC;QACb,EAAE,GAAG,WAAW,CAAC;IACnB,CAAC;SAAM,CAAC;QACN,OAAO,GAAG,WAAW,CAAC;QACtB,EAAE,GAAG,OAAQ,CAAC;IAChB,CAAC;IAED,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;IACrC,MAAM,OAAO,GAAG,OAAO,CAAC,IAAI,IAAI,IAAI,CAAC;IACrC,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC;IACpC,MAAM,cAAc,GAAG,OAAO,CAAC,QAAQ,IAAI,IAAI,CAAC;IAChD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,eAAe;IACxD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,IAAI,IAAI,CAAC;IAEpC,8DAA8D;IAC9D,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,IAAI,eAAe,CAAC,MAAM,CAAC;IACxD,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,eAAe,CAAC,SAAS,CAAC;IAEjE,4DAA4D;IAC5D,IAAI,OAAO,KAAK,MAAM,IAAI,MAAM,EAAE,CAAC;QACjC,MAAM,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;QACtF,aAAa,CAAC,GAAG,CAAC,OAAO,EAAE;YACzB,EAAE,EAAE,EAAqC;YACzC,WAAW;YACX,WAAW,EAAE,IAAI;SAClB,CAAC,CAAC;IACL,CAAC;IAED,MAAM,OAAO,GAAG,CAAC,CAAC,GAAG,IAAe,EAAE,EAAE;QACtC,MAAM,MAAM,GAAG,aAAa,EAAE,MAAM,CAAC;QACrC,IAAI,CAAC,MAAM;YAAE,OAAO,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;QAEhC,MAAM,MAAM,GAAG,OAAO,KAAK,IAAI,CAAC;QAEhC,OAAO,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YAC9C,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,IAAI,CAAC;oBACH,MAAM,aAAa,GAAG,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;oBAC3E,MAAM,MAAM,GAAG,EAAE,CAAC,GAAG,IAAI,CAAC,CAAC;oBAE3B,yBAAyB;oBACzB,IAAI,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ,IAAI,OAAO,MAAM,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC9E,OAAQ,MAA2B,CAAC,IAAI,CACtC,CAAC,QAAQ,EAAE,EAAE;4BACX,iBAAiB,CAAC,IAAI,EAAE;gCACtB,MAAM;gCACN,OAAO;gCACP,OAAO;gCACP,KAAK,EAAE,aAAa;gCACpB,MAAM,EAAE,gBAAgB,CAAC,QAAQ,CAAC;gCAClC,KAAK;gCACL,QAAQ,EAAE,cAAc;gCACxB,KAAK,EAAE,SAAS;gCAChB,MAAM;gCACN,SAAS;gCACT,KAAK;6BACN,CAAC,CAAC;4BACH,IAAI,CAAC,GAAG,EAAE,CAAC;4BACX,OAAO,QAAQ,CAAC;wBAClB,CAAC,EACD,CAAC,GAAU,EAAE,EAAE;4BACb,iBAAiB,CAAC,IAAI,EAAE;gCACtB,MAAM;gCACN,OAAO;gCACP,OAAO;gCACP,KAAK,EAAE,aAAa;gCACpB,MAAM,EAAE,IAAI;gCACZ,KAAK;gCACL,QAAQ,EAAE,cAAc;gCACxB,KAAK,EAAE,OAAO;gCACd,aAAa,EAAE,GAAG,CAAC,OAAO;gCAC1B,MAAM;gCACN,SAAS;gCACT,KAAK;6BACN,CAAC,CAAC;4BACH,IAAI,CAAC,GAAG,EAAE,CAAC;4BACX,MAAM,GAAG,CAAC;wBACZ,CAAC,CACF,CAAC;oBACJ,CAAC;oBAED,cAAc;oBACd,iBAAiB,CAAC,IAAI,EAAE;wBACtB,MAAM;wBACN,OAAO;wBACP,OAAO;wBACP,KAAK,EAAE,aAAa;wBACpB,MAAM,EAAE,gBAAgB,CAAC,MAAM,CAAC;wBAChC,KAAK;wBACL,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE,SAAS;wBAChB,MAAM;wBACN,SAAS;wBACT,KAAK;qBACN,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,OAAO,MAAM,CAAC;gBAChB,CAAC;gBAAC,OAAO,GAAG,EAAE,CAAC;oBACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;oBACjE,iBAAiB,CAAC,IAAI,EAAE;wBACtB,MAAM;wBACN,OAAO;wBACP,OAAO;wBACP,KAAK,EAAE,gBAAgB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;wBAC3D,MAAM,EAAE,IAAI;wBACZ,KAAK;wBACL,QAAQ,EAAE,cAAc;wBACxB,KAAK,EAAE,OAAO;wBACd,aAAa,EAAE,OAAO;wBACtB,MAAM;wBACN,SAAS;wBACT,KAAK;qBACN,CAAC,CAAC;oBACH,IAAI,CAAC,GAAG,EAAE,CAAC;oBACX,MAAM,GAAG,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;YAEF,OAAO,OAAO,EAAE,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC,CAAiB,CAAC;IAEnB,yBAAyB;IACzB,MAAM,CAAC,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,IAAI,IAAI,EAAE,CAAC,CAAC;IAEnE,OAAO,OAAO,CAAC;AACjB,CAAC"}

package/dist/types.d.ts

@@ -16,7 +16,7 @@ export interface TraceOptions {
     type?: ObservationType;
     /** Override observation name (defaults to the name argument). */
     name?: string;
-    /** For type="tool": register for remote invocation. Default: true. */
+    /** For type="tool": expose via dev server for dashboard invocation. Default: true. */
     invoke?: boolean;
     /** For type="generation": LLM model name. */
     model?: string;
@@ -44,43 +44,6 @@ export interface TraceEvent {
 export interface ToolRegistryEntry {
     fn: (...args: unknown[]) => unknown;
     inputSchema: Record<string, unknown> | null;
+    description?: string | null;
 }
-/** WebSocket messages from SDK to backend. */
-export type SdkMessage = {
-    type: "register";
-    sdkInstanceId: string;
-    tools: Array<{
-        name: string;
-        inputSchema: unknown;
-    }>;
-} | {
-    type: "result";
-    nonce: string;
-    output: unknown;
-    error?: string;
-    durationMs: number;
-    signature: string;
-} | {
-    type: "heartbeat";
-};
-/** WebSocket messages from backend to SDK. */
-export type ServerMessage = {
-    type: "connected";
-    sessionToken: string;
-} | {
-    type: "registered";
-    tools: string[];
-} | {
-    type: "invoke";
-    nonce: string;
-    tool: string;
-    input: unknown;
-    state?: unknown;
-    signature: string;
-} | {
-    type: "heartbeat_ack";
-} | {
-    type: "error";
-    message: string;
-};
 //# sourceMappingURL=types.d.ts.map

package/dist/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,gDAAgD;AAChD,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,YAAY,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,CAAC;AAEjF,8EAA8E;AAC9E,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAMnF,CAAC;AAEF,wDAAwD;AACxD,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAMxD,CAAC;AAEF,yDAAyD;AACzD,MAAM,WAAW,YAAY;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,uCAAuC;AACvC,MAAM,WAAW,YAAY;IAC3B,gDAAgD;IAChD,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,iEAAiE;IACjE,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sEAAsE;IACtE,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,6CAA6C;IAC7C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,0EAA0E;IAC1E,WAAW,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IACjC,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,0CAA0C;IAC1C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,+CAA+C;IAC/C,KAAK,CAAC,EAAE,YAAY,CAAC;CACtB;AAED,uEAAuE;AACvE,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,qDAAqD;AACrD,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC;IACpC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC7C;AAED,8CAA8C;AAC9C,MAAM,MAAM,UAAU,GAClB;IACE,IAAI,EAAE,UAAU,CAAC;IACjB,aAAa,EAAE,MAAM,CAAC;IACtB,KAAK,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,WAAW,EAAE,OAAO,CAAA;KAAE,CAAC,CAAC;CACtD,GACD;IACE,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,OAAO,CAAC;IAChB,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,UAAU,EAAE,MAAM,CAAC;IACnB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,WAAW,CAAA;CAAE,CAAC;AAE1B,8CAA8C;AAC9C,MAAM,MAAM,aAAa,GACrB;IAAE,IAAI,EAAE,WAAW,CAAC;IAAC,YAAY,EAAE,MAAM,CAAA;CAAE,GAC3C;IAAE,IAAI,EAAE,YAAY,CAAC;IAAC,KAAK,EAAE,MAAM,EAAE,CAAA;CAAE,GACvC;IACE,IAAI,EAAE,QAAQ,CAAC;IACf,KAAK,EAAE,MAAM,CAAC;IACd,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,OAAO,CAAC;IACf,KAAK,CAAC,EAAE,OAAO,CAAC;IAChB,SAAS,EAAE,MAAM,CAAC;CACnB,GACD;IAAE,IAAI,EAAE,eAAe,CAAA;CAAE,GACzB;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,OAAO,EAAE,MAAM,CAAA;CAAE,CAAC"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,gDAAgD;AAChD,MAAM,MAAM,eAAe,GAAG,MAAM,GAAG,YAAY,GAAG,OAAO,GAAG,MAAM,GAAG,OAAO,CAAC;AAEjF,8EAA8E;AAC9E,eAAO,MAAM,cAAc,EAAE,MAAM,CAAC,MAAM,EAAE,CAAC,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,CAMnF,CAAC;AAEF,wDAAwD;AACxD,eAAO,MAAM,qBAAqB,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAMxD,CAAC;AAEF,yDAAyD;AACzD,MAAM,WAAW,YAAY;IAC3B,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAC1B,WAAW,CAAC,EAAE,MAAM,CAAC;CACtB;AAED,uCAAuC;AACvC,MAAM,WAAW,YAAY;IAC3B,gDAAgD;IAChD,IAAI,CAAC,EAAE,eAAe,CAAC;IACvB,iEAAiE;IACjE,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,sFAAsF;IACtF,MAAM,CAAC,EAAE,OAAO,CAAC;IACjB,6CAA6C;IAC7C,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,0EAA0E;IAC1E,WAAW,CAAC,EAAE;QAAE,IAAI,CAAC,EAAE,OAAO,CAAA;KAAE,CAAC;IACjC,8CAA8C;IAC9C,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACnC,0CAA0C;IAC1C,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,6CAA6C;IAC7C,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,+CAA+C;IAC/C,KAAK,CAAC,EAAE,YAAY,CAAC;CACtB;AAED,uEAAuE;AACvE,MAAM,WAAW,UAAU;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;IAClB,IAAI,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;CAC/B;AAED,qDAAqD;AACrD,MAAM,WAAW,iBAAiB;IAChC,EAAE,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC;IACpC,WAAW,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;IAC5C,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;CAC7B"}