pyproject.toml

[tool.poetry]
name = "pr-copilot"
version = "0.0.1"
description = "Your AI pair programmer 🤖️ specialized in code review, code summary and even code completion. 🧑‍💻🐛"
authors = ["kerthcet <kerthcet@gmail.com>"]
license = "MIT License"
readme = "README.md"

[tool.poetry.dependencies]
python = "~3.10"
fastapi = "0.103.0"
ray = {extras = ["serve"], version = "2.6.3"}
torch = "2.0.0"
llmlite = "0.0.9"
retry = "^0.9.2"
requests = "^2.31.0"
bitsandbytes-cuda117 = "0.26.0.post2"
bitsandbytes = "^0.41.1"
scipy = "^1.11.3"

[tool.poetry.group.dev.dependencies]
pytest = "^7.4.2"
black = "^23.9.1"
mypy = "^1.5.1"

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

[tool.mypy]
exclude = [
    'tmp/',
    'examples/',
]