langchain 0.0.102 → 0.0.104

package/chains/openai_functions.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/chains/openai_functions/index.cjs');

package/chains/openai_functions.d.ts

@@ -0,0 +1 @@
+export * from '../dist/chains/openai_functions/index.js'

package/chains/openai_functions.js

@@ -0,0 +1 @@
+export * from '../dist/chains/openai_functions/index.js'

package/chat_models/baiduwenxin.cjs

@@ -0,0 +1 @@
+module.exports = require('../dist/chat_models/baiduwenxin.cjs');

package/chat_models/baiduwenxin.d.ts

@@ -0,0 +1 @@
+export * from '../dist/chat_models/baiduwenxin.js'

package/chat_models/baiduwenxin.js

@@ -0,0 +1 @@
+export * from '../dist/chat_models/baiduwenxin.js'

package/dist/agents/agent.d.ts

@@ -2,7 +2,7 @@ import { BaseLanguageModel } from "../base_language/index.js";
 import { CallbackManager, Callbacks } from "../callbacks/manager.js";
 import { LLMChain } from "../chains/llm_chain.js";
 import { BasePromptTemplate } from "../prompts/base.js";
-import { AgentAction, AgentFinish, AgentStep, BaseChatMessage, ChainValues } from "../schema/index.js";
+import { AgentAction, AgentFinish, AgentStep, BaseMessage, ChainValues } from "../schema/index.js";
 import { Serializable } from "../load/serializable.js";
 import { StructuredTool, Tool } from "../tools/base.js";
 import { AgentActionOutputParser, AgentInput, SerializedAgent, StoppingMethod } from "./types.js";
@@ -139,7 +139,7 @@ export declare abstract class Agent extends BaseSingleActionAgent {
     /**
      * Construct a scratchpad to let the agent continue its thought process
      */
-    constructScratchPad(steps: AgentStep[]): Promise<string | BaseChatMessage[]>;
+    constructScratchPad(steps: AgentStep[]): Promise<string | BaseMessage[]>;
     private _plan;
     /**
      * Decide what to do given some input.

package/dist/agents/chat_convo/index.cjs

@@ -46,8 +46,8 @@ class ChatConversationalAgent extends agent_js_1.Agent {
     async constructScratchPad(steps) {
         const thoughts = [];
         for (const step of steps) {
-            thoughts.push(new index_js_1.AIChatMessage(step.action.log));
-            thoughts.push(new index_js_1.HumanChatMessage((0, template_js_1.renderTemplate)(prompt_js_1.TEMPLATE_TOOL_RESPONSE, "f-string", {
+            thoughts.push(new index_js_1.AIMessage(step.action.log));
+            thoughts.push(new index_js_1.HumanMessage((0, template_js_1.renderTemplate)(prompt_js_1.TEMPLATE_TOOL_RESPONSE, "f-string", {
                 observation: step.observation,
             })));
         }

package/dist/agents/chat_convo/index.d.ts

@@ -1,6 +1,6 @@
 import { BaseLanguageModel } from "../../base_language/index.js";
 import { ChatPromptTemplate } from "../../prompts/chat.js";
-import { AgentStep, BaseChatMessage } from "../../schema/index.js";
+import { AgentStep, BaseMessage } from "../../schema/index.js";
 import { Tool } from "../../tools/base.js";
 import { Optional } from "../../types/type-utils.js";
 import { Agent, AgentArgs, OutputParserArgs } from "../agent.js";
@@ -29,7 +29,7 @@ export declare class ChatConversationalAgent extends Agent {
     llmPrefix(): string;
     _stop(): string[];
     static validateTools(tools: Tool[]): void;
-    constructScratchPad(steps: AgentStep[]): Promise<BaseChatMessage[]>;
+    constructScratchPad(steps: AgentStep[]): Promise<BaseMessage[]>;
     static getDefaultOutputParser(fields?: OutputParserArgs & {
         toolNames: string[];
     }): AgentActionOutputParser;

package/dist/agents/chat_convo/index.js

@@ -1,7 +1,7 @@
 import { LLMChain } from "../../chains/llm_chain.js";
 import { ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder, SystemMessagePromptTemplate, } from "../../prompts/chat.js";
 import { renderTemplate } from "../../prompts/template.js";
-import { AIChatMessage, HumanChatMessage, } from "../../schema/index.js";
+import { AIMessage, HumanMessage, } from "../../schema/index.js";
 import { Agent } from "../agent.js";
 import { ChatConversationalAgentOutputParserWithRetries } from "./outputParser.js";
 import { PREFIX_END, DEFAULT_PREFIX, DEFAULT_SUFFIX, TEMPLATE_TOOL_RESPONSE, } from "./prompt.js";
@@ -43,8 +43,8 @@ export class ChatConversationalAgent extends Agent {
     async constructScratchPad(steps) {
         const thoughts = [];
         for (const step of steps) {
-            thoughts.push(new AIChatMessage(step.action.log));
-            thoughts.push(new HumanChatMessage(renderTemplate(TEMPLATE_TOOL_RESPONSE, "f-string", {
+            thoughts.push(new AIMessage(step.action.log));
+            thoughts.push(new HumanMessage(renderTemplate(TEMPLATE_TOOL_RESPONSE, "f-string", {
                 observation: step.observation,
             })));
         }

package/dist/agents/openai/index.cjs

@@ -15,11 +15,11 @@ function parseOutput(message) {
             toolInput: function_call.arguments
                 ? JSON.parse(function_call.arguments)
                 : {},
-            log: message.text,
+            log: message.content,
         };
     }
     else {
-        return { returnValues: { output: message.text }, log: message.text };
+        return { returnValues: { output: message.content }, log: message.content };
     }
 }
 class OpenAIAgent extends agent_js_1.Agent {
@@ -79,13 +79,13 @@ class OpenAIAgent extends agent_js_1.Agent {
     }
     async constructScratchPad(steps) {
         return steps.flatMap(({ action, observation }) => [
-            new index_js_1.AIChatMessage("", {
+            new index_js_1.AIMessage("", {
                 function_call: {
                     name: action.tool,
                     arguments: JSON.stringify(action.toolInput),
                 },
             }),
-            new index_js_1.FunctionChatMessage(observation, action.tool),
+            new index_js_1.FunctionMessage(observation, action.tool),
         ]);
     }
     async plan(steps, inputs, callbackManager) {

package/dist/agents/openai/index.d.ts

@@ -1,6 +1,6 @@
 import { CallbackManager } from "../../callbacks/manager.js";
 import { BasePromptTemplate } from "../../prompts/base.js";
-import { AgentAction, AgentFinish, AgentStep, BaseChatMessage, ChainValues } from "../../schema/index.js";
+import { AgentAction, AgentFinish, AgentStep, BaseMessage, ChainValues } from "../../schema/index.js";
 import { StructuredTool } from "../../tools/base.js";
 import { Agent, AgentArgs } from "../agent.js";
 import { AgentInput } from "../types.js";
@@ -21,6 +21,6 @@ export declare class OpenAIAgent extends Agent {
     constructor(input: Omit<OpenAIAgentInput, "outputParser">);
     static createPrompt(_tools: StructuredTool[], fields?: OpenAIAgentCreatePromptArgs): BasePromptTemplate;
     static fromLLMAndTools(llm: BaseLanguageModel, tools: StructuredTool[], args?: OpenAIAgentCreatePromptArgs & Pick<AgentArgs, "callbacks">): OpenAIAgent;
-    constructScratchPad(steps: AgentStep[]): Promise<string | BaseChatMessage[]>;
+    constructScratchPad(steps: AgentStep[]): Promise<string | BaseMessage[]>;
     plan(steps: Array<AgentStep>, inputs: ChainValues, callbackManager?: CallbackManager): Promise<AgentAction | AgentFinish>;
 }

package/dist/agents/openai/index.js

@@ -1,4 +1,4 @@
-import { AIChatMessage, FunctionChatMessage, } from "../../schema/index.js";
+import { AIMessage, FunctionMessage, } from "../../schema/index.js";
 import { Agent } from "../agent.js";
 import { PREFIX } from "./prompt.js";
 import { ChatPromptTemplate, HumanMessagePromptTemplate, MessagesPlaceholder, SystemMessagePromptTemplate, } from "../../prompts/chat.js";
@@ -12,11 +12,11 @@ function parseOutput(message) {
             toolInput: function_call.arguments
                 ? JSON.parse(function_call.arguments)
                 : {},
-            log: message.text,
+            log: message.content,
         };
     }
     else {
-        return { returnValues: { output: message.text }, log: message.text };
+        return { returnValues: { output: message.content }, log: message.content };
     }
 }
 export class OpenAIAgent extends Agent {
@@ -76,13 +76,13 @@ export class OpenAIAgent extends Agent {
     }
     async constructScratchPad(steps) {
         return steps.flatMap(({ action, observation }) => [
-            new AIChatMessage("", {
+            new AIMessage("", {
                 function_call: {
                     name: action.tool,
                     arguments: JSON.stringify(action.toolInput),
                 },
             }),
-            new FunctionChatMessage(observation, action.tool),
+            new FunctionMessage(observation, action.tool),
         ]);
     }
     async plan(steps, inputs, callbackManager) {

package/dist/base_language/index.cjs

@@ -39,9 +39,16 @@ class BaseLangChain extends serializable_js_1.Serializable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "metadata", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.verbose = params.verbose ?? getVerbosity();
         this.callbacks = params.callbacks;
         this.tags = params.tags ?? [];
+        this.metadata = params.metadata ?? {};
     }
 }
 exports.BaseLangChain = BaseLangChain;
@@ -53,7 +60,7 @@ class BaseLanguageModel extends BaseLangChain {
      * Keys that the language model accepts as call options.
      */
     get callKeys() {
-        return ["stop", "timeout", "signal"];
+        return ["stop", "timeout", "signal", "tags", "metadata", "callbacks"];
     }
     constructor({ callbacks, callbackManager, ...params }) {
         super({

package/dist/base_language/index.d.ts

@@ -1,5 +1,5 @@
-import { BaseChatMessage, BasePromptValue, LLMResult } from "../schema/index.js";
-import { CallbackManager, Callbacks } from "../callbacks/manager.js";
+import { BaseMessage, BasePromptValue, LLMResult } from "../schema/index.js";
+import { BaseCallbackConfig, CallbackManager, Callbacks } from "../callbacks/manager.js";
 import { AsyncCaller, AsyncCallerParams } from "../util/async_caller.js";
 import { Serializable } from "../load/serializable.js";
 export type SerializedLLM = {
@@ -10,6 +10,7 @@ export interface BaseLangChainParams {
     verbose?: boolean;
     callbacks?: Callbacks;
     tags?: string[];
+    metadata?: Record<string, unknown>;
 }
 /**
  * Base class for language models, chains, tools.
@@ -21,6 +22,7 @@ export declare abstract class BaseLangChain extends Serializable implements Base
     verbose: boolean;
     callbacks?: Callbacks;
     tags?: string[];
+    metadata?: Record<string, unknown>;
     get lc_attributes(): {
         [key: string]: undefined;
     } | undefined;
@@ -37,7 +39,7 @@ export interface BaseLanguageModelParams extends AsyncCallerParams, BaseLangChai
      */
     callbackManager?: CallbackManager;
 }
-export interface BaseLanguageModelCallOptions {
+export interface BaseLanguageModelCallOptions extends BaseCallbackConfig {
     /**
      * Stop tokens to use for this call.
      * If not provided, the default stop tokens for the model will be used.
@@ -50,12 +52,9 @@ export interface BaseLanguageModelCallOptions {
     /**
      * Abort signal for this call.
      * If provided, the call will be aborted when the signal is aborted.
+     * @see https://developer.mozilla.org/en-US/docs/Web/API/AbortSignal
      */
     signal?: AbortSignal;
-    /**
-     * Tags to attach to this call.
-     */
-    tags?: string[];
 }
 /**
  * Base class for language models.
@@ -74,7 +73,7 @@ export declare abstract class BaseLanguageModel extends BaseLangChain implements
     constructor({ callbacks, callbackManager, ...params }: BaseLanguageModelParams);
     abstract generatePrompt(promptValues: BasePromptValue[], options?: string[] | this["CallOptions"], callbacks?: Callbacks): Promise<LLMResult>;
     abstract predict(text: string, options?: string[] | this["CallOptions"], callbacks?: Callbacks): Promise<string>;
-    abstract predictMessages(messages: BaseChatMessage[], options?: string[] | this["CallOptions"], callbacks?: Callbacks): Promise<BaseChatMessage>;
+    abstract predictMessages(messages: BaseMessage[], options?: string[] | this["CallOptions"], callbacks?: Callbacks): Promise<BaseMessage>;
     abstract _modelType(): string;
     abstract _llmType(): string;
     private _encoding?;

package/dist/base_language/index.js

@@ -36,9 +36,16 @@ export class BaseLangChain extends Serializable {
             writable: true,
             value: void 0
         });
+        Object.defineProperty(this, "metadata", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
         this.verbose = params.verbose ?? getVerbosity();
         this.callbacks = params.callbacks;
         this.tags = params.tags ?? [];
+        this.metadata = params.metadata ?? {};
     }
 }
 /**
@@ -49,7 +56,7 @@ export class BaseLanguageModel extends BaseLangChain {
      * Keys that the language model accepts as call options.
      */
     get callKeys() {
-        return ["stop", "timeout", "signal"];
+        return ["stop", "timeout", "signal", "tags", "metadata", "callbacks"];
     }
     constructor({ callbacks, callbackManager, ...params }) {
         super({

package/dist/callbacks/base.d.ts

@@ -1,4 +1,4 @@
-import { AgentAction, AgentFinish, BaseChatMessage, ChainValues, LLMResult } from "../schema/index.js";
+import { AgentAction, AgentFinish, BaseMessage, ChainValues, LLMResult } from "../schema/index.js";
 import { Serializable, Serialized, SerializedNotImplemented } from "../load/serializable.js";
 import { SerializedFields } from "../load/map_keys.js";
 type Error = any;
@@ -16,7 +16,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[]): Promise<void> | void;
+    handleLLMStart?(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): Promise<void> | void;
     /**
      * Called when an LLM/ChatModel in `streaming` mode produces a new token
      */
@@ -27,57 +27,57 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      * idx.completion is the index of the completion that produced the token
      *   (if multiple completions per prompt are requested)
      */
-    idx: NewTokenIndices, runId: string, parentRunId?: string): Promise<void> | void;
+    idx: NewTokenIndices, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called if an LLM/ChatModel run encounters an error
      */
-    handleLLMError?(err: Error, runId: string, parentRunId?: string): Promise<void> | void;
+    handleLLMError?(err: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called at the end of an LLM/ChatModel run, with the output and the run ID.
      */
-    handleLLMEnd?(output: LLMResult, runId: string, parentRunId?: string): Promise<void> | void;
+    handleLLMEnd?(output: LLMResult, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called at the start of a Chat Model run, with the prompt(s)
      * and the run ID.
      */
-    handleChatModelStart?(llm: Serialized, messages: BaseChatMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[]): Promise<void> | void;
+    handleChatModelStart?(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>, tags?: string[], metadata?: Record<string, unknown>): Promise<void> | void;
     /**
      * 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[]): Promise<void> | void;
+    handleChainStart?(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>): Promise<void> | void;
     /**
      * Called if a Chain run encounters an error
      */
-    handleChainError?(err: Error, runId: string, parentRunId?: string): Promise<void> | void;
+    handleChainError?(err: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called at the end of a Chain run, with the outputs and the run ID.
      */
-    handleChainEnd?(outputs: ChainValues, runId: string, parentRunId?: string): Promise<void> | void;
+    handleChainEnd?(outputs: ChainValues, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * 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[]): Promise<void> | void;
+    handleToolStart?(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: Record<string, unknown>): Promise<void> | void;
     /**
      * Called if a Tool run encounters an error
      */
-    handleToolError?(err: Error, runId: string, parentRunId?: string): Promise<void> | void;
+    handleToolError?(err: Error, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called at the end of a Tool run, with the tool output and the run ID.
      */
-    handleToolEnd?(output: string, runId: string, parentRunId?: string): Promise<void> | void;
-    handleText?(text: string, runId: string, parentRunId?: string): Promise<void> | void;
+    handleToolEnd?(output: string, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
+    handleText?(text: string, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called when an agent is about to execute an action,
      * with the action and the run ID.
      */
-    handleAgentAction?(action: AgentAction, runId: string, parentRunId?: string): Promise<void> | void;
+    handleAgentAction?(action: AgentAction, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
     /**
      * Called when an agent finishes execution, before it exits.
      * with the final output and the run ID.
      */
-    handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string): Promise<void> | void;
+    handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string, tags?: string[]): Promise<void> | void;
 }
 /**
  * Base interface for callbacks. All methods are optional. If a method is not
@@ -136,61 +136,61 @@ 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): void | Promise<void>;
+        handleLLMStart?(llm: Serialized, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): void | Promise<void>;
         /**
          * Called when an LLM/ChatModel in `streaming` mode produces a new token
          */
-        handleLLMNewToken?(token: string, idx: NewTokenIndices, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleLLMNewToken?(token: string, idx: NewTokenIndices, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called if an LLM/ChatModel run encounters an error
          */
-        handleLLMError?(err: any, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleLLMError?(err: any, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called at the end of an LLM/ChatModel run, with the output and the run ID.
          */
-        handleLLMEnd?(output: LLMResult, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleLLMEnd?(output: LLMResult, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called at the start of a Chat Model run, with the prompt(s)
          * and the run ID.
          */
-        handleChatModelStart?(llm: Serialized, messages: BaseChatMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined): void | Promise<void>;
+        handleChatModelStart?(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): void | Promise<void>;
         /**
          * 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): void | Promise<void>;
+        handleChainStart?(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): void | Promise<void>;
         /**
          * Called if a Chain run encounters an error
          */
-        handleChainError?(err: any, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleChainError?(err: any, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called at the end of a Chain run, with the outputs and the run ID.
          */
-        handleChainEnd?(outputs: ChainValues, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleChainEnd?(outputs: ChainValues, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * 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): void | Promise<void>;
+        handleToolStart?(tool: Serialized, input: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined, metadata?: Record<string, unknown> | undefined): void | Promise<void>;
         /**
          * Called if a Tool run encounters an error
          */
-        handleToolError?(err: any, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleToolError?(err: any, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called at the end of a Tool run, with the tool output and the run ID.
          */
-        handleToolEnd?(output: string, runId: string, parentRunId?: string | undefined): void | Promise<void>;
-        handleText?(text: string, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleToolEnd?(output: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
+        handleText?(text: string, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called when an agent is about to execute an action,
          * with the action and the run ID.
          */
-        handleAgentAction?(action: AgentAction, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleAgentAction?(action: AgentAction, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
         /**
          * Called when an agent finishes execution, before it exits.
          * with the final output and the run ID.
          */
-        handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        handleAgentEnd?(action: AgentFinish, runId: string, parentRunId?: string | undefined, tags?: string[] | undefined): void | Promise<void>;
     };
 }
 export {};

package/dist/callbacks/handlers/tracer.cjs

@@ -45,9 +45,12 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         }
         return parentRun.child_execution_order + 1;
     }
-    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags) {
+    async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
+        const finalExtraParams = metadata
+            ? { ...extraParams, metadata }
+            : extraParams;
         const run = {
             id: runId,
             name: llm.id[llm.id.length - 1],
@@ -65,15 +68,18 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
             child_runs: [],
             child_execution_order: execution_order,
             run_type: "llm",
-            extra: extraParams ?? {},
+            extra: finalExtraParams ?? {},
             tags: tags || [],
         };
         this._startTrace(run);
         await this.onLLMStart?.(run);
     }
-    async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags) {
+    async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
+        const finalExtraParams = metadata
+            ? { ...extraParams, metadata }
+            : extraParams;
         const run = {
             id: runId,
             name: llm.id[llm.id.length - 1],
@@ -91,7 +97,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
             child_runs: [],
             child_execution_order: execution_order,
             run_type: "llm",
-            extra: extraParams ?? {},
+            extra: finalExtraParams ?? {},
             tags: tags || [],
         };
         this._startTrace(run);
@@ -125,7 +131,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         await this.onLLMError?.(run);
         await this._endTrace(run);
     }
-    async handleChainStart(chain, inputs, runId, parentRunId, tags) {
+    async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const run = {
@@ -145,7 +151,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
             child_execution_order: execution_order,
             run_type: "chain",
             child_runs: [],
-            extra: {},
+            extra: metadata ? { metadata } : {},
             tags: tags || [],
         };
         this._startTrace(run);
@@ -179,7 +185,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         await this.onChainError?.(run);
         await this._endTrace(run);
     }
-    async handleToolStart(tool, input, runId, parentRunId, tags) {
+    async handleToolStart(tool, input, runId, parentRunId, tags, metadata) {
         const execution_order = this._getExecutionOrder(parentRunId);
         const start_time = Date.now();
         const run = {
@@ -199,7 +205,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
             child_execution_order: execution_order,
             run_type: "tool",
             child_runs: [],
-            extra: {},
+            extra: metadata ? { metadata } : {},
             tags: tags || [],
         };
         this._startTrace(run);
@@ -248,6 +254,18 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         });
         await this.onAgentAction?.(run);
     }
+    async handleAgentEnd(action, runId) {
+        const run = this.runMap.get(runId);
+        if (!run || run?.run_type !== "chain") {
+            return;
+        }
+        run.events.push({
+            name: "agent_end",
+            time: Date.now(),
+            kwargs: { action },
+        });
+        await this.onAgentEnd?.(run);
+    }
     async handleText(text, runId) {
         const run = this.runMap.get(runId);
         if (!run || run?.run_type !== "chain") {
@@ -260,5 +278,17 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         });
         await this.onText?.(run);
     }
+    async handleLLMNewToken(token, idx, runId) {
+        const run = this.runMap.get(runId);
+        if (!run || run?.run_type !== "llm") {
+            return;
+        }
+        run.events.push({
+            name: "new_token",
+            time: Date.now(),
+            kwargs: { token, idx },
+        });
+        await this.onLLMNewToken?.(run);
+    }
 }
 exports.BaseTracer = BaseTracer;

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

@@ -1,7 +1,7 @@
 import { KVMap, BaseRun } from "langchainplus-sdk/schemas";
-import { AgentAction, BaseChatMessage, ChainValues, LLMResult } from "../../schema/index.js";
+import { AgentAction, AgentFinish, BaseMessage, ChainValues, LLMResult } from "../../schema/index.js";
 import { Serialized } from "../../load/serializable.js";
-import { BaseCallbackHandler, BaseCallbackHandlerInput } from "../base.js";
+import { BaseCallbackHandler, BaseCallbackHandlerInput, NewTokenIndices } from "../base.js";
 export type RunType = "llm" | "chain" | "tool";
 export interface Run extends BaseRun {
     id: string;
@@ -27,18 +27,20 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     protected _startTrace(run: Run): void;
     protected _endTrace(run: Run): Promise<void>;
     protected _getExecutionOrder(parentRunId: string | undefined): number;
-    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[]): Promise<void>;
-    handleChatModelStart(llm: Serialized, messages: BaseChatMessage[][], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[]): Promise<void>;
+    handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[], metadata?: KVMap): Promise<void>;
+    handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[], metadata?: KVMap): Promise<void>;
     handleLLMEnd(output: LLMResult, runId: string): Promise<void>;
     handleLLMError(error: Error, runId: string): Promise<void>;
-    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap): Promise<void>;
     handleChainEnd(outputs: ChainValues, runId: string): Promise<void>;
     handleChainError(error: Error, runId: string): Promise<void>;
-    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[]): Promise<void>;
+    handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap): Promise<void>;
     handleToolEnd(output: string, runId: string): Promise<void>;
     handleToolError(error: Error, runId: string): Promise<void>;
     handleAgentAction(action: AgentAction, runId: string): Promise<void>;
+    handleAgentEnd(action: AgentFinish, runId: string): Promise<void>;
     handleText(text: string, runId: string): Promise<void>;
+    handleLLMNewToken(token: string, idx: NewTokenIndices, runId: string): Promise<void>;
     onLLMStart?(run: Run): void | Promise<void>;
     onLLMEnd?(run: Run): void | Promise<void>;
     onLLMError?(run: Run): void | Promise<void>;
@@ -49,5 +51,7 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     onToolEnd?(run: Run): void | Promise<void>;
     onToolError?(run: Run): void | Promise<void>;
     onAgentAction?(run: Run): void | Promise<void>;
+    onAgentEnd?(run: Run): void | Promise<void>;
     onText?(run: Run): void | Promise<void>;
+    onLLMNewToken?(run: Run): void | Promise<void>;
 }