pyproject.toml

[tool.ruff]

# description of all rules are available on https://docs.astral.sh/ruff/rules/
lint.select = ["D", "E", "F", "W", "C", "S", "I", "TCH", "SLOT", "RUF", "C90", "N", "YTT", "ASYNC", "A", "C4", "T10", "PGH"]

# we need to check 'mood' of all docstrings, this needs to be enabled explicitly
lint.extend-select = ["D401"]

lint.ignore = []

target-version = "py311"
lint.pydocstyle.convention = "google"
line-length = 100

[tool.coverage.report]
# unit tests fails if the total coverage measurement is under this threshold value
fail_under = 90

[tool.mypy]
disable_error_code = ["union-attr", "return-value", "arg-type", "import-untyped"]
ignore_missing_imports = true
plugins = ["pydantic.mypy"]

[tool.pydantic-mypy]
init_forbid_extra = true
init_typed = true
warn_required_dynamic_aliases = true

[tool.bandit]
skips = []
exclude_dirs = ["tests", "scripts"]

[tool.pdm]
distribution = false

[tool.pdm.dev-dependencies]
dev = [
    "black==24.10.0",
    "pydocstyle==6.3.0",
    "fastparquet==2024.5.0",  # Required for model evaluation (runtime, if parquet qna file is used)
    "httpx==0.27.0",
    "mypy==1.13.0",  # Usually needs to be set to latest stable version available
    "packaging==24.1",
    "pytest==8.3.2",
    "pytest-cov==5.0.0",
    "pytest-asyncio==0.24.0",
    "pydantic==2.9.1",
    "rouge-score==0.1.2",  # Required for model evaluation
    "ruff==0.7.0",
    "bandit==1.7.9",
    "types-requests==2.32.0.20240622",
    "gradio==4.44.0",
    "boto3==1.34.145",
    "reportportal-client==5.5.6",
    "pytest-reportportal==5.4.1",
    "pytest-benchmark[histogram]>=4.0.0",
    "typing-extensions==4.12.2",
    "pytest-subtests==0.13.1",
]

# The following section is needed only for torch[cpu] variant on Linux,
# because this variant need to be downloaded from external link, not from
# standard Python package index:
#
# [[tool.pdm.source]]
# type = "find_links"
# url = "https://download.pytorch.org/whl/cpu/torch_stable.html"
# name = "torch"


[build-system]
requires = ["pdm-backend"]
build-backend = "pdm.backend"

[project]
name = "ols"
version = "0.2.0"
description = "OpenShift LightSpeed is an AI powered assistant that runs on OpenShift and provides answers to product questions using backend LLM services."
authors = []
dependencies = [
    "pdm==2.20.0.post1",
    "torch==2.4.1",
    "pandas==2.1.4",
    "fastapi==0.115.4",
    "langchain==0.3.6",
    "langchain-ibm==0.3.2",
    "llama-index==0.10.62",
    "llama-index-vector-stores-faiss==0.1.2",
    "llama-index-embeddings-huggingface==0.2.2",
    "uvicorn==0.32.0",
    "redis==5.0.8",
    "faiss-cpu==1.8.0.post1",
    "sentence-transformers==3.1.1",
    "openai==1.54.3",
    "pyarrow==18.0.0",
    "ibm-generative-ai==3.0.0",
    "ibm-cos-sdk==2.13.6",
    "langchain-openai==0.2.5",
    "pydantic==2.9.1",
    "setuptools==72.1.0",
    "prometheus-client==0.20.0",
    "kubernetes==30.1.0",
    "psycopg2-binary==2.9.9",
    "azure-identity==1.18.0",
    "langchain-community==0.3.5",
    "SQLAlchemy==2.0.35",
    "huggingface_hub==0.26.1",
    "ibm-watsonx-ai==1.1.22",
    "certifi==2024.8.30",
    "cryptography==43.0.1",
    "urllib3==2.2.2",
    "nltk==3.9.1",
    "aiohttp==3.10.5",
    "zipp==3.20.1",
    "jinja2==3.1.4",
    "scikit-learn==1.5.1",
    "starlette==0.40.0",
    "tqdm==4.66.5",
    "findpython==0.6.2",
    "filelock==3.16.1",
]
requires-python = "==3.11.*"
readme = "README.md"
license = {file = "LICENSE"}

[tool.pdm.scripts]
start = "pdm run make run"
test = "pdm run make test"
test-unit = "pdm run make test-unit"
test-integration = "pdm run make test-integration"
test-e2e = "pdm run make test-e2e"
coverage-report = "pdm run make coverage-report"
unit-tests-coverage-report = "pdm run make unit-tests-coverage-report"
integration-tests-coverage-report = "pdm run make integration-tests-coverage-report"
check-types = "pdm run make check-types"
generate-schema = "pdm run make schema"
security-check = "pdm run make security-check"
benchmarks = "pdm run make benchmarks"

[tool.setuptools]
packages = ["ols"]

[tool.pylint."MESSAGES CONTROL"]
good-names = ["e"]
disable = ["W1203", "C0103", "C0301", "C0302", "C0415", "E0602", "E0611", "E1101", "R0902", "R0903", "R0913", "R0914", "W0102", "W0212", "W0511", "W0603", "W0613", "W0621", "W0622", "W0707", "W0718", "W0719", "W1514", "E0401", "R0801", "R0917"]