@langchain/core 0.2.17 → 0.2.18

package/dist/tracers/base.js

@@ -12,6 +12,9 @@ function convertToDottedOrderFormat(epoch, runId, executionOrder) {
     const paddedOrder = executionOrder.toFixed(0).slice(0, 3).padStart(3, "0");
     return (stripNonAlphanumeric(`${new Date(epoch).toISOString().slice(0, -1)}${paddedOrder}Z`) + runId);
 }
+export function isBaseTracer(x) {
+    return typeof x._addRunToRunMap === "function";
+}
 export class BaseTracer extends BaseCallbackHandler {
     constructor(_fields) {
         super(...arguments);
@@ -38,7 +41,7 @@ export class BaseTracer extends BaseCallbackHandler {
     _addChildRun(parentRun, childRun) {
         parentRun.child_runs.push(childRun);
     }
-    async _startTrace(run) {
+    _addRunToRunMap(run) {
         const currentDottedOrder = convertToDottedOrderFormat(run.start_time, run.id, run.execution_order);
         const storedRun = { ...run };
         if (storedRun.parent_run_id !== undefined) {
@@ -70,7 +73,7 @@ export class BaseTracer extends BaseCallbackHandler {
             storedRun.dotted_order = currentDottedOrder;
         }
         this.runMap.set(storedRun.id, storedRun);
-        await this.onRunCreate?.(storedRun);
+        return storedRun;
     }
     async _endTrace(run) {
         const parentRun = run.parent_run_id !== undefined && this.runMap.get(run.parent_run_id);
@@ -91,7 +94,12 @@ export class BaseTracer extends BaseCallbackHandler {
         }
         return parentRun.child_execution_order + 1;
     }
-    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
+    /**
+     * Create and add a run to the run map for LLM start events.
+     * This must sometimes be done synchronously to avoid race conditions
+     * when callbacks are backgrounded, so we expose it as a separate method here.
+     */
+    _createRunForLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const finalExtraParams = metadata
@@ -117,11 +125,21 @@ export class BaseTracer extends BaseCallbackHandler {
             extra: finalExtraParams ?? {},
             tags: tags || [],
         };
-        await this._startTrace(run);
+        return this._addRunToRunMap(run);
+    }
+    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
+        const run = this.runMap.get(runId) ??
+            this._createRunForLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name);
+        await this.onRunCreate?.(run);
         await this.onLLMStart?.(run);
         return run;
     }
-    async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
+    /**
+     * Create and add a run to the run map for chat model start events.
+     * This must sometimes be done synchronously to avoid race conditions
+     * when callbacks are backgrounded, so we expose it as a separate method here.
+     */
+    _createRunForChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const finalExtraParams = metadata
@@ -147,7 +165,12 @@ export class BaseTracer extends BaseCallbackHandler {
             extra: finalExtraParams ?? {},
             tags: tags || [],
         };
-        await this._startTrace(run);
+        return this._addRunToRunMap(run);
+    }
+    async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
+        const run = this.runMap.get(runId) ??
+            this._createRunForChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name);
+        await this.onRunCreate?.(run);
         await this.onLLMStart?.(run);
         return run;
     }
@@ -181,7 +204,12 @@ export class BaseTracer extends BaseCallbackHandler {
         await this._endTrace(run);
         return run;
     }
-    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
+    /**
+     * Create and add a run to the run map for chain start events.
+     * This must sometimes be done synchronously to avoid race conditions
+     * when callbacks are backgrounded, so we expose it as a separate method here.
+     */
+    _createRunForChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const run = {
@@ -204,7 +232,12 @@ export class BaseTracer extends BaseCallbackHandler {
             extra: metadata ? { metadata } : {},
             tags: tags || [],
         };
-        await this._startTrace(run);
+        return this._addRunToRunMap(run);
+    }
+    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
+        const run = this.runMap.get(runId) ??
+            this._createRunForChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name);
+        await this.onRunCreate?.(run);
         await this.onChainStart?.(run);
         return run;
     }
@@ -244,7 +277,12 @@ export class BaseTracer extends BaseCallbackHandler {
         await this._endTrace(run);
         return run;
     }
-    async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
+    /**
+     * Create and add a run to the run map for tool start events.
+     * This must sometimes be done synchronously to avoid race conditions
+     * when callbacks are backgrounded, so we expose it as a separate method here.
+     */
+    _createRunForToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const run = {
@@ -267,7 +305,12 @@ export class BaseTracer extends BaseCallbackHandler {
             extra: metadata ? { metadata } : {},
             tags: tags || [],
         };
-        await this._startTrace(run);
+        return this._addRunToRunMap(run);
+    }
+    async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
+        const run = this.runMap.get(runId) ??
+            this._createRunForToolStart(tool, input, runId, parentRunId, tags, metadata, name);
+        await this.onRunCreate?.(run);
         await this.onToolStart?.(run);
         return run;
     }
@@ -329,7 +372,12 @@ export class BaseTracer extends BaseCallbackHandler {
         });
         await this.onAgentEnd?.(run);
     }
-    async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
+    /**
+     * Create and add a run to the run map for retriever start events.
+     * This must sometimes be done synchronously to avoid race conditions
+     * when callbacks are backgrounded, so we expose it as a separate method here.
+     */
+    _createRunForRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const run = {
@@ -352,7 +400,12 @@ export class BaseTracer extends BaseCallbackHandler {
             extra: metadata ? { metadata } : {},
             tags: tags || [],
         };
-        await this._startTrace(run);
+        return this._addRunToRunMap(run);
+    }
+    async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
+        const run = this.runMap.get(runId) ??
+            this._createRunForRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name);
+        await this.onRunCreate?.(run);
         await this.onRetrieverStart?.(run);
         return run;
     }

package/dist/tracers/event_stream.cjs

@@ -166,6 +166,13 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
             yield firstChunk.value;
             return;
         }
+        // Match format from handlers below
+        function _formatOutputChunk(eventType, data) {
+            if (eventType === "llm" && typeof data === "string") {
+                return new outputs_js_1.GenerationChunk({ text: data });
+            }
+            return data;
+        }
         let tappedPromise = this.tappedPromises.get(runId);
         // if we are the first to tap, issue stream events
         if (tappedPromise === undefined) {
@@ -185,7 +192,9 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
                 };
                 await this.send({
                     ...event,
-                    data: { chunk: firstChunk.value },
+                    data: {
+                        chunk: _formatOutputChunk(runInfo.runType, firstChunk.value),
+                    },
                 }, runInfo);
                 yield firstChunk.value;
                 for await (const chunk of outputStream) {
@@ -194,7 +203,7 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
                         await this.send({
                             ...event,
                             data: {
-                                chunk,
+                                chunk: _formatOutputChunk(runInfo.runType, chunk),
                             },
                         }, runInfo);
                     }
@@ -263,6 +272,10 @@ class EventStreamCallbackHandler extends base_js_1.BaseTracer {
         if (runInfo === undefined) {
             throw new Error(`onLLMNewToken: Run ID ${run.id} not found in run map.`);
         }
+        // Top-level streaming events are covered by tapOutputIterable
+        if (run.parent_run_id === undefined) {
+            return;
+        }
         if (runInfo.runType === "chat_model") {
             eventName = "on_chat_model_stream";
             if (kwargs?.chunk === undefined) {

package/dist/tracers/event_stream.js

@@ -162,6 +162,13 @@ export class EventStreamCallbackHandler extends BaseTracer {
             yield firstChunk.value;
             return;
         }
+        // Match format from handlers below
+        function _formatOutputChunk(eventType, data) {
+            if (eventType === "llm" && typeof data === "string") {
+                return new GenerationChunk({ text: data });
+            }
+            return data;
+        }
         let tappedPromise = this.tappedPromises.get(runId);
         // if we are the first to tap, issue stream events
         if (tappedPromise === undefined) {
@@ -181,7 +188,9 @@ export class EventStreamCallbackHandler extends BaseTracer {
                 };
                 await this.send({
                     ...event,
-                    data: { chunk: firstChunk.value },
+                    data: {
+                        chunk: _formatOutputChunk(runInfo.runType, firstChunk.value),
+                    },
                 }, runInfo);
                 yield firstChunk.value;
                 for await (const chunk of outputStream) {
@@ -190,7 +199,7 @@ export class EventStreamCallbackHandler extends BaseTracer {
                         await this.send({
                             ...event,
                             data: {
-                                chunk,
+                                chunk: _formatOutputChunk(runInfo.runType, chunk),
                             },
                         }, runInfo);
                     }
@@ -259,6 +268,10 @@ export class EventStreamCallbackHandler extends BaseTracer {
         if (runInfo === undefined) {
             throw new Error(`onLLMNewToken: Run ID ${run.id} not found in run map.`);
         }
+        // Top-level streaming events are covered by tapOutputIterable
+        if (run.parent_run_id === undefined) {
+            return;
+        }
         if (runInfo.runType === "chat_model") {
             eventName = "on_chat_model_stream";
             if (kwargs?.chunk === undefined) {

package/dist/tracers/tests/langsmith_interop.test.d.ts

@@ -0,0 +1 @@
+export {};