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

add some env in ci

Browse Source
This commit is contained in:
yichuan520030910320
2025-07-28 16:11:44 -07:00
parent b124709bcd
commit 27d0d73f99
5 changed files with 16 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
.gitignore vendored
View File

@@ -86,3 +86,5 @@ packages/leann-backend-diskann/third_party/DiskANN/_deps/
*.passages.json *.passages.json
batchtest.py batchtest.py
tests/__pytest_cache__/
tests/__pycache__/

4
packages/leann-core/src/leann/__init__.py
View File

@@ -8,6 +8,10 @@ if platform.system() == "Darwin":
os.environ["MKL_NUM_THREADS"] = "1" os.environ["MKL_NUM_THREADS"] = "1"
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE" os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
os.environ["KMP_BLOCKTIME"] = "0" os.environ["KMP_BLOCKTIME"] = "0"
# Additional fixes for PyTorch/sentence-transformers on macOS ARM64 only in CI
if os.environ.get("CI") == "true":
os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "0"
os.environ["TOKENIZERS_PARALLELISM"] = "false"
from .api import LeannBuilder, LeannChat, LeannSearcher from .api import LeannBuilder, LeannChat, LeannSearcher
from .registry import BACKEND_REGISTRY, autodiscover_backends from .registry import BACKEND_REGISTRY, autodiscover_backends

5
packages/leann-core/src/leann/api.py
View File

@@ -23,6 +23,11 @@ from .registry import BACKEND_REGISTRY
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
def get_registered_backends() -> list[str]:
"""Get list of registered backend names."""
return list(BACKEND_REGISTRY.keys())
def compute_embeddings( def compute_embeddings(
chunks: list[str], chunks: list[str],
model_name: str, model_name: str,