[tool.poetry]
name = "TEMPO-BIAS"
version = "1.0.0"
description = "A pipeline for analyzing political bias in Large Language Models using vLLM serving"
authors = ["Paris-Saclay University <fairness-ai@universite-paris-saclay.fr>"]
readme = "README.md"
license = "MIT"
repository = "https://github.com/moujar/TEMPO-BIAS"
keywords = ["llm", "political-bias", "fairness", "vllm", "nlp"]

[tool.poetry.dependencies]
python = "^3.10"
torch = "^2.0.0"
transformers = "^4.36.0"
vllm = "^0.3.0"
pandas = "^2.0.0"
numpy = "^1.24.0"
scipy = "^1.11.0"
scikit-learn = "^1.3.0"
matplotlib = "^3.7.0"
seaborn = "^0.12.0"
plotly = "^5.15.0"
vaderSentiment = "^3.3.2"
textblob = "^0.17.1"
requests = "^2.31.0"
aiohttp = "^3.9.0"
pyyaml = "^6.0"
python-dotenv = "^1.0.0"
rich = "^13.0.0"
tqdm = "^4.65.0"
huggingface-hub = "^0.19.0"
datasets = "^2.14.0"
accelerate = "^0.24.0"

[tool.poetry.group.dev.dependencies]
pytest = "^7.4.0"
pytest-cov = "^4.1.0"
black = "^23.0.0"
isort = "^5.12.0"
flake8 = "^6.0.0"
mypy = "^1.5.0"

[tool.poetry.group.web.dependencies]
fastapi = "^0.100.0"
uvicorn = "^0.23.0"
gradio = "^4.0.0"

[tool.poetry.scripts]
bias-analysis = "run_pipeline:main"

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

[tool.black]
line-length = 100
target-version = ['py310']

[tool.isort]
profile = "black"
line_length = 100

[tool.mypy]
python_version = "3.10"
warn_return_any = true
warn_unused_configs = true
ignore_missing_imports = true