@langchain/community/dist/stores/doc/tests/gcs.int.test.js

Third-party integrations for LangChain.js

import { describe, test, expect } from "@jest/globals";
import { Document } from "@langchain/core/documents";
import { GoogleCloudStorageDocstore } from "../gcs.js";
describe.skip("GoogleCloudStorageDocstore", () => {
    const bucket = "INSERT_BUCKET_HERE";
    test("save", async () => {
        const name = "test1";
        const pageContent = "This is a test";
        const document = new Document({ pageContent });
        const store = new GoogleCloudStorageDocstore({
            bucket,
        });
        await store.addDocument(name, document);
    });
    test("save metadata", async () => {
        const name = "test2";
        const pageContent = "This is a metadata test";
        const metadata = {
            meta1: "one",
            meta2: "two",
        };
        const document = new Document({ pageContent, metadata });
        const store = new GoogleCloudStorageDocstore({
            bucket,
        });
        await store.addDocument(name, document);
    });
    test("save prefix", async () => {
        const prefix = "prefix/";
        const name = "test3";
        const pageContent = "This is a prefix test";
        const document = new Document({ pageContent });
        const store = new GoogleCloudStorageDocstore({
            bucket,
            prefix,
        });
        await store.addDocument(name, document);
    });
    test("load", async () => {
        const name = "test1";
        const store = new GoogleCloudStorageDocstore({
            bucket,
        });
        const document = await store.search(name);
        console.log(document);
        expect(document.pageContent).toEqual("This is a test");
    });
    test("load metadata", async () => {
        const name = "test2";
        const store = new GoogleCloudStorageDocstore({
            bucket,
        });
        const document = await store.search(name);
        console.log(document);
        expect(document.pageContent).toEqual("This is a metadata test");
        expect(document.metadata.meta1).toEqual("one");
    });
    test("load prefix", async () => {
        const prefix = "prefix/";
        const name = "test3";
        const store = new GoogleCloudStorageDocstore({
            bucket,
            prefix,
        });
        const document = await store.search(name);
        console.log(document);
        expect(document.pageContent).toEqual("This is a prefix test");
    });
});