diff --git a/buster/documents_manager/base.py b/buster/documents_manager/base.py index a1f7dd7..589ab1f 100644 --- a/buster/documents_manager/base.py +++ b/buster/documents_manager/base.py @@ -130,8 +130,9 @@ def batch_add( min_time_interval: int = 60, num_workers: int = 16, embedding_fn: callable = get_openai_embedding, - csv_filename: Optional[str] = None, - csv_overwrite: bool = False, + csv_embeddings_filename: Optional[str] = None, + csv_errors_filename: Optional[str] = None, + csv_overwrite: bool = True, **add_kwargs, ): """ @@ -174,7 +175,8 @@ def batch_add( self.add( batch_df, num_workers=num_workers, - csv_filename=csv_filename, + csv_embeddings_filename=csv_embeddings_filename, + csv_errors_filename=csv_errors_filename, csv_overwrite=csv_overwrite, embedding_fn=embedding_fn, **add_kwargs,