@lmnr-ai/lmnr 0.8.48 → 0.8.50

package/dist/index.cjs

@@ -1,6 +1,6 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_rolldown_runtime = require("./rolldown-runtime-CVvi-lCc.cjs");
-const require_utils = require("./utils-BNApHb06.cjs");
+const require_rolldown_runtime = require("./rolldown-runtime-Ctb5zv51.cjs");
+const require_utils = require("./utils-CNR1tzh5.cjs");
 const require_consts = require("./consts-9rSS0oeR.cjs");
 let path = require("path");
 path = require_rolldown_runtime.__toESM(path);
@@ -183,7 +183,7 @@ var LaminarDataset = class extends EvaluationDataset {
 	*/
 	async push(paths, recursive = false) {
 		if (!this.client) throw new Error("Client not set");
-		const { loadFromPaths } = await Promise.resolve().then(() => require("./file-utils-BvYQiFYO.cjs")).then((n) => n.file_utils_exports);
+		const { loadFromPaths } = await Promise.resolve().then(() => require("./file-utils-RXJzJyiI.cjs")).then((n) => n.file_utils_exports);
 		const data = await loadFromPaths(Array.isArray(paths) ? paths : [paths], recursive);
 		if (data.length === 0) {
 			console.warn("No data to push. Skipping");
@@ -569,7 +569,8 @@ const handleStreamResult = (originalResult, streamInfo, span, ignoreOutput, seri
 		});
 		trackStreamProcessing(async () => {
 			try {
-				const { chunks, error } = await consumeResponse(new Response(stream2));
+				const tempResponse = new Response(stream2);
+				const { chunks, error } = await consumeResponse(tempResponse);
 				if (shouldSendTraces() && !ignoreOutput) try {
 					const output = {
 						type: "response",
@@ -668,7 +669,7 @@ const writeDebugSessionFile = (file, dir = process.cwd()) => {
 * logic line-comparable with the Python `config.py`.
 */
 const logger$24 = require_utils.initializeLogger();
-const TRUTHY = new Set([
+const TRUTHY = /* @__PURE__ */ new Set([
 	"true",
 	"1",
 	"yes",
@@ -740,10 +741,12 @@ const buildDebugConfig = () => {
 	const providedSessionId = process.env.LMNR_DEBUG_SESSION_ID || existing?.session_id || void 0;
 	const sessionId = providedSessionId || (0, node_crypto.randomUUID)();
 	const sessionMinted = providedSessionId === void 0;
+	const replayTraceId = process.env.LMNR_DEBUG_REPLAY_TRACE_ID || existing?.replay_trace_id || null;
+	const cacheUntilValue = process.env.LMNR_DEBUG_CACHE_UNTIL ?? existing?.cache_until ?? void 0;
 	return {
 		sessionId,
-		replayTraceId: process.env.LMNR_DEBUG_REPLAY_TRACE_ID || existing?.replay_trace_id || null,
-		cacheUntilSpanId: parseCacheUntil(process.env.LMNR_DEBUG_CACHE_UNTIL ?? existing?.cache_until ?? void 0),
+		replayTraceId,
+		cacheUntilSpanId: parseCacheUntil(cacheUntilValue),
 		localOrigin: true,
 		sessionMinted
 	};
@@ -964,14 +967,15 @@ var DebugRuntime = class {
 		if (!this._config.localOrigin) return;
 		if (this._emitted) return;
 		this._emitted = true;
-		emitPointer(buildDebugSessionFile({
+		const file = buildDebugSessionFile({
 			sessionId: this._config.sessionId,
 			traceId: this._traceId,
 			replayTraceId: this._config.replayTraceId,
 			cacheUntil: this._config.cacheUntilSpanId,
 			debuggerUrl: this.debuggerSessionUrl(),
 			startedAt: this._startedAt
-		}));
+		});
+		emitPointer(file);
 	}
 };
 let runtime = null;
@@ -1180,7 +1184,7 @@ var LaminarSpan = class LaminarSpan {
 	}
 	addTags(tags) {
 		const currentTags = this.tags;
-		this.setTags(Array.from(new Set([...currentTags, ...tags])));
+		this.setTags(Array.from(/* @__PURE__ */ new Set([...currentTags, ...tags])));
 	}
 	getLaminarSpanContext() {
 		let spanPath = [];
@@ -1568,7 +1572,8 @@ const flagSettingsEnv = (existing, cwd) => {
 		} catch {}
 		return {};
 	}
-	return settingsEnvBlock(loadSettingsFile(path.isAbsolute(trimmed) ? trimmed : path.join(cwd ?? process.cwd(), trimmed)) ?? {});
+	const resolved = path.isAbsolute(trimmed) ? trimmed : path.join(cwd ?? process.cwd(), trimmed);
+	return settingsEnvBlock(loadSettingsFile(resolved) ?? {});
 };
 /**
 * Build the `--settings` value that forces the CLI through our proxy.
@@ -1602,7 +1607,8 @@ const buildProxyFlagSettings = (existing, proxyUrl, cwd, settingSources) => {
 			if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) return null;
 			settingsObj = parsed;
 		} else {
-			const loaded = loadSettingsFile(path.isAbsolute(trimmed) ? trimmed : path.join(cwd ?? process.cwd(), trimmed));
+			const resolved = path.isAbsolute(trimmed) ? trimmed : path.join(cwd ?? process.cwd(), trimmed);
+			const loaded = loadSettingsFile(resolved);
 			if (loaded === null) return null;
 			settingsObj = loaded;
 		}
@@ -1693,7 +1699,10 @@ const resolveTargetUrlFromEnv = (envDict, fallback = DEFAULT_ANTHROPIC_BASE_URL,
 		const bedrockBaseUrl = getEnvValue(BEDROCK_BASE_URL_ENV);
 		if (bedrockBaseUrl) return bedrockBaseUrl.replace(/\/$/, "");
 		let region = getEnvValue(BEDROCK_AWS_REGION_ENV);
-		if (!region) region = getRegionFromAwsConfig(getEnvValue("AWS_PROFILE") || "default") ?? void 0;
+		if (!region) {
+			const awsProfile = getEnvValue("AWS_PROFILE") || "default";
+			region = getRegionFromAwsConfig(awsProfile) ?? void 0;
+		}
 		if (region) return `https://bedrock-runtime.${region}.amazonaws.com`;
 		logger$22.error(`${BEDROCK_USE_ENV} is set but could not determine AWS region. Set ${BEDROCK_AWS_REGION_ENV} or configure a region in ~/.aws/config for the active profile.`);
 		return null;
@@ -2232,10 +2241,13 @@ var Laminar = class Laminar {
 					runtime.recordProjectId(projectId);
 					const sessionUrl = runtime.debuggerSessionUrl();
 					logger$20.info(`Laminar debugger session: ${sessionUrl}`);
-					if (runtime.shouldOpenBrowser) (0, node_child_process.spawn)(process.platform === "win32" ? "start" : process.platform === "darwin" ? "open" : "xdg-open", [sessionUrl], {
-						detached: true,
-						stdio: "ignore"
-					}).unref();
+					if (runtime.shouldOpenBrowser) {
+						const opener = process.platform === "win32" ? "start" : process.platform === "darwin" ? "open" : "xdg-open";
+						(0, node_child_process.spawn)(opener, [sessionUrl], {
+							detached: true,
+							stdio: "ignore"
+						}).unref();
+					}
 				}
 			}).catch((e) => {
 				logger$20.warn("Failed to register debug session: " + require_utils.errorMessage(e));
@@ -2282,11 +2294,13 @@ var Laminar = class Laminar {
 			if (!debug || debug.enabled !== true || !debug.sessionId) return;
 			const existing = getRuntime();
 			if (existing !== null && existing.localOrigin) return;
-			const { runtime, configChanged } = initDebugRuntimeFromContext(debug, existing?.rolloutSessions ?? new require_utils.LaminarClient({
+			const rolloutSessions = existing?.rolloutSessions ?? new require_utils.LaminarClient({
 				baseUrl: this.baseUrlForDebug,
 				projectApiKey: this.projectApiKey,
 				port: this.httpPortForDebug
-			}).rolloutSessions, process?.env?.LMNR_FRONTEND_URL ?? require_utils.getFrontendUrl(this.baseUrlForDebug));
+			}).rolloutSessions;
+			const debuggerUrl = process?.env?.LMNR_FRONTEND_URL ?? require_utils.getFrontendUrl(this.baseUrlForDebug);
+			const { runtime, configChanged } = initDebugRuntimeFromContext(debug, rolloutSessions, debuggerUrl);
 			if (runtime === null) return;
 			if (configChanged) {
 				this.debugRunLive = false;
@@ -2590,12 +2604,13 @@ var Laminar = class Laminar {
 		const sessionIdProperties = sessionIdValue ? { [require_utils.SESSION_ID]: sessionIdValue } : {};
 		const traceTypeProperties = traceTypeValue ? { [require_utils.TRACE_TYPE]: traceTypeValue } : {};
 		const tracingLevelProperties = tracingLevelValue ? { [`${require_utils.ASSOCIATION_PROPERTIES}.tracing_level`]: tracingLevelValue } : {};
-		const metadataProperties = require_utils.metadataToAttributes({
+		const mergedMetadata = {
 			...this.globalMetadata,
 			...ctxMetadata,
 			...parentMetadata,
 			...metadata
-		});
+		};
+		const metadataProperties = require_utils.metadataToAttributes(mergedMetadata);
 		const attributes = {
 			[require_utils.SPAN_TYPE]: spanType ?? "DEFAULT",
 			...tagProperties,
@@ -2959,7 +2974,8 @@ const prettyPrintZodSchema = (schema, indent = 2) => {
 			if (typeof description === "string") result += `.describe('${description.replace(/'/g, "\\'")}')`;
 			return result;
 		};
-		return `${indentString}${key}: ${applyModifiers(value, getBaseType(value))},`;
+		const finalType = applyModifiers(value, getBaseType(value));
+		return `${indentString}${key}: ${finalType},`;
 	}).join("\n")}\n})`;
 };
 const modelToProviderMap = {
@@ -3015,7 +3031,7 @@ const modelToProvider = (model) => {
 const injectScript = (sessionRecordingOptions, stringifyCallbackArgs) => {
 	const BATCH_TIMEOUT = 2e3;
 	const HEARTBEAT_INTERVAL = 1e3;
-	const CHUNK_SIZE = 256 * 1024;
+	const CHUNK_SIZE = 262144;
 	window.lmnrRrwebEventsBatch = [];
 	window.lmnrChunkSequence = 0;
 	window.lmnrSendInProgress = false;
@@ -3076,7 +3092,9 @@ const injectScript = (sessionRecordingOptions, stringifyCallbackArgs) => {
 		try {
 			const compressedEvents = [];
 			for (const event of events) try {
-				const base64Data = await bufferToBase64(await gzipCompress(JSON.stringify(event.data)));
+				const dataString = JSON.stringify(event.data);
+				const compressed = await gzipCompress(dataString);
+				const base64Data = await bufferToBase64(compressed);
 				compressedEvents.push({
 					...event,
 					data: base64Data
@@ -5326,15 +5344,17 @@ const applyGenerationSpanData = (lmnrSpan, spanData) => {
 	setGenAiInputMessages(lmnrSpan, inputData, getCurrentSystemInstructions());
 	const outputData = spanData?.output;
 	setGenAiOutputMessages(lmnrSpan, outputData);
-	applyLlmAttributes(lmnrSpan, {
+	const llmData = {
 		model: spanData?.model,
 		usage: spanData?.usage,
 		response_id: spanData?.response_id ?? spanData?.id
-	});
+	};
+	applyLlmAttributes(lmnrSpan, llmData);
 };
 const applyResponseSpanData = (lmnrSpan, spanData) => {
 	const response = spanData?._response ?? spanData?.response;
-	setGenAiInputMessages(lmnrSpan, spanData?._input ?? spanData?.input, getCurrentSystemInstructions());
+	const responseInput = spanData?._input ?? spanData?.input;
+	setGenAiInputMessages(lmnrSpan, responseInput, getCurrentSystemInstructions());
 	if (response != null) {
 		setGenAiOutputMessagesFromResponse(lmnrSpan, response);
 		setToolDefinitionsFromResponse(lmnrSpan, response);
@@ -5375,8 +5395,10 @@ const applySpeechSpanData = (lmnrSpan, spanData) => {
 	const inputText = spanData?.input;
 	if (inputText) setGenAiInputMessages(lmnrSpan, inputText);
 	const outputData = spanData?.output;
-	if (outputData) if (typeof outputData === "object" && !Array.isArray(outputData)) setGenAiOutputMessages(lmnrSpan, outputData.data);
-	else setGenAiOutputMessages(lmnrSpan, outputData);
+	if (outputData) {
+		if (typeof outputData === "object" && !Array.isArray(outputData)) setGenAiOutputMessages(lmnrSpan, outputData.data);
+		else setGenAiOutputMessages(lmnrSpan, outputData);
+	}
 };
 const applyTranscriptionSpanData = (lmnrSpan, spanData) => {
 	const model = spanData?.model;
@@ -5386,8 +5408,10 @@ const applyTranscriptionSpanData = (lmnrSpan, spanData) => {
 		lmnrSpan.setAttribute(require_utils.LaminarAttributes.PROVIDER, "openai");
 	}
 	const inputData = spanData?.input;
-	if (inputData) if (typeof inputData === "object" && !Array.isArray(inputData)) setGenAiInputMessages(lmnrSpan, inputData.data);
-	else setGenAiInputMessages(lmnrSpan, inputData);
+	if (inputData) {
+		if (typeof inputData === "object" && !Array.isArray(inputData)) setGenAiInputMessages(lmnrSpan, inputData.data);
+		else setGenAiInputMessages(lmnrSpan, inputData);
+	}
 	const outputText = spanData?.output;
 	if (outputText) setGenAiOutputMessages(lmnrSpan, outputText);
 };
@@ -5610,7 +5634,8 @@ const wrapGetResponse = (original) => function(request, ...rest) {
 };
 const wrapGetStreamedResponse = (original) => function(request, ...rest) {
 	const systemInstructions = request?.systemInstructions;
-	return wrapStreamWithSystemInstructions(systemInstructions, original.call(this, request, ...rest));
+	const source = original.call(this, request, ...rest);
+	return wrapStreamWithSystemInstructions(systemInstructions, source);
 };
 const MODEL_METHODS = ["getResponse", "getStreamedResponse"];
 var OpenAIAgentsInstrumentation = class extends _opentelemetry_instrumentation.InstrumentationBase {
@@ -5755,12 +5780,12 @@ var OpencodeInstrumentation = class extends _opentelemetry_instrumentation.Instr
 };
 //#endregion
 //#region src/opentelemetry-lib/instrumentation/openrouter/utils.ts
-const TERMINAL_RESPONSE_EVENTS = new Set([
+const TERMINAL_RESPONSE_EVENTS = /* @__PURE__ */ new Set([
 	"response.completed",
 	"response.incomplete",
 	"response.failed"
 ]);
-const ERROR_RESPONSE_STATUSES = new Set(["failed", "incomplete"]);
+const ERROR_RESPONSE_STATUSES = /* @__PURE__ */ new Set(["failed", "incomplete"]);
 const safeSetAttribute$1 = (span, key, value) => {
 	if (value === null || value === void 0) return;
 	if (typeof value === "string" && value.length === 0) return;
@@ -5926,7 +5951,8 @@ const wrapStream = (span, kind, stream, traceContent) => {
 			throw error;
 		} finally {
 			try {
-				setResponseAttributes$1(span, kind, kind === "chat" ? aggregateChatChunks(chunks) : responseFromStreamEvents(chunks), traceContent);
+				const response = kind === "chat" ? aggregateChatChunks(chunks) : responseFromStreamEvents(chunks);
+				setResponseAttributes$1(span, kind, response, traceContent);
 			} finally {
 				span.end();
 			}
@@ -6492,7 +6518,9 @@ const buildHeaders = (existing) => {
 	const headers = { ...existing };
 	if (laminarCtx) {
 		headers[require_consts.LAMINAR_SPAN_CONTEXT_HEADER] = encodePayload(JSON.stringify(laminarCtx));
-		headers[require_consts.TRACEPARENT_HEADER] = encodePayload(`00-${laminarCtx.traceId.replace(/-/g, "")}-${laminarCtx.spanId.replace(/-/g, "").slice(16)}-01`);
+		const traceHex = laminarCtx.traceId.replace(/-/g, "");
+		const spanHex = laminarCtx.spanId.replace(/-/g, "").slice(16);
+		headers[require_consts.TRACEPARENT_HEADER] = encodePayload(`00-${traceHex}-${spanHex}-01`);
 	} else {
 		const ctx = rawSpan.spanContext();
 		headers[require_consts.TRACEPARENT_HEADER] = encodePayload(`00-${ctx.traceId}-${ctx.spanId}-01`);
@@ -6956,10 +6984,11 @@ const patchTemporalWorker = (workerModule, options = {}) => {
 			activity: [ActivityInterceptorFactory(options), ...rawOpts?.interceptors?.activity ?? []]
 		};
 		if (!usesWorkflowBundle) interceptors.workflowModules = [require.resolve("@lmnr-ai/lmnr/temporal-workflow-interceptors"), ...rawOpts?.interceptors?.workflowModules ?? []];
-		return originalCreate({
+		const patched = {
 			...rawOpts,
 			interceptors
-		}, ...rest);
+		};
+		return originalCreate(patched, ...rest);
 	};
 };
 /**
@@ -7168,8 +7197,9 @@ const initializeLaminarInstrumentations = (options = {}) => {
 	if (apiKey) {
 		const url = options.baseUrl ?? process?.env?.LMNR_BASE_URL ?? "https://api.lmnr.ai";
 		const port = options.httpPort ?? (url.match(/:\d{1,5}$/g) ? parseInt(url.match(/:\d{1,5}$/g)[0].slice(1)) : 443);
+		const urlWithoutSlash = url.replace(/\/$/, "").replace(/:\d{1,5}$/g, "");
 		client = new require_utils.LaminarClient({
-			baseUrl: `${url.replace(/\/$/, "").replace(/:\d{1,5}$/g, "")}:${port}`,
+			baseUrl: `${urlWithoutSlash}:${port}`,
 			projectApiKey: apiKey
 		});
 	}
@@ -7449,14 +7479,15 @@ const startTracing = (options) => {
 	});
 };
 const patchModules = (modules) => {
+	const instrumentations = initializeLaminarInstrumentations({
+		baseUrl: _baseHttpUrl,
+		apiKey: _apiKey,
+		instrumentModules: modules,
+		suppressContentTracing: !shouldSendTraces(),
+		sessionRecordingOptions: _configuration?.sessionRecordingOptions
+	});
 	(0, _opentelemetry_instrumentation.registerInstrumentations)({
-		instrumentations: initializeLaminarInstrumentations({
-			baseUrl: _baseHttpUrl,
-			apiKey: _apiKey,
-			instrumentModules: modules,
-			suppressContentTracing: !shouldSendTraces(),
-			sessionRecordingOptions: _configuration?.sessionRecordingOptions
-		}),
+		instrumentations,
 		tracerProvider
 	});
 };
@@ -8875,8 +8906,10 @@ const stripSystemMessages = (messages) => messages.filter((m) => !(m !== null &&
 * hex-encoded (64 chars).
 */
 const debugInputHash = (messages) => {
-	const canonical = canonicalJson(stripSystemMessages(messages));
-	return (0, _noble_hashes_utils_js.bytesToHex)((0, _noble_hashes_blake3_js.blake3)(new TextEncoder().encode(canonical)));
+	const stripped = stripSystemMessages(messages);
+	const canonical = canonicalJson(stripped);
+	const digest = (0, _noble_hashes_blake3_js.blake3)(new TextEncoder().encode(canonical));
+	return (0, _noble_hashes_utils_js.bytesToHex)(digest);
 };
 //#endregion
 //#region src/debug/replay.ts
@@ -8992,6 +9025,28 @@ const applyUsageToSpan = (span, usage) => {
 	applyUsage(attrs, usage);
 	for (const [k, v] of Object.entries(attrs)) if (v !== void 0) span.setAttribute(k, v);
 };
+const resolveAssociationProperties = (event, parent) => {
+	const { userId, sessionId, tags, ...metadata } = {
+		...event?.toolsContext ?? {},
+		...event?.runtimeContext ?? {}
+	};
+	return {
+		associationProperties: {
+			userId: typeof userId === "string" && userId.length > 0 ? userId : parent.userId,
+			sessionId: typeof sessionId === "string" && sessionId.length > 0 ? sessionId : parent.sessionId,
+			metadata: {
+				...parent.metadata ?? {},
+				...metadata
+			}
+		},
+		tags: Array.isArray(tags) && tags.every((tag) => typeof tag === "string") ? tags : void 0
+	};
+};
+const applyAssociationProperties = (span, properties) => {
+	if (properties.userId) span.setAttribute(require_utils.USER_ID, properties.userId);
+	if (properties.sessionId) span.setAttribute(require_utils.SESSION_ID, properties.sessionId);
+	span.setAttributes(require_utils.metadataToAttributes(properties.metadata));
+};
 const applyFinishReason = (span, finishReason) => {
 	span.setAttribute("gen_ai.response.finish_reason", finishReason);
 	span.setAttribute("ai.response.finishReason", finishReason);
@@ -9025,6 +9080,17 @@ const verbatimPromptString = (prompt) => {
 		return null;
 	}
 };
+const buildDecisionInputMessages = (state) => {
+	if (state === void 0 || state === null) return void 0;
+	if (Array.isArray(state)) return verbatimPromptString([{
+		role: "user",
+		content: state
+	}]) ?? void 0;
+	return serializeJSON$1([{
+		role: "user",
+		content: typeof state === "string" ? state : serializeJSON$1(state)
+	}]);
+};
 const verbatimPromptMessages = (prompt) => {
 	const serialized = verbatimPromptString(prompt);
 	if (serialized === null) return null;
@@ -9680,7 +9746,6 @@ var LaminarAiSdkTelemetry = class {
 					} : {}
 				}
 			}, parentCtx);
-			const spanCtx = _opentelemetry_api.trace.setSpan(parentCtx, span);
 			span.setAttribute(require_utils.SPAN_TYPE, "DEFAULT");
 			if (typeof event.operationId === "string") span.setAttribute("ai.operation", event.operationId);
 			if (typeof event.provider === "string") {
@@ -9693,6 +9758,14 @@ var LaminarAiSdkTelemetry = class {
 			}
 			if (typeof event.functionId === "string") span.setAttribute("ai.functionId", event.functionId);
 			if (event.operationId === "ai.embed" || event.operationId === "ai.embedMany" || event.operationId === "ai.rerank") span.setAttribute(require_utils.SPAN_TYPE, "LLM");
+			const parentAssociationProperties = parentCtx.getValue(ASSOCIATION_PROPERTIES_KEY) ?? {};
+			const { associationProperties, tags } = resolveAssociationProperties(event, parentAssociationProperties);
+			applyAssociationProperties(span, associationProperties);
+			if (tags) span.setAttribute(`${require_utils.ASSOCIATION_PROPERTIES}.tags`, tags);
+			const spanCtx = _opentelemetry_api.trace.setSpan(parentCtx.setValue(ASSOCIATION_PROPERTIES_KEY, {
+				...parentAssociationProperties,
+				...Object.fromEntries(Object.entries(associationProperties).filter(([, value]) => value !== void 0))
+			}), span);
 			if (this.recordInputs) {
 				if (Array.isArray(event.messages) || event.system !== void 0 || event.instructions !== void 0) {
 					const msgs = verbatimStandardizedMessages(event);
@@ -9708,7 +9781,8 @@ var LaminarAiSdkTelemetry = class {
 				ctx: spanCtx,
 				operationId: event.operationId ?? "unknown",
 				provider: event.provider,
-				modelId: event.modelId
+				modelId: event.modelId,
+				associationProperties
 			});
 		};
 		this.onStepStart = (event) => {
@@ -9726,6 +9800,7 @@ var LaminarAiSdkTelemetry = class {
 			const spanCtx = _opentelemetry_api.trace.setSpan(op.ctx, span);
 			span.setAttribute(require_utils.SPAN_TYPE, "DEFAULT");
 			span.setAttributes(buildAiSdkInstrumentationAttributes());
+			applyAssociationProperties(span, op.associationProperties);
 			span.setAttribute("ai.step.number", stepNumber);
 			if (this.recordInputs) {
 				const msgs = verbatimStandardizedMessages(event);
@@ -9742,11 +9817,13 @@ var LaminarAiSdkTelemetry = class {
 			if (!callId) return;
 			const step = this.findLatestStep(callId);
 			const stepNumber = step?.stepNumber ?? event?.stepNumber ?? 0;
-			const parentCtx = step?.ctx ?? this.operationByCallId.get(callId)?.ctx;
-			if (!parentCtx) return;
+			const op = this.operationByCallId.get(callId);
+			const parentCtx = step?.ctx ?? op?.ctx;
+			if (!op || !parentCtx) return;
 			const span = getTracer().startSpan(`ai.llm ${event.provider ?? ""}:${event.modelId ?? ""}`, { kind: _opentelemetry_api.SpanKind.CLIENT }, parentCtx);
 			span.setAttribute(require_utils.SPAN_TYPE, "LLM");
 			span.setAttributes(buildAiSdkInstrumentationAttributes());
+			applyAssociationProperties(span, op.associationProperties);
 			applyRequestModelAttributes(span, event);
 			if (this.recordInputs) {
 				if (event.tools) try {
@@ -9843,6 +9920,7 @@ var LaminarAiSdkTelemetry = class {
 			const span = getTracer().startSpan(`ai.llm ${event.provider ?? ""}:${event.modelId ?? ""}`, { kind: _opentelemetry_api.SpanKind.CLIENT }, op.ctx);
 			span.setAttribute(require_utils.SPAN_TYPE, "LLM");
 			span.setAttributes(buildAiSdkInstrumentationAttributes());
+			applyAssociationProperties(span, op.associationProperties);
 			applyRequestModelAttributes(span, event);
 			if (this.recordInputs && Array.isArray(event?.promptMessages)) {
 				const serialized = verbatimPromptString(event.promptMessages);
@@ -9881,12 +9959,15 @@ var LaminarAiSdkTelemetry = class {
 			const toolCallId = toolCall?.toolCallId;
 			const toolName = toolCall?.toolName;
 			if (!callId || !toolCallId) return;
-			const parentCtx = this.findLatestStep(callId)?.ctx ?? this.operationByCallId.get(callId)?.ctx;
-			if (!parentCtx) return;
+			const step = this.findLatestStep(callId);
+			const op = this.operationByCallId.get(callId);
+			const parentCtx = step?.ctx ?? op?.ctx;
+			if (!op || !parentCtx) return;
 			const span = getTracer().startSpan(`ai.tool ${toolName ?? "unknown"}`, { kind: _opentelemetry_api.SpanKind.CLIENT }, parentCtx);
 			const spanCtx = _opentelemetry_api.trace.setSpan(parentCtx, span);
 			span.setAttribute(require_utils.SPAN_TYPE, "TOOL");
 			span.setAttributes(buildAiSdkInstrumentationAttributes());
+			applyAssociationProperties(span, op.associationProperties);
 			if (toolName) span.setAttribute("ai.toolCall.name", toolName);
 			span.setAttribute("ai.toolCall.id", toolCallId);
 			if (this.recordInputs && toolCall?.input !== void 0) {
@@ -9956,6 +10037,35 @@ var LaminarAiSdkTelemetry = class {
 		};
 		this.onRerankStart = () => {};
 		this.onRerankEnd = () => {};
+		this.experimental_onDecideStart = (event) => {
+			const callId = event?.callId;
+			if (!callId) return;
+			this.onStart(event);
+			const op = this.operationByCallId.get(callId);
+			if (!op) return;
+			op.span.setAttribute(require_utils.SPAN_TYPE, "LLM");
+			applyRequestModelAttributes(op.span, event);
+			if (this.recordInputs) {
+				const input = buildDecisionInputMessages(event.state);
+				if (input !== void 0) op.span.setAttribute("gen_ai.input.messages", input);
+				if (event.questions !== void 0) op.span.setAttribute("gen_ai.request.structured_output_schema", serializeJSON$1(event.questions));
+			}
+		};
+		this.experimental_onDecideEnd = (event) => {
+			const callId = event?.callId;
+			if (!callId) return;
+			const op = this.operationByCallId.get(callId);
+			if (!op) return;
+			if (typeof event.response?.modelId === "string") op.span.setAttribute(require_utils.LaminarAttributes.RESPONSE_MODEL, event.response.modelId);
+			if (typeof event.response?.id === "string") op.span.setAttribute("gen_ai.response.id", event.response.id);
+			applyUsageToSpan(op.span, event.usage);
+			if (this.recordOutputs && event.answers !== void 0) op.span.setAttribute("gen_ai.output.messages", serializeJSON$1([{
+				role: "assistant",
+				content: serializeJSON$1(event.answers)
+			}]));
+			op.span.setStatus({ code: _opentelemetry_api.SpanStatusCode.OK });
+			this.closeOperation(callId, op);
+		};
 		this.onEnd = (event) => {
 			const callId = event?.callId;
 			if (!callId) return;