@langchain/community
Version:
Third-party integrations for LangChain.js
21 lines (20 loc) • 743 B
JavaScript
import { test, expect } from "@jest/globals";
import { GoogleVertexAIMultimodalEmbeddings, } from "../googlevertexai.js";
test("mediaToInstance text", async () => {
const e = new GoogleVertexAIMultimodalEmbeddings();
const media = {
text: "just text",
};
const instance = e.mediaToInstance(media);
expect(instance.text).toEqual("just text");
expect(instance.image).toBeUndefined();
});
test("mediaToInstance image", async () => {
const e = new GoogleVertexAIMultimodalEmbeddings();
const media = {
image: Buffer.from("abcd"),
};
const instance = e.mediaToInstance(media);
expect(instance.image?.bytesBase64Encoded).toEqual("YWJjZA==");
expect(instance.text).toBeUndefined();
});