169 lines
4.2 KiB
TOML
169 lines
4.2 KiB
TOML
[tool.poetry]
|
|
name = "embedchain"
|
|
version = "0.0.72"
|
|
description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data"
|
|
authors = ["Taranjeet Singh, Deshraj Yadav"]
|
|
license = "Apache License"
|
|
readme = "README.md"
|
|
exclude = [
|
|
"db",
|
|
"configs",
|
|
"notebooks"
|
|
]
|
|
|
|
[build-system]
|
|
requires = ["setuptools", "wheel"]
|
|
build-backend = "setuptools.build_meta"
|
|
|
|
[tool.ruff]
|
|
select = ["E", "F"]
|
|
ignore = []
|
|
fixable = ["ALL"]
|
|
unfixable = []
|
|
exclude = [
|
|
".bzr",
|
|
".direnv",
|
|
".eggs",
|
|
".git",
|
|
".git-rewrite",
|
|
".hg",
|
|
".mypy_cache",
|
|
".nox",
|
|
".pants.d",
|
|
".pytype",
|
|
".ruff_cache",
|
|
".svn",
|
|
".tox",
|
|
".venv",
|
|
"__pypackages__",
|
|
"_build",
|
|
"buck-out",
|
|
"build",
|
|
"dist",
|
|
"node_modules",
|
|
"venv"
|
|
]
|
|
line-length = 120
|
|
dummy-variable-rgx = "^(_+|(_+[a-zA-Z0-9_]*[a-zA-Z0-9]+?))$"
|
|
target-version = "py38"
|
|
|
|
[tool.ruff.mccabe]
|
|
max-complexity = 10
|
|
|
|
# Ignore `E402` (import violations) in all `__init__.py` files, and in `path/to/file.py`.
|
|
[tool.ruff.per-file-ignores]
|
|
"embedchain/__init__.py" = ["E401"]
|
|
|
|
[tool.black]
|
|
line-length = 120
|
|
target-version = ["py38", "py39", "py310", "py311"]
|
|
include = '\.pyi?$'
|
|
exclude = '''
|
|
/(
|
|
\.eggs
|
|
| \.git
|
|
| \.hg
|
|
| \.mypy_cache
|
|
| \.nox
|
|
| \.pants.d
|
|
| \.pytype
|
|
| \.ruff_cache
|
|
| \.svn
|
|
| \.tox
|
|
| \.venv
|
|
| __pypackages__
|
|
| _build
|
|
| buck-out
|
|
| build
|
|
| dist
|
|
| node_modules
|
|
| venv
|
|
)/
|
|
'''
|
|
|
|
[tool.black.format]
|
|
color = true
|
|
|
|
[tool.poetry.dependencies]
|
|
python = ">=3.9,<3.9.7 || >3.9.7,<4.0"
|
|
python-dotenv = "^1.0.0"
|
|
langchain = "^0.0.279"
|
|
requests = "^2.31.0"
|
|
openai = "^0.27.5"
|
|
tiktoken = { version="^0.4.0", optional=true }
|
|
chromadb ="^0.4.8"
|
|
youtube-transcript-api = { version="^0.6.1", optional=true }
|
|
beautifulsoup4 = { version="^4.12.2", optional=true }
|
|
pypdf = { version="^3.11.0", optional=true }
|
|
pytube = { version="^15.0.0", optional=true }
|
|
duckduckgo-search = { version="^3.8.5", optional=true }
|
|
llama-hub = { version = "^0.0.29", optional = true }
|
|
sentence-transformers = { version = "^2.2.2", optional = true }
|
|
torch = { version = "2.0.0", optional = true }
|
|
# Torch 2.0.1 is not compatible with poetry (https://github.com/pytorch/pytorch/issues/100974)
|
|
gpt4all = { version = "1.0.8", optional = true }
|
|
# 1.0.9 is not working for some users (https://github.com/nomic-ai/gpt4all/issues/1394)
|
|
opensearch-py = { version = "2.3.1", optional = true }
|
|
elasticsearch = { version = "^8.9.0", optional = true }
|
|
flask = { version = "^2.3.3", optional = true }
|
|
twilio = { version = "^8.5.0", optional = true }
|
|
fastapi-poe = { version = "0.0.16", optional = true }
|
|
discord = { version = "^2.3.2", optional = true }
|
|
slack-sdk = { version = "3.21.3", optional = true }
|
|
cohere = { version = "^4.27", optional= true }
|
|
docx2txt = { version="^0.8", optional=true }
|
|
pinecone-client = { version = "^2.2.4", optional = true }
|
|
unstructured = {extras = ["local-inference"], version = "^0.10.18", optional=true}
|
|
pillow = { version = "10.0.1", optional = true }
|
|
torchvision = { version = ">=0.15.1, !=0.15.2", optional = true }
|
|
ftfy = { version = "6.1.1", optional = true }
|
|
regex = { version = "2023.8.8", optional = true }
|
|
huggingface_hub = { version = "^0.17.3", optional = true }
|
|
pymilvus = { version="2.3.1", optional = true }
|
|
|
|
[tool.poetry.group.dev.dependencies]
|
|
black = "^23.3.0"
|
|
pre-commit = "^3.2.2"
|
|
ruff = "^0.0.220"
|
|
pytest = "^7.3.1"
|
|
pytest-mock = "^3.10.0"
|
|
pytest-env = "^0.8.1"
|
|
click = "^8.1.3"
|
|
isort = "^5.12.0"
|
|
pytest-cov = "^4.1.0"
|
|
responses = "^0.23.3"
|
|
mock = "^5.1.0"
|
|
pytest-asyncio = "^0.21.1"
|
|
|
|
[tool.poetry.extras]
|
|
streamlit = ["streamlit"]
|
|
community = ["llama-hub"]
|
|
opensource = ["sentence-transformers", "torch", "gpt4all"]
|
|
elasticsearch = ["elasticsearch"]
|
|
opensearch = ["opensearch-py"]
|
|
poe = ["fastapi-poe"]
|
|
discord = ["discord"]
|
|
slack = ["slack-sdk", "flask"]
|
|
whatsapp = ["twilio", "flask"]
|
|
pinecone = ["pinecone-client"]
|
|
images = ["torch", "ftfy", "regex", "pillow", "torchvision"]
|
|
huggingface_hub=["huggingface_hub"]
|
|
cohere = ["cohere"]
|
|
milvus = ["pymilvus"]
|
|
dataloaders=[
|
|
"youtube-transcripts-api",
|
|
"beautifulsoup4",
|
|
"pypdf",
|
|
"pytube",
|
|
"duckduckgo-search",
|
|
"docx2txt",
|
|
"unstructured",
|
|
"sentence-transformers",
|
|
]
|
|
|
|
[tool.poetry.group.docs.dependencies]
|
|
|
|
|
|
|
|
[tool.poetry.scripts]
|