pyproject.toml

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

[tool.poetry]
name = "llama-likes"
version = "0.1.0"
description = ""
authors = ["niklas <niklasfinken@me.com>"]
readme = "README.md"
packages = [{from="src", include = "llama_likes"}]

[tool.poetry.dependencies]
python = "^3.11"
openai = "^1.3.3"
pydantic = "^2.5.1"
python-dotenv = "^1.0.0"
transformers = "^4.35.2"
datasets = {version="^2.15.0", optional=true}
ipykernel = {version="^6.27.0", optional=true}
scikit-learn = {version="^1.3.2", optional=true}

[tool.poetry.extras]
notebook = ["datasets", "ipykernel", "scikit-learn"]

[tool.poetry.group.dev.dependencies]
mypy = "^1.6.1"
types-requests = "^2.31.0.10"
pytest = "*"
black = "*"
pre-commit = "*"

[tool.mypy]
files = "src,tests"
strict = "True"

[tool.pytest.ini_options]
markers = ["internal: marks tests as internal (deselect with '-k \"not internal\"')"]
addopts = "--capture=tee-sys"