langchain 0.0.161 → 0.0.163

package/README.md

@@ -7,8 +7,9 @@
 
 Looking for the Python version? Check out [LangChain](https://github.com/hwchase17/langchain).
 
-**Production Support:** As you move your LangChains into production, we'd love to offer more hands-on support.
-Fill out [this form](https://airtable.com/appwQzlErAS2qiP0L/shrGtGaVBVAz7NcV2) to share more about what you're building, and our team will get in touch.
+To help you ship LangChain apps to production faster, check out [LangSmith](https://smith.langchain.com). 
+[LangSmith](https://smith.langchain.com) is a unified developer platform for building, testing, and monitoring LLM applications. 
+Fill out [this form](https://airtable.com/appwQzlErAS2qiP0L/shrGtGaVBVAz7NcV2) to get off the waitlist or speak with our sales team
 
 ## Quick Install
 

package/dist/callbacks/base.d.ts

@@ -35,7 +35,7 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      * Called at the start of an LLM or Chat Model run, with the prompt(s)
      * and the run ID.
      */
-    handleLLMStart?(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handleLLMStart?(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, name?: string): // eslint-disable-next-line @typescript-eslint/no-explicit-any
     Promise<any> | any;
     /**
      * Called when an LLM/ChatModel in `streaming` mode produces a new token
@@ -63,13 +63,13 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      * Called at the start of a Chat Model run, with the prompt(s)
      * and the run ID.
      */
-    handleChatModelStart?(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handleChatModelStart?(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>, name?: string): // eslint-disable-next-line @typescript-eslint/no-explicit-any
     Promise<any> | any;
     /**
      * Called at the start of a Chain run, with the chain name and inputs
      * and the run ID.
      */
-    handleChainStart?(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string): // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handleChainStart?(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, runType?: string, name?: string): // eslint-disable-next-line @typescript-eslint/no-explicit-any
     Promise<any> | any;
     /**
      * Called if a Chain run encounters an error
@@ -89,7 +89,7 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      * Called at the start of a Tool run, with the tool name and input
      * and the run ID.
      */
-    handleToolStart?(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>): // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handleToolStart?(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, name?: string): // eslint-disable-next-line @typescript-eslint/no-explicit-any
     Promise<any> | any;
     /**
      * Called if a Tool run encounters an error
@@ -112,7 +112,7 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      * with the final output and the run ID.
      */
     handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
-    handleRetrieverStart?(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>): // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    handleRetrieverStart?(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>, name?: string): // eslint-disable-next-line @typescript-eslint/no-explicit-any
     Promise<any> | any;
     handleRetrieverEnd?(documents: Document[], runId: string, parentRunId?: string, tags?: string[]): // eslint-disable-next-line @typescript-eslint/no-explicit-any
     Promise<any> | any;
@@ -199,7 +199,7 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
          * Called at the start of an LLM or Chat Model run, with the prompt(s)
          * and the run ID.
          */
-        handleLLMStart?(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): any;
+        handleLLMStart?(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string | undefined): any;
         /**
          * Called when an LLM/ChatModel in `streaming` mode produces a new token
          */
@@ -216,12 +216,12 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
          * Called at the start of a Chat Model run, with the prompt(s)
          * and the run ID.
          */
-        handleChatModelStart?(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): any;
+        handleChatModelStart?(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string | undefined): any;
         /**
          * Called at the start of a Chain run, with the chain name and inputs
          * and the run ID.
          */
-        handleChainStart?(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, runType?: string | undefined): any;
+        handleChainStart?(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, runType?: string | undefined, name?: string | undefined): any;
         /**
          * Called if a Chain run encounters an error
          */
@@ -238,7 +238,7 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
          * Called at the start of a Tool run, with the tool name and input
          * and the run ID.
          */
-        handleToolStart?(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): any;
+        handleToolStart?(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string | undefined): any;
         /**
          * Called if a Tool run encounters an error
          */
@@ -258,7 +258,7 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
          * with the final output and the run ID.
          */
         handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
-        handleRetrieverStart?(retriever: Serialized, query: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): any;
+        handleRetrieverStart?(retriever: Serialized, query: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined, name?: string | undefined): any;
         handleRetrieverEnd?(documents: Document<Record<string, any>>[], runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): any;
         handleRetrieverError?(err: any, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): any;
     };

package/dist/callbacks/handlers/log_stream.cjs

@@ -101,7 +101,13 @@ class LogStreamCallbackHandler extends tracer_js_1.BaseTracer {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "indexMap", {
+        Object.defineProperty(this, "keyMapByRunId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: {}
+        });
+        Object.defineProperty(this, "counterMapByRunName", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -191,7 +197,7 @@ class LogStreamCallbackHandler extends tracer_js_1.BaseTracer {
                             id: run.id,
                             streamed_output: [],
                             final_output: undefined,
-                            logs: [],
+                            logs: {},
                         },
                     },
                 ],
@@ -200,7 +206,13 @@ class LogStreamCallbackHandler extends tracer_js_1.BaseTracer {
         if (!this._includeRun(run)) {
             return;
         }
-        this.indexMap[run.id] = Math.max(...Object.values(this.indexMap), -1) + 1;
+        if (this.counterMapByRunName[run.name] === undefined) {
+            this.counterMapByRunName[run.name] = 0;
+        }
+        this.counterMapByRunName[run.name] += 1;
+        const count = this.counterMapByRunName[run.name];
+        this.keyMapByRunId[run.id] =
+            count === 1 ? run.name : `${run.name}:${count}`;
         const logEntry = {
             id: run.id,
             name: run.name,
@@ -216,7 +228,7 @@ class LogStreamCallbackHandler extends tracer_js_1.BaseTracer {
             ops: [
                 {
                     op: "add",
-                    path: `/logs/${this.indexMap[run.id]}`,
+                    path: `/logs/${this.keyMapByRunId[run.id]}`,
                     value: logEntry,
                 },
             ],
@@ -224,21 +236,21 @@ class LogStreamCallbackHandler extends tracer_js_1.BaseTracer {
     }
     async onRunUpdate(run) {
         try {
-            const index = this.indexMap[run.id];
-            if (index === undefined) {
+            const runName = this.keyMapByRunId[run.id];
+            if (runName === undefined) {
                 return;
             }
             const ops = [
                 {
                     op: "add",
-                    path: `/logs/${index}/final_output`,
+                    path: `/logs/${runName}/final_output`,
                     value: run.outputs,
                 },
             ];
             if (run.end_time !== undefined) {
                 ops.push({
                     op: "add",
-                    path: `/logs/${index}/end_time`,
+                    path: `/logs/${runName}/end_time`,
                     value: new Date(run.end_time).toISOString(),
                 });
             }
@@ -264,15 +276,15 @@ class LogStreamCallbackHandler extends tracer_js_1.BaseTracer {
         }
     }
     async onLLMNewToken(run, token) {
-        const index = this.indexMap[run.id];
-        if (index === undefined) {
+        const runName = this.keyMapByRunId[run.id];
+        if (runName === undefined) {
             return;
         }
         const patch = new RunLogPatch({
             ops: [
                 {
                     op: "add",
-                    path: `/logs/${index}/streamed_output_str/-`,
+                    path: `/logs/${runName}/streamed_output_str/-`,
                     value: token,
                 },
             ],

package/dist/callbacks/handlers/log_stream.d.ts

@@ -37,7 +37,7 @@ export type RunState = {
      * List of sub-runs contained in this run, if any, in the order they were started.
      * If filters were supplied, this list will contain only the runs that matched the filters.
      */
-    logs: LogEntry[];
+    logs: Record<string, LogEntry>;
 };
 /**
  * List of jsonpatch JSONPatchOperations, which describe how to create the run state
@@ -84,7 +84,8 @@ export declare class LogStreamCallbackHandler extends BaseTracer {
     protected excludeNames?: string[];
     protected excludeTypes?: string[];
     protected excludeTags?: string[];
-    private indexMap;
+    private keyMapByRunId;
+    private counterMapByRunName;
     protected transformStream: TransformStream;
     writer: WritableStreamDefaultWriter;
     receiveStream: IterableReadableStream<RunLogPatch>;

package/dist/callbacks/handlers/log_stream.js

@@ -96,7 +96,13 @@ export class LogStreamCallbackHandler extends BaseTracer {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "indexMap", {
+        Object.defineProperty(this, "keyMapByRunId", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: {}
+        });
+        Object.defineProperty(this, "counterMapByRunName", {
             enumerable: true,
             configurable: true,
             writable: true,
@@ -186,7 +192,7 @@ export class LogStreamCallbackHandler extends BaseTracer {
                             id: run.id,
                             streamed_output: [],
                             final_output: undefined,
-                            logs: [],
+                            logs: {},
                         },
                     },
                 ],
@@ -195,7 +201,13 @@ export class LogStreamCallbackHandler extends BaseTracer {
         if (!this._includeRun(run)) {
             return;
         }
-        this.indexMap[run.id] = Math.max(...Object.values(this.indexMap), -1) + 1;
+        if (this.counterMapByRunName[run.name] === undefined) {
+            this.counterMapByRunName[run.name] = 0;
+        }
+        this.counterMapByRunName[run.name] += 1;
+        const count = this.counterMapByRunName[run.name];
+        this.keyMapByRunId[run.id] =
+            count === 1 ? run.name : `${run.name}:${count}`;
         const logEntry = {
             id: run.id,
             name: run.name,
@@ -211,7 +223,7 @@ export class LogStreamCallbackHandler extends BaseTracer {
             ops: [
                 {
                     op: "add",
-                    path: `/logs/${this.indexMap[run.id]}`,
+                    path: `/logs/${this.keyMapByRunId[run.id]}`,
                     value: logEntry,
                 },
             ],
@@ -219,21 +231,21 @@ export class LogStreamCallbackHandler extends BaseTracer {
     }
     async onRunUpdate(run) {
         try {
-            const index = this.indexMap[run.id];
-            if (index === undefined) {
+            const runName = this.keyMapByRunId[run.id];
+            if (runName === undefined) {
                 return;
             }
             const ops = [
                 {
                     op: "add",
-                    path: `/logs/${index}/final_output`,
+                    path: `/logs/${runName}/final_output`,
                     value: run.outputs,
                 },
             ];
             if (run.end_time !== undefined) {
                 ops.push({
                     op: "add",
-                    path: `/logs/${index}/end_time`,
+                    path: `/logs/${runName}/end_time`,
                     value: new Date(run.end_time).toISOString(),
                 });
             }
@@ -259,15 +271,15 @@ export class LogStreamCallbackHandler extends BaseTracer {
         }
     }
     async onLLMNewToken(run, token) {
-        const index = this.indexMap[run.id];
-        if (index === undefined) {
+        const runName = this.keyMapByRunId[run.id];
+        if (runName === undefined) {
             return;
         }
         const patch = new RunLogPatch({
             ops: [
                 {
                     op: "add",
-                    path: `/logs/${index}/streamed_output_str/-`,
+                    path: `/logs/${runName}/streamed_output_str/-`,
                     value: token,
                 },
             ],

package/dist/callbacks/manager.cjs

@@ -330,13 +330,13 @@ class CallbackManager extends BaseCallbackManager {
         this.inheritableHandlers = [];
         this._parentRunId = parentRunId;
     }
-    async handleLLMStart(llm, prompts, _runId = undefined, _parentRunId = undefined, extraParams = undefined) {
+    async handleLLMStart(llm, prompts, _runId = undefined, _parentRunId = undefined, extraParams = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         return Promise.all(prompts.map(async (prompt) => {
             const runId = (0, uuid_1.v4)();
             await Promise.all(this.handlers.map((handler) => (0, promises_js_1.consumeCallback)(async () => {
                 if (!handler.ignoreLLM) {
                     try {
-                        await handler.handleLLMStart?.(llm, [prompt], runId, this._parentRunId, extraParams, this.tags, this.metadata);
+                        await handler.handleLLMStart?.(llm, [prompt], runId, this._parentRunId, extraParams, this.tags, this.metadata, runName);
                     }
                     catch (err) {
                         console.error(`Error in handler ${handler.constructor.name}, handleLLMStart: ${err}`);
@@ -346,17 +346,18 @@ class CallbackManager extends BaseCallbackManager {
             return new CallbackManagerForLLMRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
         }));
     }
-    async handleChatModelStart(llm, messages, _runId = undefined, _parentRunId = undefined, extraParams = undefined) {
+    async handleChatModelStart(llm, messages, _runId = undefined, _parentRunId = undefined, extraParams = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         return Promise.all(messages.map(async (messageGroup) => {
             const runId = (0, uuid_1.v4)();
             await Promise.all(this.handlers.map((handler) => (0, promises_js_1.consumeCallback)(async () => {
                 if (!handler.ignoreLLM) {
                     try {
-                        if (handler.handleChatModelStart)
-                            await handler.handleChatModelStart?.(llm, [messageGroup], runId, this._parentRunId, extraParams, this.tags, this.metadata);
+                        if (handler.handleChatModelStart) {
+                            await handler.handleChatModelStart?.(llm, [messageGroup], runId, this._parentRunId, extraParams, this.tags, this.metadata, runName);
+                        }
                         else if (handler.handleLLMStart) {
                             const messageString = (0, base_js_2.getBufferString)(messageGroup);
-                            await handler.handleLLMStart?.(llm, [messageString], runId, this._parentRunId, extraParams, this.tags, this.metadata);
+                            await handler.handleLLMStart?.(llm, [messageString], runId, this._parentRunId, extraParams, this.tags, this.metadata, runName);
                         }
                     }
                     catch (err) {
@@ -367,11 +368,11 @@ class CallbackManager extends BaseCallbackManager {
             return new CallbackManagerForLLMRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
         }));
     }
-    async handleChainStart(chain, inputs, runId = (0, uuid_1.v4)(), runType = undefined) {
+    async handleChainStart(chain, inputs, runId = (0, uuid_1.v4)(), runType = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         await Promise.all(this.handlers.map((handler) => (0, promises_js_1.consumeCallback)(async () => {
             if (!handler.ignoreChain) {
                 try {
-                    await handler.handleChainStart?.(chain, inputs, runId, this._parentRunId, this.tags, this.metadata, runType);
+                    await handler.handleChainStart?.(chain, inputs, runId, this._parentRunId, this.tags, this.metadata, runType, runName);
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleChainStart: ${err}`);
@@ -380,11 +381,11 @@ class CallbackManager extends BaseCallbackManager {
         }, handler.awaitHandlers)));
         return new CallbackManagerForChainRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
     }
-    async handleToolStart(tool, input, runId = (0, uuid_1.v4)()) {
+    async handleToolStart(tool, input, runId = (0, uuid_1.v4)(), _parentRunId = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         await Promise.all(this.handlers.map((handler) => (0, promises_js_1.consumeCallback)(async () => {
             if (!handler.ignoreAgent) {
                 try {
-                    await handler.handleToolStart?.(tool, input, runId, this._parentRunId, this.tags, this.metadata);
+                    await handler.handleToolStart?.(tool, input, runId, this._parentRunId, this.tags, this.metadata, runName);
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleToolStart: ${err}`);
@@ -393,11 +394,11 @@ class CallbackManager extends BaseCallbackManager {
         }, handler.awaitHandlers)));
         return new CallbackManagerForToolRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
     }
-    async handleRetrieverStart(retriever, query, runId = (0, uuid_1.v4)(), _parentRunId = undefined) {
+    async handleRetrieverStart(retriever, query, runId = (0, uuid_1.v4)(), _parentRunId = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         await Promise.all(this.handlers.map((handler) => (0, promises_js_1.consumeCallback)(async () => {
             if (!handler.ignoreRetriever) {
                 try {
-                    await handler.handleRetrieverStart?.(retriever, query, runId, this._parentRunId, this.tags, this.metadata);
+                    await handler.handleRetrieverStart?.(retriever, query, runId, this._parentRunId, this.tags, this.metadata, runName);
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleRetrieverStart: ${err}`);

package/dist/callbacks/manager.d.ts

@@ -12,6 +12,10 @@ export interface CallbackManagerOptions {
 }
 export type Callbacks = CallbackManager | (BaseCallbackHandler | CallbackHandlerMethods)[];
 export interface BaseCallbackConfig {
+    /**
+     * Name for the tracer run for this call. Defaults to the name of the class.
+     */
+    runName?: string;
     /**
      * Tags for this call and any sub-calls (eg. a Chain calling an LLM).
      * You can use these to filter calls.
@@ -92,11 +96,11 @@ export declare class CallbackManager extends BaseCallbackManager implements Base
     name: string;
     private readonly _parentRunId?;
     constructor(parentRunId?: string);
-    handleLLMStart(llm: Serialized, prompts: string[], _runId?: string | undefined, _parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined): Promise<CallbackManagerForLLMRun[]>;
-    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], _runId?: string | undefined, _parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined): Promise<CallbackManagerForLLMRun[]>;
-    handleChainStart(chain: Serialized, inputs: ChainValues, runId?: string, runType?: string | undefined): Promise<CallbackManagerForChainRun>;
-    handleToolStart(tool: Serialized, input: string, runId?: string): Promise<CallbackManagerForToolRun>;
-    handleRetrieverStart(retriever: Serialized, query: string, runId?: string, _parentRunId?: string | undefined): Promise<CallbackManagerForRetrieverRun>;
+    handleLLMStart(llm: Serialized, prompts: string[], _runId?: string | undefined, _parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, _tags?: string[] | undefined, _metadata?: Record<string, unknown> | undefined, runName?: string | undefined): Promise<CallbackManagerForLLMRun[]>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], _runId?: string | undefined, _parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, _tags?: string[] | undefined, _metadata?: Record<string, unknown> | undefined, runName?: string | undefined): Promise<CallbackManagerForLLMRun[]>;
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId?: string, runType?: string | undefined, _tags?: string[] | undefined, _metadata?: Record<string, unknown> | undefined, runName?: string | undefined): Promise<CallbackManagerForChainRun>;
+    handleToolStart(tool: Serialized, input: string, runId?: string, _parentRunId?: string | undefined, _tags?: string[] | undefined, _metadata?: Record<string, unknown> | undefined, runName?: string | undefined): Promise<CallbackManagerForToolRun>;
+    handleRetrieverStart(retriever: Serialized, query: string, runId?: string, _parentRunId?: string | undefined, _tags?: string[] | undefined, _metadata?: Record<string, unknown> | undefined, runName?: string | undefined): Promise<CallbackManagerForRetrieverRun>;
     addHandler(handler: BaseCallbackHandler, inherit?: boolean): void;
     removeHandler(handler: BaseCallbackHandler): void;
     setHandlers(handlers: BaseCallbackHandler[], inherit?: boolean): void;

package/dist/callbacks/manager.js

@@ -321,13 +321,13 @@ export class CallbackManager extends BaseCallbackManager {
         this.inheritableHandlers = [];
         this._parentRunId = parentRunId;
     }
-    async handleLLMStart(llm, prompts, _runId = undefined, _parentRunId = undefined, extraParams = undefined) {
+    async handleLLMStart(llm, prompts, _runId = undefined, _parentRunId = undefined, extraParams = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         return Promise.all(prompts.map(async (prompt) => {
             const runId = uuidv4();
             await Promise.all(this.handlers.map((handler) => consumeCallback(async () => {
                 if (!handler.ignoreLLM) {
                     try {
-                        await handler.handleLLMStart?.(llm, [prompt], runId, this._parentRunId, extraParams, this.tags, this.metadata);
+                        await handler.handleLLMStart?.(llm, [prompt], runId, this._parentRunId, extraParams, this.tags, this.metadata, runName);
                     }
                     catch (err) {
                         console.error(`Error in handler ${handler.constructor.name}, handleLLMStart: ${err}`);
@@ -337,17 +337,18 @@ export class CallbackManager extends BaseCallbackManager {
             return new CallbackManagerForLLMRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
         }));
     }
-    async handleChatModelStart(llm, messages, _runId = undefined, _parentRunId = undefined, extraParams = undefined) {
+    async handleChatModelStart(llm, messages, _runId = undefined, _parentRunId = undefined, extraParams = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         return Promise.all(messages.map(async (messageGroup) => {
             const runId = uuidv4();
             await Promise.all(this.handlers.map((handler) => consumeCallback(async () => {
                 if (!handler.ignoreLLM) {
                     try {
-                        if (handler.handleChatModelStart)
-                            await handler.handleChatModelStart?.(llm, [messageGroup], runId, this._parentRunId, extraParams, this.tags, this.metadata);
+                        if (handler.handleChatModelStart) {
+                            await handler.handleChatModelStart?.(llm, [messageGroup], runId, this._parentRunId, extraParams, this.tags, this.metadata, runName);
+                        }
                         else if (handler.handleLLMStart) {
                             const messageString = getBufferString(messageGroup);
-                            await handler.handleLLMStart?.(llm, [messageString], runId, this._parentRunId, extraParams, this.tags, this.metadata);
+                            await handler.handleLLMStart?.(llm, [messageString], runId, this._parentRunId, extraParams, this.tags, this.metadata, runName);
                         }
                     }
                     catch (err) {
@@ -358,11 +359,11 @@ export class CallbackManager extends BaseCallbackManager {
             return new CallbackManagerForLLMRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
         }));
     }
-    async handleChainStart(chain, inputs, runId = uuidv4(), runType = undefined) {
+    async handleChainStart(chain, inputs, runId = uuidv4(), runType = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         await Promise.all(this.handlers.map((handler) => consumeCallback(async () => {
             if (!handler.ignoreChain) {
                 try {
-                    await handler.handleChainStart?.(chain, inputs, runId, this._parentRunId, this.tags, this.metadata, runType);
+                    await handler.handleChainStart?.(chain, inputs, runId, this._parentRunId, this.tags, this.metadata, runType, runName);
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleChainStart: ${err}`);
@@ -371,11 +372,11 @@ export class CallbackManager extends BaseCallbackManager {
         }, handler.awaitHandlers)));
         return new CallbackManagerForChainRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
     }
-    async handleToolStart(tool, input, runId = uuidv4()) {
+    async handleToolStart(tool, input, runId = uuidv4(), _parentRunId = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         await Promise.all(this.handlers.map((handler) => consumeCallback(async () => {
             if (!handler.ignoreAgent) {
                 try {
-                    await handler.handleToolStart?.(tool, input, runId, this._parentRunId, this.tags, this.metadata);
+                    await handler.handleToolStart?.(tool, input, runId, this._parentRunId, this.tags, this.metadata, runName);
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleToolStart: ${err}`);
@@ -384,11 +385,11 @@ export class CallbackManager extends BaseCallbackManager {
         }, handler.awaitHandlers)));
         return new CallbackManagerForToolRun(runId, this.handlers, this.inheritableHandlers, this.tags, this.inheritableTags, this.metadata, this.inheritableMetadata, this._parentRunId);
     }
-    async handleRetrieverStart(retriever, query, runId = uuidv4(), _parentRunId = undefined) {
+    async handleRetrieverStart(retriever, query, runId = uuidv4(), _parentRunId = undefined, _tags = undefined, _metadata = undefined, runName = undefined) {
         await Promise.all(this.handlers.map((handler) => consumeCallback(async () => {
             if (!handler.ignoreRetriever) {
                 try {
-                    await handler.handleRetrieverStart?.(retriever, query, runId, this._parentRunId, this.tags, this.metadata);
+                    await handler.handleRetrieverStart?.(retriever, query, runId, this._parentRunId, this.tags, this.metadata, runName);
                 }
                 catch (err) {
                     console.error(`Error in handler ${handler.constructor.name}, handleRetrieverStart: ${err}`);

package/dist/chains/base.cjs

@@ -98,7 +98,7 @@ class BaseChain extends index_js_2.BaseLangChain {
         const fullValues = await this._formatValues(values);
         const parsedConfig = (0, manager_js_1.parseCallbackConfigArg)(config);
         const callbackManager_ = await manager_js_1.CallbackManager.configure(parsedConfig.callbacks, this.callbacks, parsedConfig.tags || tags, this.tags, parsedConfig.metadata, this.metadata, { verbose: this.verbose });
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), fullValues);
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), fullValues, undefined, undefined, undefined, undefined, parsedConfig.runName);
         let outputValues;
         try {
             outputValues = await (values.signal

package/dist/chains/base.js

@@ -95,7 +95,7 @@ export class BaseChain extends BaseLangChain {
         const fullValues = await this._formatValues(values);
         const parsedConfig = parseCallbackConfigArg(config);
         const callbackManager_ = await CallbackManager.configure(parsedConfig.callbacks, this.callbacks, parsedConfig.tags || tags, this.tags, parsedConfig.metadata, this.metadata, { verbose: this.verbose });
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), fullValues);
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), fullValues, undefined, undefined, undefined, undefined, parsedConfig.runName);
         let outputValues;
         try {
             outputValues = await (values.signal

package/dist/chat_models/base.cjs

@@ -71,7 +71,7 @@ class BaseChatModel extends index_js_2.BaseLanguageModel {
                 options: callOptions,
                 invocation_params: this?.invocationParams(callOptions),
             };
-            const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), [messages], undefined, undefined, extra);
+            const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), [messages], undefined, undefined, extra, undefined, undefined, runnableConfig.runName);
             let generationChunk;
             try {
                 for await (const chunk of this._streamResponseChunks(messages, callOptions, runManagers?.[0])) {
@@ -103,7 +103,7 @@ class BaseChatModel extends index_js_2.BaseLanguageModel {
             options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
-        const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), baseMessages, undefined, undefined, extra);
+        const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), baseMessages, undefined, undefined, extra, undefined, undefined, handledOptions.runName);
         // generate results
         const results = await Promise.allSettled(baseMessages.map((messageList, i) => this._generate(messageList, { ...parsedOptions, promptIndex: i }, runManagers?.[i])));
         // handle results

package/dist/chat_models/base.js

@@ -67,7 +67,7 @@ export class BaseChatModel extends BaseLanguageModel {
                 options: callOptions,
                 invocation_params: this?.invocationParams(callOptions),
             };
-            const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), [messages], undefined, undefined, extra);
+            const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), [messages], undefined, undefined, extra, undefined, undefined, runnableConfig.runName);
             let generationChunk;
             try {
                 for await (const chunk of this._streamResponseChunks(messages, callOptions, runManagers?.[0])) {
@@ -99,7 +99,7 @@ export class BaseChatModel extends BaseLanguageModel {
             options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
-        const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), baseMessages, undefined, undefined, extra);
+        const runManagers = await callbackManager_?.handleChatModelStart(this.toJSON(), baseMessages, undefined, undefined, extra, undefined, undefined, handledOptions.runName);
         // generate results
         const results = await Promise.allSettled(baseMessages.map((messageList, i) => this._generate(messageList, { ...parsedOptions, promptIndex: i }, runManagers?.[i])));
         // handle results

package/dist/llms/base.cjs

@@ -55,7 +55,7 @@ class BaseLLM extends index_js_2.BaseLanguageModel {
                 options: callOptions,
                 invocation_params: this?.invocationParams(callOptions),
             };
-            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra);
+            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra, undefined, undefined, runnableConfig.runName);
             let generation = new index_js_1.GenerationChunk({
                 text: "",
             });
@@ -129,7 +129,7 @@ class BaseLLM extends index_js_2.BaseLanguageModel {
             options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
-        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra);
+        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra, undefined, undefined, handledOptions?.runName);
         let output;
         try {
             output = await this._generate(prompts, parsedOptions, runManagers?.[0]);

package/dist/llms/base.js

@@ -52,7 +52,7 @@ export class BaseLLM extends BaseLanguageModel {
                 options: callOptions,
                 invocation_params: this?.invocationParams(callOptions),
             };
-            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra);
+            const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), [prompt.toString()], undefined, undefined, extra, undefined, undefined, runnableConfig.runName);
             let generation = new GenerationChunk({
                 text: "",
             });
@@ -126,7 +126,7 @@ export class BaseLLM extends BaseLanguageModel {
             options: parsedOptions,
             invocation_params: this?.invocationParams(parsedOptions),
         };
-        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra);
+        const runManagers = await callbackManager_?.handleLLMStart(this.toJSON(), prompts, undefined, undefined, extra, undefined, undefined, handledOptions?.runName);
         let output;
         try {
             output = await this._generate(prompts, parsedOptions, runManagers?.[0]);

package/dist/prompts/chat.cjs

@@ -3,9 +3,10 @@
 // Replace with "string" when we are comfortable with a breaking change.
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ChatPromptTemplate = exports.SystemMessagePromptTemplate = exports.AIMessagePromptTemplate = exports.HumanMessagePromptTemplate = exports.ChatMessagePromptTemplate = exports.BaseChatPromptTemplate = exports.BaseMessageStringPromptTemplate = exports.MessagesPlaceholder = exports.ChatPromptValue = exports.BaseMessagePromptTemplate = void 0;
+const base_js_1 = require("../memory/base.cjs");
 const index_js_1 = require("../schema/index.cjs");
 const index_js_2 = require("../schema/runnable/index.cjs");
-const base_js_1 = require("./base.cjs");
+const base_js_2 = require("./base.cjs");
 const prompt_js_1 = require("./prompt.cjs");
 /**
  * Abstract class that serves as a base for creating message prompt
@@ -74,7 +75,7 @@ class ChatPromptValue extends index_js_1.BasePromptValue {
         this.messages = fields.messages;
     }
     toString() {
-        return JSON.stringify(this.messages);
+        return (0, base_js_1.getBufferString)(this.messages);
     }
     toChatMessages() {
         return this.messages;
@@ -142,7 +143,7 @@ exports.BaseMessageStringPromptTemplate = BaseMessageStringPromptTemplate;
  * Abstract class that serves as a base for creating chat prompt
  * templates. It extends the BasePromptTemplate.
  */
-class BaseChatPromptTemplate extends base_js_1.BasePromptTemplate {
+class BaseChatPromptTemplate extends base_js_2.BasePromptTemplate {
     constructor(input) {
         super(input);
     }

package/dist/prompts/chat.js

@@ -1,5 +1,6 @@
 // Default generic "any" values are for backwards compatibility.
 // Replace with "string" when we are comfortable with a breaking change.
+import { getBufferString } from "../memory/base.js";
 import { AIMessage, BaseMessage, BasePromptValue, ChatMessage, HumanMessage, SystemMessage, coerceMessageLikeToMessage, isBaseMessage, } from "../schema/index.js";
 import { Runnable } from "../schema/runnable/index.js";
 import { BasePromptTemplate, } from "./base.js";
@@ -70,7 +71,7 @@ export class ChatPromptValue extends BasePromptValue {
         this.messages = fields.messages;
     }
     toString() {
-        return JSON.stringify(this.messages);
+        return getBufferString(this.messages);
     }
     toChatMessages() {
         return this.messages;

package/dist/prompts/selectors/SemanticSimilarityExampleSelector.cjs

@@ -39,10 +39,17 @@ class SemanticSimilarityExampleSelector extends base_js_1.BaseExampleSelector {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "filter", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.vectorStore = data.vectorStore;
         this.k = data.k ?? 4;
         this.exampleKeys = data.exampleKeys;
         this.inputKeys = data.inputKeys;
+        this.filter = data.filter;
     }
     /**
      * Method that adds a new example to the vectorStore. The example is
@@ -56,7 +63,7 @@ class SemanticSimilarityExampleSelector extends base_js_1.BaseExampleSelector {
         await this.vectorStore.addDocuments([
             new document_js_1.Document({
                 pageContent: stringExample,
-                metadata: { example },
+                metadata: example,
             }),
         ]);
     }
@@ -70,7 +77,7 @@ class SemanticSimilarityExampleSelector extends base_js_1.BaseExampleSelector {
     async selectExamples(inputVariables) {
         const inputKeys = this.inputKeys ?? Object.keys(inputVariables);
         const query = sortedValues(inputKeys.reduce((acc, key) => ({ ...acc, [key]: inputVariables[key] }), {})).join(" ");
-        const exampleDocs = await this.vectorStore.similaritySearch(query, this.k);
+        const exampleDocs = await this.vectorStore.similaritySearch(query, this.k, this.filter);
         const examples = exampleDocs.map((doc) => doc.metadata);
         if (this.exampleKeys) {
             // If example keys are provided, filter examples to those keys.

package/dist/prompts/selectors/SemanticSimilarityExampleSelector.d.ts

@@ -6,9 +6,10 @@ import { BaseExampleSelector } from "../base.js";
  * Interface for the input data of the SemanticSimilarityExampleSelector
  * class.
  */
-export interface SemanticSimilarityExampleSelectorInput {
-    vectorStore: VectorStore;
+export interface SemanticSimilarityExampleSelectorInput<V extends VectorStore = VectorStore> {
+    vectorStore: V;
     k?: number;
+    filter?: V["FilterType"];
     exampleKeys?: string[];
     inputKeys?: string[];
 }
@@ -16,12 +17,13 @@ export interface SemanticSimilarityExampleSelectorInput {
  * Class that selects examples based on semantic similarity. It extends
  * the BaseExampleSelector class.
  */
-export declare class SemanticSimilarityExampleSelector extends BaseExampleSelector {
-    vectorStore: VectorStore;
+export declare class SemanticSimilarityExampleSelector<V extends VectorStore = VectorStore> extends BaseExampleSelector {
+    vectorStore: V;
     k: number;
     exampleKeys?: string[];
     inputKeys?: string[];
-    constructor(data: SemanticSimilarityExampleSelectorInput);
+    filter?: V["FilterType"];
+    constructor(data: SemanticSimilarityExampleSelectorInput<V>);
     /**
      * Method that adds a new example to the vectorStore. The example is
      * converted to a string and added to the vectorStore as a document.

package/dist/prompts/selectors/SemanticSimilarityExampleSelector.js

@@ -36,10 +36,17 @@ export class SemanticSimilarityExampleSelector extends BaseExampleSelector {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "filter", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.vectorStore = data.vectorStore;
         this.k = data.k ?? 4;
         this.exampleKeys = data.exampleKeys;
         this.inputKeys = data.inputKeys;
+        this.filter = data.filter;
     }
     /**
      * Method that adds a new example to the vectorStore. The example is
@@ -53,7 +60,7 @@ export class SemanticSimilarityExampleSelector extends BaseExampleSelector {
         await this.vectorStore.addDocuments([
             new Document({
                 pageContent: stringExample,
-                metadata: { example },
+                metadata: example,
             }),
         ]);
     }
@@ -67,7 +74,7 @@ export class SemanticSimilarityExampleSelector extends BaseExampleSelector {
     async selectExamples(inputVariables) {
         const inputKeys = this.inputKeys ?? Object.keys(inputVariables);
         const query = sortedValues(inputKeys.reduce((acc, key) => ({ ...acc, [key]: inputVariables[key] }), {})).join(" ");
-        const exampleDocs = await this.vectorStore.similaritySearch(query, this.k);
+        const exampleDocs = await this.vectorStore.similaritySearch(query, this.k, this.filter);
         const examples = exampleDocs.map((doc) => doc.metadata);
         if (this.exampleKeys) {
             // If example keys are provided, filter examples to those keys.

package/dist/schema/retriever.cjs

@@ -64,7 +64,7 @@ class BaseRetriever extends index_js_1.Runnable {
     async getRelevantDocuments(query, config) {
         const parsedConfig = (0, manager_js_1.parseCallbackConfigArg)(config);
         const callbackManager_ = await manager_js_1.CallbackManager.configure(parsedConfig.callbacks, this.callbacks, parsedConfig.tags, this.tags, parsedConfig.metadata, this.metadata, { verbose: this.verbose });
-        const runManager = await callbackManager_?.handleRetrieverStart(this.toJSON(), query);
+        const runManager = await callbackManager_?.handleRetrieverStart(this.toJSON(), query, undefined, undefined, undefined, undefined, parsedConfig.runName);
         try {
             const results = await this._getRelevantDocuments(query, runManager);
             await runManager?.handleRetrieverEnd(results);

package/dist/schema/retriever.js

@@ -61,7 +61,7 @@ export class BaseRetriever extends Runnable {
     async getRelevantDocuments(query, config) {
         const parsedConfig = parseCallbackConfigArg(config);
         const callbackManager_ = await CallbackManager.configure(parsedConfig.callbacks, this.callbacks, parsedConfig.tags, this.tags, parsedConfig.metadata, this.metadata, { verbose: this.verbose });
-        const runManager = await callbackManager_?.handleRetrieverStart(this.toJSON(), query);
+        const runManager = await callbackManager_?.handleRetrieverStart(this.toJSON(), query, undefined, undefined, undefined, undefined, parsedConfig.runName);
         try {
             const results = await this._getRelevantDocuments(query, runManager);
             await runManager?.handleRetrieverEnd(results);

package/dist/schema/runnable/base.cjs

@@ -143,16 +143,18 @@ class Runnable extends serializable_js_1.Serializable {
             callbacks: options.callbacks,
             tags: options.tags,
             metadata: options.metadata,
+            runName: options.runName,
         };
         const callOptions = { ...options };
         delete callOptions.callbacks;
         delete callOptions.tags;
         delete callOptions.metadata;
+        delete callOptions.runName;
         return [runnableConfig, callOptions];
     }
     async _callWithConfig(func, input, options) {
         const callbackManager_ = await (0, config_js_1.getCallbackMangerForConfig)(options);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, options?.runType);
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, options?.runType, undefined, undefined, options?.runName);
         let output;
         try {
             output = await func.bind(this)(input, options, runManager);
@@ -176,7 +178,7 @@ class Runnable extends serializable_js_1.Serializable {
     async _batchWithConfig(func, inputs, options, batchOptions) {
         const configs = this._getOptionsList((options ?? {}), inputs.length);
         const callbackManagers = await Promise.all(configs.map(config_js_1.getCallbackMangerForConfig));
-        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"))));
+        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"), undefined, configs[i].runType, undefined, undefined, configs[i].runName)));
         let outputs;
         try {
             outputs = await func(inputs, configs, runManagers, batchOptions);
@@ -206,7 +208,7 @@ class Runnable extends serializable_js_1.Serializable {
                 if (!runManager) {
                     // Start the run manager AFTER the iterator starts to preserve
                     // tracing order
-                    runManager = await callbackManager_?.handleChainStart(serializedRepresentation, { input: "" }, undefined, options?.runType);
+                    runManager = await callbackManager_?.handleChainStart(serializedRepresentation, { input: "" }, undefined, options?.runType, undefined, undefined, options?.runName);
                 }
                 if (finalInputSupported) {
                     if (finalInput === undefined) {
@@ -257,7 +259,16 @@ class Runnable extends serializable_js_1.Serializable {
         await runManager?.handleChainEnd(finalOutput ?? {}, undefined, undefined, undefined, { inputs: _coerceToDict(finalInput, "input") });
     }
     _patchConfig(config = {}, callbackManager = undefined) {
-        return { ...config, callbacks: callbackManager };
+        const newConfig = { ...config };
+        if (callbackManager !== undefined) {
+            /**
+             * If we're replacing callbacks we need to unset runName
+             * since that should apply only to the same run as the original callbacks
+             */
+            delete newConfig.runName;
+            return { ...newConfig, callbacks: callbackManager };
+        }
+        return newConfig;
     }
     /**
      * Create a new runnable sequence that runs each individual runnable in series,
@@ -690,7 +701,7 @@ class RunnableSequence extends Runnable {
     }
     async invoke(input, options) {
         const callbackManager_ = await (0, config_js_1.getCallbackMangerForConfig)(options);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"));
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, options?.runName);
         let nextStepInput = input;
         let finalOutput;
         try {
@@ -712,7 +723,7 @@ class RunnableSequence extends Runnable {
     async batch(inputs, options, batchOptions) {
         const configList = this._getOptionsList(options ?? {}, inputs.length);
         const callbackManagers = await Promise.all(configList.map(config_js_1.getCallbackMangerForConfig));
-        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"))));
+        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"), undefined, undefined, undefined, undefined, configList[i].runName)));
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let nextStepInputs = inputs;
         let finalOutputs;
@@ -733,7 +744,7 @@ class RunnableSequence extends Runnable {
     }
     async *_streamIterator(input, options) {
         const callbackManager_ = await (0, config_js_1.getCallbackMangerForConfig)(options);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"));
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, options?.runName);
         let nextStepInput = input;
         const steps = [this.first, ...this.middle, this.last];
         // Find the index of the last runnable in the sequence that doesn't have an overridden .transform() method
@@ -867,7 +878,7 @@ class RunnableMap extends Runnable {
         const callbackManager_ = await (0, config_js_1.getCallbackMangerForConfig)(options);
         const runManager = await callbackManager_?.handleChainStart(this.toJSON(), {
             input,
-        });
+        }, undefined, undefined, undefined, undefined, options?.runName);
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         const output = {};
         try {
@@ -907,6 +918,11 @@ class RunnableLambda extends Runnable {
         });
         this.func = fields.func;
     }
+    static from(func) {
+        return new RunnableLambda({
+            func,
+        });
+    }
     async _invoke(input, config, runManager) {
         let output = await this.func(input);
         if (output && Runnable.isRunnable(output)) {
@@ -963,7 +979,7 @@ class RunnableWithFallbacks extends Runnable {
     }
     async invoke(input, options) {
         const callbackManager_ = await manager_js_1.CallbackManager.configure(options?.callbacks, undefined, options?.tags, undefined, options?.metadata);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"));
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, options?.runName);
         let firstError;
         for (const runnable of this.runnables()) {
             try {
@@ -989,7 +1005,7 @@ class RunnableWithFallbacks extends Runnable {
         }
         const configList = this._getOptionsList(options ?? {}, inputs.length);
         const callbackManagers = await Promise.all(configList.map((config) => manager_js_1.CallbackManager.configure(config?.callbacks, undefined, config?.tags, undefined, config?.metadata)));
-        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"))));
+        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"), undefined, undefined, undefined, undefined, configList[i].runName)));
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let firstError;
         for (const runnable of this.runnables()) {

package/dist/schema/runnable/base.d.ts

@@ -57,7 +57,9 @@ export declare abstract class Runnable<RunInput = any, RunOutput = any, CallOpti
     withFallbacks(fields: {
         fallbacks: Runnable<RunInput, RunOutput>[];
     }): RunnableWithFallbacks<RunInput, RunOutput>;
-    protected _getOptionsList(options: Partial<CallOptions> | Partial<CallOptions>[], length?: number): Partial<CallOptions>[];
+    protected _getOptionsList(options: Partial<CallOptions> | Partial<CallOptions>[], length?: number): Partial<CallOptions & {
+        runType?: string;
+    }>[];
     /**
      * Default implementation of batch, which calls invoke N times.
      * Subclasses should override this method if they can batch more efficiently.
@@ -300,6 +302,7 @@ export declare class RunnableLambda<RunInput, RunOutput> extends Runnable<RunInp
     constructor(fields: {
         func: RunnableFunc<RunInput, RunOutput>;
     });
+    static from<RunInput, RunOutput>(func: RunnableFunc<RunInput, RunOutput>): RunnableLambda<RunInput, RunOutput>;
     _invoke(input: RunInput, config?: Partial<BaseCallbackConfig>, runManager?: CallbackManagerForChainRun): Promise<RunOutput>;
     invoke(input: RunInput, options?: Partial<BaseCallbackConfig>): Promise<RunOutput>;
 }

package/dist/schema/runnable/base.js

@@ -137,16 +137,18 @@ export class Runnable extends Serializable {
             callbacks: options.callbacks,
             tags: options.tags,
             metadata: options.metadata,
+            runName: options.runName,
         };
         const callOptions = { ...options };
         delete callOptions.callbacks;
         delete callOptions.tags;
         delete callOptions.metadata;
+        delete callOptions.runName;
         return [runnableConfig, callOptions];
     }
     async _callWithConfig(func, input, options) {
         const callbackManager_ = await getCallbackMangerForConfig(options);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, options?.runType);
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, options?.runType, undefined, undefined, options?.runName);
         let output;
         try {
             output = await func.bind(this)(input, options, runManager);
@@ -170,7 +172,7 @@ export class Runnable extends Serializable {
     async _batchWithConfig(func, inputs, options, batchOptions) {
         const configs = this._getOptionsList((options ?? {}), inputs.length);
         const callbackManagers = await Promise.all(configs.map(getCallbackMangerForConfig));
-        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"))));
+        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"), undefined, configs[i].runType, undefined, undefined, configs[i].runName)));
         let outputs;
         try {
             outputs = await func(inputs, configs, runManagers, batchOptions);
@@ -200,7 +202,7 @@ export class Runnable extends Serializable {
                 if (!runManager) {
                     // Start the run manager AFTER the iterator starts to preserve
                     // tracing order
-                    runManager = await callbackManager_?.handleChainStart(serializedRepresentation, { input: "" }, undefined, options?.runType);
+                    runManager = await callbackManager_?.handleChainStart(serializedRepresentation, { input: "" }, undefined, options?.runType, undefined, undefined, options?.runName);
                 }
                 if (finalInputSupported) {
                     if (finalInput === undefined) {
@@ -251,7 +253,16 @@ export class Runnable extends Serializable {
         await runManager?.handleChainEnd(finalOutput ?? {}, undefined, undefined, undefined, { inputs: _coerceToDict(finalInput, "input") });
     }
     _patchConfig(config = {}, callbackManager = undefined) {
-        return { ...config, callbacks: callbackManager };
+        const newConfig = { ...config };
+        if (callbackManager !== undefined) {
+            /**
+             * If we're replacing callbacks we need to unset runName
+             * since that should apply only to the same run as the original callbacks
+             */
+            delete newConfig.runName;
+            return { ...newConfig, callbacks: callbackManager };
+        }
+        return newConfig;
     }
     /**
      * Create a new runnable sequence that runs each individual runnable in series,
@@ -680,7 +691,7 @@ export class RunnableSequence extends Runnable {
     }
     async invoke(input, options) {
         const callbackManager_ = await getCallbackMangerForConfig(options);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"));
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, options?.runName);
         let nextStepInput = input;
         let finalOutput;
         try {
@@ -702,7 +713,7 @@ export class RunnableSequence extends Runnable {
     async batch(inputs, options, batchOptions) {
         const configList = this._getOptionsList(options ?? {}, inputs.length);
         const callbackManagers = await Promise.all(configList.map(getCallbackMangerForConfig));
-        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"))));
+        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"), undefined, undefined, undefined, undefined, configList[i].runName)));
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let nextStepInputs = inputs;
         let finalOutputs;
@@ -723,7 +734,7 @@ export class RunnableSequence extends Runnable {
     }
     async *_streamIterator(input, options) {
         const callbackManager_ = await getCallbackMangerForConfig(options);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"));
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, options?.runName);
         let nextStepInput = input;
         const steps = [this.first, ...this.middle, this.last];
         // Find the index of the last runnable in the sequence that doesn't have an overridden .transform() method
@@ -856,7 +867,7 @@ export class RunnableMap extends Runnable {
         const callbackManager_ = await getCallbackMangerForConfig(options);
         const runManager = await callbackManager_?.handleChainStart(this.toJSON(), {
             input,
-        });
+        }, undefined, undefined, undefined, undefined, options?.runName);
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         const output = {};
         try {
@@ -895,6 +906,11 @@ export class RunnableLambda extends Runnable {
         });
         this.func = fields.func;
     }
+    static from(func) {
+        return new RunnableLambda({
+            func,
+        });
+    }
     async _invoke(input, config, runManager) {
         let output = await this.func(input);
         if (output && Runnable.isRunnable(output)) {
@@ -950,7 +966,7 @@ export class RunnableWithFallbacks extends Runnable {
     }
     async invoke(input, options) {
         const callbackManager_ = await CallbackManager.configure(options?.callbacks, undefined, options?.tags, undefined, options?.metadata);
-        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"));
+        const runManager = await callbackManager_?.handleChainStart(this.toJSON(), _coerceToDict(input, "input"), undefined, undefined, undefined, undefined, options?.runName);
         let firstError;
         for (const runnable of this.runnables()) {
             try {
@@ -976,7 +992,7 @@ export class RunnableWithFallbacks extends Runnable {
         }
         const configList = this._getOptionsList(options ?? {}, inputs.length);
         const callbackManagers = await Promise.all(configList.map((config) => CallbackManager.configure(config?.callbacks, undefined, config?.tags, undefined, config?.metadata)));
-        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"))));
+        const runManagers = await Promise.all(callbackManagers.map((callbackManager, i) => callbackManager?.handleChainStart(this.toJSON(), _coerceToDict(inputs[i], "input"), undefined, undefined, undefined, undefined, configList[i].runName)));
         // eslint-disable-next-line @typescript-eslint/no-explicit-any
         let firstError;
         for (const runnable of this.runnables()) {

package/dist/stores/message/dynamodb.cjs

@@ -17,7 +17,7 @@ class DynamoDBChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
             "config.credentials.sessionToken": "AWS_SESSION_TOKEN",
         };
     }
-    constructor({ tableName, sessionId, partitionKey, sortKey, messageAttributeName, config, }) {
+    constructor({ tableName, sessionId, partitionKey, sortKey, messageAttributeName, config, key = {}, }) {
         super();
         Object.defineProperty(this, "lc_namespace", {
             enumerable: true,
@@ -65,7 +65,7 @@ class DynamoDBChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: void 0
+            value: {}
         });
         this.tableName = tableName;
         this.sessionId = sessionId;
@@ -74,10 +74,13 @@ class DynamoDBChatMessageHistory extends index_js_1.BaseListChatMessageHistory {
         this.sortKey = sortKey;
         this.messageAttributeName =
             messageAttributeName ?? this.messageAttributeName;
-        this.dynamoKey = {};
-        this.dynamoKey[this.partitionKey] = { S: this.sessionId };
-        if (this.sortKey) {
-            this.dynamoKey[this.sortKey] = { S: this.sortKey };
+        this.dynamoKey = key;
+        // override dynamoKey with partition key and sort key when key not specified
+        if (Object.keys(this.dynamoKey).length === 0) {
+            this.dynamoKey[this.partitionKey] = { S: this.sessionId };
+            if (this.sortKey) {
+                this.dynamoKey[this.sortKey] = { S: this.sortKey };
+            }
         }
     }
     /**

package/dist/stores/message/dynamodb.d.ts

@@ -1,4 +1,4 @@
-import { DynamoDBClientConfig } from "@aws-sdk/client-dynamodb";
+import { DynamoDBClientConfig, AttributeValue } from "@aws-sdk/client-dynamodb";
 import { BaseMessage, BaseListChatMessageHistory } from "../../schema/index.js";
 /**
  * Interface defining the fields required to create an instance of
@@ -13,6 +13,7 @@ export interface DynamoDBChatMessageHistoryFields {
     sortKey?: string;
     messageAttributeName?: string;
     config?: DynamoDBClientConfig;
+    key?: Record<string, AttributeValue>;
 }
 /**
  * Class providing methods to interact with a DynamoDB table to store and
@@ -31,7 +32,7 @@ export declare class DynamoDBChatMessageHistory extends BaseListChatMessageHisto
     private sortKey?;
     private messageAttributeName;
     private dynamoKey;
-    constructor({ tableName, sessionId, partitionKey, sortKey, messageAttributeName, config, }: DynamoDBChatMessageHistoryFields);
+    constructor({ tableName, sessionId, partitionKey, sortKey, messageAttributeName, config, key, }: DynamoDBChatMessageHistoryFields);
     /**
      * Retrieves all messages from the DynamoDB table and returns them as an
      * array of `BaseMessage` instances.

package/dist/stores/message/dynamodb.js

@@ -14,7 +14,7 @@ export class DynamoDBChatMessageHistory extends BaseListChatMessageHistory {
             "config.credentials.sessionToken": "AWS_SESSION_TOKEN",
         };
     }
-    constructor({ tableName, sessionId, partitionKey, sortKey, messageAttributeName, config, }) {
+    constructor({ tableName, sessionId, partitionKey, sortKey, messageAttributeName, config, key = {}, }) {
         super();
         Object.defineProperty(this, "lc_namespace", {
             enumerable: true,
@@ -62,7 +62,7 @@ export class DynamoDBChatMessageHistory extends BaseListChatMessageHistory {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: void 0
+            value: {}
         });
         this.tableName = tableName;
         this.sessionId = sessionId;
@@ -71,10 +71,13 @@ export class DynamoDBChatMessageHistory extends BaseListChatMessageHistory {
         this.sortKey = sortKey;
         this.messageAttributeName =
             messageAttributeName ?? this.messageAttributeName;
-        this.dynamoKey = {};
-        this.dynamoKey[this.partitionKey] = { S: this.sessionId };
-        if (this.sortKey) {
-            this.dynamoKey[this.sortKey] = { S: this.sortKey };
+        this.dynamoKey = key;
+        // override dynamoKey with partition key and sort key when key not specified
+        if (Object.keys(this.dynamoKey).length === 0) {
+            this.dynamoKey[this.partitionKey] = { S: this.sessionId };
+            if (this.sortKey) {
+                this.dynamoKey[this.sortKey] = { S: this.sortKey };
+            }
         }
     }
     /**

package/dist/tools/base.cjs

@@ -68,7 +68,7 @@ class StructuredTool extends index_js_1.BaseLangChain {
         }
         const config = (0, manager_js_1.parseCallbackConfigArg)(configArg);
         const callbackManager_ = await manager_js_1.CallbackManager.configure(config.callbacks, this.callbacks, config.tags || tags, this.tags, config.metadata, this.metadata, { verbose: this.verbose });
-        const runManager = await callbackManager_?.handleToolStart(this.toJSON(), typeof parsed === "string" ? parsed : JSON.stringify(parsed));
+        const runManager = await callbackManager_?.handleToolStart(this.toJSON(), typeof parsed === "string" ? parsed : JSON.stringify(parsed), undefined, undefined, undefined, undefined, config.runName);
         let result;
         try {
             result = await this._call(parsed, runManager);

package/dist/tools/base.js

@@ -64,7 +64,7 @@ export class StructuredTool extends BaseLangChain {
         }
         const config = parseCallbackConfigArg(configArg);
         const callbackManager_ = await CallbackManager.configure(config.callbacks, this.callbacks, config.tags || tags, this.tags, config.metadata, this.metadata, { verbose: this.verbose });
-        const runManager = await callbackManager_?.handleToolStart(this.toJSON(), typeof parsed === "string" ? parsed : JSON.stringify(parsed));
+        const runManager = await callbackManager_?.handleToolStart(this.toJSON(), typeof parsed === "string" ? parsed : JSON.stringify(parsed), undefined, undefined, undefined, undefined, config.runName);
         let result;
         try {
             result = await this._call(parsed, runManager);

package/dist/util/bedrock.cjs

@@ -48,7 +48,8 @@ class BedrockLLMInputOutputAdapter {
         else if (provider === "ai21") {
             return responseBody?.completions?.[0]?.data?.text ?? "";
         }
-        return responseBody.outputText;
+        // I haven't been able to get a response with more than one result in it.
+        return responseBody.results?.[0]?.outputText;
     }
 }
 exports.BedrockLLMInputOutputAdapter = BedrockLLMInputOutputAdapter;

package/dist/util/bedrock.js

@@ -45,6 +45,7 @@ export class BedrockLLMInputOutputAdapter {
         else if (provider === "ai21") {
             return responseBody?.completions?.[0]?.data?.text ?? "";
         }
-        return responseBody.outputText;
+        // I haven't been able to get a response with more than one result in it.
+        return responseBody.results?.[0]?.outputText;
     }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langchain",
-  "version": "0.0.161",
+  "version": "0.0.163",
   "description": "Typescript bindings for langchain",
   "type": "module",
   "engines": {