langsmith 0.1.31-rc.0 → 0.1.31

package/dist/run_trees.cjs

@@ -412,7 +412,13 @@ class RunTree {
         return this.fromHeaders({ "langsmith-trace": dottedOrder });
     }
     static fromHeaders(headers, inheritArgs) {
-        const headerTrace = headers["langsmith-trace"];
+        const rawHeaders = "get" in headers && typeof headers.get === "function"
+            ? {
+                "langsmith-trace": headers.get("langsmith-trace"),
+                baggage: headers.get("baggage"),
+            }
+            : headers;
+        const headerTrace = rawHeaders["langsmith-trace"];
         if (!headerTrace || typeof headerTrace !== "string")
             return undefined;
         const parentDottedOrder = headerTrace.trim();
@@ -430,18 +436,24 @@ class RunTree {
             trace_id: traceId,
             dotted_order: parentDottedOrder,
         };
-        if (headers["baggage"]) {
-            const baggage = Baggage.fromHeader(headers["baggage"]);
+        if (rawHeaders["baggage"] && typeof rawHeaders["baggage"] === "string") {
+            const baggage = Baggage.fromHeader(rawHeaders["baggage"]);
             config.metadata = baggage.metadata;
             config.tags = baggage.tags;
         }
         return new RunTree(config);
     }
-    toHeaders() {
-        return {
+    toHeaders(headers) {
+        const result = {
             "langsmith-trace": this.dotted_order,
             baggage: new Baggage(this.extra?.metadata, this.tags).toHeader(),
         };
+        if (headers) {
+            for (const [key, value] of Object.entries(result)) {
+                headers.set(key, value);
+            }
+        }
+        return result;
     }
 }
 exports.RunTree = RunTree;

package/dist/run_trees.d.ts

@@ -44,6 +44,10 @@ export interface RunnableConfigLike {
      */
     callbacks?: any;
 }
+interface HeadersLike {
+    get(name: string): string | null;
+    set(name: string, value: string): void;
+}
 export declare class RunTree implements BaseRun {
     id: string;
     name: RunTreeConfig["name"];
@@ -77,11 +81,12 @@ export declare class RunTree implements BaseRun {
     toJSON(): RunCreate;
     static fromRunnableConfig(parentConfig: RunnableConfigLike, props: RunTreeConfig): RunTree;
     static fromDottedOrder(dottedOrder: string): RunTree | undefined;
-    static fromHeaders(headers: Record<string, string>, inheritArgs?: RunTreeConfig): RunTree | undefined;
-    toHeaders(): {
+    static fromHeaders(headers: Record<string, string | string[]> | HeadersLike, inheritArgs?: RunTreeConfig): RunTree | undefined;
+    toHeaders(headers?: HeadersLike): {
         "langsmith-trace": string;
         baggage: string;
     };
 }
 export declare function isRunTree(x?: unknown): x is RunTree;
 export declare function isRunnableConfigLike(x?: unknown): x is RunnableConfigLike;
+export {};

package/dist/run_trees.js

@@ -385,7 +385,13 @@ export class RunTree {
         return this.fromHeaders({ "langsmith-trace": dottedOrder });
     }
     static fromHeaders(headers, inheritArgs) {
-        const headerTrace = headers["langsmith-trace"];
+        const rawHeaders = "get" in headers && typeof headers.get === "function"
+            ? {
+                "langsmith-trace": headers.get("langsmith-trace"),
+                baggage: headers.get("baggage"),
+            }
+            : headers;
+        const headerTrace = rawHeaders["langsmith-trace"];
         if (!headerTrace || typeof headerTrace !== "string")
             return undefined;
         const parentDottedOrder = headerTrace.trim();
@@ -403,18 +409,24 @@ export class RunTree {
             trace_id: traceId,
             dotted_order: parentDottedOrder,
         };
-        if (headers["baggage"]) {
-            const baggage = Baggage.fromHeader(headers["baggage"]);
+        if (rawHeaders["baggage"] && typeof rawHeaders["baggage"] === "string") {
+            const baggage = Baggage.fromHeader(rawHeaders["baggage"]);
             config.metadata = baggage.metadata;
             config.tags = baggage.tags;
         }
         return new RunTree(config);
     }
-    toHeaders() {
-        return {
+    toHeaders(headers) {
+        const result = {
             "langsmith-trace": this.dotted_order,
             baggage: new Baggage(this.extra?.metadata, this.tags).toHeader(),
         };
+        if (headers) {
+            for (const [key, value] of Object.entries(result)) {
+                headers.set(key, value);
+            }
+        }
+        return result;
     }
 }
 export function isRunTree(x) {

package/dist/singletons/traceable.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.isTraceableFunction = exports.ROOT = exports.getCurrentRunTree = exports.AsyncLocalStorageProviderSingleton = void 0;
+exports.isTraceableFunction = exports.ROOT = exports.withRunTree = exports.getCurrentRunTree = exports.AsyncLocalStorageProviderSingleton = void 0;
 class MockAsyncLocalStorage {
     getStore() {
         return undefined;
@@ -47,12 +47,20 @@ const getCurrentRunTree = () => {
         throw new Error([
             "Could not get the current run tree.",
             "",
-            "Please make sure you are calling this method within a traceable function.",
+            "Please make sure you are calling this method within a traceable function or the tracing is enabled.",
         ].join("\n"));
     }
     return runTree;
 };
 exports.getCurrentRunTree = getCurrentRunTree;
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+function withRunTree(runTree, fn) {
+    const storage = exports.AsyncLocalStorageProviderSingleton.getInstance();
+    return new Promise((resolve, reject) => {
+        storage.run(runTree, () => void Promise.resolve(fn()).then(resolve).catch(reject));
+    });
+}
+exports.withRunTree = withRunTree;
 exports.ROOT = Symbol.for("langsmith:traceable:root");
 function isTraceableFunction(x
 // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/singletons/traceable.d.ts

@@ -18,6 +18,7 @@ export declare const AsyncLocalStorageProviderSingleton: AsyncLocalStorageProvid
  * @returns The run tree for the given context.
  */
 export declare const getCurrentRunTree: () => RunTree;
+export declare function withRunTree<Fn extends (...args: any[]) => any>(runTree: RunTree, fn: Fn): Promise<Awaited<ReturnType<Fn>>>;
 export declare const ROOT: unique symbol;
 export declare function isTraceableFunction(x: unknown): x is TraceableFunction<any>;
 export type { TraceableFunction } from "./types.js";

package/dist/singletons/traceable.js

@@ -44,11 +44,18 @@ export const getCurrentRunTree = () => {
         throw new Error([
             "Could not get the current run tree.",
             "",
-            "Please make sure you are calling this method within a traceable function.",
+            "Please make sure you are calling this method within a traceable function or the tracing is enabled.",
         ].join("\n"));
     }
     return runTree;
 };
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+export function withRunTree(runTree, fn) {
+    const storage = AsyncLocalStorageProviderSingleton.getInstance();
+    return new Promise((resolve, reject) => {
+        storage.run(runTree, () => void Promise.resolve(fn()).then(resolve).catch(reject));
+    });
+}
 export const ROOT = Symbol.for("langsmith:traceable:root");
 export function isTraceableFunction(x
 // eslint-disable-next-line @typescript-eslint/no-explicit-any

package/dist/traceable.cjs

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ROOT = exports.isTraceableFunction = exports.getCurrentRunTree = exports.traceable = void 0;
+exports.ROOT = exports.withRunTree = exports.isTraceableFunction = exports.getCurrentRunTree = exports.traceable = void 0;
 const node_async_hooks_1 = require("node:async_hooks");
 const run_trees_js_1 = require("./run_trees.cjs");
 const env_js_1 = require("./env.cjs");
@@ -35,8 +35,8 @@ const getTracingRunTree = (runTree, inputs, getInvocationParams) => {
     if (invocationParams != null) {
         runTree.extra ??= {};
         runTree.extra.metadata = {
-            ...runTree.extra.metadata,
             ...invocationParams,
+            ...runTree.extra.metadata,
         };
     }
     return runTree;
@@ -418,4 +418,5 @@ exports.traceable = traceable;
 var traceable_js_2 = require("./singletons/traceable.cjs");
 Object.defineProperty(exports, "getCurrentRunTree", { enumerable: true, get: function () { return traceable_js_2.getCurrentRunTree; } });
 Object.defineProperty(exports, "isTraceableFunction", { enumerable: true, get: function () { return traceable_js_2.isTraceableFunction; } });
+Object.defineProperty(exports, "withRunTree", { enumerable: true, get: function () { return traceable_js_2.withRunTree; } });
 Object.defineProperty(exports, "ROOT", { enumerable: true, get: function () { return traceable_js_2.ROOT; } });

package/dist/traceable.d.ts

@@ -28,5 +28,5 @@ export declare function traceable<Func extends (...args: any[]) => any>(wrappedF
      */
     getInvocationParams?: (...args: Parameters<Func>) => InvocationParamsSchema | undefined;
 }): TraceableFunction<Func>;
-export { getCurrentRunTree, isTraceableFunction, ROOT, } from "./singletons/traceable.js";
+export { getCurrentRunTree, isTraceableFunction, withRunTree, ROOT, } from "./singletons/traceable.js";
 export type { RunTreeLike, TraceableFunction } from "./singletons/types.js";

package/dist/traceable.js

@@ -32,8 +32,8 @@ const getTracingRunTree = (runTree, inputs, getInvocationParams) => {
     if (invocationParams != null) {
         runTree.extra ??= {};
         runTree.extra.metadata = {
-            ...runTree.extra.metadata,
             ...invocationParams,
+            ...runTree.extra.metadata,
         };
     }
     return runTree;
@@ -411,4 +411,4 @@ export function traceable(wrappedFunc, config) {
     });
     return traceableFunc;
 }
-export { getCurrentRunTree, isTraceableFunction, ROOT, } from "./singletons/traceable.js";
+export { getCurrentRunTree, isTraceableFunction, withRunTree, ROOT, } from "./singletons/traceable.js";

package/package.json

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