langchain 0.2.10 → 0.2.11

package/dist/memory/tests/entity_memory.int.test.js

@@ -1,79 +0,0 @@
-import { test } from "@jest/globals";
-import { ChatOpenAI, OpenAI } from "@langchain/openai";
-import { EntityMemory } from "../entity_memory.js";
-import { LLMChain } from "../../chains/llm_chain.js";
-import { ENTITY_MEMORY_CONVERSATION_TEMPLATE } from "../prompt.js";
-test.skip("Test entity memory in a chain", async () => {
-    const memory = new EntityMemory({
-        llm: new OpenAI({ temperature: 0 }),
-    });
-    const model = new OpenAI({ temperature: 0.9 });
-    const chain = new LLMChain({
-        llm: model,
-        prompt: ENTITY_MEMORY_CONVERSATION_TEMPLATE,
-        memory,
-    });
-    const res1 = await chain.call({ input: "Hi! I'm Jim." });
-    console.log({
-        res1,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-    const res2 = await chain.call({
-        input: "My office is the Scranton branch of Dunder Mifflin. What about you?",
-    });
-    console.log({
-        res2,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-    const res3 = await chain.call({
-        input: "I am Jim.",
-    });
-    console.log({
-        res3,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-    const res4 = await chain.call({
-        input: "What have I told you about Jim so far?",
-    });
-    console.log({
-        res4,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-}, 120000);
-test.skip("Test entity memory with a chat model in a chain", async () => {
-    const memory = new EntityMemory({
-        llm: new ChatOpenAI({ temperature: 0 }),
-    });
-    const model = new ChatOpenAI({ temperature: 0.9 });
-    const chain = new LLMChain({
-        llm: model,
-        prompt: ENTITY_MEMORY_CONVERSATION_TEMPLATE,
-        memory,
-    });
-    const res1 = await chain.call({ input: "Hi! I'm Jim." });
-    console.log({
-        res1,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-    const res2 = await chain.call({
-        input: "My office is the Utica branch of Dunder Mifflin. What about you?",
-    });
-    console.log({
-        res2,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-    const res3 = await chain.call({
-        input: "I am Jim.",
-    });
-    console.log({
-        res3,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-    const res4 = await chain.call({
-        input: "What have I told you about Jim so far?",
-    });
-    console.log({
-        res4,
-        memory: await memory.loadMemoryVariables({ input: "Who is Jim?" }),
-    });
-}, 120000);

package/dist/memory/tests/entity_memory.test.d.ts

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

package/dist/memory/tests/entity_memory.test.js

@@ -1,48 +0,0 @@
-import { test, expect } from "@jest/globals";
-import { AIMessage, HumanMessage } from "@langchain/core/messages";
-import { BaseLLM } from "@langchain/core/language_models/llms";
-import { InMemoryChatMessageHistory as ChatMessageHistory } from "@langchain/core/chat_history";
-import { EntityMemory } from "../entity_memory.js";
-class FakeLLM extends BaseLLM {
-    _llmType() {
-        return "fake";
-    }
-    async _generate(_prompts) {
-        const mockVal = { generations: [[{ text: "foo" }]] };
-        return mockVal;
-    }
-}
-test("Test entity memory", async () => {
-    const model = new FakeLLM({});
-    const memory = new EntityMemory({ llm: model });
-    const result1 = await memory.loadMemoryVariables({ input: "foo" });
-    const expectedResult1 = {
-        history: "",
-        entities: { foo: "No current information known." },
-    };
-    expect(result1).toStrictEqual(expectedResult1);
-    await memory.saveContext({ foo: "bar" }, { bar: "foo" });
-    const expectedString = "Human: bar\nAI: foo";
-    const result2 = await memory.loadMemoryVariables({ input: "foo" });
-    expect(result2).toStrictEqual({
-        history: expectedString,
-        entities: { foo: "foo" },
-    });
-});
-test("Test entity memory with pre-loaded history", async () => {
-    const model = new FakeLLM({});
-    const pastMessages = [
-        new HumanMessage("My name's foo"),
-        new AIMessage("Nice to meet you, foo!"),
-    ];
-    const memory = new EntityMemory({
-        returnMessages: true,
-        chatHistory: new ChatMessageHistory(pastMessages),
-        llm: model,
-    });
-    const result = await memory.loadMemoryVariables({ input: "foo" });
-    expect(result).toStrictEqual({
-        history: pastMessages,
-        entities: { foo: "No current information known." },
-    });
-});

package/dist/memory/tests/summary.int.test.d.ts

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

package/dist/memory/tests/summary.int.test.js

@@ -1,50 +0,0 @@
-import { test, expect } from "@jest/globals";
-import { ChatOpenAI, OpenAIChat } from "@langchain/openai";
-import { SystemMessage } from "@langchain/core/messages";
-import { ConversationSummaryMemory } from "../summary.js";
-test("Test summary memory", async () => {
-    const memory = new ConversationSummaryMemory({
-        llm: new OpenAIChat({ modelName: "gpt-3.5-turbo", temperature: 0 }),
-    });
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-    await memory.saveContext({ input: "How's it going?" }, { response: "Hello! I'm doing fine. and you?" });
-    const result2 = await memory.loadMemoryVariables({});
-    console.log("result2", result2);
-    await memory.clear();
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-});
-test("Test summary memory with chat model", async () => {
-    const memory = new ConversationSummaryMemory({
-        llm: new ChatOpenAI({ temperature: 0 }),
-    });
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-    await memory.saveContext({ input: "How's it going?" }, { response: "Hello! I'm doing fine. and you?" });
-    const result2 = await memory.loadMemoryVariables({});
-    console.log("result2", result2);
-    await memory.clear();
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-});
-test("Test summary memory return messages", async () => {
-    const memory = new ConversationSummaryMemory({
-        llm: new OpenAIChat({ modelName: "gpt-3.5-turbo", temperature: 0 }),
-        returnMessages: true,
-    });
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: [new SystemMessage("")],
-    });
-    await memory.saveContext({ input: "How's it going?" }, { response: "Hello! I'm doing fine. and you?" });
-    const result2 = await memory.loadMemoryVariables({});
-    console.log("result2", result2);
-    await memory.clear();
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: [new SystemMessage("")],
-    });
-});

package/dist/memory/tests/summary_buffer.int.test.d.ts

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

package/dist/memory/tests/summary_buffer.int.test.js

@@ -1,55 +0,0 @@
-import { test, expect } from "@jest/globals";
-import { OpenAI, ChatOpenAI } from "@langchain/openai";
-import { SystemMessage } from "@langchain/core/messages";
-import { ConversationSummaryBufferMemory } from "../summary_buffer.js";
-test("Test summary buffer memory", async () => {
-    const memory = new ConversationSummaryBufferMemory({
-        llm: new OpenAI({ modelName: "gpt-3.5-turbo-instruct", temperature: 0 }),
-        maxTokenLimit: 10,
-    });
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-    await memory.saveContext({ input: "How's it going?" }, { response: "Hello! I'm doing fine. and you?" });
-    const result = await memory.loadMemoryVariables({});
-    console.log("result", result);
-    await memory.clear();
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-});
-test("Test summary buffer memory with chat model", async () => {
-    const memory = new ConversationSummaryBufferMemory({
-        llm: new ChatOpenAI({ temperature: 0 }),
-        maxTokenLimit: 10,
-    });
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-    await memory.saveContext({ input: "How's it going?" }, { response: "Hello! I'm doing fine. and you?" });
-    const result = await memory.loadMemoryVariables({});
-    console.log("result", result);
-    await memory.clear();
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: "",
-    });
-});
-test("Test summary buffer memory return messages", async () => {
-    const memory = new ConversationSummaryBufferMemory({
-        llm: new OpenAI({ modelName: "gpt-3.5-turbo-instruct", temperature: 0 }),
-        returnMessages: true,
-        maxTokenLimit: 10,
-    });
-    const exampleBuffer = "hello summary buffer";
-    memory.movingSummaryBuffer = exampleBuffer;
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: [new SystemMessage(exampleBuffer)],
-    });
-    await memory.saveContext({ input: "How's it going?" }, { response: "Hello! I'm doing fine. and you?" });
-    const result = await memory.loadMemoryVariables({});
-    console.log("result", result);
-    await memory.clear();
-    expect(await memory.loadMemoryVariables({})).toEqual({
-        history: [],
-    });
-});

package/dist/memory/tests/vector_store_memory.int.test.d.ts

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

package/dist/memory/tests/vector_store_memory.int.test.js

@@ -1,55 +0,0 @@
-import { test, expect } from "@jest/globals";
-import { OpenAIEmbeddings } from "@langchain/openai";
-import { Document } from "@langchain/core/documents";
-import { VectorStoreRetrieverMemory } from "../vector_store.js";
-import { MemoryVectorStore } from "../../vectorstores/memory.js";
-test("Test vector store memory", async () => {
-    const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());
-    const memory = new VectorStoreRetrieverMemory({
-        vectorStoreRetriever: vectorStore.asRetriever(),
-    });
-    const result1 = await memory.loadMemoryVariables({ input: "foo" });
-    expect(result1).toStrictEqual({ memory: "" });
-    await memory.saveContext({ foo: "bar" }, { bar: "foo" });
-    const expectedString = "foo: bar\nbar: foo";
-    const result2 = await memory.loadMemoryVariables({ input: "foo" });
-    expect(result2).toStrictEqual({ memory: expectedString });
-});
-test("Test vector store memory return docs", async () => {
-    const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());
-    const memory = new VectorStoreRetrieverMemory({
-        vectorStoreRetriever: vectorStore.asRetriever(),
-        returnDocs: true,
-    });
-    const result1 = await memory.loadMemoryVariables({ input: "foo" });
-    expect(result1).toStrictEqual({ memory: [] });
-    await memory.saveContext({ foo: "bar" }, { bar: "foo" });
-    const expectedResult = [new Document({ pageContent: "foo: bar\nbar: foo" })];
-    const result2 = await memory.loadMemoryVariables({ input: "foo" });
-    expect(result2).toStrictEqual({ memory: expectedResult });
-});
-test("Test vector store memory metadata object", async () => {
-    const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());
-    const memory = new VectorStoreRetrieverMemory({
-        vectorStoreRetriever: vectorStore.asRetriever(),
-        metadata: { foo: "bar" },
-    });
-    await memory.saveContext({ foo: "bar" }, { bar: "foo" });
-    vectorStore.memoryVectors.forEach((vector) => {
-        expect(vector.metadata).toStrictEqual({ foo: "bar" });
-    });
-});
-test("Test vector store memory metadata function", async () => {
-    const vectorStore = new MemoryVectorStore(new OpenAIEmbeddings());
-    const memory = new VectorStoreRetrieverMemory({
-        vectorStoreRetriever: vectorStore.asRetriever(),
-        metadata: (inputValues, outputValues) => ({
-            foo: `${inputValues?.foo} ${inputValues?.foo}`,
-            bar: `${outputValues?.bar} ${outputValues?.bar}`, // "foo foo"
-        }),
-    });
-    await memory.saveContext({ foo: "bar" }, { bar: "foo" });
-    vectorStore.memoryVectors.forEach((vector) => {
-        expect(vector.metadata).toStrictEqual({ foo: "bar bar", bar: "foo foo" });
-    });
-});

package/dist/output_parsers/tests/combining.int.test.d.ts

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

package/dist/output_parsers/tests/combining.int.test.js

@@ -1,26 +0,0 @@
-import { test } from "@jest/globals";
-import { PromptTemplate } from "@langchain/core/prompts";
-import { ChatOpenAI } from "@langchain/openai";
-import { StructuredOutputParser, RegexParser, CombiningOutputParser, } from "../index.js";
-test("CombiningOutputParser", async () => {
-    const answerParser = StructuredOutputParser.fromNamesAndDescriptions({
-        answer: "answer to the user's question",
-        source: "source used to answer the user's question, should be a website.",
-    });
-    const confidenceParser = new RegexParser(/Confidence: (A|B|C), Explanation: (.*)/, ["confidence", "explanation"], "noConfidence");
-    const parser = new CombiningOutputParser(answerParser, confidenceParser);
-    const formatInstructions = parser.getFormatInstructions();
-    const prompt = new PromptTemplate({
-        template: "Answer the users question as best as possible.\n{format_instructions}\n{question}",
-        inputVariables: ["question"],
-        partialVariables: { format_instructions: formatInstructions },
-    });
-    const model = new ChatOpenAI({ temperature: 0 });
-    const input = await prompt.format({
-        question: "What is the capital of France?",
-    });
-    console.log(input);
-    const response = await model.invoke(input);
-    console.log(response);
-    console.log(await parser.parse(response.content));
-});

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

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

package/dist/output_parsers/tests/combining.test.js

@@ -1,54 +0,0 @@
-import { expect, test } from "@jest/globals";
-import { CombiningOutputParser } from "../combining.js";
-import { StructuredOutputParser } from "../structured.js";
-import { RegexParser } from "../regex.js";
-test("CombiningOutputParser", async () => {
-    const parser = new CombiningOutputParser(StructuredOutputParser.fromNamesAndDescriptions({
-        url: "A link to the resource",
-    }), new RegexParser(/Confidence: (A|B|C), Explanation: (.*)/, ["confidence", "explanation"], "noConfidence"));
-    expect(parser.getFormatInstructions()).toMatchInlineSnapshot(`
-    "Return the following 2 outputs, each formatted as described below. Include the delimiter characters "-----" in your response:
-
-    -----Output 1-----
-    You must format your output as a JSON value that adheres to a given "JSON Schema" instance.
-
-    "JSON Schema" is a declarative language that allows you to annotate and validate JSON documents.
-
-    For example, the example "JSON Schema" instance {{"properties": {{"foo": {{"description": "a list of test words", "type": "array", "items": {{"type": "string"}}}}}}, "required": ["foo"]}}}}
-    would match an object with one required property, "foo". The "type" property specifies "foo" must be an "array", and the "description" property semantically describes it as "a list of test words". The items within "foo" must be strings.
-    Thus, the object {{"foo": ["bar", "baz"]}} is a well-formatted instance of this example "JSON Schema". The object {{"properties": {{"foo": ["bar", "baz"]}}}} is not well-formatted.
-
-    Your output will be parsed and type-checked according to the provided schema instance, so make sure all fields in your output match the schema exactly and there are no trailing commas!
-
-    Here is the JSON Schema instance your output must adhere to. Include the enclosing markdown codeblock:
-    \`\`\`json
-    {"type":"object","properties":{"url":{"type":"string","description":"A link to the resource"}},"required":["url"],"additionalProperties":false,"$schema":"http://json-schema.org/draft-07/schema#"}
-    \`\`\`
-    -----
-
-    -----Output 2-----
-    Your response should match the following regex: /Confidence: (A|B|C), Explanation: (.*)/
-    -----
-    "
-  `);
-    expect(await parser.parse(`-----Output 0-----
-{"url": "https://en.wikipedia.org/wiki/Paris"}
------
-
------Output 1-----
-Confidence: A, Explanation: Because it is the capital of France.
------`)).toMatchInlineSnapshot(`
-    {
-      "confidence": "A",
-      "explanation": "Because it is the capital of France.",
-      "url": "https://en.wikipedia.org/wiki/Paris",
-    }
-  `);
-    expect(await parser.parse('```\n{"url": "https://en.wikipedia.org/wiki/Paris"}\n```\nConfidence: A, Explanation: Because it is the capital of France.')).toMatchInlineSnapshot(`
-    {
-      "confidence": "A",
-      "explanation": "Because it is the capital of France.",
-      "url": "https://en.wikipedia.org/wiki/Paris",
-    }
-  `);
-});

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

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

package/dist/output_parsers/tests/datetime.test.js

@@ -1,14 +0,0 @@
-import { test, expect } from "@jest/globals";
-import { OutputParserException } from "@langchain/core/output_parsers";
-import { DatetimeOutputParser } from "../datetime.js";
-test("DatetimeOutputParser", async () => {
-    const parser = new DatetimeOutputParser();
-    const currentDate = new Date();
-    const currentDateStr = currentDate.toISOString();
-    expect(await parser.parse(currentDateStr)).toEqual(currentDate);
-    const isoDateStr = "05 October 2011 14:48Z";
-    const isoDate = new Date(Date.UTC(2011, 9, 5, 14, 48));
-    expect(await parser.parse(isoDateStr)).toEqual(isoDate);
-    const invalidDateStr = "Vespertine was released by Bjork.";
-    await expect(() => parser.parse(invalidDateStr)).rejects.toThrow(OutputParserException);
-});

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

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