TBNilles/LEANN
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add todo

Browse Source
This commit is contained in:
yichuan520030910320
2025-07-21 20:59:09 -07:00
parent 530f6e4af5
commit 727724990e
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
packages/leann-core/src/leann/embedding_compute.py
View File

@@ -154,6 +154,7 @@ def compute_embeddings_sentence_transformers(
def compute_embeddings_openai(texts: List[str], model_name: str) -> np.ndarray: 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""" """Compute embeddings using OpenAI API"""
try: try:
import openai import openai
@@ -208,6 +209,7 @@ def compute_embeddings_openai(texts: List[str], model_name: str) -> np.ndarray:
def compute_embeddings_mlx( def compute_embeddings_mlx(
chunks: List[str], model_name: str, batch_size: int = 16 chunks: List[str], model_name: str, batch_size: int = 16
) -> np.ndarray: ) -> np.ndarray:
# TODO: @yichuan-w add progress bar only in build mode
"""Computes embeddings using an MLX model.""" """Computes embeddings using an MLX model."""
try: try:
import mlx.core as mx import mlx.core as mx