import { test, expect } from "@jest/globals";
import { CohereEmbeddings } from "../cohere.js";

test("Test CohereEmbeddings.embedQuery", async () => {
  const embeddings = new CohereEmbeddings();
  const res = await embeddings.embedQuery("Hello world");
  expect(typeof res[0]).toBe("number");
});

test("Test CohereEmbeddings.embedDocuments", async () => {
  const embeddings = new CohereEmbeddings();
  const res = await embeddings.embedDocuments(["Hello world", "Bye bye"]);
  expect(res).toHaveLength(2);
  expect(typeof res[0][0]).toBe("number");
  expect(typeof res[1][0]).toBe("number");
});