wehub-resource-sync
124 lines
3.2 KiB
TOML
124 lines
3.2 KiB
TOML
[project]
|
|
name = "pixelrag"
|
|
version = "0.3.0"
|
|
description = "Visual Retrieval-Augmented Generation — render, embed, index, search"
|
|
requires-python = ">=3.12"
|
|
authors = [{ name = "Zhifei Li", email = "andylizf@outlook.com" }]
|
|
license = "Apache-2.0"
|
|
readme = "README.md"
|
|
keywords = ["rag", "retrieval", "vision-language", "screenshots", "embeddings", "faiss"]
|
|
classifiers = [
|
|
"License :: OSI Approved :: Apache Software License",
|
|
"Programming Language :: Python :: 3.12",
|
|
"Topic :: Scientific/Engineering :: Artificial Intelligence",
|
|
]
|
|
# Core stays light — rendering / screenshots only (no torch). The `pixelshot` command and
|
|
# the `pixelrag` umbrella work on this alone. Heavy ML stages are opt-in extras below.
|
|
dependencies = [
|
|
"pillow>=10.0.0",
|
|
"websockets>=12.0",
|
|
"pymupdf>=1.27.2.3",
|
|
"pyturbojpeg>=2.2.0",
|
|
"cef-capi-py>=131.3.5",
|
|
"anthropic>=0.102.0",
|
|
]
|
|
|
|
[project.optional-dependencies]
|
|
embed = [
|
|
"torch>=2.9.0",
|
|
"torchvision>=0.24.0",
|
|
"transformers>=4.57.0",
|
|
"faiss-cpu>=1.9.0",
|
|
"numpy>=1.26.0",
|
|
"tqdm>=4.60.0",
|
|
]
|
|
serve = [
|
|
"fastapi>=0.115.0",
|
|
"uvicorn>=0.30.0",
|
|
"numpy>=1.26.0",
|
|
"faiss-cpu>=1.9.0",
|
|
"transformers>=4.57.0",
|
|
"torch>=2.9.0",
|
|
"qwen-vl-utils",
|
|
"pydantic>=2.0.0",
|
|
]
|
|
index = ["pixelrag[embed]", "pyyaml>=6.0", "markdown>=3.4"]
|
|
all = ["pixelrag[embed,serve,index]"]
|
|
gpu = ["faiss-gpu-cu12>=1.13.2; sys_platform == 'linux'"]
|
|
playwright = ["playwright>=1.40.0"]
|
|
pdf = ["pdf2image>=1.16.0"]
|
|
kiwix = ["libzim>=3.6.0"]
|
|
distributed = ["boto3>=1.42.0"]
|
|
eval = [
|
|
"pandas>=2.0",
|
|
"Pillow>=10.0",
|
|
"tqdm>=4.60",
|
|
"trafilatura>=1.6",
|
|
"openai>=1.0",
|
|
"aiohttp>=3.9",
|
|
"datasets>=2.14",
|
|
"huggingface-hub>=0.20",
|
|
]
|
|
dev = ["pytest>=8.0"]
|
|
|
|
[project.scripts]
|
|
pixelshot = "pixelrag_render.render:main"
|
|
pixelrag = "pixelrag.cli:main"
|
|
|
|
[project.urls]
|
|
Homepage = "https://pixelrag.ai"
|
|
Repository = "https://github.com/StarTrail-org/PixelRAG"
|
|
|
|
[build-system]
|
|
requires = ["hatchling"]
|
|
build-backend = "hatchling.build"
|
|
|
|
[tool.hatch.build.targets.wheel]
|
|
# One distribution bundles the umbrella + every stage module.
|
|
packages = [
|
|
"src/pixelrag",
|
|
"render/src/pixelrag_render",
|
|
"embed/src/pixelrag_embed",
|
|
"index/src/pixelrag_index",
|
|
"serve/src/pixelrag_serve",
|
|
]
|
|
|
|
[tool.hatch.build.targets.sdist]
|
|
# Repo root holds multi-GB data dirs (.venv, tiles, arxiv, …); ship only the package sources.
|
|
include = [
|
|
"/src/pixelrag",
|
|
"/render/src",
|
|
"/embed/src",
|
|
"/index/src",
|
|
"/serve/src",
|
|
"/README.md",
|
|
"/LICENSE",
|
|
]
|
|
|
|
[tool.uv]
|
|
override-dependencies = ["nvidia-cudnn-cu12==9.20.0.48; sys_platform == 'linux'"]
|
|
environments = ["sys_platform == 'linux'", "sys_platform == 'darwin'"]
|
|
|
|
[tool.uv.sources]
|
|
torch = [{ index = "pytorch-cu129", marker = "sys_platform == 'linux'" }]
|
|
torchvision = [{ index = "pytorch-cu129", marker = "sys_platform == 'linux'" }]
|
|
|
|
[[tool.uv.index]]
|
|
name = "pytorch-cu129"
|
|
url = "https://download.pytorch.org/whl/cu129"
|
|
explicit = true
|
|
|
|
[tool.ruff]
|
|
exclude = ["tmp"]
|
|
|
|
[tool.ruff.lint]
|
|
ignore = [
|
|
"E402", # module-level import not at top — common in ML codebases
|
|
"E741", # ambiguous variable name (l, O, I)
|
|
]
|
|
|
|
[tool.pyright]
|
|
extraPaths = ["render/src", "embed/src", "index/src", "serve/src", "train/src"]
|
|
venvPath = "."
|
|
venv = ".venv"
|