diff --git a/nodejs/__test__/embedding_registry.test.ts b/nodejs/__test__/embedding_registry.test.ts index 74696b356..48ba48f96 100644 --- a/nodejs/__test__/embedding_registry.test.ts +++ b/nodejs/__test__/embedding_registry.test.ts @@ -16,6 +16,18 @@ describe("embedding function registry", () => { registries.length = 0; }); + it("registers built-in providers through the public embedding API", () => { + jest.isolateModules(() => { + const { getRegistry } = + require("../lancedb/embedding") as EmbeddingModule; + const registry = getRegistry(); + registries.push(registry); + + expect(registry.get("openai")).toBeDefined(); + expect(registry.get("huggingface")).toBeDefined(); + }); + }); + it("shares registrations across duplicated provider module graphs", () => { let registeringRegistry: EmbeddingFunctionRegistry | undefined; diff --git a/nodejs/lancedb/embedding/index.ts b/nodejs/lancedb/embedding/index.ts index 4a1aa9a7b..f8b453211 100644 --- a/nodejs/lancedb/embedding/index.ts +++ b/nodejs/lancedb/embedding/index.ts @@ -5,6 +5,8 @@ import { Field, Schema } from "../arrow"; import { sanitizeType } from "../sanitize"; import { EmbeddingFunction } from "./embedding_function"; import { EmbeddingFunctionConfig, getRegistry } from "./registry"; +import "./openai"; +import "./transformers"; export { FieldOptions,