langsmith 0.3.85 → 0.3.86

package/dist/index.cjs

@@ -13,4 +13,4 @@ var uuid_js_1 = require("./uuid.cjs");
 Object.defineProperty(exports, "uuid7", { enumerable: true, get: function () { return uuid_js_1.uuid7; } });
 Object.defineProperty(exports, "uuid7FromTime", { enumerable: true, get: function () { return uuid_js_1.uuid7FromTime; } });
 // Update using yarn bump-version
-exports.__version__ = "0.3.85";
+exports.__version__ = "0.3.86";

package/dist/index.d.ts

@@ -4,4 +4,4 @@ export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 export { uuid7, uuid7FromTime } from "./uuid.js";
-export declare const __version__ = "0.3.85";
+export declare const __version__ = "0.3.86";

package/dist/index.js

@@ -4,4 +4,4 @@ export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 export { uuid7, uuid7FromTime } from "./uuid.js";
 // Update using yarn bump-version
-export const __version__ = "0.3.85";
+export const __version__ = "0.3.86";

package/dist/run_trees.cjs

@@ -909,12 +909,14 @@ function isRunnableConfigLike(x) {
     // Check that it's an object with a callbacks arg
     // that has either a CallbackManagerLike object with a langchain tracer within it
     // or an array with a LangChainTracerLike object within it
+    const callbacks = x?.callbacks;
     return (x != null &&
-        typeof x.callbacks === "object" &&
+        typeof callbacks === "object" &&
         // Callback manager with a langchain tracer
-        (containsLangChainTracerLike(x.callbacks?.handlers) ||
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (containsLangChainTracerLike(callbacks?.handlers) ||
             // Or it's an array with a LangChainTracerLike object within it
-            containsLangChainTracerLike(x.callbacks)));
+            containsLangChainTracerLike(callbacks)));
 }
 function _getWriteReplicasFromEnv() {
     const envVar = (0, env_js_2.getEnvironmentVariable)("LANGSMITH_RUNS_ENDPOINTS");

package/dist/run_trees.d.ts

@@ -48,7 +48,7 @@ export interface RunnableConfigLike {
      * Callbacks for this call and any sub-calls (eg. a Chain calling an LLM).
      * Tags are passed to all callbacks, metadata is passed to handle*Start callbacks.
      */
-    callbacks?: any;
+    callbacks?: Record<string, any> | any[];
 }
 interface HeadersLike {
     get(name: string): string | null;

package/dist/run_trees.js

@@ -902,12 +902,14 @@ export function isRunnableConfigLike(x) {
     // Check that it's an object with a callbacks arg
     // that has either a CallbackManagerLike object with a langchain tracer within it
     // or an array with a LangChainTracerLike object within it
+    const callbacks = x?.callbacks;
     return (x != null &&
-        typeof x.callbacks === "object" &&
+        typeof callbacks === "object" &&
         // Callback manager with a langchain tracer
-        (containsLangChainTracerLike(x.callbacks?.handlers) ||
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        (containsLangChainTracerLike(callbacks?.handlers) ||
             // Or it's an array with a LangChainTracerLike object within it
-            containsLangChainTracerLike(x.callbacks)));
+            containsLangChainTracerLike(callbacks)));
 }
 function _getWriteReplicasFromEnv() {
     const envVar = getEnvironmentVariable("LANGSMITH_RUNS_ENDPOINTS");

package/dist/singletons/types.d.ts

@@ -1,16 +1,18 @@
 import { RunTree, RunnableConfigLike } from "../run_trees.js";
 import { ROOT } from "./traceable.js";
 import { _LC_CONTEXT_VARIABLES_KEY } from "./constants.js";
-type SmartPromise<T> = T extends AsyncGenerator ? T : T extends Promise<unknown> ? T : Promise<T>;
+type SmartPromise<T> = T extends AsyncIterable<any, any, any> ? T : T extends Promise<unknown> ? T : Promise<T>;
 type WrapArgReturnPair<Pair> = Pair extends [
     infer Args extends any[],
     infer Return
 ] ? Args extends [RunTree, ...infer RestArgs] ? {
     (runTree: RunTree | typeof ROOT, ...args: RestArgs): SmartPromise<Return>;
+    /** @deprecated Will be removed in 0.4 */
     (config: RunnableConfigLike, ...args: RestArgs): SmartPromise<Return>;
 } : {
     (...args: Args): SmartPromise<Return>;
-    (runTree: RunTree, ...rest: Args): SmartPromise<Return>;
+    (runTree: RunTree | typeof ROOT, ...rest: Args): SmartPromise<Return>;
+    /** @deprecated Will be removed in 0.4 */
     (config: RunnableConfigLike, ...args: Args): SmartPromise<Return>;
 } : never;
 type UnionToIntersection<U> = (U extends any ? (x: U) => void : never) extends (x: infer I) => void ? I : never;

package/dist/wrappers/openai.cjs

@@ -313,6 +313,36 @@ const wrapOpenAI = (openai, options) => {
             tracedOpenAIClient.beta.chat.completions.parse = (0, traceable_js_1.traceable)(openai.beta.chat.completions.parse.bind(openai.beta.chat.completions), chatCompletionParseMetadata);
         }
     }
+    // Shared function to wrap stream methods (similar to wrapAnthropic)
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const wrapStreamMethod = (originalStreamFn) => {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return function (...args) {
+            const stream = originalStreamFn(...args);
+            // Helper to ensure stream is fully consumed before calling final methods
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const ensureStreamConsumed = (methodName) => {
+                if (methodName in stream && typeof stream[methodName] === "function") {
+                    const originalMethod = stream[methodName].bind(stream);
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    stream[methodName] = async (...args) => {
+                        if ("done" in stream && typeof stream.done === "function") {
+                            await stream.done();
+                        }
+                        for await (const _ of stream) {
+                            // Finish consuming the stream if it has not already been consumed
+                        }
+                        return originalMethod(...args);
+                    };
+                }
+            };
+            // Ensure stream is consumed for final methods
+            ensureStreamConsumed("finalChatCompletion");
+            ensureStreamConsumed("finalMessage");
+            ensureStreamConsumed("finalResponse");
+            return stream;
+        };
+    };
     tracedOpenAIClient.chat = {
         ...openai.chat,
         completions: Object.create(Object.getPrototypeOf(openai.chat.completions)),
@@ -325,6 +355,17 @@ const wrapOpenAI = (openai, options) => {
     if (typeof openai.chat.completions.parse === "function") {
         tracedOpenAIClient.chat.completions.parse = (0, traceable_js_1.traceable)(openai.chat.completions.parse.bind(openai.chat.completions), chatCompletionParseMetadata);
     }
+    // Wrap chat.completions.stream if it exists
+    if (typeof openai.chat.completions.stream === "function") {
+        tracedOpenAIClient.chat.completions.stream = (0, traceable_js_1.traceable)(wrapStreamMethod(openai.chat.completions.stream.bind(openai.chat.completions)), chatCompletionParseMetadata);
+    }
+    // Wrap beta.chat.completions.stream if it exists
+    if (openai.beta &&
+        openai.beta.chat &&
+        openai.beta.chat.completions &&
+        typeof openai.beta.chat.completions.stream === "function") {
+        tracedOpenAIClient.beta.chat.completions.stream = (0, traceable_js_1.traceable)(wrapStreamMethod(openai.beta.chat.completions.stream.bind(openai.beta.chat.completions)), chatCompletionParseMetadata);
+    }
     tracedOpenAIClient.completions = {
         ...openai.completions,
         create: (0, traceable_js_1.traceable)(openai.completions.create.bind(openai.completions), {
@@ -413,7 +454,7 @@ const wrapOpenAI = (openai, options) => {
         }
         if (tracedOpenAIClient.responses &&
             typeof tracedOpenAIClient.responses.stream === "function") {
-            tracedOpenAIClient.responses.stream = (0, traceable_js_1.traceable)(openai.responses.stream.bind(openai.responses), {
+            tracedOpenAIClient.responses.stream = (0, traceable_js_1.traceable)(wrapStreamMethod(openai.responses.stream.bind(openai.responses)), {
                 name: chatName,
                 run_type: "llm",
                 aggregator: responsesAggregator,

package/dist/wrappers/openai.d.ts

@@ -6,7 +6,8 @@ type OpenAIType = {
     chat: {
         completions: {
             create: (...args: any[]) => any;
-            parse: (...args: any[]) => any;
+            parse?: (...args: any[]) => any;
+            stream?: (...args: any[]) => any;
         };
     };
     completions: {

package/dist/wrappers/openai.js

@@ -310,6 +310,36 @@ export const wrapOpenAI = (openai, options) => {
             tracedOpenAIClient.beta.chat.completions.parse = traceable(openai.beta.chat.completions.parse.bind(openai.beta.chat.completions), chatCompletionParseMetadata);
         }
     }
+    // Shared function to wrap stream methods (similar to wrapAnthropic)
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    const wrapStreamMethod = (originalStreamFn) => {
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        return function (...args) {
+            const stream = originalStreamFn(...args);
+            // Helper to ensure stream is fully consumed before calling final methods
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            const ensureStreamConsumed = (methodName) => {
+                if (methodName in stream && typeof stream[methodName] === "function") {
+                    const originalMethod = stream[methodName].bind(stream);
+                    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+                    stream[methodName] = async (...args) => {
+                        if ("done" in stream && typeof stream.done === "function") {
+                            await stream.done();
+                        }
+                        for await (const _ of stream) {
+                            // Finish consuming the stream if it has not already been consumed
+                        }
+                        return originalMethod(...args);
+                    };
+                }
+            };
+            // Ensure stream is consumed for final methods
+            ensureStreamConsumed("finalChatCompletion");
+            ensureStreamConsumed("finalMessage");
+            ensureStreamConsumed("finalResponse");
+            return stream;
+        };
+    };
     tracedOpenAIClient.chat = {
         ...openai.chat,
         completions: Object.create(Object.getPrototypeOf(openai.chat.completions)),
@@ -322,6 +352,17 @@ export const wrapOpenAI = (openai, options) => {
     if (typeof openai.chat.completions.parse === "function") {
         tracedOpenAIClient.chat.completions.parse = traceable(openai.chat.completions.parse.bind(openai.chat.completions), chatCompletionParseMetadata);
     }
+    // Wrap chat.completions.stream if it exists
+    if (typeof openai.chat.completions.stream === "function") {
+        tracedOpenAIClient.chat.completions.stream = traceable(wrapStreamMethod(openai.chat.completions.stream.bind(openai.chat.completions)), chatCompletionParseMetadata);
+    }
+    // Wrap beta.chat.completions.stream if it exists
+    if (openai.beta &&
+        openai.beta.chat &&
+        openai.beta.chat.completions &&
+        typeof openai.beta.chat.completions.stream === "function") {
+        tracedOpenAIClient.beta.chat.completions.stream = traceable(wrapStreamMethod(openai.beta.chat.completions.stream.bind(openai.beta.chat.completions)), chatCompletionParseMetadata);
+    }
     tracedOpenAIClient.completions = {
         ...openai.completions,
         create: traceable(openai.completions.create.bind(openai.completions), {
@@ -410,7 +451,7 @@ export const wrapOpenAI = (openai, options) => {
         }
         if (tracedOpenAIClient.responses &&
             typeof tracedOpenAIClient.responses.stream === "function") {
-            tracedOpenAIClient.responses.stream = traceable(openai.responses.stream.bind(openai.responses), {
+            tracedOpenAIClient.responses.stream = traceable(wrapStreamMethod(openai.responses.stream.bind(openai.responses)), {
                 name: chatName,
                 run_type: "llm",
                 aggregator: responsesAggregator,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.3.85",
+  "version": "0.3.86",
   "description": "Client library to connect to the LangSmith Observability and Evaluation Platform.",
   "packageManager": "yarn@1.22.19",
   "files": [