langchain 0.0.146 → 0.0.148

package/dist/evaluation/agents/trajectory.cjs

@@ -0,0 +1,189 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TrajectoryEvalChain = exports.TrajectoryOutputParser = void 0;
+const output_parser_js_1 = require("../../schema/output_parser.cjs");
+const base_js_1 = require("../base.cjs");
+const index_js_1 = require("../../schema/index.cjs");
+const prompt_js_1 = require("./prompt.cjs");
+/**
+ * A parser for the output of the TrajectoryEvalChain.
+ */
+class TrajectoryOutputParser extends output_parser_js_1.BaseLLMOutputParser {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "evaluation", "agents"]
+        });
+    }
+    static lc_name() {
+        return "TrajectoryOutputParser";
+    }
+    parseResult(generations, _callbacks) {
+        const { text } = generations[0];
+        if (!text.includes("Score:")) {
+            throw new Error(`Could not find score in model eval output: ${text}`);
+        }
+        let [reasoning, scoreStr] = text.split("Score:", 2);
+        reasoning = reasoning.trim();
+        scoreStr = scoreStr.trim();
+        // Use regex to extract the score.
+        // This will get the number in the string, even if it is a float or more than 10.
+        // E.g. "Score: 1" will return 1, "Score: 3.5" will return 3.5, and
+        // "Score: 10" will return 10.
+        // The score should be an integer digit in the range 1-5.
+        const scoreMatch = scoreStr.match(/(\d+(\.\d+)?)/);
+        if (scoreMatch === null || scoreMatch[1].includes(".")) {
+            throw new Error(`Score is not an integer digit in the range 1-5: ${text}`);
+        }
+        const score = +scoreMatch[1];
+        if (score < 1 || score > 5) {
+            throw new Error(`Score is not a digit in the range 1-5: ${text}`);
+        }
+        const normalizedScore = (score - 1) / 4;
+        return Promise.resolve({
+            reasoning,
+            score: normalizedScore,
+        });
+    }
+}
+exports.TrajectoryOutputParser = TrajectoryOutputParser;
+/**
+ * A chain for evaluating ReAct style agents.
+ *
+ * This chain is used to evaluate ReAct style agents by reasoning about
+ * the sequence of actions taken and their outcomes.
+ */
+class TrajectoryEvalChain extends base_js_1.AgentTrajectoryEvaluator {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "criterionName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.criterionName
+        });
+        Object.defineProperty(this, "requiresInput", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        Object.defineProperty(this, "requiresReference", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "outputParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new TrajectoryOutputParser()
+        });
+    }
+    static lc_name() {
+        return "TrajectoryEvalChain";
+    }
+    static resolveTrajectoryPrompt(prompt, agentTools) {
+        let _prompt;
+        if (prompt) {
+            _prompt = prompt;
+        }
+        else if (agentTools) {
+            _prompt = prompt_js_1.EVAL_CHAT_PROMPT;
+        }
+        else {
+            _prompt = prompt_js_1.TOOL_FREE_EVAL_CHAT_PROMPT;
+        }
+        return _prompt;
+    }
+    /**
+     * Get the description of the agent tools.
+     *
+     * @returns The description of the agent tools.
+     */
+    static toolsDescription(agentTools) {
+        return agentTools
+            .map((tool, i) => `Tool ${i + 1}: ${tool.name}\n Description: ${tool.description}`)
+            .join("\n\n");
+    }
+    /**
+     * Create a new TrajectoryEvalChain.
+     * @param llm
+     * @param agentTools - The tools used by the agent.
+     * @param chainOptions - The options for the chain.
+     */
+    static async fromLLM(llm, agentTools, chainOptions) {
+        let prompt = this.resolveTrajectoryPrompt(chainOptions?.prompt, agentTools);
+        if (agentTools) {
+            const toolDescriptions = this.toolsDescription(agentTools);
+            prompt = await prompt.partial({ toolDescriptions });
+        }
+        const options = chainOptions;
+        if (options) {
+            // remove prompt from chainOptions
+            delete options.prompt;
+        }
+        return new this({
+            llm,
+            prompt,
+            ...options,
+        });
+    }
+    _prepareOutput(result) {
+        const parsed = result[this.outputKey];
+        if (index_js_1.RUN_KEY in result && result[index_js_1.RUN_KEY]) {
+            parsed[index_js_1.RUN_KEY] = result[index_js_1.RUN_KEY];
+        }
+        return parsed;
+    }
+    /**
+     * Get the agent trajectory as a formatted string.
+     *
+     * @param steps - The agent trajectory.
+     * @returns The formatted agent trajectory.
+     */
+    getAgentTrajectory(steps) {
+        return steps
+            .map((step, i) => {
+            const { action, observation } = step;
+            return (`Step ${i + 1}:\n` +
+                `Tool used: ${action.tool}\n` +
+                `Tool input: ${action.toolInput}\n` +
+                `Tool output: ${observation}`);
+        })
+            .join("\n\n");
+    }
+    formatReference(reference) {
+        if (!reference) {
+            return "";
+        }
+        return `
+The following is the expected answer. Use this to measure correctness:
+[GROUND_TRUTH]
+${reference}
+[END_GROUND_TRUTH]
+        `;
+    }
+    async _evaluateAgentTrajectory(args, callOptions, config) {
+        const { input, prediction, reference, agentTrajectory } = args;
+        const inputs = {
+            question: input,
+            agentTrajectory: this.getAgentTrajectory(agentTrajectory),
+            answer: prediction,
+            reference: this.formatReference(reference),
+        };
+        const result = await this.call({ ...inputs, ...callOptions }, config);
+        return this._prepareOutput(result);
+    }
+}
+exports.TrajectoryEvalChain = TrajectoryEvalChain;

package/dist/evaluation/agents/trajectory.d.ts

@@ -0,0 +1,54 @@
+import { BaseLLMOutputParser } from "../../schema/output_parser.js";
+import { AgentTrajectoryEvaluator, EvalOutputType, LLMEvalChainInput, LLMTrajectoryEvaluatorArgs } from "../base.js";
+import { AgentStep, ChainValues, ChatGeneration, Generation } from "../../schema/index.js";
+import { Callbacks } from "../../callbacks/index.js";
+import { BaseCallbackConfig } from "../../callbacks/manager.js";
+import { BasePromptTemplate } from "../../prompts/index.js";
+import { StructuredTool } from "../../tools/index.js";
+import { BaseChatModel } from "../../chat_models/base.js";
+/**
+ * A parser for the output of the TrajectoryEvalChain.
+ */
+export declare class TrajectoryOutputParser extends BaseLLMOutputParser<EvalOutputType> {
+    static lc_name(): string;
+    lc_namespace: string[];
+    parseResult(generations: Generation[] | ChatGeneration[], _callbacks: Callbacks | undefined): Promise<EvalOutputType>;
+}
+/**
+ * A chain for evaluating ReAct style agents.
+ *
+ * This chain is used to evaluate ReAct style agents by reasoning about
+ * the sequence of actions taken and their outcomes.
+ */
+export declare class TrajectoryEvalChain extends AgentTrajectoryEvaluator {
+    static lc_name(): string;
+    criterionName?: string;
+    evaluationName?: string;
+    requiresInput: boolean;
+    requiresReference: boolean;
+    outputParser: TrajectoryOutputParser;
+    static resolveTrajectoryPrompt(prompt?: BasePromptTemplate | undefined, agentTools?: StructuredTool[]): import("../../prompts/chat.js").ChatPromptTemplate<any, any> | BasePromptTemplate<any, import("../../schema/index.js").BasePromptValue, any>;
+    /**
+     * Get the description of the agent tools.
+     *
+     * @returns The description of the agent tools.
+     */
+    static toolsDescription(agentTools: StructuredTool[]): string;
+    /**
+     * Create a new TrajectoryEvalChain.
+     * @param llm
+     * @param agentTools - The tools used by the agent.
+     * @param chainOptions - The options for the chain.
+     */
+    static fromLLM(llm: BaseChatModel, agentTools?: StructuredTool[], chainOptions?: Partial<Omit<LLMEvalChainInput, "llm">>): Promise<TrajectoryEvalChain>;
+    _prepareOutput(result: ChainValues): any;
+    /**
+     * Get the agent trajectory as a formatted string.
+     *
+     * @param steps - The agent trajectory.
+     * @returns The formatted agent trajectory.
+     */
+    getAgentTrajectory(steps: AgentStep[]): string;
+    formatReference(reference?: string): string;
+    _evaluateAgentTrajectory(args: LLMTrajectoryEvaluatorArgs, callOptions: this["llm"]["CallOptions"], config?: Callbacks | BaseCallbackConfig): Promise<ChainValues>;
+}

package/dist/evaluation/agents/trajectory.js

@@ -0,0 +1,184 @@
+import { BaseLLMOutputParser } from "../../schema/output_parser.js";
+import { AgentTrajectoryEvaluator, } from "../base.js";
+import { RUN_KEY, } from "../../schema/index.js";
+import { EVAL_CHAT_PROMPT, TOOL_FREE_EVAL_CHAT_PROMPT } from "./prompt.js";
+/**
+ * A parser for the output of the TrajectoryEvalChain.
+ */
+export class TrajectoryOutputParser extends BaseLLMOutputParser {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "lc_namespace", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: ["langchain", "evaluation", "agents"]
+        });
+    }
+    static lc_name() {
+        return "TrajectoryOutputParser";
+    }
+    parseResult(generations, _callbacks) {
+        const { text } = generations[0];
+        if (!text.includes("Score:")) {
+            throw new Error(`Could not find score in model eval output: ${text}`);
+        }
+        let [reasoning, scoreStr] = text.split("Score:", 2);
+        reasoning = reasoning.trim();
+        scoreStr = scoreStr.trim();
+        // Use regex to extract the score.
+        // This will get the number in the string, even if it is a float or more than 10.
+        // E.g. "Score: 1" will return 1, "Score: 3.5" will return 3.5, and
+        // "Score: 10" will return 10.
+        // The score should be an integer digit in the range 1-5.
+        const scoreMatch = scoreStr.match(/(\d+(\.\d+)?)/);
+        if (scoreMatch === null || scoreMatch[1].includes(".")) {
+            throw new Error(`Score is not an integer digit in the range 1-5: ${text}`);
+        }
+        const score = +scoreMatch[1];
+        if (score < 1 || score > 5) {
+            throw new Error(`Score is not a digit in the range 1-5: ${text}`);
+        }
+        const normalizedScore = (score - 1) / 4;
+        return Promise.resolve({
+            reasoning,
+            score: normalizedScore,
+        });
+    }
+}
+/**
+ * A chain for evaluating ReAct style agents.
+ *
+ * This chain is used to evaluate ReAct style agents by reasoning about
+ * the sequence of actions taken and their outcomes.
+ */
+export class TrajectoryEvalChain extends AgentTrajectoryEvaluator {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "criterionName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.criterionName
+        });
+        Object.defineProperty(this, "requiresInput", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        Object.defineProperty(this, "requiresReference", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "outputParser", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new TrajectoryOutputParser()
+        });
+    }
+    static lc_name() {
+        return "TrajectoryEvalChain";
+    }
+    static resolveTrajectoryPrompt(prompt, agentTools) {
+        let _prompt;
+        if (prompt) {
+            _prompt = prompt;
+        }
+        else if (agentTools) {
+            _prompt = EVAL_CHAT_PROMPT;
+        }
+        else {
+            _prompt = TOOL_FREE_EVAL_CHAT_PROMPT;
+        }
+        return _prompt;
+    }
+    /**
+     * Get the description of the agent tools.
+     *
+     * @returns The description of the agent tools.
+     */
+    static toolsDescription(agentTools) {
+        return agentTools
+            .map((tool, i) => `Tool ${i + 1}: ${tool.name}\n Description: ${tool.description}`)
+            .join("\n\n");
+    }
+    /**
+     * Create a new TrajectoryEvalChain.
+     * @param llm
+     * @param agentTools - The tools used by the agent.
+     * @param chainOptions - The options for the chain.
+     */
+    static async fromLLM(llm, agentTools, chainOptions) {
+        let prompt = this.resolveTrajectoryPrompt(chainOptions?.prompt, agentTools);
+        if (agentTools) {
+            const toolDescriptions = this.toolsDescription(agentTools);
+            prompt = await prompt.partial({ toolDescriptions });
+        }
+        const options = chainOptions;
+        if (options) {
+            // remove prompt from chainOptions
+            delete options.prompt;
+        }
+        return new this({
+            llm,
+            prompt,
+            ...options,
+        });
+    }
+    _prepareOutput(result) {
+        const parsed = result[this.outputKey];
+        if (RUN_KEY in result && result[RUN_KEY]) {
+            parsed[RUN_KEY] = result[RUN_KEY];
+        }
+        return parsed;
+    }
+    /**
+     * Get the agent trajectory as a formatted string.
+     *
+     * @param steps - The agent trajectory.
+     * @returns The formatted agent trajectory.
+     */
+    getAgentTrajectory(steps) {
+        return steps
+            .map((step, i) => {
+            const { action, observation } = step;
+            return (`Step ${i + 1}:\n` +
+                `Tool used: ${action.tool}\n` +
+                `Tool input: ${action.toolInput}\n` +
+                `Tool output: ${observation}`);
+        })
+            .join("\n\n");
+    }
+    formatReference(reference) {
+        if (!reference) {
+            return "";
+        }
+        return `
+The following is the expected answer. Use this to measure correctness:
+[GROUND_TRUTH]
+${reference}
+[END_GROUND_TRUTH]
+        `;
+    }
+    async _evaluateAgentTrajectory(args, callOptions, config) {
+        const { input, prediction, reference, agentTrajectory } = args;
+        const inputs = {
+            question: input,
+            agentTrajectory: this.getAgentTrajectory(agentTrajectory),
+            answer: prediction,
+            reference: this.formatReference(reference),
+        };
+        const result = await this.call({ ...inputs, ...callOptions }, config);
+        return this._prepareOutput(result);
+    }
+}

package/dist/evaluation/base.cjs

@@ -0,0 +1,274 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentTrajectoryEvaluator = exports.LLMPairwiseStringEvaluator = exports.PairwiseStringEvaluator = exports.StringEvaluator = exports.LLMStringEvaluator = exports.EvalChain = exports.LLMEvalChain = exports.eqSet = void 0;
+const index_js_1 = require("../chains/index.cjs");
+/**
+ * Compare two sets for equality
+ *
+ * @param xs
+ * @param ys
+ */
+const eqSet = (xs, ys) => xs.size === ys.size && [...xs].every((x) => ys.has(x));
+exports.eqSet = eqSet;
+/**
+ * Base llm chain class for evaluators.
+ */
+class LLMEvalChain extends index_js_1.LLMChain {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "requiresInput", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "requiresReference", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "skipInputWarning", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: `Ignoring input in ${this.constructor.name}, as it is not expected.`
+        });
+        Object.defineProperty(this, "skipReferenceWarning", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: `Ignoring reference in ${this.constructor.name}, as it is not expected.`
+        });
+    }
+    /**
+     * Check if the evaluation arguments are valid.
+     * @param reference  The reference label.
+     * @param input The input string.
+     * @throws {Error} If the evaluator requires an input string but none is provided, or if the evaluator requires a reference label but none is provided.
+     */
+    checkEvaluationArgs(reference, input) {
+        if (this.requiresInput && input == null) {
+            throw new Error(`${this.constructor.name} requires an input string.`);
+        }
+        else if (input != null && !this.requiresInput) {
+            console.warn(this.skipInputWarning);
+        }
+        if (this.requiresReference && reference == null) {
+            throw new Error(`${this.constructor.name} requires a reference string.`);
+        }
+        else if (reference != null && !this.requiresReference) {
+            console.warn(this.skipReferenceWarning);
+        }
+    }
+}
+exports.LLMEvalChain = LLMEvalChain;
+/**
+ * Base chain class for evaluators.
+ */
+class EvalChain extends index_js_1.BaseChain {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "requiresInput", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "requiresReference", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
+        Object.defineProperty(this, "skipInputWarning", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: `Ignoring input in ${this.constructor.name}, as it is not expected.`
+        });
+        Object.defineProperty(this, "skipReferenceWarning", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: `Ignoring reference in ${this.constructor.name}, as it is not expected.`
+        });
+    }
+    /**
+     * Check if the evaluation arguments are valid.
+     * @param reference  The reference label.
+     * @param input The input string.
+     * @throws {Error} If the evaluator requires an input string but none is provided, or if the evaluator requires a reference label but none is provided.
+     */
+    checkEvaluationArgs(reference, input) {
+        if (this.requiresInput && input == null) {
+            throw new Error(`${this.constructor.name} requires an input string.`);
+        }
+        else if (input != null && !this.requiresInput) {
+            console.warn(this.skipInputWarning);
+        }
+        if (this.requiresReference && reference == null) {
+            throw new Error(`${this.constructor.name} requires a reference string.`);
+        }
+        else if (reference != null && !this.requiresReference) {
+            console.warn(this.skipReferenceWarning);
+        }
+    }
+}
+exports.EvalChain = EvalChain;
+/**
+ * Grade, tag, or otherwise evaluate predictions relative to their inputs
+ * and/or reference labels
+ */
+class LLMStringEvaluator extends LLMEvalChain {
+    constructor() {
+        super(...arguments);
+        /**
+         * The name of the evaluation.
+         */
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.constructor.name
+        });
+    }
+    /**
+     * Evaluate Chain or LLM output, based on optional input and label.
+     * @returns The evaluation results containing the score or value. It is recommended that the dictionary contain the following keys:
+     * - score: the score of the evaluation, if applicable.
+     * - value: the string value of the evaluation, if applicable.
+     * - reasoning: the reasoning for the evaluation, if applicable.
+     * @param args
+     * @param callOptions
+     * @param config
+     */
+    evaluateStrings(args, callOptions, config) {
+        this.checkEvaluationArgs(args.reference, args.input);
+        return this._evaluateStrings(args, callOptions, config);
+    }
+}
+exports.LLMStringEvaluator = LLMStringEvaluator;
+/**
+ * Grade, tag, or otherwise evaluate predictions relative to their inputs
+ * and/or reference labels
+ */
+class StringEvaluator extends EvalChain {
+    constructor() {
+        super(...arguments);
+        /**
+         * The name of the evaluation.
+         */
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.constructor.name
+        });
+    }
+    /**
+     * Evaluate Chain or LLM output, based on optional input and label.
+     * @returns The evaluation results containing the score or value. It is recommended that the dictionary contain the following keys:
+     * - score: the score of the evaluation, if applicable.
+     * - value: the string value of the evaluation, if applicable.
+     * - reasoning: the reasoning for the evaluation, if applicable.
+     * @param args
+     * @param config
+     */
+    evaluateStrings(args, config) {
+        this.checkEvaluationArgs(args.reference, args.input);
+        return this._evaluateStrings(args, config);
+    }
+}
+exports.StringEvaluator = StringEvaluator;
+/**
+ * Compare the output of two models (or two outputs of the same model).
+ */
+class PairwiseStringEvaluator extends EvalChain {
+    constructor() {
+        super(...arguments);
+        /**
+         * The name of the evaluation.
+         */
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.constructor.name
+        });
+    }
+    /**
+     * Evaluate the output string pairs.
+     * @param args
+     * @param config
+     * @return A dictionary containing the preference, scores, and/or other information.
+     */
+    evaluateStringPairs(args, config) {
+        return this._evaluateStringPairs(args, config);
+    }
+}
+exports.PairwiseStringEvaluator = PairwiseStringEvaluator;
+/**
+ * Compare the output of two models (or two outputs of the same model).
+ */
+class LLMPairwiseStringEvaluator extends LLMEvalChain {
+    constructor() {
+        super(...arguments);
+        /**
+         * The name of the evaluation.
+         */
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.constructor.name
+        });
+    }
+    /**
+     * Evaluate the output string pairs.
+     * @param args
+     * @param callOptions
+     * @param config
+     * @return A dictionary containing the preference, scores, and/or other information.
+     */
+    evaluateStringPairs(args, callOptions, config) {
+        this.checkEvaluationArgs(args.reference, args.input);
+        return this._evaluateStringPairs(args, callOptions, config);
+    }
+}
+exports.LLMPairwiseStringEvaluator = LLMPairwiseStringEvaluator;
+/**
+ * Interface for evaluating agent trajectories.
+ */
+class AgentTrajectoryEvaluator extends LLMEvalChain {
+    constructor() {
+        super(...arguments);
+        Object.defineProperty(this, "requiresInput", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: true
+        });
+        /**
+         * The name of the evaluation.
+         */
+        Object.defineProperty(this, "evaluationName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: this.constructor.name
+        });
+    }
+    /**
+     * Evaluate a trajectory.
+     * @return The evaluation result.
+     * @param args
+     * @param callOptions
+     * @param config
+     */
+    evaluateAgentTrajectory(args, callOptions, config) {
+        this.checkEvaluationArgs(args.reference, args.input);
+        return this._evaluateAgentTrajectory(args, callOptions, config);
+    }
+}
+exports.AgentTrajectoryEvaluator = AgentTrajectoryEvaluator;