RAGatouille/pyproject.toml at main · 1st2024/RAGatouille · GitHub

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
[build-system]
requires = ["setuptools"]
build-backend = "setuptools.build_meta"

[tool.setuptools]
packages = [
    "ragatouille",
    "ragatouille.models",
    "ragatouille.data",
    "ragatouille.integrations",
    "ragatouille.negative_miners",
]

[project]
name = "RAGatouille"
version = "0.0.9post2"
description = "Library to facilitate the use of state-of-the-art retrieval models in common RAG contexts."
keywords = ["reranking", "retrieval", "rag", "nlp"]
authors = [
  {name = "Ben Clavié", email = "bc@answer.ai" }
]
maintainers = [
  {name = "Ben Clavié", email = "bc@answer.ai" }
]
license = {file = "LICENSE"}
readme = "README.md"

dependencies = [
  "llama-index",
  "faiss-cpu",
  "langchain_core",
  "colbert-ai>=0.2.19",
  "langchain",
  "onnx",
  "srsly",
  "voyager",
  "torch>=1.13",
  "fast-pytorch-kmeans",
  "sentence-transformers",
]

[project.optional-dependencies]
all = [
    "llama-index",
    "langchain",
    "rerankers",
    "voyager",
]
llamaindex = ["llama-index"]
langchain = ["langchain"]
train = ["sentence-transformers", "pylate", "rerankers"]

[project.urls]
"Homepage" = "https://github.com/answerdotai/ragatouille"

[tool.pytest.ini_options]
filterwarnings = [
    "ignore::Warning"
]

target-version = "py39"