Repository navigation
Expand file tree
/
Copy pathpyproject_ARM.toml
More file actions
126 lines (115 loc) · 3.33 KB
/
Copy pathpyproject_ARM.toml
File metadata and controls
126 lines (115 loc) · 3.33 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
[project]
name = "colette"
version = "0.1.0"
authors = [{ name = "Colette Team", email = "contact@jolibrain.com" }]
license = "Apache-2.0"
readme = "README.md"
description = "Colette is an open-source local RAG for your documents"
classifiers = [
"Programming Language :: Python :: 3.12",
"Intended Audience :: Developers",
"Intended Audience :: Information Technology",
"Intended Audience :: Science/Research",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Scientific/Engineering :: Information Analysis",
]
requires-python = ">=3.12"
dependencies = [
"click==8.3.0",
"chromadb==1.1.0",
"fastapi==0.118.0",
"pydantic==2.11.9",
"uvicorn==0.37.0",
"vllm==0.11.0",
"bitsandbytes==0.48.0",
"qwen_vl_utils==0.0.14",
"pypandoc_binary==1.15",
"python-doctr==1.0.1",
"ujson==5.11.0",
"GitPython==3.1.45",
"transformers==5.3.0",
"flash-attn==2.8.3",
"diffusers==0.35.1",
"plotly==6.3.1",
"seaborn==0.13.2",
"scikit-learn==1.7.2",
"nbformat==5.10.4",
"umap-learn==0.5.9.post2",
]
[project.optional-dependencies]
dev = [
"pytest==8.4.2",
"pytest-asyncio==1.2.0",
"ruff==0.13.2",
"pre-commit==4.3.0",
]
trag = [
"huggingface_hub[cli]>=1.3.0,<2.0",
"rank_bm25==0.2.2",
"unstructured[docx]==0.18.15",
"unstructured[pptx]==0.18.15",
"unstructured[xlsx]==0.18.15",
"unstructured[pdf]==0.18.15",
"unstructured-inference==1.0.5",
"langchain-core==0.3.77",
"langchain-community==0.3.30",
"langchain-chroma==0.2.6",
"langchain-huggingface==0.3.1",
"langchain-openai==0.3.33",
"langchain-ollama==0.3.8",
"langchain-unstructured==0.1.6",
"llama-cpp-python==0.3.1",
"pymupdf==1.26.4",
"tantivy==0.25.1",
]
[project.scripts]
colette_cli = "colette.colette_cli:cli"
[build-system]
requires = [
"setuptools>=61.0",
"ninja",
"packaging",
"setuptools-scm>=8.0",
"wheel",
"jinja2",
]
build-backend = "setuptools.build_meta"
[tool.setuptools.package-dir]
"" = "src"
[tool.setuptools.packages.find]
where = ["src"]
[tool.ruff]
line-length = 120
target-version = "py310"
[tool.ruff.lint]
select = [
"E", # pycodestyle errors
"F", # pyflakes
"I", # isort (imports)
"UP", # pyupgrade
"B", # flake8-bugbear
]
[tool.ruff.lint.per-file-ignores]
"src/colette/apidata.py" = ["E501"]
"src/colette/backends/hf/query_rephraser.py" = ["E501"]
"src/colette/backends/coldb/collection.py" = ["E501"]
"src/colette/backends/coldb/indexing/collection_encoder.py" = ["E501"]
"src/colette/backends/coldb/indexing/collection_indexer.py" = ["E501"]
"src/colette/backends/coldb/infra/config/base_config.py" = ["E501"]
"src/colette/backends/coldb/infra/launcher.py" = ["E501"]
"src/colette/backends/coldb/infra/run.py" = ["E501"]
"src/colette/backends/coldb/utility/utils/save_metadata.py" = ["E501"]
"src/colette/backends/coldb/modeling/checkpoint.py" = ["E501"]
"src/colette/ui/utils/namesgenerator.py" = ["E501"]
"src/colette/ui/utils/config.py" = ["E501"]
"src/colette/ui/utils/api.py" = ["E501"]
[tool.ruff.format]
quote-style = "double"
indent-style = "space"
docstring-code-format = true
docstring-code-line-length = 120
[tool.uv]
default-groups = []
no-build-isolation-package = ["flash-attn"]
[tool.uv.sources]
colpali-engine = { git = "https://github.com/illuin-tech/colpali" }