Add batch_size in config for VectorDB (#1448)

2024-06-27 14:45:58 -07:00
parent edaeb78ccf
commit 0a78198bb5
10 changed files with 28 additions and 34 deletions
@@ -124,7 +124,7 @@ class TestWeaviateDb(unittest.TestCase):
        db = WeaviateDB()
        app_config = AppConfig(collect_metrics=False)
        App(config=app_config, db=db, embedding_model=embedder)
-        db.BATCH_SIZE = 1
+        db.config.batch_size = 1

        documents = ["This is test document"]
        metadatas = [None]