langsmith 0.3.35-rc.0 → 0.3.35-rc.2

package/dist/client.cjs

@@ -391,14 +391,7 @@ class Client {
         this.fetchOptions = config.fetchOptions || {};
         this.manualFlushMode = config.manualFlushMode ?? this.manualFlushMode;
         if ((0, env_js_1.getEnvironmentVariable)("OTEL_ENABLED") === "true") {
-            const otel_trace = (0, otel_js_1.getOTELTrace)();
-            const existingTracerProvider = otel_trace.getTracerProvider();
-            const { tracerProvider: langSmithTracerProvider } = (0, otel_js_1.getDefaultOTLPTracerComponents)() ?? {};
-            // If user has set global tracer before, this fails and returns false
-            const globalSuccessfullyOverridden = otel_trace.setGlobalTracerProvider(langSmithTracerProvider);
-            this.langSmithToOTELTranslator = new translator_js_1.LangSmithToOTELTranslator(globalSuccessfullyOverridden
-                ? langSmithTracerProvider
-                : existingTracerProvider);
+            this.langSmithToOTELTranslator = new translator_js_1.LangSmithToOTELTranslator();
         }
     }
     static getDefaultClientConfig() {

package/dist/client.js

@@ -353,14 +353,7 @@ export class Client {
         this.fetchOptions = config.fetchOptions || {};
         this.manualFlushMode = config.manualFlushMode ?? this.manualFlushMode;
         if (getEnvironmentVariable("OTEL_ENABLED") === "true") {
-            const otel_trace = getOTELTrace();
-            const existingTracerProvider = otel_trace.getTracerProvider();
-            const { tracerProvider: langSmithTracerProvider } = getDefaultOTLPTracerComponents() ?? {};
-            // If user has set global tracer before, this fails and returns false
-            const globalSuccessfullyOverridden = otel_trace.setGlobalTracerProvider(langSmithTracerProvider);
-            this.langSmithToOTELTranslator = new LangSmithToOTELTranslator(globalSuccessfullyOverridden
-                ? langSmithTracerProvider
-                : existingTracerProvider);
+            this.langSmithToOTELTranslator = new LangSmithToOTELTranslator();
         }
     }
     static getDefaultClientConfig() {

package/dist/experimental/otel/exporter.cjs

@@ -40,7 +40,7 @@ const env_js_1 = require("../../env.cjs");
 class LangSmithOTLPTraceExporter extends exporter_trace_otlp_proto_1.OTLPTraceExporter {
     export(spans, resultCallback) {
         if (!(0, env_js_1.isTracingEnabled)()) {
-            return;
+            return resultCallback({ code: 0 });
         }
         for (const span of spans) {
             if (!span.attributes[constants.GENAI_PROMPT]) {

package/dist/experimental/otel/exporter.js

@@ -4,7 +4,7 @@ import { isTracingEnabled } from "../../env.js";
 export class LangSmithOTLPTraceExporter extends OTLPTraceExporter {
     export(spans, resultCallback) {
         if (!isTracingEnabled()) {
-            return;
+            return resultCallback({ code: 0 });
         }
         for (const span of spans) {
             if (!span.attributes[constants.GENAI_PROMPT]) {

package/dist/experimental/otel/setup.cjs

@@ -9,14 +9,6 @@ const sdk_trace_base_1 = require("@opentelemetry/sdk-trace-base");
 const env_js_1 = require("../../utils/env.cjs");
 const exporter_js_1 = require("./exporter.cjs");
 const otel_js_1 = require("../../singletons/otel.cjs");
-const otel = {
-    trace: api_1.trace,
-    context: api_1.context,
-};
-(0, otel_js_1.setOTELInstances)(otel);
-const contextManager = new context_async_hooks_1.AsyncHooksContextManager();
-contextManager.enable();
-api_1.context.setGlobalContextManager(contextManager);
 /**
  * Convert headers string in format "name=value,name2=value2" to object
  */
@@ -32,40 +24,50 @@ function parseHeadersString(headersStr) {
     });
     return headers;
 }
-function getDefaultOTLPTracerComponents() {
-    // Set LangSmith-specific defaults if not already set in environment
-    if (!(0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_ENDPOINT")) {
-        const lsEndpoint = (0, env_js_1.getLangSmithEnvironmentVariable)("ENDPOINT") ||
-            "https://api.smith.langchain.com";
-        const baseUrl = lsEndpoint.replace(/\/$/, "");
-        (0, env_js_1.setEnvironmentVariable)("OTEL_EXPORTER_OTLP_ENDPOINT", `${baseUrl}/otel/v1/traces`);
+// Set LangSmith-specific defaults if not already set in environment
+if (!(0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_ENDPOINT")) {
+    const lsEndpoint = (0, env_js_1.getLangSmithEnvironmentVariable)("ENDPOINT") ||
+        "https://api.smith.langchain.com";
+    const baseUrl = lsEndpoint.replace(/\/$/, "");
+    (0, env_js_1.setEnvironmentVariable)("OTEL_EXPORTER_OTLP_ENDPOINT", `${baseUrl}/otel/v1/traces`);
+}
+// Configure headers with API key and project if available
+if (!(0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_HEADERS")) {
+    const apiKey = (0, env_js_1.getLangSmithEnvironmentVariable)("API_KEY");
+    if (!apiKey) {
+        throw new Error("LANGSMITH_API_KEY or LANGCHAIN_API_KEY environment variable is required");
     }
-    // Configure headers with API key and project if available
-    if (!(0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_HEADERS")) {
-        const apiKey = (0, env_js_1.getLangSmithEnvironmentVariable)("API_KEY");
-        if (!apiKey) {
-            throw new Error("LANGSMITH_API_KEY or LANGCHAIN_API_KEY environment variable is required");
-        }
-        let headers = `x-api-key=${apiKey}`;
-        const project = (0, env_js_1.getLangSmithEnvironmentVariable)("PROJECT");
-        if (project) {
-            headers += `,Langsmith-Project=${project}`;
-        }
-        (0, env_js_1.setEnvironmentVariable)("OTEL_EXPORTER_OTLP_HEADERS", headers);
+    let headers = `x-api-key=${apiKey}`;
+    const project = (0, env_js_1.getLangSmithEnvironmentVariable)("PROJECT");
+    if (project) {
+        headers += `,Langsmith-Project=${project}`;
     }
-    const headersStr = (0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_HEADERS") || "";
-    const headersObj = parseHeadersString(headersStr);
-    const langsmithSpanExporter = new exporter_js_1.LangSmithOTLPTraceExporter({
-        url: (0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_ENDPOINT"),
-        headers: headersObj,
-    });
-    const spanProcessor = new sdk_trace_base_1.BatchSpanProcessor(langsmithSpanExporter);
-    return {
-        tracerProvider: new sdk_trace_base_1.BasicTracerProvider({
-            spanProcessors: [spanProcessor],
-        }),
-        spanProcessor,
-        langsmithSpanExporter,
-    };
+    (0, env_js_1.setEnvironmentVariable)("OTEL_EXPORTER_OTLP_HEADERS", headers);
+}
+const headersStr = (0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_HEADERS") || "";
+const headersObj = parseHeadersString(headersStr);
+const langsmithSpanExporter = new exporter_js_1.LangSmithOTLPTraceExporter({
+    url: (0, env_js_1.getEnvironmentVariable)("OTEL_EXPORTER_OTLP_ENDPOINT"),
+    headers: headersObj,
+});
+const spanProcessor = new sdk_trace_base_1.BatchSpanProcessor(langsmithSpanExporter);
+const otel = {
+    trace: api_1.trace,
+    context: api_1.context,
+};
+(0, otel_js_1.setOTELInstances)(otel);
+const contextManager = new context_async_hooks_1.AsyncHooksContextManager();
+contextManager.enable();
+api_1.context.setGlobalContextManager(contextManager);
+const defaultComponents = {
+    spanProcessor,
+    tracerProvider: new sdk_trace_base_1.BasicTracerProvider({
+        spanProcessors: [spanProcessor],
+    }),
+    langsmithSpanExporter,
+};
+// If user has set global tracer before, this fails and returns false
+const globalSuccessfullyOverridden = api_1.trace.setGlobalTracerProvider(defaultComponents.tracerProvider);
+if (globalSuccessfullyOverridden) {
+    (0, otel_js_1.setDefaultOTLPTracerComponents)(defaultComponents);
 }
-(0, otel_js_1.setDefaultOTLPTracerComponents)(getDefaultOTLPTracerComponents());

package/dist/experimental/otel/setup.js

@@ -7,14 +7,6 @@ import { BatchSpanProcessor, BasicTracerProvider, } from "@opentelemetry/sdk-tra
 import { getEnvironmentVariable, getLangSmithEnvironmentVariable, setEnvironmentVariable, } from "../../utils/env.js";
 import { LangSmithOTLPTraceExporter } from "./exporter.js";
 import { setDefaultOTLPTracerComponents, setOTELInstances, } from "../../singletons/otel.js";
-const otel = {
-    trace: otel_trace,
-    context: otel_context,
-};
-setOTELInstances(otel);
-const contextManager = new AsyncHooksContextManager();
-contextManager.enable();
-otel_context.setGlobalContextManager(contextManager);
 /**
  * Convert headers string in format "name=value,name2=value2" to object
  */
@@ -30,40 +22,50 @@ function parseHeadersString(headersStr) {
     });
     return headers;
 }
-function getDefaultOTLPTracerComponents() {
-    // Set LangSmith-specific defaults if not already set in environment
-    if (!getEnvironmentVariable("OTEL_EXPORTER_OTLP_ENDPOINT")) {
-        const lsEndpoint = getLangSmithEnvironmentVariable("ENDPOINT") ||
-            "https://api.smith.langchain.com";
-        const baseUrl = lsEndpoint.replace(/\/$/, "");
-        setEnvironmentVariable("OTEL_EXPORTER_OTLP_ENDPOINT", `${baseUrl}/otel/v1/traces`);
+// Set LangSmith-specific defaults if not already set in environment
+if (!getEnvironmentVariable("OTEL_EXPORTER_OTLP_ENDPOINT")) {
+    const lsEndpoint = getLangSmithEnvironmentVariable("ENDPOINT") ||
+        "https://api.smith.langchain.com";
+    const baseUrl = lsEndpoint.replace(/\/$/, "");
+    setEnvironmentVariable("OTEL_EXPORTER_OTLP_ENDPOINT", `${baseUrl}/otel/v1/traces`);
+}
+// Configure headers with API key and project if available
+if (!getEnvironmentVariable("OTEL_EXPORTER_OTLP_HEADERS")) {
+    const apiKey = getLangSmithEnvironmentVariable("API_KEY");
+    if (!apiKey) {
+        throw new Error("LANGSMITH_API_KEY or LANGCHAIN_API_KEY environment variable is required");
     }
-    // Configure headers with API key and project if available
-    if (!getEnvironmentVariable("OTEL_EXPORTER_OTLP_HEADERS")) {
-        const apiKey = getLangSmithEnvironmentVariable("API_KEY");
-        if (!apiKey) {
-            throw new Error("LANGSMITH_API_KEY or LANGCHAIN_API_KEY environment variable is required");
-        }
-        let headers = `x-api-key=${apiKey}`;
-        const project = getLangSmithEnvironmentVariable("PROJECT");
-        if (project) {
-            headers += `,Langsmith-Project=${project}`;
-        }
-        setEnvironmentVariable("OTEL_EXPORTER_OTLP_HEADERS", headers);
+    let headers = `x-api-key=${apiKey}`;
+    const project = getLangSmithEnvironmentVariable("PROJECT");
+    if (project) {
+        headers += `,Langsmith-Project=${project}`;
     }
-    const headersStr = getEnvironmentVariable("OTEL_EXPORTER_OTLP_HEADERS") || "";
-    const headersObj = parseHeadersString(headersStr);
-    const langsmithSpanExporter = new LangSmithOTLPTraceExporter({
-        url: getEnvironmentVariable("OTEL_EXPORTER_OTLP_ENDPOINT"),
-        headers: headersObj,
-    });
-    const spanProcessor = new BatchSpanProcessor(langsmithSpanExporter);
-    return {
-        tracerProvider: new BasicTracerProvider({
-            spanProcessors: [spanProcessor],
-        }),
-        spanProcessor,
-        langsmithSpanExporter,
-    };
+    setEnvironmentVariable("OTEL_EXPORTER_OTLP_HEADERS", headers);
+}
+const headersStr = getEnvironmentVariable("OTEL_EXPORTER_OTLP_HEADERS") || "";
+const headersObj = parseHeadersString(headersStr);
+const langsmithSpanExporter = new LangSmithOTLPTraceExporter({
+    url: getEnvironmentVariable("OTEL_EXPORTER_OTLP_ENDPOINT"),
+    headers: headersObj,
+});
+const spanProcessor = new BatchSpanProcessor(langsmithSpanExporter);
+const otel = {
+    trace: otel_trace,
+    context: otel_context,
+};
+setOTELInstances(otel);
+const contextManager = new AsyncHooksContextManager();
+contextManager.enable();
+otel_context.setGlobalContextManager(contextManager);
+const defaultComponents = {
+    spanProcessor,
+    tracerProvider: new BasicTracerProvider({
+        spanProcessors: [spanProcessor],
+    }),
+    langsmithSpanExporter,
+};
+// If user has set global tracer before, this fails and returns false
+const globalSuccessfullyOverridden = otel_trace.setGlobalTracerProvider(defaultComponents.tracerProvider);
+if (globalSuccessfullyOverridden) {
+    setDefaultOTLPTracerComponents(defaultComponents);
 }
-setDefaultOTLPTracerComponents(getDefaultOTLPTracerComponents());

package/dist/experimental/otel/translator.cjs

@@ -34,7 +34,6 @@ var __importStar = (this && this.__importStar) || (function () {
 })();
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LangSmithToOTELTranslator = void 0;
-const index_js_1 = require("../../index.cjs");
 const constants = __importStar(require("./constants.cjs"));
 const otel_js_1 = require("../../singletons/otel.cjs");
 const WELL_KNOWN_OPERATION_NAMES = {
@@ -48,25 +47,15 @@ function getOperationName(runType) {
     return WELL_KNOWN_OPERATION_NAMES[runType] || runType;
 }
 class LangSmithToOTELTranslator {
-    constructor(tracerProvider) {
-        Object.defineProperty(this, "tracer", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
+    constructor() {
         Object.defineProperty(this, "spans", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: new Map()
         });
-        this.tracer = (tracerProvider ?? (0, otel_js_1.getOTELTrace)()).getTracer("langsmith", index_js_1.__version__);
     }
     exportBatch(operations, otelContextMap) {
-        if (!this.tracer) {
-            return;
-        }
         for (const op of operations) {
             try {
                 if (!op.run) {
@@ -74,7 +63,7 @@ class LangSmithToOTELTranslator {
                 }
                 if (op.operation === "post") {
                     const span = this.createSpanForRun(op, op.run, otelContextMap.get(op.id));
-                    if (span) {
+                    if (span && !op.run.end_time) {
                         this.spans.set(op.id, span);
                     }
                 }
@@ -88,23 +77,19 @@ class LangSmithToOTELTranslator {
         }
     }
     createSpanForRun(op, runInfo, otelContext) {
-        if (!this.tracer) {
-            return;
-        }
         const activeSpan = otelContext && (0, otel_js_1.getOTELTrace)().getSpan(otelContext);
         if (!activeSpan) {
             return;
         }
         try {
-            const endTime = runInfo.end_time;
-            return this.finishSpanSetup(activeSpan, runInfo, op, endTime);
+            return this.finishSpanSetup(activeSpan, runInfo, op);
         }
         catch (e) {
             console.error(`Failed to create span for run ${op.id}:`, e);
             return undefined;
         }
     }
-    finishSpanSetup(span, runInfo, op, endTime) {
+    finishSpanSetup(span, runInfo, op) {
         // Set all attributes
         this.setSpanAttributes(span, runInfo, op);
         // Set status based on error
@@ -116,8 +101,8 @@ class LangSmithToOTELTranslator {
             span.setStatus({ code: 1 }); // OK status
         }
         // End the span if end_time is present
-        if (endTime) {
-            span.end(endTime);
+        if (runInfo.end_time) {
+            span.end(runInfo.end_time);
         }
         return span;
     }
@@ -142,7 +127,6 @@ class LangSmithToOTELTranslator {
             const endTime = runInfo.end_time;
             if (endTime) {
                 span.end(endTime);
-                // Remove the span from our dictionary
                 this.spans.delete(op.id);
             }
         }

package/dist/experimental/otel/translator.d.ts

@@ -1,4 +1,4 @@
-import type { OTELContext, OTELTracerProvider } from "./types.js";
+import type { OTELContext } from "./types.js";
 import type { RunCreate, RunUpdate } from "../../schemas.js";
 export type SerializedRunOperation<T extends "post" | "patch" = "post" | "patch"> = {
     operation: T;
@@ -7,9 +7,7 @@ export type SerializedRunOperation<T extends "post" | "patch" = "post" | "patch"
     run: T extends "post" ? RunCreate : RunUpdate;
 };
 export declare class LangSmithToOTELTranslator {
-    private tracer?;
     private spans;
-    constructor(tracerProvider?: OTELTracerProvider);
     exportBatch(operations: SerializedRunOperation[], otelContextMap: Map<string, OTELContext>): void;
     private createSpanForRun;
     private finishSpanSetup;

package/dist/experimental/otel/translator.js

@@ -1,4 +1,3 @@
-import { __version__ } from "../../index.js";
 import * as constants from "./constants.js";
 import { getOTELTrace } from "../../singletons/otel.js";
 const WELL_KNOWN_OPERATION_NAMES = {
@@ -12,25 +11,15 @@ function getOperationName(runType) {
     return WELL_KNOWN_OPERATION_NAMES[runType] || runType;
 }
 export class LangSmithToOTELTranslator {
-    constructor(tracerProvider) {
-        Object.defineProperty(this, "tracer", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
+    constructor() {
         Object.defineProperty(this, "spans", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: new Map()
         });
-        this.tracer = (tracerProvider ?? getOTELTrace()).getTracer("langsmith", __version__);
     }
     exportBatch(operations, otelContextMap) {
-        if (!this.tracer) {
-            return;
-        }
         for (const op of operations) {
             try {
                 if (!op.run) {
@@ -38,7 +27,7 @@ export class LangSmithToOTELTranslator {
                 }
                 if (op.operation === "post") {
                     const span = this.createSpanForRun(op, op.run, otelContextMap.get(op.id));
-                    if (span) {
+                    if (span && !op.run.end_time) {
                         this.spans.set(op.id, span);
                     }
                 }
@@ -52,23 +41,19 @@ export class LangSmithToOTELTranslator {
         }
     }
     createSpanForRun(op, runInfo, otelContext) {
-        if (!this.tracer) {
-            return;
-        }
         const activeSpan = otelContext && getOTELTrace().getSpan(otelContext);
         if (!activeSpan) {
             return;
         }
         try {
-            const endTime = runInfo.end_time;
-            return this.finishSpanSetup(activeSpan, runInfo, op, endTime);
+            return this.finishSpanSetup(activeSpan, runInfo, op);
         }
         catch (e) {
             console.error(`Failed to create span for run ${op.id}:`, e);
             return undefined;
         }
     }
-    finishSpanSetup(span, runInfo, op, endTime) {
+    finishSpanSetup(span, runInfo, op) {
         // Set all attributes
         this.setSpanAttributes(span, runInfo, op);
         // Set status based on error
@@ -80,8 +65,8 @@ export class LangSmithToOTELTranslator {
             span.setStatus({ code: 1 }); // OK status
         }
         // End the span if end_time is present
-        if (endTime) {
-            span.end(endTime);
+        if (runInfo.end_time) {
+            span.end(runInfo.end_time);
         }
         return span;
     }
@@ -106,7 +91,6 @@ export class LangSmithToOTELTranslator {
             const endTime = runInfo.end_time;
             if (endTime) {
                 span.end(endTime);
-                // Remove the span from our dictionary
                 this.spans.delete(op.id);
             }
         }

package/dist/index.cjs

@@ -10,4 +10,4 @@ Object.defineProperty(exports, "overrideFetchImplementation", { enumerable: true
 var project_js_1 = require("./utils/project.cjs");
 Object.defineProperty(exports, "getDefaultProjectName", { enumerable: true, get: function () { return project_js_1.getDefaultProjectName; } });
 // Update using yarn bump-version
-exports.__version__ = "0.3.35-rc.0";
+exports.__version__ = "0.3.35-rc.2";

package/dist/index.d.ts

@@ -3,4 +3,4 @@ export type { Dataset, Example, TracerSession, Run, Feedback, RetrieverOutput, }
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
-export declare const __version__ = "0.3.35-rc.0";
+export declare const __version__ = "0.3.35-rc.2";

package/dist/index.js

@@ -3,4 +3,4 @@ export { RunTree } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 // Update using yarn bump-version
-export const __version__ = "0.3.35-rc.0";
+export const __version__ = "0.3.35-rc.2";

package/dist/traceable.cjs

@@ -9,14 +9,14 @@ const traceable_js_1 = require("./singletons/traceable.cjs");
 const constants_js_1 = require("./singletons/constants.cjs");
 const asserts_js_1 = require("./utils/asserts.cjs");
 const env_js_2 = require("./utils/env.cjs");
-const utils_js_1 = require("./experimental/otel/utils.cjs");
 const index_js_1 = require("./index.cjs");
 const otel_js_1 = require("./singletons/otel.cjs");
+const utils_js_1 = require("./experimental/otel/utils.cjs");
 traceable_js_1.AsyncLocalStorageProviderSingleton.initializeGlobalInstance(new node_async_hooks_1.AsyncLocalStorage());
 /**
  * Create OpenTelemetry context manager from RunTree if OTEL is enabled.
  */
-function maybeCreateOtelContext(runTree) {
+function maybeCreateOtelContext(runTree, tracer) {
     if (!runTree || (0, env_js_2.getEnvironmentVariable)("OTEL_ENABLED") !== "true") {
         return;
     }
@@ -25,7 +25,7 @@ function maybeCreateOtelContext(runTree) {
     try {
         const spanContext = (0, utils_js_1.createOtelSpanContextFromRun)(runTree);
         return (fn) => {
-            const resolvedTracer = otel_trace.getTracer("langsmith", index_js_1.__version__);
+            const resolvedTracer = tracer ?? otel_trace.getTracer("langsmith", index_js_1.__version__);
             return resolvedTracer.startActiveSpan(runTree.name, {
                 attributes: {
                     "langsmith.traceable": "true",
@@ -383,7 +383,7 @@ function traceable(wrappedFunc, config) {
                 const currentRunTree = getTracingRunTree(firstArg === traceable_js_1.ROOT
                     ? new run_trees_js_1.RunTree(ensuredConfig)
                     : firstArg.createChild(ensuredConfig), restArgs, config?.getInvocationParams, processInputsFn, extractAttachmentsFn);
-                return [currentRunTree, restArgs];
+                return [currentRunTree, [currentRunTree, ...restArgs]];
             }
             // Node.JS uses AsyncLocalStorage (ALS) and AsyncResource
             // to allow storing context
@@ -406,7 +406,7 @@ function traceable(wrappedFunc, config) {
             }
             return [currentRunTree, processedArgs];
         })();
-        const otelContextManager = maybeCreateOtelContext(currentRunTree);
+        const otelContextManager = maybeCreateOtelContext(currentRunTree, config?.tracer);
         const otel_context = (0, otel_js_1.getOTELContext)();
         const runWithContext = () => {
             const postRunPromise = currentRunTree?.postRun();

package/dist/traceable.d.ts

@@ -1,6 +1,7 @@
 import { RunTreeConfig } from "./run_trees.js";
 import { Attachments, InvocationParamsSchema, KVMap } from "./schemas.js";
 import { TraceableFunction } from "./singletons/types.js";
+import { OTELTracer } from "./experimental/otel/types.js";
 /**
  * Higher-order function that takes function as input and returns a
  * "TraceableFunction" - a wrapped version of the input that
@@ -18,6 +19,7 @@ import { TraceableFunction } from "./singletons/types.js";
 export declare function traceable<Func extends (...args: any[]) => any>(wrappedFunc: Func, config?: Partial<Omit<RunTreeConfig, "inputs" | "outputs">> & {
     aggregator?: (args: any[]) => any;
     argsConfigPath?: [number] | [number, string];
+    tracer?: OTELTracer;
     __finalTracedIteratorKey?: string;
     /**
      * Extract attachments from args and return remaining args.

package/dist/traceable.js

@@ -5,14 +5,14 @@ import { ROOT, AsyncLocalStorageProviderSingleton, } from "./singletons/traceabl
 import { _LC_CONTEXT_VARIABLES_KEY } from "./singletons/constants.js";
 import { isKVMap, isReadableStream, isAsyncIterable, isIteratorLike, isThenable, isGenerator, isPromiseMethod, } from "./utils/asserts.js";
 import { getEnvironmentVariable } from "./utils/env.js";
-import { createOtelSpanContextFromRun } from "./experimental/otel/utils.js";
 import { __version__ } from "./index.js";
 import { getOTELTrace, getOTELContext } from "./singletons/otel.js";
+import { createOtelSpanContextFromRun } from "./experimental/otel/utils.js";
 AsyncLocalStorageProviderSingleton.initializeGlobalInstance(new AsyncLocalStorage());
 /**
  * Create OpenTelemetry context manager from RunTree if OTEL is enabled.
  */
-function maybeCreateOtelContext(runTree) {
+function maybeCreateOtelContext(runTree, tracer) {
     if (!runTree || getEnvironmentVariable("OTEL_ENABLED") !== "true") {
         return;
     }
@@ -21,7 +21,7 @@ function maybeCreateOtelContext(runTree) {
     try {
         const spanContext = createOtelSpanContextFromRun(runTree);
         return (fn) => {
-            const resolvedTracer = otel_trace.getTracer("langsmith", __version__);
+            const resolvedTracer = tracer ?? otel_trace.getTracer("langsmith", __version__);
             return resolvedTracer.startActiveSpan(runTree.name, {
                 attributes: {
                     "langsmith.traceable": "true",
@@ -379,7 +379,7 @@ export function traceable(wrappedFunc, config) {
                 const currentRunTree = getTracingRunTree(firstArg === ROOT
                     ? new RunTree(ensuredConfig)
                     : firstArg.createChild(ensuredConfig), restArgs, config?.getInvocationParams, processInputsFn, extractAttachmentsFn);
-                return [currentRunTree, restArgs];
+                return [currentRunTree, [currentRunTree, ...restArgs]];
             }
             // Node.JS uses AsyncLocalStorage (ALS) and AsyncResource
             // to allow storing context
@@ -402,7 +402,7 @@ export function traceable(wrappedFunc, config) {
             }
             return [currentRunTree, processedArgs];
         })();
-        const otelContextManager = maybeCreateOtelContext(currentRunTree);
+        const otelContextManager = maybeCreateOtelContext(currentRunTree, config?.tracer);
         const otel_context = getOTELContext();
         const runWithContext = () => {
             const postRunPromise = currentRunTree?.postRun();

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.3.35-rc.0",
+  "version": "0.3.35-rc.2",
   "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
   "packageManager": "yarn@1.22.19",
   "files": [