pyproject.toml

# Note: Ths file is generated by the cookiecutter template and modify with poetry commands. It's not neccessary to
# edit it manually (advanced).
# The pyproject.toml file with Poetry: https://python-poetry.org/docs/pyproject/
# Fast Reference: https://py-pkgs.org/03-how-to-package-a-python#summary-and-next-steps

[tool.poetry]
name = "mrag"
version = "0.1.0"
description = "RAG"
authors = ["Pilar Madariaga <pilarmadariaga.ii@gmail.com>"]
license = "MIT"
readme = "README.md"


[tool.poetry.dependencies]
# Core Packages:  poetry add <package>  (to remove: poetry remove <package>)
# Common issue: In case of incompatibility error after 'poetry add <package>', just set the --python property to the one
# suggested at the end of the error message. e.g.:
# output message: ... a possible solution would be to set the `python` property to ">=3.11,<3.12"
# solution: $ poetry add pandas-profiling --python ">=3.11,<3.12"

python = "^3.11"

pandas = ">=2.1"
pyarrow = ">=13"
python-dotenv = ">=1.0.0"
py-cpuinfo = ">=9.0.0"

# Docs
# py7zr = ">=0.20.6"
# pypdfium2 = ">=4.22.0"
pymupdf = ">=1.23.5"
docx2txt = ">=0.8"


# Retrieval / Database
faiss-cpu = ">=1.7.4"
chromadb = ">=0.4.18"


# LLM
langchain = ">=0.0.352"
openai = ">=1.6.1"
tiktoken = ">=0.5.1"


[tool.poetry.extras]
# Advanced: recommended for large Building Blocks only
# Detailed Dependence Groups with Poetry: https://python-poetry.org/docs/managing-dependencies/#dependency-groups
# e.g.: $ poetry add tensorflow --extras "deep_learning" --optional

[tool.poetry.group.dev.dependencies]
# Edition Only Packages:  poetry add <package> --group dev (to remove: poetry remove <package>)
# They won't be on the final package (pip install <my-package>), but will be installed when developing the package
pytest-cov = ">=4.1"
pre-commit = ">=3.4.0"
mypy = ">=1.5"
flake8 = ">=6.1.0"
pylint = ">=2.17"
isort = ">=5.12"
black = ">=23.9"
pyupgrade = ">=3.12"
snakeviz = ">=2.2"

notebook = ">=7"
jupyter = ">=1.0.0"
jupyter-contrib-nbextensions = ">=0.7.0"
ipython = "8.16.1"                       # 8.17.1 has a bug with autoreload


matplotlib = ">=3.7.2"
plotly = ">=5.17.0"

# API
uvicorn = ">=0.23.1"
fastapi = ">=0.100.0"

# Front Lite
streamlit = ">=1.27.2"
gradio = ">=4.7.1"

# new packages
openpyxl = "^3.1.2"
seaborn = ">=0.13.0"
scikit-learn = "^1.3.2"
umap = "^0.1.1"
altair = "^5.2.0"
deeplake = "^3.8.13"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"