diff --git a/src/collections/config/types/vectorizer.ts b/src/collections/config/types/vectorizer.ts index 1a0e97d3..17433419 100644 --- a/src/collections/config/types/vectorizer.ts +++ b/src/collections/config/types/vectorizer.ts @@ -381,6 +381,8 @@ export type Text2VecAWSConfig = { region: string; /** The AWS service to use. */ service: 'sagemaker' | 'bedrock' | string; + /** The number of dimensions for the generated embeddings. */ + dimensions?: number; /** Whether the collection name is vectorized. */ vectorizeCollectionName?: boolean; }; diff --git a/src/collections/configure/unit.test.ts b/src/collections/configure/unit.test.ts index 637821f3..8e877f92 100644 --- a/src/collections/configure/unit.test.ts +++ b/src/collections/configure/unit.test.ts @@ -961,6 +961,7 @@ describe('Unit testing of the vectorizer factory class', () => { model: 'model', region: 'region', service: 'service', + dimensions: 512, }); expect(config).toEqual>({ name: 'test', @@ -971,6 +972,7 @@ describe('Unit testing of the vectorizer factory class', () => { model: 'model', region: 'region', service: 'service', + dimensions: 512, }, }, });