@lmnr-ai/lmnr 0.7.13 → 0.7.15

package/dist/index.mjs

@@ -1,5 +1,5 @@
-import { a as CONTEXT_SPAN_PATH_KEY, c as createResource, d as TracingLevel, f as __require, i as ASSOCIATION_PROPERTIES_KEY, l as getParentSpanId, n as getLangVersion, o as LaminarContextManager, r as version, s as LaminarSpan, t as LaminarClient, u as makeSpanOtelV2Compatible } from "./client-D_P5R4sW.mjs";
-import { A as SPAN_SDK_VERSION, C as SESSION_ID, D as SPAN_LANGUAGE_VERSION, E as SPAN_INSTRUMENTATION_SOURCE, M as TRACE_HAS_BROWSER_SESSION, N as TRACE_TYPE, O as SPAN_OUTPUT, P as USER_ID, S as PARENT_SPAN_PATH, T as SPAN_INPUT, _ as ASSOCIATION_PROPERTIES, a as getOtelEnvVar, b as LaminarAttributes, d as otelSpanIdToUUID, f as otelTraceIdToUUID, g as validateTracingConfig, h as tryToOtelSpanContext, j as SPAN_TYPE, k as SPAN_PATH, l as metadataToAttributes, n as Semaphore, o as initializeLogger, p as parseOtelHeaders, r as deserializeLaminarSpanContext, s as isOtelAttributeValueType, t as NIL_UUID, u as newUUID, v as ASSOCIATION_PROPERTIES_OVERRIDES, w as SPAN_IDS_PATH, x as PARENT_SPAN_IDS_PATH, y as HUMAN_EVALUATOR_OPTIONS } from "./utils-viuxL6V7.mjs";
+import { a as CONTEXT_SPAN_PATH_KEY, c as createResource, d as TracingLevel, f as __require, i as ASSOCIATION_PROPERTIES_KEY, l as getParentSpanId, n as getLangVersion, o as LaminarContextManager, r as version, s as LaminarSpan, t as LaminarClient, u as makeSpanOtelV2Compatible } from "./client-Bp0oS3Ur.mjs";
+import { A as SPAN_SDK_VERSION, C as SESSION_ID, D as SPAN_LANGUAGE_VERSION, E as SPAN_INSTRUMENTATION_SOURCE, M as TRACE_HAS_BROWSER_SESSION, N as TRACE_TYPE, O as SPAN_OUTPUT, P as USER_ID, S as PARENT_SPAN_PATH, T as SPAN_INPUT, _ as ASSOCIATION_PROPERTIES, a as getOtelEnvVar, b as LaminarAttributes, d as otelSpanIdToUUID, f as otelTraceIdToUUID, g as validateTracingConfig, h as tryToOtelSpanContext, j as SPAN_TYPE, k as SPAN_PATH, l as metadataToAttributes, n as Semaphore, o as initializeLogger, p as parseOtelHeaders, r as deserializeLaminarSpanContext, s as isOtelAttributeValueType, t as NIL_UUID, u as newUUID, v as ASSOCIATION_PROPERTIES_OVERRIDES, w as SPAN_IDS_PATH, x as PARENT_SPAN_IDS_PATH, y as HUMAN_EVALUATOR_OPTIONS } from "./utils-ChSbRj0P.mjs";
 import { config } from "dotenv";
 import { DiagConsoleLogger, DiagLogLevel, ROOT_CONTEXT, context, createContextKey, diag, isSpanContextValid, trace } from "@opentelemetry/api";
 import { AsyncLocalStorageContextManager } from "@opentelemetry/context-async-hooks";
@@ -80,7 +80,7 @@ var LaminarDataset = class extends EvaluationDataset {
 	*/
 	async push(paths, recursive = false) {
 		if (!this.client) throw new Error("Client not set");
-		const { loadFromPaths } = await import("./file-utils-D3qI1vPq.mjs");
+		const { loadFromPaths } = await import("./file-utils-ACjf7y05.mjs");
 		const data = await loadFromPaths(Array.isArray(paths) ? paths : [paths], recursive);
 		if (data.length === 0) {
 			console.warn("No data to push. Skipping");
@@ -166,6 +166,7 @@ function waitForPort(port, timeoutMs = 5e3) {
 function stopCcProxy() {
 	try {
 		const { stopServer } = __require("@lmnr-ai/claude-code-proxy");
+		logger$11.debug("Stopping cc-proxy...");
 		stopServer();
 	} catch (e) {
 		logger$11.debug(`Unable to stop cc-proxy: ${e instanceof Error ? e.message : String(e)}`);
@@ -182,6 +183,7 @@ function stopCcProxy() {
 function registerProxyShutdown() {
 	if (!ccProxyShutdownRegistered) {
 		process.on("exit", () => {
+			logger$11.debug("process.on(\"exit\") called");
 			stopCcProxy();
 		});
 		ccProxyShutdownRegistered = true;
@@ -196,7 +198,7 @@ function getProxyBaseUrl() {
 /**
 * Start the claude-code proxy server with reference counting
 */
-async function startProxy() {
+async function startProxy({ env }) {
 	if (ccProxyStartupPromise !== null) {
 		logger$11.debug("Waiting for ongoing proxy startup to complete");
 		await ccProxyStartupPromise;
@@ -213,7 +215,7 @@ async function startProxy() {
 				logger$11.warn("Unable to allocate port for cc-proxy.");
 				return null;
 			}
-			const targetUrl = ccProxyTargetUrl || process.env.ANTHROPIC_ORIGINAL_BASE_URL || process.env.ANTHROPIC_BASE_URL || DEFAULT_ANTHROPIC_BASE_URL;
+			const targetUrl = ccProxyTargetUrl || process.env.ANTHROPIC_ORIGINAL_BASE_URL || env.ANTHROPIC_BASE_URL || process.env.ANTHROPIC_BASE_URL || DEFAULT_ANTHROPIC_BASE_URL;
 			ccProxyTargetUrl = targetUrl;
 			process.env.ANTHROPIC_ORIGINAL_BASE_URL = targetUrl;
 			try {
@@ -340,11 +342,16 @@ function instrumentClaudeAgentQuery(originalQuery) {
 		const generator = async function* () {
 			const collected = [];
 			try {
-				await startProxy();
-				if (getProxyBaseUrl()) await Laminar.withSpan(span, () => {
-					setTraceToProxy();
-				});
-				else logger$10.debug("No claude proxy server found. Skipping span context publication.");
+				await startProxy({ env: params.options?.env ?? process.env });
+				const proxyBaseUrl = getProxyBaseUrl();
+				logger$10.debug(`getProxyBaseUrl() result: ${proxyBaseUrl}`);
+				if (proxyBaseUrl) {
+					await Laminar.withSpan(span, () => {
+						logger$10.debug("Setting trace to proxy...");
+						setTraceToProxy();
+					});
+					if (params.options?.env) params.options.env.ANTHROPIC_BASE_URL = proxyBaseUrl;
+				} else logger$10.debug("No claude proxy server found. Skipping span context publication.");
 				const originalGenerator = originalQuery(params);
 				for await (const message of originalGenerator) {
 					collected.push(message);