@lmnr-ai/lmnr 0.8.21 → 0.8.22

package/dist/index.mjs

@@ -1,6 +1,6 @@
-import { n as version, t as LaminarClient } from "./dist-B5CmOQ4s.mjs";
+import { n as version, t as LaminarClient } from "./dist-BXwORFgR.mjs";
 import { A as SPAN_OUTPUT, D as SPAN_INPUT, E as SPAN_IDS_PATH, I as USER_ID, M as SPAN_SDK_VERSION, N as SPAN_TYPE, O as SPAN_INSTRUMENTATION_SOURCE, T as SESSION_ID, a as getFrontendUrl, b as HUMAN_EVALUATOR_OPTIONS, c as loadEnv, d as normalizeOtelSpanId, f as normalizeOtelTraceId, j as SPAN_PATH, k as SPAN_LANGUAGE_VERSION, m as otelTraceIdToUUID, n as Semaphore, p as otelSpanIdToUUID, s as initializeLogger, u as newUUID, v as ASSOCIATION_PROPERTIES, x as LaminarAttributes } from "./utils-CHJ0KZUR.mjs";
-import { a as getSpanProcessor, c as LaminarSpanProcessor, d as Laminar, f as instrumentClaudeAgentQuery, g as stringifyPromptForTelemetry, i as withTracingLevel, l as getLangVersion, m as TracingLevel, n as observeDecorator, o as getTracer, p as LaminarContextManager, r as observeExperimentalDecorator, s as getTracerProvider, t as observe, u as initializeLaminarInstrumentations } from "./decorators-BDFgXfgI.mjs";
+import { _ as stringifyPromptForTelemetry, a as getSpanProcessor, c as LaminarSpanProcessor, d as Laminar, f as instrumentClaudeAgentQuery, h as TracingLevel, i as withTracingLevel, l as getLangVersion, m as LaminarContextManager, n as observeDecorator, o as getTracer, p as ASSOCIATION_PROPERTIES_KEY, r as observeExperimentalDecorator, s as getTracerProvider, t as observe, u as initializeLaminarInstrumentations } from "./decorators-DrWG-1Po.mjs";
 import { SpanKind, SpanStatusCode, TraceFlags, context, trace } from "@opentelemetry/api";
 import * as cliProgress from "cli-progress";
 //#region src/datasets.ts
@@ -232,23 +232,34 @@ var Evaluation = class {
 		}
 	}
 	async evaluateInBatches(evalId) {
-		const semaphore = new Semaphore(this.concurrencyLimit);
-		const tasks = [];
-		const evaluateTask = async (datapoint, index) => {
-			try {
-				const result = await this.evaluateDatapoint(evalId, datapoint, index);
-				this.progressReporter.update(1);
-				return [index, result];
-			} finally {
-				semaphore.release();
+		const baseContext = LaminarContextManager.getContext();
+		const currentAssociationProperties = baseContext.getValue(ASSOCIATION_PROPERTIES_KEY) ?? {};
+		const entityContext = baseContext.setValue(ASSOCIATION_PROPERTIES_KEY, {
+			...currentAssociationProperties,
+			metadata: {
+				...currentAssociationProperties.metadata ?? {},
+				evaluation_id: evalId
 			}
-		};
-		for (let i = 0; i < await this.getLength(); i++) {
-			await semaphore.acquire();
-			const datapoint = Array.isArray(this.data) ? this.data[i] : await this.data.get(i);
-			tasks.push(evaluateTask(datapoint, i));
-		}
-		return (await Promise.all(tasks)).sort((a, b) => a[0] - b[0]).map(([, result]) => result);
+		});
+		return LaminarContextManager.runWithIsolatedContext([entityContext], async () => {
+			const semaphore = new Semaphore(this.concurrencyLimit);
+			const tasks = [];
+			const evaluateTask = async (datapoint, index) => {
+				try {
+					const result = await this.evaluateDatapoint(evalId, datapoint, index);
+					this.progressReporter.update(1);
+					return [index, result];
+				} finally {
+					semaphore.release();
+				}
+			};
+			for (let i = 0; i < await this.getLength(); i++) {
+				await semaphore.acquire();
+				const datapoint = Array.isArray(this.data) ? this.data[i] : await this.data.get(i);
+				tasks.push(evaluateTask(datapoint, i));
+			}
+			return (await Promise.all(tasks)).sort((a, b) => a[0] - b[0]).map(([, result]) => result);
+		});
 	}
 	async evaluateDatapoint(evalId, datapoint, index) {
 		return observe({