fix batch_add method

jerpint · Nov 9, 2023 · 9025609 · 9025609
1 parent f6f11c1
commit 9025609
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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,