mirror of
https://github.com/lancedb/lancedb.git
synced 2026-08-18 12:08:35 +00:00
fix(node): initialize built-in embedding providers
This commit is contained in:
@@ -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;
|
||||
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user