-
Notifications
You must be signed in to change notification settings - Fork 17
/
pyproject.toml
56 lines (50 loc) · 1.61 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
[tool.poetry]
name = "graphrag_sdk"
version = "0.4.0"
license = "MIT"
description = "Enables graph-based RAG and integration with LLMs for advanced querying and knowledge graph creation."
authors = ["FalkorDB <[email protected]>"]
readme = "README.md"
repository = "https://github.com/FalkorDB/GraphRAG-SDK"
homepage = "http://RAG-SDK.readthedocs.io"
keywords = ['RAG', 'Graph-RAG', 'KnowledgeGraph']
[tool.poetry.dependencies]
python = "^3.9.0"
falkordb = "^1.0.9"
typing-extensions = "^4.12.1"
bs4 = "^0.0.2"
pypdf = "^4.2.0"
backoff = "^2.2.1"
requests = "^2.32.3"
python-abc = "^0.2.0"
ratelimit = "^2.2.1"
python-dotenv = "^1.0.1"
fix-busted-json = "^0.0.18"
ipykernel = "^6.29.5"
vertexai = { version = "^1.49.0", optional = true }
ollama = { version = "^0.2.1", optional = true }
openai = { version = "^1.35.9", optional = true }
google-generativeai = { version = "^0.8.1", optional = true }
litellm = { version = "^1.52.14", optional = true }
[tool.poetry.extras]
all = ["vertexai", "ollama", "openai", "google-generativeai", "litellm"]
ollama = ["ollama"]
openai = ["openai"]
vertexai = ["vertexai"]
google-generativeai = ["google-generativeai"]
litellm = ["litellm", "ollama"]
[tool.poetry.group.test.dependencies]
pytest = "^8.2.1"
[tool.poetry.group.dev.dependencies]
sphinx = "^7.3.7"
sphinx-rtd-theme = "^2.0.0"
pandas = "^2.2.2"
jupyter = "^1.0.0"
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"
[tool.pytest.ini_options]
log_cli = true
log_cli_level = "INFO"
log_cli_format = "%(asctime)s [%(levelname)8s] %(message)s (%(filename)s:%(lineno)s)"
log_cli_date_format = "%Y-%m-%d %H:%M:%S"