Feat - Batch Size

src/main.py

@@ -371,6 +371,8 @@ async def create_embedding(
 
         dataset = Dataset.from_dict(result)
 
+        embedding_service.batch_size = request.batch_size
+
         # Step 2: Generate embeddings
         dataset = await embedding_service.create_embeddings(
             dataset, request.target_column, request.output_column