@langchain/core 0.2.18 → 0.2.20

package/dist/messages/tests/message_utils.test.js

@@ -1,434 +0,0 @@
-import { it, expect } from "@jest/globals";
-import { filterMessages, mergeMessageRuns, trimMessages, } from "../transformers.js";
-import { AIMessage } from "../ai.js";
-import { HumanMessage } from "../human.js";
-import { SystemMessage } from "../system.js";
-import { getBufferString } from "../utils.js";
-describe("filterMessage", () => {
-    const getMessages = () => [
-        new SystemMessage("you're a good assistant."),
-        new HumanMessage({
-            content: "what's your name",
-            id: "foo",
-            name: "example_user",
-        }),
-        new AIMessage({ content: "steve-o", id: "bar", name: "example_assistant" }),
-        new HumanMessage({ content: "what's your favorite color", id: "baz" }),
-        new AIMessage({ content: "silicon blue", id: "blah" }),
-    ];
-    it("works", () => {
-        const messages = getMessages();
-        const filteredMessages = filterMessages(messages, {
-            includeNames: ["example_user", "example_assistant"],
-            includeTypes: ["system"],
-            excludeIds: ["bar"],
-        });
-        expect(filteredMessages).toEqual([
-            new SystemMessage("you're a good assistant."),
-            new HumanMessage({
-                content: "what's your name",
-                id: "foo",
-                name: "example_user",
-            }),
-        ]);
-    });
-    it("can filter messages based on class types", () => {
-        const messages = getMessages();
-        const filteredMessages = filterMessages(messages, {
-            includeTypes: [HumanMessage, AIMessage],
-        });
-        expect(filteredMessages).toHaveLength(4);
-        expect(filteredMessages).toEqual([
-            new HumanMessage({
-                content: "what's your name",
-                id: "foo",
-                name: "example_user",
-            }),
-            new AIMessage({
-                content: "steve-o",
-                id: "bar",
-                name: "example_assistant",
-            }),
-            new HumanMessage({ content: "what's your favorite color", id: "baz" }),
-            new AIMessage({ content: "silicon blue", id: "blah" }),
-        ]);
-    });
-    it("returns a runnable if no messages are passed", () => {
-        const filteredMessagesRunnable = filterMessages();
-        expect(filteredMessagesRunnable).toBeDefined();
-        expect(filteredMessagesRunnable.lc_namespace).toEqual([
-            "langchain_core",
-            "runnables",
-        ]);
-        expect("func" in filteredMessagesRunnable).toBeTruthy();
-        // `func` is protected, so we need to cast it to any to access it
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        expect(typeof filteredMessagesRunnable.func).toBe("function");
-    });
-});
-describe("mergeMessageRuns", () => {
-    const getMessages = () => [
-        new SystemMessage("you're a good assistant."),
-        new HumanMessage({ content: "what's your favorite color", id: "foo" }),
-        new HumanMessage({ content: "wait your favorite food", id: "bar" }),
-        new AIMessage({
-            content: "my favorite colo",
-            tool_calls: [{ name: "blah_tool", args: { x: 2 }, id: "123" }],
-            id: "baz",
-        }),
-        new AIMessage({
-            content: [{ type: "text", text: "my favorite dish is lasagna" }],
-            tool_calls: [{ name: "blah_tool", args: { x: -10 }, id: "456" }],
-            id: "blur",
-        }),
-    ];
-    it("works", () => {
-        const messages = getMessages();
-        const mergedMessages = mergeMessageRuns(messages);
-        expect(mergedMessages).toHaveLength(3);
-        expect(mergedMessages).toEqual([
-            new SystemMessage("you're a good assistant."),
-            new HumanMessage({
-                content: "what's your favorite color\nwait your favorite food",
-                id: "foo",
-            }),
-            new AIMessage({
-                content: [
-                    { type: "text", text: "my favorite colo" },
-                    { type: "text", text: "my favorite dish is lasagna" },
-                ],
-                tool_calls: [
-                    { name: "blah_tool", args: { x: 2 }, id: "123", type: "tool_call" },
-                    { name: "blah_tool", args: { x: -10 }, id: "456", type: "tool_call" },
-                ],
-                id: "baz",
-            }),
-        ]);
-    });
-    it("returns a runnable if no messages are passed", () => {
-        const mergedMessages = mergeMessageRuns();
-        expect(mergedMessages).toBeDefined();
-        expect(mergedMessages.lc_namespace).toEqual([
-            "langchain_core",
-            "runnables",
-        ]);
-        expect("func" in mergedMessages).toBeTruthy();
-        // `func` is protected, so we need to cast it to any to access it
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        expect(typeof mergedMessages.func).toBe("function");
-    });
-});
-describe("trimMessages can trim", () => {
-    const messagesAndTokenCounterFactory = () => {
-        const messages = [
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "first",
-            }),
-            new AIMessage({
-                content: [
-                    { type: "text", text: "This is the FIRST 4 token block." },
-                    { type: "text", text: "This is the SECOND 4 token block." },
-                ],
-                id: "second",
-            }),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-            new AIMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "fourth",
-            }),
-        ];
-        const dummyTokenCounter = (messages) => {
-            // treat each message like it adds 3 default tokens at the beginning
-            // of the message and at the end of the message. 3 + 4 + 3 = 10 tokens
-            // per message.
-            const defaultContentLen = 4;
-            const defaultMsgPrefixLen = 3;
-            const defaultMsgSuffixLen = 3;
-            let count = 0;
-            for (const msg of messages) {
-                if (typeof msg.content === "string") {
-                    count +=
-                        defaultMsgPrefixLen + defaultContentLen + defaultMsgSuffixLen;
-                }
-                if (Array.isArray(msg.content)) {
-                    count +=
-                        defaultMsgPrefixLen +
-                            msg.content.length * defaultContentLen +
-                            defaultMsgSuffixLen;
-                }
-            }
-            return count;
-        };
-        return {
-            messages,
-            dummyTokenCounter,
-        };
-    };
-    it("First 30 tokens, not allowing partial messages", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 30,
-            tokenCounter: dummyTokenCounter,
-            strategy: "first",
-        });
-        expect(trimmedMessages).toHaveLength(2);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "first",
-            }),
-        ]);
-    });
-    it("First 30 tokens, allowing partial messages", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 30,
-            tokenCounter: dummyTokenCounter,
-            strategy: "first",
-            allowPartial: true,
-        });
-        expect(trimmedMessages).toHaveLength(3);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "first",
-            }),
-            new AIMessage({
-                content: [{ type: "text", text: "This is the FIRST 4 token block." }],
-                id: "second",
-            }),
-        ]);
-    });
-    it("First 30 tokens, allowing partial messages, have to end on HumanMessage", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 30,
-            tokenCounter: dummyTokenCounter,
-            strategy: "first",
-            allowPartial: true,
-            endOn: "human",
-        });
-        expect(trimmedMessages).toHaveLength(2);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "first",
-            }),
-        ]);
-    });
-    it("Last 30 tokens, including system message, not allowing partial messages", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 30,
-            includeSystem: true,
-            tokenCounter: dummyTokenCounter,
-            strategy: "last",
-        });
-        expect(trimmedMessages).toHaveLength(3);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-            new AIMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "fourth",
-            }),
-        ]);
-    });
-    it("Last 40 tokens, including system message, allowing partial messages", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 40,
-            tokenCounter: dummyTokenCounter,
-            strategy: "last",
-            allowPartial: true,
-            includeSystem: true,
-        });
-        expect(trimmedMessages).toHaveLength(4);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new AIMessage({
-                content: [{ type: "text", text: "This is the FIRST 4 token block." }],
-                id: "second",
-            }),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-            new AIMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "fourth",
-            }),
-        ]);
-    });
-    it("Last 30 tokens, including system message, allowing partial messages, end on HumanMessage", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 30,
-            tokenCounter: dummyTokenCounter,
-            strategy: "last",
-            endOn: "human",
-            includeSystem: true,
-            allowPartial: true,
-        });
-        expect(trimmedMessages).toHaveLength(3);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new AIMessage({
-                content: [{ type: "text", text: "This is the FIRST 4 token block." }],
-                id: "second",
-            }),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-        ]);
-    });
-    it("Last 40 tokens, including system message, allowing partial messages, start on HumanMessage", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 40,
-            tokenCounter: dummyTokenCounter,
-            strategy: "last",
-            includeSystem: true,
-            allowPartial: true,
-            startOn: "human",
-        });
-        expect(trimmedMessages).toHaveLength(3);
-        console.log(trimmedMessages);
-        expect(trimmedMessages).toEqual([
-            new SystemMessage("This is a 4 token text. The full message is 10 tokens."),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-            new AIMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "fourth",
-            }),
-        ]);
-    });
-    it("can filter (startOn) with message classes", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 40,
-            tokenCounter: dummyTokenCounter,
-            startOn: [HumanMessage],
-        });
-        expect(trimmedMessages).toHaveLength(2);
-        expect(trimmedMessages).toEqual([
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-            new AIMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "fourth",
-            }),
-        ]);
-    });
-    it("can filter (endOn) with message classes", async () => {
-        const { messages, dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = await trimMessages(messages, {
-            maxTokens: 40,
-            tokenCounter: dummyTokenCounter,
-            endOn: [HumanMessage],
-        });
-        console.log(trimmedMessages);
-        expect(trimmedMessages).toHaveLength(3);
-        expect(trimmedMessages).toEqual([
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "first",
-            }),
-            new AIMessage({
-                content: [
-                    { type: "text", text: "This is the FIRST 4 token block." },
-                    { type: "text", text: "This is the SECOND 4 token block." },
-                ],
-                id: "second",
-            }),
-            new HumanMessage({
-                content: "This is a 4 token text. The full message is 10 tokens.",
-                id: "third",
-            }),
-        ]);
-    });
-    it("can return a runnable if empty array is passed", () => {
-        const { dummyTokenCounter } = messagesAndTokenCounterFactory();
-        const trimmedMessages = trimMessages({
-            maxTokens: 40,
-            tokenCounter: dummyTokenCounter,
-        });
-        expect(trimmedMessages).toBeDefined();
-        expect(trimmedMessages.lc_namespace).toEqual([
-            "langchain_core",
-            "runnables",
-        ]);
-        expect("func" in trimmedMessages).toBeTruthy();
-        // `func` is protected, so we need to cast it to any to access it
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        expect(typeof trimmedMessages.func).toBe("function");
-    });
-});
-test("getBufferString can handle complex messages", () => {
-    const messageArr1 = [new HumanMessage("Hello there!")];
-    const messageArr2 = [
-        new AIMessage({
-            content: [
-                {
-                    type: "text",
-                    text: "Hello there!",
-                },
-            ],
-        }),
-    ];
-    const messageArr3 = [
-        new HumanMessage({
-            content: [
-                {
-                    type: "image_url",
-                    image_url: {
-                        url: "https://example.com/image.jpg",
-                    },
-                },
-                {
-                    type: "image_url",
-                    image_url: "https://example.com/image.jpg",
-                },
-            ],
-        }),
-    ];
-    const bufferString1 = getBufferString(messageArr1);
-    expect(bufferString1).toBe("Human: Hello there!");
-    const bufferString2 = getBufferString(messageArr2);
-    expect(bufferString2).toBe(`AI: ${JSON.stringify([
-        {
-            type: "text",
-            text: "Hello there!",
-        },
-    ], null, 2)}`);
-    const bufferString3 = getBufferString(messageArr3);
-    expect(bufferString3).toBe(`Human: ${JSON.stringify([
-        {
-            type: "image_url",
-            image_url: {
-                url: "https://example.com/image.jpg",
-            },
-        },
-        {
-            type: "image_url",
-            image_url: "https://example.com/image.jpg",
-        },
-    ], null, 2)}`);
-});

package/dist/output_parsers/openai_tools/tests/json_output_tools_parser.test.d.ts

@@ -1 +0,0 @@
-export {};

package/dist/output_parsers/openai_tools/tests/json_output_tools_parser.test.js

@@ -1,81 +0,0 @@
-import { test } from "@jest/globals";
-import { z } from "zod";
-import { JsonOutputKeyToolsParser } from "../json_output_tools_parsers.js";
-import { AIMessage } from "../../../messages/index.js";
-import { OutputParserException } from "../../base.js";
-test("JSONOutputKeyToolsParser invoke", async () => {
-    const outputParser = new JsonOutputKeyToolsParser({
-        keyName: "testing",
-        returnSingle: true,
-    });
-    const result = await outputParser.invoke(new AIMessage({
-        content: "",
-        additional_kwargs: {
-            tool_calls: [
-                {
-                    id: "test",
-                    type: "function",
-                    function: {
-                        name: "testing",
-                        arguments: JSON.stringify({ testKey: 9 }),
-                    },
-                },
-            ],
-        },
-    }));
-    expect(result).toEqual({ testKey: 9 });
-});
-test("JSONOutputKeyToolsParser with a passed schema throws", async () => {
-    const outputParser = new JsonOutputKeyToolsParser({
-        keyName: "testing",
-        returnSingle: true,
-        zodSchema: z.object({
-            testKey: z.string(),
-        }),
-    });
-    try {
-        await outputParser.invoke(new AIMessage({
-            content: "",
-            additional_kwargs: {
-                tool_calls: [
-                    {
-                        id: "test",
-                        type: "function",
-                        function: {
-                            name: "testing",
-                            arguments: JSON.stringify({ testKey: 9 }),
-                        },
-                    },
-                ],
-            },
-        }));
-    }
-    catch (e) {
-        expect(e).toBeInstanceOf(OutputParserException);
-    }
-});
-test("JSONOutputKeyToolsParser can validate a proper input", async () => {
-    const outputParser = new JsonOutputKeyToolsParser({
-        keyName: "testing",
-        returnSingle: true,
-        zodSchema: z.object({
-            testKey: z.string(),
-        }),
-    });
-    const result = await outputParser.invoke(new AIMessage({
-        content: "",
-        additional_kwargs: {
-            tool_calls: [
-                {
-                    id: "test",
-                    type: "function",
-                    function: {
-                        name: "testing",
-                        arguments: JSON.stringify({ testKey: "testval" }),
-                    },
-                },
-            ],
-        },
-    }));
-    expect(result).toEqual({ testKey: "testval" });
-});

package/dist/output_parsers/tests/json.test.d.ts

@@ -1 +0,0 @@
-export {};