pyproject.toml

# https://setuptools.pypa.io/en/latest/userguide/quickstart.html
[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[tool.black]
line-length = 88
target-version = ['py310']
include = '\.pyi?$'
exclude = '''

(
  /(
      \.eggs         # exclude a few common directories in the
    | \.git          # root of the project
    | \.hg
    | \.mypy_cache
    | \.tox
    | \.venv
    | _build
    | buck-out
    | build
    | dist
  )/
  | foo.py           # also separately exclude a file named foo.py in
                     # the root of the project
)
'''

[tool.interrogate]
ignore-init-method = true
ignore-init-module = false
ignore-magic = false
ignore-semiprivate = false
ignore-private = false
ignore-property-decorators = false
ignore-module = false
fail-under = 100
exclude = ["setup.py", "docs", "build", "examples"]
ignore-regex = ["^get$", "^mock_.*", ".*BaseClass.*"]
verbose = 2
quiet = false
whitelist-regex = []
color = true

[tool.pytest.ini_options]
addopts = "-v --cov --cov-report term-missing -m 'not llm_eval' --durations=10"
testpaths = [
    "tests",
]
norecursedirs = ["archive/*"]

[tool.isort]
profile = "black"
float-to-top = true

[tool.setuptools]
include-package-data = true

[tool.setuptools.packages.find]
where = ["."]
namespaces = false

[project]
name = "llamabot"
version = "0.11.2"
# Runtime dependencies below
dependencies = [
    "openai",
    "loguru",
    "python-dotenv",
    "typer",
    "pyprojroot",
    "beautifulsoup4",
    "litellm>=1.59.1", # this is necessary to guarantee that ollama_chat models are supported for structured outputs
    "python-slugify",
    "pydantic>=2.0",
    "numpy", # https://github.com/ericmjl/llamabot/issues/56
    "jinja2",
    "fastapi>=0.104.0",  # This version requires Python >=3.8
    "uvicorn",
    "tenacity",
    "python-multipart",
    "httpx",
    "tqdm",
    "sqlalchemy",
    "pdfminer.six",
]
requires-python = ">=3.9,<3.13"
description = "A Pythonic interface to LLMs."
readme = "README.md"

[project.urls]
"Documentation" = "https://ericmjl.github.io/llamabot"
"Source Code" = "https://github.com/ericmjl/llamabot"

[project.scripts]
llamabot = "llamabot.cli:app"

[project.optional-dependencies]
notebooks = [
  "ics",
  "tzlocal",
  "chonkie[all]>=0.2.2,<0.3"
]
ui = [
    "panel",
    "jupyter_bokeh",
    "bokeh"
]
rag = [
    "lancedb",
    "chromadb",
    "tantivy",
    "rank-bm25",
]
agent = [
    "docker",
    "duckduckgo-search",
    "markdownify",
]
cli = [
    "pyzotero",
    "nbformat",
    "python-frontmatter",
    "rich",
    "gitpython",
    "prompt-toolkit",
    "case-converter",
    "pyperclip",
    "astor",
]
all = [
    "llamabot[notebooks,ui,rag,agent,cli]"
]

[dependency-groups]
notebooks = ["ollama>=0.4.4,<0.5"]
ui = []
agent = []
cli = []

[tool.pixi.project]
name = "llamabot"
description = "A Pythonic interface to LLMs"
channels = ["conda-forge"]
platforms = ["osx-arm64", "linux-64", "osx-64", "linux-aarch64"]

[tool.pixi.pypi-dependencies]
llamabot = { path = ".", editable = true }

[tool.pixi.feature.tests.dependencies]
pytest = "*"
hypothesis = "*"
pytest-cov = "*"
pytest-mock = "*"
tuna = "*"

[tool.pixi.feature.devtools.dependencies]
pre-commit = "*"
ipython = "*"
bump2version = "*"

[tool.pixi.feature.docs.dependencies]
mkdocs = "*"
mkdocs-material = "*"
mknotebooks = "*"

# Stuff that is referenced _only_ in notebooks go here.
[tool.pixi.feature.notebooks.dependencies]
jupyter = "*"
ipykernel = "*"
jupyterlab = "*"
pixi-kernel = "*"
biopython = "*"
lxml = ">=5.3.0,<6"
hvplot = ">=0.11.1,<0.12"
pip = ">=24.3.1,<25"
selenium = ">=4.11.2,<5"
python-chromedriver-binary = ">=133.0.6882.0.0,<134"

[tool.pixi.feature.notebooks.pypi-dependencies]
pdf2image = "*"

# TASKS BELOW!
[tool.pixi.tasks]
llamabot-cli = "llamabot --help"

[tool.pixi.feature.notebooks.tasks]
jlab = "jupyter lab"

[tool.pixi.feature.tests.tasks]
test = "pytest"

[tool.pixi.feature.docs.tasks]
docs = "mkdocs serve"
build-docs = "mkdocs build"

[tool.pixi.environments]
default = ["tests", "devtools", "docs", "notebooks", "ui", "rag", "agent", "cli"]
docs = ["docs"]
notebooks = ["notebooks"] # for those who just want to run notebooks
bare = ["devtools"]

[tool.pixi.dependencies]
python-multipart = "*"
numpydoc = ">=1.8.0,<2"