[project]
name = "private-gpt"
version = "1.0.1"
description = "Private GPT"
authors = [{ name = "Zylon", email = "hi@zylon.ai" }]
requires-python = ">=3.11,<3.12"
dependencies = [
    # Web framework
    "fastapi[all]>=0.136.0",
    "python-multipart>=0.0.26",
    # DI + config
    "injector>=0.24.0,<0.25",
    "pyyaml>=6.0.3,<7",
    "watchdog>=4.0.1,<5",
    # Core framework
    "llama-index-core>=0.14.21,<0.15.0",
    # Utilities
    "cachetools>=7.0.0",
    "partial-json-parser>=0.2.1.1.post7,<0.3",
    "requests>=2.33.0,<3",
    "retry-async>=0.1.4,<0.2",
    "nest-asyncio>=1.6.0",
    "nltk>=3.9.1,<4",
    "arq>=0.26,<0.27",
]

[project.scripts]
private-gpt = "private_gpt.cli.main:main"

[project.optional-dependencies]

# ---------------------------------------------------------------------------
# LLM providers
# ---------------------------------------------------------------------------
# sdk-openai: shared base for llm-openai* and embedding-openai*
sdk-openai = [
    "openai>=2.0.0,<3",
    "openai-harmony>=0.0.8",
]
llm-openai = [
    "private-gpt[sdk-openai]",
    "llama-index-llms-openai",
]
llm-openai-compatible = [
    "private-gpt[sdk-openai]",
    "llama-index-llms-openai-like",
]
llm-mistral = [
    "mistral-common[image,audio,hf-hub]>=1.11.1,<2",
    "huggingface_hub",
]
llm-anthropic = [
    "langchain-anthropic>=1.4.3",
    "anthropic>=0.116.0",
]

# ---------------------------------------------------------------------------
# Tokenizers
# ---------------------------------------------------------------------------
tokenizer-local = [
    "transformers>=5.7.0",
    "huggingface-hub",
    "hf-transfer>=0.1.9",
]
tokenizer = [
    "private-gpt[tokenizer-local]",
]

# ---------------------------------------------------------------------------
# Embedding providers
# ---------------------------------------------------------------------------
embedding-openai = [
    "private-gpt[sdk-openai]",
    "llama-index-embeddings-openai",
]
embedding-openai-compatible = [
    "private-gpt[sdk-openai]",
    "llama-index-embeddings-openai-like",
]


# ---------------------------------------------------------------------------
# Model providers
# ---------------------------------------------------------------------------
model-openai = [
    "private-gpt[llm-openai]",
    "private-gpt[embedding-openai]",
]
model-openai-compatible = [
    "private-gpt[llm-openai-compatible]",
    "private-gpt[embedding-openai-compatible]",
]

# ---------------------------------------------------------------------------
# Ingestion
# ---------------------------------------------------------------------------
# ingest-core: shared base for all ingest-* extras
ingest-core = [
    "chardet>=5.2.0,<6",
    "python-magic>=0.4.27,<0.5",
    "llama-index-readers-file",
    "llama-index-utils-workflow",
    "levenshtein>=0.27.3",
]
ingest-documents = [
    "private-gpt[ingest-core]",
    "langdetect>=1.0.9,<2",
    "python-pptx>=1.0.2,<2",
    "openpyxl>=3.1.5,<4",
    "pptx2md>=1.7.1",
    "scipy>=1.10.0,<2",
]
ingest-markitdown = [
    "private-gpt[ingest-core]",
    "markitdown[docx,pdf,pptx,xls,xlsx]>=0.1.5",
]
ingest-markup = [
    "private-gpt[ingest-core]",
    "mistune>=3.2.1,<4",
    "beautifulsoup4>=4.14.3,<5",
    "html2text>=2024.2.26,<2025",
    "marko>=2.2.0,<3",
    "markdownify>=0.13.1,<0.14",
    "lxml-html-clean>=0.4.4",
]
ingest = [
    "private-gpt[ingest-documents,ingest-markup, ingest-markitdown]",
]

# ---------------------------------------------------------------------------
# Storage backends
# ---------------------------------------------------------------------------
storage-s3 = [
    "boto3>=1.35.0,<1.43.2",
    "aiobotocore>=3.6.0,<3.7",
]
storage-smb = [
    "smbprotocol>=1.16.1,<2",
]
storage = [
    "private-gpt[storage-s3,storage-smb]",
]

# ---------------------------------------------------------------------------
# Vector stores
# ---------------------------------------------------------------------------
vectorstore-qdrant = [
    "llama-index-vector-stores-qdrant",
    "qdrant-client>=1.17.1,<2",
]
nodestore-postgres = [
    "private-gpt[database-postgres]",
    "llama-index-storage-docstore-postgres",
]
indexstore-postgres = [
    "private-gpt[database-postgres]",
    "llama-index-storage-index-store-postgres",
]

# ---------------------------------------------------------------------------
# Redis
# ---------------------------------------------------------------------------
# redis-core: shared base for redis
redis-core = [
    "redis>=5.0.1,<6",
]
redis = [
    "private-gpt[redis-core]",
    "redis-semaphore-async>=0.0.4",
]

# ---------------------------------------------------------------------------
# Database providers
# ---------------------------------------------------------------------------
# database-core: shared base for all database-* extras
database-core = [
    "sqlalchemy>=2.0.49,<3",
    "sqlgpt-parser>=0.0.1a5",
]
database-postgres = [
    "private-gpt[database-core]",
    "psycopg2-binary>=2.9.12,<3",
    "asyncpg>=0.31.0,<0.32",
]
database-mysql = [
    "private-gpt[database-core]",
    "pymysql>=1.1.1,<2",
]
database-mssql = [
    "private-gpt[database-core]",
    "pyodbc>=5.3.0",
]
database-db2 = [
    "private-gpt[database-core]",
    "ibm-db>=3.2.9; platform_machine != 'aarch64'",
    "ibm-db-sa>=0.4.4; platform_machine != 'aarch64'",
]
database = [
    "private-gpt[database-postgres,database-mysql,database-mssql,database-db2]",
]

# ---------------------------------------------------------------------------
# Queue / Task processing
# ---------------------------------------------------------------------------
queue-celery = [
    "celery==5.4.0,<6",
    "flower>=2.0.1,<3",
]
queue-rabbitmq = [
    "pika>=1.3.2,<2",
]
queue = [
    "private-gpt[queue-celery,queue-rabbitmq]",
]

# ---------------------------------------------------------------------------
# Data / Tabular
# ---------------------------------------------------------------------------
data-pandas = [
    "pandas>=2.2.3,<3",
]
data-pandasai = [
    "private-gpt[data-pandas]",
    "pandasai>=3.0.0",
]

# ---------------------------------------------------------------------------
# Tool integrations
# ---------------------------------------------------------------------------
tool-mcp = [
    "llama-index-tools-mcp",
    "mcp>=1.27.0,<2",
]
tool-tabular = [
    "private-gpt[data-pandasai]",
]
tool-database = [
    "private-gpt[database]",
]
tool-web-scraping = [
    "playwright==1.60.0",
]
tools = [
    "private-gpt[tool-mcp,tool-tabular,tool-database,tool-web-scraping]",
]

# ---------------------------------------------------------------------------
# Media processing
# ---------------------------------------------------------------------------
# media-core: shared torch base for all media-* extras
media-core = [
    "torch>=2.11.0,<3",
]
media-image = [
    "private-gpt[media-core]",
    "torchvision>=0.26.0,<1",
    "opencv-python-headless>=4.10.0.84,<4.13.0",
    "imagehash>=4.3.2",
]
media-audio = [
    "private-gpt[media-core]",
    "torchaudio>=2.11.0,<3",
    "pydub>=0.25.1",
    "noisereduce>=3.0.3",
]
media-video = [
    "private-gpt[media-core]",
    "ffmpeg>=1.4",
    "ffprobe>=0.5",
]
media = [
    "private-gpt[media-image,media-audio,media-video]",
]

# ---------------------------------------------------------------------------
# Observability
# ---------------------------------------------------------------------------
observability-arize-phoenix = [
    "openinference-instrumentation-llama-index>=4.3.9,<5",
    "opentelemetry-proto>=1.41.0,<2",
    "opentelemetry-exporter-otlp>=1.41.0,<2",
    "opentelemetry-sdk>=1.41.0,<2",
]
observability-opik = [
    "opik>=2.0.21,<3",
]
observability = [
    "private-gpt[observability-arize-phoenix,observability-opik]",
]

# ---------------------------------------------------------------------------
# Install flavors
# ---------------------------------------------------------------------------
core = [
    "private-gpt[model-openai,model-openai-compatible,tokenizer,vectorstore-qdrant,ingest,tools]",
]

# ---------------------------------------------------------------------------
# Dev tooling
# ---------------------------------------------------------------------------
lint = [
    "pre-commit>=4.5.0,<5",
    "ruff>=0,<1",
]
test = [
    "pytest>=9,<10",
    "pytest-xdist>=3.6.1",
    "pytest-asyncio>=1.3.0,<2",
    "pytest-cov>=7,<8",
    "pytest-httpx>=0.36.0",
]
typecheck = [
    "types-pyyaml>=6.0.12.12,<7",
    "pika-stubs>=0.1.3,<0.2",
    "types-retry>=0.9.9.4,<0.10",
    "celery-types>=0.20.0,<0.21",
    "types-psycopg2>=2.9.21.20241019,<3",
    "pandas-stubs>=2.2.3.241009,<3",
    "types-requests>=2.32.0.20241016,<3",
    "types-markdown>=3.7.0.20240822,<4",
    "types-beautifulsoup4>=4.12.0.20241020,<5",
    "types-openpyxl>=3.1.5.20241126,<4",
    "types-psutil>=6.1.0.20241221,<7",
    "types-tqdm>=4.67.0.20250809",
    "types-cachetools>=6.2.0.20251022,<7",
    "mypy-boto3-s3>=1.42.37",
    "types-regex>=2026.1.15.20260116",
    "ty==0.0.59",
]
dev = [
    "private-gpt[lint,test,typecheck]",
]

[tool.uv]
default-groups = []

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

[tool.hatch.build.targets.sdist]
exclude = [
  ".uv-cache",
  ".dist",
]

[tool.hatch.build.targets.wheel]
packages = ["private_gpt"]

[tool.hatch.build.targets.wheel.force-include]
"settings.yaml" = "settings.yaml"
"version.txt" = "version.txt"
"ui" = "ui"
"scripts" = "scripts"

[tool.coverage.run]
branch = true

[tool.coverage.report]
skip_empty = true
precision = 2

[tool.ruff]
target-version = 'py311'

lint.select = [
    "E",
    "W",
    "F",
    "B",
    "C4",
    "D",
    "I",
    "SIM",
    "TCH",
    "TID",
    "Q",
    "UP",
    "PT",
    "RUF",
]

lint.ignore = [
    "E501",
    "PT011",
    "SIM102",
    "D100",
    "D101",
    "D102",
    "D103",
    "D104",
    "D105",
    "D106",
    "D107",
    "TC006",
]

[tool.ruff.lint.pydocstyle]
convention = "google"

[tool.ruff.lint.pycodestyle]
max-doc-length = 88

[tool.ruff.lint.flake8-tidy-imports]
ban-relative-imports = "all"

[tool.ruff.lint.flake8-type-checking]
strict = true
runtime-evaluated-base-classes = ["pydantic.BaseModel"]
runtime-evaluated-decorators = ["injector.inject"]

[tool.ruff.lint.per-file-ignores]
"tests/**/*.py" = ["D"]

[tool.ty.environment]
python-version = "3.11"

[tool.ty.rules]
redundant-cast = "ignore"
unresolved-import = "ignore"
unused-ignore-comment = "ignore"
unused-type-ignore-comment = "ignore"

[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["tests"]
addopts = [
    "--import-mode=importlib",
]