ai-dial-mind-map-backend/pyproject.toml at development · epam/ai-dial-mind-map-backend · 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
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
[tool.poetry]
name = "mindmap"
version = "0.13.1"
description = "The project is a backend part of the DIAL Mind Map Studio"
authors = ["Uladzislau Vishneuski <Uladzislau_Vishneuski@epam.com>", "Viachaslau Shpak <Viachaslau_Shpak@epam.com>"]
homepage = "https://dialx.ai/"
keywords = ["ai", "mind-map"]

[tool.poetry.dependencies]
python = ">=3.10,<3.12"
fastapi = "^0.115.0"
uvicorn = "^0.31.0"
aiohttp = "^3.10.8"
langchain-community = "^0.3.1"
beautifulsoup4 = "^4.12.3"
markdownify = "^0.13.1"
langchain-openai = "^0.3.0"
sentence-transformers = {version = "3.3.1", extras = ["openvino"]}
faiss-cpu = "^1.9.0"
aidial-sdk = "^0.20.0"
aidial-client = "^0.0.1"
torch = [
    { markers = "sys_platform == 'darwin'", version = "2.7.0", source = "PyPI" },
    { markers = "sys_platform != 'darwin'", version = "2.7.0+cpu", source = "pytorch" },
]
torchvision = [
    { markers = "sys_platform == 'darwin'", version = "0.22.0", source = "PyPI" },
    { markers = "sys_platform != 'darwin'", version = "0.22.0+cpu", source = "pytorch" },
]
pyjwt = "^2.10.1"
langchain = "^0.3.13"
langchain-core = "^0.3.28"
numpy = "^1.22.4"
pandas = "^2.2.3"
scikit-learn = "1.7.1"
networkx = "3.3"
python-dotenv = "^1.0.1"
python-frontmatter = "^1.1.0"
python-pptx = "^1.0.2"
pdf2image = "^1.17.0"
pymupdf = "^1.25.3"
more-itertools = "10.1.0"
docarray = "0.39.1"
unstructured = {version = "0.16.16", extras = ["doc", "docx", "pdf", "ppt", "pptx", "xlsx"]}
cachetools = "5.3.1"
rank-bm25 = "^0.2.2"
optimum = {extras = ["openvino"], version = "^1.27.0"}
nltk = "3.9.1"
openvino = "2025.2.0"
pdfplumber = "0.11.9"
theine = "^0.4.4"
xgboost = "^3.0.2"

[[tool.poetry.source]]
name = "PyPI"
priority = "primary"

[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cpu"
priority = "explicit"


[tool.poetry.group.lint.dependencies]
flake8 = ">=6.0.0"
black = { version = "^25.1.0", extras = ["jupyter"] }
isort = "^5.12.0"
pyright = "^1.1.324"
autoflake = "^2.2.0"


[tool.poetry.group.dev.dependencies]
nox = "^2024.4.15"
pandas-stubs = "^2.3.0.250703"
pytest = "^8.4.1"
pytest-asyncio = "^1.1.0"
httpx = "^0.28.1"
pytest-mock = "^3.14.1"
streamlit = "^1.49.1"

[tool.pyright]
typeCheckingMode = "basic"
reportUnusedVariable = "error"
reportIncompatibleMethodOverride = "error"
exclude = [
    ".git",
    ".venv",
    ".nox",
    "**/__pycache__"
]

[tool.black]
line-length = 80
exclude = '''
/(
    \.git
  | \.venv
  | \.nox
  | \.__pycache__
)/
'''

[tool.isort]
line_length = 80
profile = "black"

[tool.autoflake]
ignore_init_module_imports = true
remove_all_unused_imports = true
in_place = true
recursive = true
quiet = true
exclude = [
    "\\.venv",
    "\\.nox",
]

[tool.pytest.ini_options]
addopts = "--doctest-modules"