[tool.poetry]
name = "capstone"
version = "0.1.0"
description = ""
authors = ["PhucVu <vuhongphuc24601@gmail.com>"]
readme = "README.md"

[tool.poetry.dependencies]
python = "^3.10"
llama-index = "^0.9.22"
weaviate-client = "^3.26.0"
pydantic-settings = "^2.1.0"
fastapi = "^0.108.0"
uvicorn = "^0.25.0"
pydantic = "^2.5.3"
gradio = "^4.12.0"

# reranker
torch = {version="^2.3.0", optional=true}
sentence-transformers = {version="^2.7.0", optional=true}

[tool.poetry.group.local]
optional = true
[tool.poetry.group.local.dependencies]
transformers = "^4.36.2"
torch = "^2.1.2"
llama-cpp-python = "^0.2.29"

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