@lmnr-ai/lmnr 0.7.14 → 0.7.15

package/dist/index.mjs

@@ -1,4 +1,4 @@
-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-BaOYuI3V.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";
@@ -198,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;
@@ -215,16 +215,11 @@ 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;
-			logger$11.debug(`process.env.ANTHROPIC_ORIGINAL_BASE_URL: ${process.env.ANTHROPIC_ORIGINAL_BASE_URL}`);
-			logger$11.debug(`process.env.ANTHROPIC_BASE_URL: ${process.env.ANTHROPIC_BASE_URL}`);
-			logger$11.debug(`DEFAULT_ANTHROPIC_BASE_URL: ${DEFAULT_ANTHROPIC_BASE_URL}`);
-			logger$11.debug(`Using anthropic base url: ${targetUrl}`);
+			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 {
 				const { runServer } = __require("@lmnr-ai/claude-code-proxy");
-				logger$11.debug(`Running cc-proxy server on port ${port} with target url ${targetUrl}`);
 				runServer(targetUrl, port);
 			} catch (e) {
 				logger$11.warn(`Unable to start cc-proxy: ${e instanceof Error ? e.message : String(e)}`);
@@ -239,7 +234,6 @@ async function startProxy() {
 			const proxyBaseUrl = `http://127.0.0.1:${port}`;
 			ccProxyBaseUrl = proxyBaseUrl;
 			ccProxyRefCount = 1;
-			logger$11.debug(`Setting ANTHROPIC_BASE_URL to: ${proxyBaseUrl}`);
 			process.env.ANTHROPIC_BASE_URL = proxyBaseUrl;
 			registerProxyShutdown();
 			logger$11.info(`Started claude proxy server on: ${proxyBaseUrl}`);
@@ -348,15 +342,16 @@ function instrumentClaudeAgentQuery(originalQuery) {
 		const generator = async function* () {
 			const collected = [];
 			try {
-				await startProxy();
+				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();
-				});
-				else logger$10.debug("No claude proxy server found. Skipping span context publication.");
-				logger$10.debug(`ANTHROPIC_BASE_URL before calling originalQuery: ${process.env.ANTHROPIC_BASE_URL}`);
+				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);