From 5bd5032343adf6ba113f1231b03de80cea601bf1 Mon Sep 17 00:00:00 2001 From: Gabriele Venturi Date: Tue, 15 Oct 2024 01:07:47 +0200 Subject: [PATCH] feat: reduce batch size when adding docs to 5 --- backend/app/vectorstore/chroma.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/backend/app/vectorstore/chroma.py b/backend/app/vectorstore/chroma.py index d65652f..d1742db 100644 --- a/backend/app/vectorstore/chroma.py +++ b/backend/app/vectorstore/chroma.py @@ -63,7 +63,7 @@ def add_docs( docs: Iterable[str], ids: Optional[Iterable[str]] = None, metadatas: Optional[List[dict]] = None, - batch_size=50, + batch_size=5, ) -> List[str]: """ Add docs to the training set