diff --git a/src/collections/config/types/vectorizer.ts b/src/collections/config/types/vectorizer.ts index ed75f606..0c403531 100644 --- a/src/collections/config/types/vectorizer.ts +++ b/src/collections/config/types/vectorizer.ts @@ -586,6 +586,8 @@ export type Text2VecVoyageAIConfig = { truncate?: boolean; /** Whether to vectorize the collection name. */ vectorizeCollectionName?: boolean; + /** The number of dimensions for the generated embeddings. */ + dimensions?: number; }; /** diff --git a/src/collections/configure/unit.test.ts b/src/collections/configure/unit.test.ts index 30ec320d..76018fb8 100644 --- a/src/collections/configure/unit.test.ts +++ b/src/collections/configure/unit.test.ts @@ -1542,6 +1542,7 @@ describe('Unit testing of the vectorizer factory class', () => { baseURL: 'base-url', model: 'model', truncate: true, + dimensions: 256, }); expect(config).toEqual>({ name: 'test', @@ -1555,6 +1556,7 @@ describe('Unit testing of the vectorizer factory class', () => { baseURL: 'base-url', model: 'model', truncate: true, + dimensions: 256, }, }, });