langchain 0.0.74 → 0.0.76

package/client.cjs

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

package/client.d.ts

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

package/client.js

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

package/dist/base_language/count_tokens.cjs

@@ -1,7 +1,8 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateMaxTokens = exports.importTiktoken = exports.getModelContextSize = exports.getEmbeddingContextSize = exports.getModelNameForTiktoken = void 0;
-// https://www.npmjs.com/package/@dqbd/tiktoken
+exports.calculateMaxTokens = exports.getModelContextSize = exports.getEmbeddingContextSize = exports.getModelNameForTiktoken = void 0;
+const tiktoken_js_1 = require("../util/tiktoken.cjs");
+// https://www.npmjs.com/package/js-tiktoken
 const getModelNameForTiktoken = (modelName) => {
     if (modelName.startsWith("gpt-3.5-turbo-")) {
         return "gpt-3.5-turbo";
@@ -49,31 +50,14 @@ const getModelContextSize = (modelName) => {
     }
 };
 exports.getModelContextSize = getModelContextSize;
-const importTiktoken = async () => {
-    try {
-        const { encoding_for_model } = await import("@dqbd/tiktoken");
-        return { encoding_for_model };
-    }
-    catch (error) {
-        console.log(error);
-        return { encoding_for_model: null };
-    }
-};
-exports.importTiktoken = importTiktoken;
 const calculateMaxTokens = async ({ prompt, modelName, }) => {
-    const { encoding_for_model } = await (0, exports.importTiktoken)();
     // fallback to approximate calculation if tiktoken is not available
     let numTokens = Math.ceil(prompt.length / 4);
     try {
-        if (encoding_for_model) {
-            const encoding = encoding_for_model((0, exports.getModelNameForTiktoken)(modelName));
-            const tokenized = encoding.encode(prompt);
-            numTokens = tokenized.length;
-            encoding.free();
-        }
+        numTokens = (await (0, tiktoken_js_1.encodingForModel)(modelName)).encode(prompt).length;
     }
     catch (error) {
-        console.warn("Failed to calculate number of tokens with tiktoken, falling back to approximate count", error);
+        console.warn("Failed to calculate number of tokens, falling back to approximate count");
     }
     const maxTokens = (0, exports.getModelContextSize)(modelName);
     return maxTokens - numTokens;

package/dist/base_language/count_tokens.d.ts

@@ -1,4 +1,4 @@
-import type { TiktokenModel } from "@dqbd/tiktoken";
+import { type TiktokenModel } from "js-tiktoken/lite";
 export declare const getModelNameForTiktoken: (modelName: string) => TiktokenModel;
 export declare const getEmbeddingContextSize: (modelName?: string) => number;
 export declare const getModelContextSize: (modelName: string) => number;
@@ -6,10 +6,5 @@ interface CalculateMaxTokenProps {
     prompt: string;
     modelName: TiktokenModel;
 }
-export declare const importTiktoken: () => Promise<{
-    encoding_for_model: typeof import("@dqbd/tiktoken").encoding_for_model;
-} | {
-    encoding_for_model: null;
-}>;
 export declare const calculateMaxTokens: ({ prompt, modelName, }: CalculateMaxTokenProps) => Promise<number>;
 export {};

package/dist/base_language/count_tokens.js

@@ -1,4 +1,5 @@
-// https://www.npmjs.com/package/@dqbd/tiktoken
+import { encodingForModel } from "../util/tiktoken.js";
+// https://www.npmjs.com/package/js-tiktoken
 export const getModelNameForTiktoken = (modelName) => {
     if (modelName.startsWith("gpt-3.5-turbo-")) {
         return "gpt-3.5-turbo";
@@ -43,30 +44,14 @@ export const getModelContextSize = (modelName) => {
             return 4097;
     }
 };
-export const importTiktoken = async () => {
-    try {
-        const { encoding_for_model } = await import("@dqbd/tiktoken");
-        return { encoding_for_model };
-    }
-    catch (error) {
-        console.log(error);
-        return { encoding_for_model: null };
-    }
-};
 export const calculateMaxTokens = async ({ prompt, modelName, }) => {
-    const { encoding_for_model } = await importTiktoken();
     // fallback to approximate calculation if tiktoken is not available
     let numTokens = Math.ceil(prompt.length / 4);
     try {
-        if (encoding_for_model) {
-            const encoding = encoding_for_model(getModelNameForTiktoken(modelName));
-            const tokenized = encoding.encode(prompt);
-            numTokens = tokenized.length;
-            encoding.free();
-        }
+        numTokens = (await encodingForModel(modelName)).encode(prompt).length;
     }
     catch (error) {
-        console.warn("Failed to calculate number of tokens with tiktoken, falling back to approximate count", error);
+        console.warn("Failed to calculate number of tokens, falling back to approximate count");
     }
     const maxTokens = getModelContextSize(modelName);
     return maxTokens - numTokens;

package/dist/base_language/index.cjs

@@ -3,6 +3,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.BaseLanguageModel = exports.BaseLangChain = void 0;
 const async_caller_js_1 = require("../util/async_caller.cjs");
 const count_tokens_js_1 = require("./count_tokens.cjs");
+const tiktoken_js_1 = require("../util/tiktoken.cjs");
 const getVerbosity = () => false;
 /**
  * Base class for language models, chains, tools.
@@ -33,6 +34,12 @@ exports.BaseLangChain = BaseLangChain;
  * Base class for language models.
  */
 class BaseLanguageModel extends BaseLangChain {
+    /**
+     * Keys that the language model accepts as call options.
+     */
+    get callKeys() {
+        return ["stop", "timeout", "signal"];
+    }
     constructor(params) {
         super({
             verbose: params.verbose,
@@ -54,38 +61,23 @@ class BaseLanguageModel extends BaseLangChain {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "_registry", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         this.caller = new async_caller_js_1.AsyncCaller(params ?? {});
     }
     async getNumTokens(text) {
         // fallback to approximate calculation if tiktoken is not available
         let numTokens = Math.ceil(text.length / 4);
-        try {
-            if (!this._encoding) {
-                const { encoding_for_model } = await (0, count_tokens_js_1.importTiktoken)();
-                // modelName only exists in openai subclasses, but tiktoken only supports
-                // openai tokenisers anyway, so for other subclasses we default to gpt2
-                if (encoding_for_model) {
-                    this._encoding = encoding_for_model("modelName" in this
-                        ? (0, count_tokens_js_1.getModelNameForTiktoken)(this.modelName)
-                        : "gpt2");
-                    // We need to register a finalizer to free the tokenizer when the
-                    // model is garbage collected.
-                    this._registry = new FinalizationRegistry((t) => t.free());
-                    this._registry.register(this, this._encoding);
-                }
+        if (!this._encoding) {
+            try {
+                this._encoding = await (0, tiktoken_js_1.encodingForModel)("modelName" in this
+                    ? (0, count_tokens_js_1.getModelNameForTiktoken)(this.modelName)
+                    : "gpt2");
             }
-            if (this._encoding) {
-                numTokens = this._encoding.encode(text).length;
+            catch (error) {
+                console.warn("Failed to calculate number of tokens, falling back to approximate count", error);
             }
         }
-        catch (error) {
-            console.warn("Failed to calculate number of tokens with tiktoken, falling back to approximate count", error);
+        if (this._encoding) {
+            numTokens = this._encoding.encode(text).length;
         }
         return numTokens;
     }

package/dist/base_language/index.d.ts

@@ -1,4 +1,4 @@
-import { BasePromptValue, LLMResult } from "../schema/index.js";
+import { BaseChatMessage, BasePromptValue, LLMResult } from "../schema/index.js";
 import { CallbackManager, Callbacks } from "../callbacks/manager.js";
 import { AsyncCaller, AsyncCallerParams } from "../util/async_caller.js";
 export type SerializedLLM = {
@@ -32,23 +32,42 @@ export interface BaseLanguageModelParams extends AsyncCallerParams, BaseLangChai
     callbackManager?: CallbackManager;
 }
 export interface BaseLanguageModelCallOptions {
+    /**
+     * Stop tokens to use for this call.
+     * If not provided, the default stop tokens for the model will be used.
+     */
+    stop?: string[];
+    /**
+     * Timeout for this call in milliseconds.
+     */
+    timeout?: number;
+    /**
+     * Abort signal for this call.
+     * If provided, the call will be aborted when the signal is aborted.
+     */
+    signal?: AbortSignal;
 }
 /**
  * Base class for language models.
  */
 export declare abstract class BaseLanguageModel extends BaseLangChain implements BaseLanguageModelParams {
     CallOptions: BaseLanguageModelCallOptions;
+    /**
+     * Keys that the language model accepts as call options.
+     */
+    get callKeys(): string[];
     /**
      * The async caller should be used by subclasses to make any async calls,
      * which will thus benefit from the concurrency and retry logic.
      */
     caller: AsyncCaller;
     constructor(params: BaseLanguageModelParams);
-    abstract generatePrompt(promptValues: BasePromptValue[], stop?: string[] | this["CallOptions"], callbacks?: Callbacks): Promise<LLMResult>;
+    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 _modelType(): string;
     abstract _llmType(): string;
     private _encoding?;
-    private _registry?;
     getNumTokens(text: string): Promise<number>;
     /**
      * Get the identifying parameters of the LLM.

package/dist/base_language/index.js

@@ -1,5 +1,6 @@
 import { AsyncCaller } from "../util/async_caller.js";
-import { getModelNameForTiktoken, importTiktoken } from "./count_tokens.js";
+import { getModelNameForTiktoken } from "./count_tokens.js";
+import { encodingForModel } from "../util/tiktoken.js";
 const getVerbosity = () => false;
 /**
  * Base class for language models, chains, tools.
@@ -29,6 +30,12 @@ export class BaseLangChain {
  * Base class for language models.
  */
 export class BaseLanguageModel extends BaseLangChain {
+    /**
+     * Keys that the language model accepts as call options.
+     */
+    get callKeys() {
+        return ["stop", "timeout", "signal"];
+    }
     constructor(params) {
         super({
             verbose: params.verbose,
@@ -50,38 +57,23 @@ export class BaseLanguageModel extends BaseLangChain {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "_registry", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
         this.caller = new AsyncCaller(params ?? {});
     }
     async getNumTokens(text) {
         // fallback to approximate calculation if tiktoken is not available
         let numTokens = Math.ceil(text.length / 4);
-        try {
-            if (!this._encoding) {
-                const { encoding_for_model } = await importTiktoken();
-                // modelName only exists in openai subclasses, but tiktoken only supports
-                // openai tokenisers anyway, so for other subclasses we default to gpt2
-                if (encoding_for_model) {
-                    this._encoding = encoding_for_model("modelName" in this
-                        ? getModelNameForTiktoken(this.modelName)
-                        : "gpt2");
-                    // We need to register a finalizer to free the tokenizer when the
-                    // model is garbage collected.
-                    this._registry = new FinalizationRegistry((t) => t.free());
-                    this._registry.register(this, this._encoding);
-                }
+        if (!this._encoding) {
+            try {
+                this._encoding = await encodingForModel("modelName" in this
+                    ? getModelNameForTiktoken(this.modelName)
+                    : "gpt2");
             }
-            if (this._encoding) {
-                numTokens = this._encoding.encode(text).length;
+            catch (error) {
+                console.warn("Failed to calculate number of tokens, falling back to approximate count", error);
             }
         }
-        catch (error) {
-            console.warn("Failed to calculate number of tokens with tiktoken, falling back to approximate count", error);
+        if (this._encoding) {
+            numTokens = this._encoding.encode(text).length;
         }
         return numTokens;
     }

package/dist/cache/redis.d.ts

@@ -1,8 +1,10 @@
-import type { RedisClientType } from "redis";
+import type { createCluster, createClient } from "redis";
 import { BaseCache, Generation } from "../schema/index.js";
+type RedisClientType = ReturnType<typeof createClient> | ReturnType<typeof createCluster>;
 export declare class RedisCache extends BaseCache {
     private redisClient;
     constructor(redisClient: RedisClientType);
     lookup(prompt: string, llmKey: string): Promise<Generation[] | null>;
     update(prompt: string, llmKey: string, value: Generation[]): Promise<void>;
 }
+export {};

package/dist/callbacks/base.d.ts

@@ -1,4 +1,4 @@
-import { AgentAction, AgentFinish, ChainValues, LLMResult } from "../schema/index.js";
+import { AgentAction, AgentFinish, BaseChatMessage, ChainValues, LLMResult } from "../schema/index.js";
 type Error = any;
 export interface BaseCallbackHandlerInput {
     ignoreLLM?: boolean;
@@ -12,7 +12,7 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      */
     handleLLMStart?(llm: {
         name: string;
-    }, prompts: string[], runId: string, parentRunId?: string): Promise<void> | void;
+    }, prompts: string[], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>): Promise<void> | void;
     /**
      * Called when an LLM/ChatModel in `streaming` mode produces a new token
      */
@@ -25,6 +25,13 @@ declare abstract class BaseCallbackHandlerMethodsClass {
      * 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;
+    /**
+     * Called at the start of a Chat Model run, with the prompt(s)
+     * and the run ID.
+     */
+    handleChatModelStart?(llm: {
+        name: string;
+    }, messages: BaseChatMessage[][], runId: string, parentRunId?: string, extraParams?: Record<string, unknown>): Promise<void> | void;
     /**
      * Called at the start of a Chain run, with the chain name and inputs
      * and the run ID.
@@ -96,7 +103,7 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
          */
         handleLLMStart?(llm: {
             name: string;
-        }, prompts: string[], runId: string, parentRunId?: string | undefined): void | Promise<void>;
+        }, prompts: string[], runId: string, parentRunId?: string | undefined, extraParams?: Record<string, unknown> | undefined): void | Promise<void>;
         /**
          * Called when an LLM/ChatModel in `streaming` mode produces a new token
          */
@@ -109,6 +116,13 @@ export declare abstract class BaseCallbackHandler extends BaseCallbackHandlerMet
          * 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>;
+        /**
+         * Called at the start of a Chat Model run, with the prompt(s)
+         * and the run ID.
+         */
+        handleChatModelStart?(llm: {
+            name: string;
+        }, messages: BaseChatMessage[][], runId: string, parentRunId?: string | undefined, extraParams?: 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.

package/dist/callbacks/handlers/console.cjs

@@ -5,7 +5,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ConsoleCallbackHandler = void 0;
 const ansi_styles_1 = __importDefault(require("ansi-styles"));
-const tracers_js_1 = require("./tracers.cjs");
+const tracer_js_1 = require("./tracer.cjs");
 function wrap(style, text) {
     return `${style.open}${text}${style.close}`;
 }
@@ -25,42 +25,25 @@ function elapsed(run) {
     return `${(elapsed / 1000).toFixed(2)}s`;
 }
 const { color } = ansi_styles_1.default;
-class ConsoleCallbackHandler extends tracers_js_1.BaseTracer {
-    // boilerplate to work with the base tracer class
+class ConsoleCallbackHandler extends tracer_js_1.BaseTracer {
     constructor() {
-        super();
+        super(...arguments);
         Object.defineProperty(this, "name", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: "console_callback_handler"
         });
-        Object.defineProperty(this, "i", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-    }
-    persistSession(session) {
-        // eslint-disable-next-line no-plusplus
-        return Promise.resolve({ ...session, id: this.i++ });
     }
     persistRun(_run) {
         return Promise.resolve();
     }
-    loadDefaultSession() {
-        return this.newSession();
-    }
-    loadSession(sessionName) {
-        return this.newSession(sessionName);
-    }
     // utility methods
     getParents(run) {
         const parents = [];
         let currentRun = run;
-        while (currentRun.parent_uuid) {
-            const parent = this.runMap.get(currentRun.parent_uuid);
+        while (currentRun.parent_run_id) {
+            const parent = this.runMap.get(currentRun.parent_run_id);
             if (parent) {
                 parents.push(parent);
                 currentRun = parent;
@@ -75,7 +58,7 @@ class ConsoleCallbackHandler extends tracers_js_1.BaseTracer {
         const parents = this.getParents(run).reverse();
         const string = [...parents, run]
             .map((parent, i, arr) => {
-            const name = `${parent.execution_order}:${parent.type}:${parent.serialized?.name}`;
+            const name = `${parent.execution_order}:${parent.run_type}:${parent.name}`;
             return i === arr.length - 1 ? wrap(ansi_styles_1.default.bold, name) : name;
         })
             .join(" > ");
@@ -96,11 +79,14 @@ class ConsoleCallbackHandler extends tracers_js_1.BaseTracer {
     }
     onLLMStart(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.green, "[llm/start]")} [${crumbs}] Entering LLM run with input: ${tryJsonStringify({ prompts: run.prompts.map((p) => p.trim()) }, "[inputs]")}`);
+        const inputs = "prompts" in run.inputs
+            ? { prompts: run.inputs.prompts.map((p) => p.trim()) }
+            : run.inputs;
+        console.log(`${wrap(color.green, "[llm/start]")} [${crumbs}] Entering LLM run with input: ${tryJsonStringify(inputs, "[inputs]")}`);
     }
     onLLMEnd(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.cyan, "[llm/end]")} [${crumbs}] [${elapsed(run)}] Exiting LLM run with output: ${tryJsonStringify(run.response, "[response]")}`);
+        console.log(`${wrap(color.cyan, "[llm/end]")} [${crumbs}] [${elapsed(run)}] Exiting LLM run with output: ${tryJsonStringify(run.outputs, "[response]")}`);
     }
     onLLMError(run) {
         const crumbs = this.getBreadcrumbs(run);
@@ -108,19 +94,20 @@ class ConsoleCallbackHandler extends tracers_js_1.BaseTracer {
     }
     onToolStart(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.green, "[tool/start]")} [${crumbs}] Entering Tool run with input: "${run.tool_input?.trim()}"`);
+        console.log(`${wrap(color.green, "[tool/start]")} [${crumbs}] Entering Tool run with input: "${run.inputs.input?.trim()}"`);
     }
     onToolEnd(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.cyan, "[tool/end]")} [${crumbs}] [${elapsed(run)}] Exiting Tool run with output: "${run.output?.trim()}"`);
+        console.log(`${wrap(color.cyan, "[tool/end]")} [${crumbs}] [${elapsed(run)}] Exiting Tool run with output: "${run.outputs?.output?.trim()}"`);
     }
     onToolError(run) {
         const crumbs = this.getBreadcrumbs(run);
         console.log(`${wrap(color.red, "[tool/error]")} [${crumbs}] [${elapsed(run)}] Tool run errored with error: ${tryJsonStringify(run.error, "[error]")}`);
     }
     onAgentAction(run) {
+        const agentRun = run;
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.blue, "[agent/action]")} [${crumbs}] Agent selected action: ${tryJsonStringify(run.actions[run.actions.length - 1], "[action]")}`);
+        console.log(`${wrap(color.blue, "[agent/action]")} [${crumbs}] Agent selected action: ${tryJsonStringify(agentRun.actions[agentRun.actions.length - 1], "[action]")}`);
     }
 }
 exports.ConsoleCallbackHandler = ConsoleCallbackHandler;

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

@@ -1,26 +1,17 @@
-import { AgentRun, BaseTracer, BaseTracerSession, ChainRun, LLMRun, Run, ToolRun } from "./tracers.js";
+import { BaseTracer, Run } from "./tracer.js";
 export declare class ConsoleCallbackHandler extends BaseTracer {
     name: "console_callback_handler";
-    constructor();
-    i: number;
-    protected persistSession(session: BaseTracerSession): Promise<{
-        id: number;
-        start_time: number;
-        name?: string | undefined;
-    }>;
     protected persistRun(_run: Run): Promise<void>;
-    loadDefaultSession(): Promise<import("./tracers.js").TracerSession>;
-    loadSession(sessionName: string): Promise<import("./tracers.js").TracerSession>;
     getParents(run: Run): Run[];
     getBreadcrumbs(run: Run): string;
-    onChainStart(run: ChainRun): void;
-    onChainEnd(run: ChainRun): void;
-    onChainError(run: ChainRun): void;
-    onLLMStart(run: LLMRun): void;
-    onLLMEnd(run: LLMRun): void;
-    onLLMError(run: LLMRun): void;
-    onToolStart(run: ToolRun): void;
-    onToolEnd(run: ToolRun): void;
-    onToolError(run: ToolRun): void;
-    onAgentAction(run: AgentRun): void;
+    onChainStart(run: Run): void;
+    onChainEnd(run: Run): void;
+    onChainError(run: Run): void;
+    onLLMStart(run: Run): void;
+    onLLMEnd(run: Run): void;
+    onLLMError(run: Run): void;
+    onToolStart(run: Run): void;
+    onToolEnd(run: Run): void;
+    onToolError(run: Run): void;
+    onAgentAction(run: Run): void;
 }

package/dist/callbacks/handlers/console.js

@@ -1,5 +1,5 @@
 import styles from "ansi-styles";
-import { BaseTracer, } from "./tracers.js";
+import { BaseTracer } from "./tracer.js";
 function wrap(style, text) {
     return `${style.open}${text}${style.close}`;
 }
@@ -20,41 +20,24 @@ function elapsed(run) {
 }
 const { color } = styles;
 export class ConsoleCallbackHandler extends BaseTracer {
-    // boilerplate to work with the base tracer class
     constructor() {
-        super();
+        super(...arguments);
         Object.defineProperty(this, "name", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: "console_callback_handler"
         });
-        Object.defineProperty(this, "i", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 0
-        });
-    }
-    persistSession(session) {
-        // eslint-disable-next-line no-plusplus
-        return Promise.resolve({ ...session, id: this.i++ });
     }
     persistRun(_run) {
         return Promise.resolve();
     }
-    loadDefaultSession() {
-        return this.newSession();
-    }
-    loadSession(sessionName) {
-        return this.newSession(sessionName);
-    }
     // utility methods
     getParents(run) {
         const parents = [];
         let currentRun = run;
-        while (currentRun.parent_uuid) {
-            const parent = this.runMap.get(currentRun.parent_uuid);
+        while (currentRun.parent_run_id) {
+            const parent = this.runMap.get(currentRun.parent_run_id);
             if (parent) {
                 parents.push(parent);
                 currentRun = parent;
@@ -69,7 +52,7 @@ export class ConsoleCallbackHandler extends BaseTracer {
         const parents = this.getParents(run).reverse();
         const string = [...parents, run]
             .map((parent, i, arr) => {
-            const name = `${parent.execution_order}:${parent.type}:${parent.serialized?.name}`;
+            const name = `${parent.execution_order}:${parent.run_type}:${parent.name}`;
             return i === arr.length - 1 ? wrap(styles.bold, name) : name;
         })
             .join(" > ");
@@ -90,11 +73,14 @@ export class ConsoleCallbackHandler extends BaseTracer {
     }
     onLLMStart(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.green, "[llm/start]")} [${crumbs}] Entering LLM run with input: ${tryJsonStringify({ prompts: run.prompts.map((p) => p.trim()) }, "[inputs]")}`);
+        const inputs = "prompts" in run.inputs
+            ? { prompts: run.inputs.prompts.map((p) => p.trim()) }
+            : run.inputs;
+        console.log(`${wrap(color.green, "[llm/start]")} [${crumbs}] Entering LLM run with input: ${tryJsonStringify(inputs, "[inputs]")}`);
     }
     onLLMEnd(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.cyan, "[llm/end]")} [${crumbs}] [${elapsed(run)}] Exiting LLM run with output: ${tryJsonStringify(run.response, "[response]")}`);
+        console.log(`${wrap(color.cyan, "[llm/end]")} [${crumbs}] [${elapsed(run)}] Exiting LLM run with output: ${tryJsonStringify(run.outputs, "[response]")}`);
     }
     onLLMError(run) {
         const crumbs = this.getBreadcrumbs(run);
@@ -102,18 +88,19 @@ export class ConsoleCallbackHandler extends BaseTracer {
     }
     onToolStart(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.green, "[tool/start]")} [${crumbs}] Entering Tool run with input: "${run.tool_input?.trim()}"`);
+        console.log(`${wrap(color.green, "[tool/start]")} [${crumbs}] Entering Tool run with input: "${run.inputs.input?.trim()}"`);
     }
     onToolEnd(run) {
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.cyan, "[tool/end]")} [${crumbs}] [${elapsed(run)}] Exiting Tool run with output: "${run.output?.trim()}"`);
+        console.log(`${wrap(color.cyan, "[tool/end]")} [${crumbs}] [${elapsed(run)}] Exiting Tool run with output: "${run.outputs?.output?.trim()}"`);
     }
     onToolError(run) {
         const crumbs = this.getBreadcrumbs(run);
         console.log(`${wrap(color.red, "[tool/error]")} [${crumbs}] [${elapsed(run)}] Tool run errored with error: ${tryJsonStringify(run.error, "[error]")}`);
     }
     onAgentAction(run) {
+        const agentRun = run;
         const crumbs = this.getBreadcrumbs(run);
-        console.log(`${wrap(color.blue, "[agent/action]")} [${crumbs}] Agent selected action: ${tryJsonStringify(run.actions[run.actions.length - 1], "[action]")}`);
+        console.log(`${wrap(color.blue, "[agent/action]")} [${crumbs}] Agent selected action: ${tryJsonStringify(agentRun.actions[agentRun.actions.length - 1], "[action]")}`);
     }
 }

package/dist/callbacks/handlers/initialize.cjs

@@ -1,9 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getTracingCallbackHandler = void 0;
-const tracers_js_1 = require("./tracers.cjs");
+exports.getTracingV2CallbackHandler = exports.getTracingCallbackHandler = void 0;
+const tracer_langchain_js_1 = require("./tracer_langchain.cjs");
+const tracer_langchain_v1_js_1 = require("./tracer_langchain_v1.cjs");
 async function getTracingCallbackHandler(session) {
-    const tracer = new tracers_js_1.LangChainTracer();
+    const tracer = new tracer_langchain_v1_js_1.LangChainTracerV1();
     if (session) {
         await tracer.loadSession(session);
     }
@@ -13,3 +14,7 @@ async function getTracingCallbackHandler(session) {
     return tracer;
 }
 exports.getTracingCallbackHandler = getTracingCallbackHandler;
+async function getTracingV2CallbackHandler() {
+    return new tracer_langchain_js_1.LangChainTracer();
+}
+exports.getTracingV2CallbackHandler = getTracingV2CallbackHandler;

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

@@ -1,2 +1,4 @@
-import { LangChainTracer } from "./tracers.js";
-export declare function getTracingCallbackHandler(session?: string): Promise<LangChainTracer>;
+import { LangChainTracer } from "./tracer_langchain.js";
+import { LangChainTracerV1 } from "./tracer_langchain_v1.js";
+export declare function getTracingCallbackHandler(session?: string): Promise<LangChainTracerV1>;
+export declare function getTracingV2CallbackHandler(): Promise<LangChainTracer>;