langchain 0.0.101 → 0.0.103

package/agents/toolkits/aws_sfn.cjs

@@ -0,0 +1 @@
+module.exports = require('../../dist/agents/toolkits/aws_sfn.cjs');

package/agents/toolkits/aws_sfn.d.ts

@@ -0,0 +1 @@
+export * from '../../dist/agents/toolkits/aws_sfn.js'

package/agents/toolkits/aws_sfn.js

@@ -0,0 +1 @@
+export * from '../../dist/agents/toolkits/aws_sfn.js'

package/agents/toolkits/sql.cjs

@@ -0,0 +1 @@
+module.exports = require('../../dist/agents/toolkits/sql/index.cjs');

package/agents/toolkits/sql.d.ts

@@ -0,0 +1 @@
+export * from '../../dist/agents/toolkits/sql/index.js'

package/agents/toolkits/sql.js

@@ -0,0 +1 @@
+export * from '../../dist/agents/toolkits/sql/index.js'

package/chains/sql_db.cjs

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

package/chains/sql_db.d.ts

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

package/chains/sql_db.js

@@ -0,0 +1 @@
+export * from '../dist/chains/sql_db/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/index.cjs

@@ -1,24 +1,22 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAIAgent = exports.StructuredChatOutputParserWithRetries = exports.StructuredChatOutputParser = exports.StructuredChatAgent = exports.AgentActionOutputParser = exports.ZeroShotAgentOutputParser = exports.ZeroShotAgent = exports.initializeAgentExecutorWithOptions = exports.initializeAgentExecutor = exports.AgentExecutor = exports.ChatConversationalAgentOutputParserWithRetries = exports.ChatConversationalAgentOutputParser = exports.ChatConversationalAgent = exports.ChatAgentOutputParser = exports.ChatAgent = exports.Toolkit = exports.createVectorStoreRouterAgent = exports.createVectorStoreAgent = exports.createSqlAgent = exports.createOpenApiAgent = exports.createJsonAgent = exports.ZapierToolKit = exports.VectorStoreToolkit = exports.VectorStoreRouterToolkit = exports.SqlToolkit = exports.RequestsToolkit = exports.OpenApiToolkit = exports.JsonToolkit = exports.LLMSingleActionAgent = exports.BaseSingleActionAgent = exports.Agent = void 0;
+exports.OpenAIAgent = exports.StructuredChatOutputParserWithRetries = exports.StructuredChatOutputParser = exports.StructuredChatAgent = exports.AgentActionOutputParser = exports.ZeroShotAgentOutputParser = exports.ZeroShotAgent = exports.initializeAgentExecutorWithOptions = exports.initializeAgentExecutor = exports.AgentExecutor = exports.ChatConversationalAgentOutputParserWithRetries = exports.ChatConversationalAgentOutputParser = exports.ChatConversationalAgent = exports.ChatAgentOutputParser = exports.ChatAgent = exports.Toolkit = exports.createVectorStoreRouterAgent = exports.createVectorStoreAgent = exports.createOpenApiAgent = exports.createJsonAgent = exports.ZapierToolKit = exports.VectorStoreToolkit = exports.VectorStoreRouterToolkit = exports.RequestsToolkit = exports.OpenApiToolkit = exports.JsonToolkit = exports.LLMSingleActionAgent = exports.BaseSingleActionAgent = exports.Agent = void 0;
 var agent_js_1 = require("./agent.cjs");
 Object.defineProperty(exports, "Agent", { enumerable: true, get: function () { return agent_js_1.Agent; } });
 Object.defineProperty(exports, "BaseSingleActionAgent", { enumerable: true, get: function () { return agent_js_1.BaseSingleActionAgent; } });
 Object.defineProperty(exports, "LLMSingleActionAgent", { enumerable: true, get: function () { return agent_js_1.LLMSingleActionAgent; } });
-var index_js_1 = require("./agent_toolkits/index.cjs");
+var index_js_1 = require("./toolkits/index.cjs");
 Object.defineProperty(exports, "JsonToolkit", { enumerable: true, get: function () { return index_js_1.JsonToolkit; } });
 Object.defineProperty(exports, "OpenApiToolkit", { enumerable: true, get: function () { return index_js_1.OpenApiToolkit; } });
 Object.defineProperty(exports, "RequestsToolkit", { enumerable: true, get: function () { return index_js_1.RequestsToolkit; } });
-Object.defineProperty(exports, "SqlToolkit", { enumerable: true, get: function () { return index_js_1.SqlToolkit; } });
 Object.defineProperty(exports, "VectorStoreRouterToolkit", { enumerable: true, get: function () { return index_js_1.VectorStoreRouterToolkit; } });
 Object.defineProperty(exports, "VectorStoreToolkit", { enumerable: true, get: function () { return index_js_1.VectorStoreToolkit; } });
 Object.defineProperty(exports, "ZapierToolKit", { enumerable: true, get: function () { return index_js_1.ZapierToolKit; } });
 Object.defineProperty(exports, "createJsonAgent", { enumerable: true, get: function () { return index_js_1.createJsonAgent; } });
 Object.defineProperty(exports, "createOpenApiAgent", { enumerable: true, get: function () { return index_js_1.createOpenApiAgent; } });
-Object.defineProperty(exports, "createSqlAgent", { enumerable: true, get: function () { return index_js_1.createSqlAgent; } });
 Object.defineProperty(exports, "createVectorStoreAgent", { enumerable: true, get: function () { return index_js_1.createVectorStoreAgent; } });
 Object.defineProperty(exports, "createVectorStoreRouterAgent", { enumerable: true, get: function () { return index_js_1.createVectorStoreRouterAgent; } });
-var base_js_1 = require("./agent_toolkits/base.cjs");
+var base_js_1 = require("./toolkits/base.cjs");
 Object.defineProperty(exports, "Toolkit", { enumerable: true, get: function () { return base_js_1.Toolkit; } });
 var index_js_2 = require("./chat/index.cjs");
 Object.defineProperty(exports, "ChatAgent", { enumerable: true, get: function () { return index_js_2.ChatAgent; } });

package/dist/agents/index.d.ts

@@ -1,6 +1,6 @@
 export { Agent, AgentArgs, BaseSingleActionAgent, LLMSingleActionAgent, LLMSingleActionAgentInput, OutputParserArgs, } from "./agent.js";
-export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreInfo, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, SqlCreatePromptArgs, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./agent_toolkits/index.js";
-export { Toolkit } from "./agent_toolkits/base.js";
+export { JsonToolkit, OpenApiToolkit, RequestsToolkit, VectorStoreInfo, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./toolkits/index.js";
+export { Toolkit } from "./toolkits/base.js";
 export { ChatAgent, ChatAgentInput, ChatCreatePromptArgs, } from "./chat/index.js";
 export { ChatAgentOutputParser } from "./chat/outputParser.js";
 export { ChatConversationalAgent, ChatConversationalAgentInput, ChatConversationalCreatePromptArgs, } from "./chat_convo/index.js";

package/dist/agents/index.js

@@ -1,6 +1,6 @@
 export { Agent, BaseSingleActionAgent, LLMSingleActionAgent, } from "./agent.js";
-export { JsonToolkit, OpenApiToolkit, RequestsToolkit, SqlToolkit, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createSqlAgent, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./agent_toolkits/index.js";
-export { Toolkit } from "./agent_toolkits/base.js";
+export { JsonToolkit, OpenApiToolkit, RequestsToolkit, VectorStoreRouterToolkit, VectorStoreToolkit, ZapierToolKit, createJsonAgent, createOpenApiAgent, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./toolkits/index.js";
+export { Toolkit } from "./toolkits/base.js";
 export { ChatAgent, } from "./chat/index.js";
 export { ChatAgentOutputParser } from "./chat/outputParser.js";
 export { ChatConversationalAgent, } from "./chat_convo/index.js";

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/agents/{agent_toolkits → toolkits}/index.cjs

@@ -1,12 +1,9 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.ZapierToolKit = exports.createVectorStoreRouterAgent = exports.createVectorStoreAgent = exports.VectorStoreRouterToolkit = exports.VectorStoreToolkit = exports.createOpenApiAgent = exports.OpenApiToolkit = exports.RequestsToolkit = exports.createSqlAgent = exports.SqlToolkit = exports.createJsonAgent = exports.JsonToolkit = void 0;
+exports.ZapierToolKit = exports.createVectorStoreRouterAgent = exports.createVectorStoreAgent = exports.VectorStoreRouterToolkit = exports.VectorStoreToolkit = exports.createOpenApiAgent = exports.OpenApiToolkit = exports.RequestsToolkit = exports.createJsonAgent = exports.JsonToolkit = void 0;
 var json_js_1 = require("./json/json.cjs");
 Object.defineProperty(exports, "JsonToolkit", { enumerable: true, get: function () { return json_js_1.JsonToolkit; } });
 Object.defineProperty(exports, "createJsonAgent", { enumerable: true, get: function () { return json_js_1.createJsonAgent; } });
-var sql_js_1 = require("./sql/sql.cjs");
-Object.defineProperty(exports, "SqlToolkit", { enumerable: true, get: function () { return sql_js_1.SqlToolkit; } });
-Object.defineProperty(exports, "createSqlAgent", { enumerable: true, get: function () { return sql_js_1.createSqlAgent; } });
 var openapi_js_1 = require("./openapi/openapi.cjs");
 Object.defineProperty(exports, "RequestsToolkit", { enumerable: true, get: function () { return openapi_js_1.RequestsToolkit; } });
 Object.defineProperty(exports, "OpenApiToolkit", { enumerable: true, get: function () { return openapi_js_1.OpenApiToolkit; } });

package/dist/agents/{agent_toolkits → toolkits}/index.d.ts

@@ -1,5 +1,4 @@
 export { JsonToolkit, createJsonAgent } from "./json/json.js";
-export { SqlToolkit, createSqlAgent, SqlCreatePromptArgs } from "./sql/sql.js";
 export { RequestsToolkit, OpenApiToolkit, createOpenApiAgent, } from "./openapi/openapi.js";
 export { VectorStoreInfo, VectorStoreToolkit, VectorStoreRouterToolkit, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./vectorstore/vectorstore.js";
 export { ZapierToolKit } from "./zapier/zapier.js";

package/dist/agents/{agent_toolkits → toolkits}/index.js

@@ -1,5 +1,4 @@
 export { JsonToolkit, createJsonAgent } from "./json/json.js";
-export { SqlToolkit, createSqlAgent } from "./sql/sql.js";
 export { RequestsToolkit, OpenApiToolkit, createOpenApiAgent, } from "./openapi/openapi.js";
 export { VectorStoreToolkit, VectorStoreRouterToolkit, createVectorStoreAgent, createVectorStoreRouterAgent, } from "./vectorstore/vectorstore.js";
 export { ZapierToolKit } from "./zapier/zapier.js";

package/dist/agents/toolkits/sql/index.cjs

@@ -0,0 +1,6 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.createSqlAgent = exports.SqlToolkit = void 0;
+var sql_js_1 = require("./sql.cjs");
+Object.defineProperty(exports, "SqlToolkit", { enumerable: true, get: function () { return sql_js_1.SqlToolkit; } });
+Object.defineProperty(exports, "createSqlAgent", { enumerable: true, get: function () { return sql_js_1.createSqlAgent; } });

package/dist/agents/toolkits/sql/index.d.ts

@@ -0,0 +1 @@
+export { SqlCreatePromptArgs, SqlToolkit, createSqlAgent } from "./sql.js";

package/dist/agents/toolkits/sql/index.js

@@ -0,0 +1 @@
+export { SqlToolkit, createSqlAgent } from "./sql.js";

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 {};