pyproject.toml

[tool.poetry]
name = "langbatch"
version = "0.1.1"
description = "LangBatch is a python package with unified API for AI batch processing workloads. Supports OpenAI, Anthropic, Azure OpenAI, Vertex AI"
authors = ["KarthikeyanVijayanEasyLLM <karthikeyan.vijayan@easyllm.tech>"]
readme = "README.md"
license = "Apache-2.0"
homepage = "https://pypi.org/project/langbatch/"
repository = "https://github.com/EasyLLM/langbatch"
documentation = "https://github.com/EasyLLM/langbatch/blob/main/README.md"

[tool.poetry.dependencies]
# These packages are mandatory and form the core of this package’s distribution.
python = "^3.11"
openai = "^1.43.0"
jsonlines = "^4.0.0"

# These packages are optional and provide additional functionality.
google-cloud-aiplatform = { version = "^1.65.0", optional = true }
google-cloud-bigquery-storage = { version = "2.26.0", optional = true }
fastavro = { version = "^1.9.7", optional = true }
redis = { version = "^5.0.8", optional = true }
anthropic = { version = "^0.36.1", optional = true }

[tool.poetry.extras]
VertexAI = ["google-cloud-aiplatform", "google-cloud-bigquery-storage", "fastavro"]
Anthropic = ["anthropic"]
redis = ["redis"]
all = ["google-cloud-aiplatform", "google-cloud-bigquery-storage", "fastavro", "redis", "anthropic"]

[tool.poetry.group.dev.dependencies]
ipykernel = "^6.29.5"
pytest = "^8.3.3"
pytest-sugar = "^1.0.0"
pytest-asyncio = "^0.24.0"
pytest-xdist = "^3.6.1"
pytest-cov = "^5.0.0"
pytest-skip-slow = "^0.0.5"
mkdocstrings = {extras = ["python"], version = "^0.26.1"}
mkdocs-material = "^9.5.37"
mkdocs-glightbox = "^0.4.0"
mike = "^2.1.3"

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