From 3b12f68c3aadb55a4fe06d8e328de9b210d7a290 Mon Sep 17 00:00:00 2001
From: Ivan Despot <66276597+g-despot@users.noreply.github.com>
Date: Mon, 22 Jun 2026 09:56:21 +0200
Subject: [PATCH] feat: add optional dimensions to text2vec-aws vectorizer

---
 src/collections/config/types/vectorizer.ts | 2 ++
 src/collections/configure/unit.test.ts     | 2 ++
 2 files changed, 4 insertions(+)

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<VectorConfigCreate<never, 'test', 'hnsw', 'text2vec-aws'>>({
       name: 'test',
@@ -971,6 +972,7 @@ describe('Unit testing of the vectorizer factory class', () => {
           model: 'model',
           region: 'region',
           service: 'service',
+          dimensions: 512,
         },
       },
     });