From 727724990e9dd9d04843bbb5cccf792e5eb69339 Mon Sep 17 00:00:00 2001 From: yichuan520030910320 Date: Mon, 21 Jul 2025 20:59:09 -0700 Subject: [PATCH] add todo --- packages/leann-core/src/leann/embedding_compute.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/packages/leann-core/src/leann/embedding_compute.py b/packages/leann-core/src/leann/embedding_compute.py index c41f009..25e505c 100644 --- a/packages/leann-core/src/leann/embedding_compute.py +++ b/packages/leann-core/src/leann/embedding_compute.py @@ -154,6 +154,7 @@ def compute_embeddings_sentence_transformers( def compute_embeddings_openai(texts: List[str], model_name: str) -> np.ndarray: + # TODO: @yichuan-w add progress bar only in build mode """Compute embeddings using OpenAI API""" try: import openai @@ -208,6 +209,7 @@ def compute_embeddings_openai(texts: List[str], model_name: str) -> np.ndarray: def compute_embeddings_mlx( chunks: List[str], model_name: str, batch_size: int = 16 ) -> np.ndarray: + # TODO: @yichuan-w add progress bar only in build mode """Computes embeddings using an MLX model.""" try: import mlx.core as mx