add progress bar in build

This commit is contained in:
yichuan520030910320
2025-07-21 20:55:18 -07:00
parent 2f224f5793
commit 530f6e4af5
4 changed files with 10 additions and 6 deletions

View File

@@ -20,6 +20,7 @@ def compute_embeddings(
mode: str = "sentence-transformers",
use_server: bool = True,
port: Optional[int] = None,
is_build=False,
) -> np.ndarray:
"""
Computes embeddings using different backends.
@@ -51,6 +52,7 @@ def compute_embeddings(
chunks,
model_name,
mode=mode,
is_build=is_build,
)
@@ -209,6 +211,7 @@ class LeannBuilder:
self.embedding_model,
self.embedding_mode,
use_server=False,
is_build=True,
)
string_ids = [chunk["id"] for chunk in self.chunks]
current_backend_kwargs = {**self.backend_kwargs, "dimensions": self.dimensions}