b3e9ee96fa
- Fix ambiguous fullwidth characters (commas, parentheses) in strings and comments - Replace Chinese comments with English equivalents - Fix unused imports with proper noqa annotations for intentional imports - Fix bare except clauses with specific exception types - Fix redefined variables and undefined names - Add ruff noqa annotations for generated protobuf files - Add lint and format check to GitHub Actions CI pipeline
47 lines
1.1 KiB
TOML
47 lines
1.1 KiB
TOML
[build-system]
|
|
requires = ["setuptools>=61.0"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[project]
|
|
name = "leann-core"
|
|
version = "0.1.13"
|
|
description = "Core API and plugin system for LEANN"
|
|
readme = "README.md"
|
|
requires-python = ">=3.9"
|
|
license = { text = "MIT" }
|
|
|
|
# All required dependencies included
|
|
dependencies = [
|
|
"numpy>=1.20.0",
|
|
"tqdm>=4.60.0",
|
|
"psutil>=5.8.0",
|
|
"pyzmq>=23.0.0",
|
|
"msgpack>=1.0.0",
|
|
"torch>=2.0.0",
|
|
"sentence-transformers>=2.2.0",
|
|
"llama-index-core>=0.12.0",
|
|
"python-dotenv>=1.0.0",
|
|
"openai>=1.0.0",
|
|
"huggingface-hub>=0.20.0",
|
|
"transformers>=4.30.0",
|
|
"requests>=2.25.0",
|
|
"accelerate>=0.20.0",
|
|
"PyPDF2>=3.0.0",
|
|
"pymupdf>=1.23.0",
|
|
"pdfplumber>=0.10.0",
|
|
"mlx>=0.26.3; sys_platform == 'darwin'",
|
|
"mlx-lm>=0.26.0; sys_platform == 'darwin'",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
colab = [
|
|
"torch>=2.0.0,<3.0.0", # Limit torch version to avoid conflicts
|
|
"transformers>=4.30.0,<5.0.0", # Limit transformers version
|
|
"accelerate>=0.20.0,<1.0.0", # Limit accelerate version
|
|
]
|
|
|
|
[project.scripts]
|
|
leann = "leann.cli:main"
|
|
|
|
[tool.setuptools.packages.find]
|
|
where = ["src"] |