forked from decodingml/llm-twin-course
-
Notifications
You must be signed in to change notification settings - Fork 0
/
pyproject.toml
62 lines (56 loc) · 1.27 KB
/
pyproject.toml
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
[tool.poetry]
name = "rag-system"
description = ""
version = "0.1.0"
authors = [
"Vlad Adumitracesei <[email protected]>",
"Paul Iusztin <[email protected]>",
"Alex Vesa <[email protected]>",
"Rares Istoc <[email protected]>"
]
readme = "README.md"
[tool.ruff]
line-length = 88
select = [
"F401",
"F403",
]
[tool.poetry.dependencies]
python = ">=3.10, <3.12"
pydantic = "^2.6.3"
pydantic-settings = "^2.1.0"
pika = "^1.3.2"
qdrant-client = "^1.8.0"
langchain = "^0.1.13"
aws-lambda-powertools = "^2.38.1"
selenium = "^4.21.0"
instructorembedding = "^1.0.1"
numpy = "^1.26.4"
langchain-openai = "^0.1.3"
gdown = "^5.1.0"
pymongo = "^4.7.1"
structlog = "^24.1.0"
rich = "^13.7.1"
pip = "^24.0"
install = "^1.3.5"
comet-ml = "^3.41.0"
ruff = "^0.4.3"
pandas = "^2.2.2"
datasets = "^2.19.1"
transformers = "^4.40.2"
safetensors = "^0.4.3"
bitsandbytes = "^0.42.0"
scikit-learn = "^1.4.2"
unstructured = "^0.14.2"
[tool.poetry.group.3-feature-pipeline.dependencies]
bytewax = "0.18.2"
[tool.poetry.group.ml.dependencies]
qwak-inference = "^0.1.17"
comet-llm = "^2.2.4"
qwak-sdk = "^0.5.69"
peft = "^0.11.1"
sentence-transformers = "^2.6.1"
accelerate = "^0.30.1"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"