kg-llm-interface/pyproject.toml at main · sdsc-ordes/kg-llm-interface · 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
[project]
authors = [
    {name = "cyril.matthey-doret", email = "cyril.matthey-doret@epfl.ch"},
]
license = {text = "MIT"}
requires-python = "<4.0,>=3.10"
dependencies = [
    "tqdm<5.0.0,>=4.65.0",
    "sentence-transformers<3.0.0,>=2.2.2",
    "python-dotenv<2.0.0,>=1.0.0",
    "ipykernel<7.0.0,>=6.22.0",
    "rdflib==6.3.0",
    "chromadb<1.0.0,>=0.4.22",
    "more-itertools<10.0.0,>=9.1.0",
    "prefect<3.0.0,>=2.10.6",
    "typer<1.0.0,>=0.9.0",
    "bokeh==2.4.3",
    "sparqlwrapper<3.0.0,>=2.0.0",
    "fastapi<1.0.0,>=0.95.1",
    "uvicorn<1.0.0,>=0.22.0",
    "typing-extensions<5.0.0,>=4.6.3",
    "protobuf==3.20",
    "jupyterlab<5.0.0,>=4.0.2",
    "langchain<1.0.0,>=0.0.230",
    "openai<1.0.0,>=0.27.8",
    "poethepoet<1.0.0,>=0.21.0",
    "html5lib<2.0,>=1.1",
    "anyio==3.7.1",
    "testcontainers<4.0.0,>=3.7.1",
    "torch==2.6.0+cpu",
    "torchvision==0.21.0+cpu",
]
name = "aikg"
version = "0.1.0"
description = "Langchain-powered natural language interface to RDF knowledge-graphs"
readme = "README.md"

[dependency-groups]
local = [
    "transformers<5.0.0,>=4.28.1",
]
dev = [
    "black<24.0.0,>=23.3.0",
    "pytest<8.0.0,>=7.3.1",
    "pre-commit<4.0.0,>=3.2.2",
]

[tool.uv.sources]
torch = [
    { index = "pytorch-cpu" },
]
torchvision = [
    { index = "pytorch-cpu" },
]

[[tool.uv.index]]
name = "pytorch-cpu"
url = "https://download.pytorch.org/whl/cpu"
explicit = true