pyproject.toml

[project]
name = "agentic-rag"
version = "0.1.0"
description = "Production-grade Agentic RAG framework with state-of-the-art retrieval techniques"
readme = "README.md"
license = { text = "MIT" }
requires-python = ">=3.12"
authors = [{ name = "Hesham Salama", email = "hesham@autonomouslab.io" }]
keywords = [
    "rag",
    "retrieval-augmented-generation",
    "agentic",
    "llm",
    "vector-search",
    "embeddings",
    "colbert",
    "graphrag",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Developers",
    "License :: OSI Approved :: MIT License",
    "Programming Language :: Python :: 3.12",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]

dependencies = [
    # Vector DB
    "qdrant-client>=1.12.0",

    # Embeddings
    "sentence-transformers>=3.3.0",
    "transformers>=4.46.0",
    "torch>=2.5.0",

    # LLM Providers (multi-provider support)
    "anthropic>=0.39.0",
    "openai>=1.58.0",
    "google-generativeai>=0.8.0",  # Gemini 2.5
    "google-genai>=1.0.0",         # Gemini 3 (new SDK)
    "groq>=0.11.0",                # Groq (ultra-fast inference for contextual chunking)

    # Sparse retrieval
    "rank-bm25>=0.2.2",

    # Core
    "pydantic>=2.10.0",
    "pydantic-settings>=2.6.0",
    "typer>=0.15.0",
    "rich>=13.9.0",
    "anyio>=4.8.0",
    "tenacity>=9.0.0",

    # Async
    "aiofiles>=24.1.0",
    "httpx>=0.28.0",

    # Observability
    "opentelemetry-api>=1.28.0",
    "opentelemetry-sdk>=1.28.0",

    # Caching
    "diskcache>=5.6.0",

    # Document processing
    "pypdf>=5.1.0",
    "python-docx>=1.1.0",
    "beautifulsoup4>=4.12.0",

    # Utilities
    "orjson>=3.10.0",
    "tqdm>=4.67.0",
    "numpy>=2.0.0",

    # GraphRAG
    "networkx>=3.4.0",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.3.0",
    "pytest-asyncio>=0.24.0",
    "pytest-cov>=6.0.0",
    "ruff>=0.8.0",
    "mypy>=1.13.0",
]

api = [
    "fastapi>=0.115.0",
    "uvicorn>=0.32.0",
]

local = [
    "vllm>=0.6.0",
]

all = [
    "agentic-rag[dev,api,local]",
]

[project.scripts]
agentic-rag = "agentic_rag.cli:app"

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

[tool.hatch.build.targets.wheel]
packages = ["src/agentic_rag"]

[tool.ruff]
target-version = "py312"
line-length = 100

[tool.ruff.lint]
select = [
    "E",     # pycodestyle errors
    "W",     # pycodestyle warnings
    "F",     # Pyflakes
    "I",     # isort
    "B",     # flake8-bugbear
    "C4",    # flake8-comprehensions
    "UP",    # pyupgrade
    "ARG",   # flake8-unused-arguments
    "SIM",   # flake8-simplify
]
ignore = [
    "E501",   # line too long (handled by formatter)
    "UP046",  # Generic[T] style is fine for compatibility
    "B008",   # Function call in default arg (FastAPI pattern)
    "SIM117", # Nested with statements (sometimes clearer)
    "SIM102", # Nested if statements (sometimes clearer)
    "SIM108", # Ternary operator (sometimes less readable)
    "B904",   # raise from - nice to have but not critical
    "B027",   # Empty methods in ABC - intentional for optional overrides
    "B007",   # Unused loop variable - rename to _ handled separately
    "E741",   # Ambiguous variable names (l, O, I) - context makes clear
]

[tool.ruff.lint.per-file-ignores]
"tests/*" = ["ARG001", "ARG002", "ARG005", "B017"]  # Test-specific patterns
"src/agentic_rag/agents/*.py" = ["ARG002"]  # Agent interface methods
"src/agentic_rag/api.py" = ["ARG001", "B008", "B904"]  # FastAPI patterns
"src/agentic_rag/retrieval/*.py" = ["ARG002"]  # Retriever interface compliance
"src/agentic_rag/reranking/*.py" = ["ARG002"]  # Reranker interface compliance
"src/agentic_rag/vectordb/*.py" = ["ARG002"]  # VectorDB interface compliance
"src/agentic_rag/chunking/*.py" = ["ARG002"]  # Chunker interface compliance
"src/agentic_rag/pipeline/*.py" = ["ARG002"]  # Pipeline interface compliance
"src/agentic_rag/generation/*.py" = ["ARG002"]  # Generator interface compliance
"src/agentic_rag/ingestion/*.py" = ["ARG002"]  # Ingestion interface compliance
"src/agentic_rag/graph/*.py" = ["ARG002"]  # Graph interface compliance
"src/agentic_rag/evaluation/*.py" = ["ARG002", "ARG004"]  # Evaluation interface compliance

[tool.ruff.lint.isort]
known-first-party = ["agentic_rag"]

[tool.mypy]
python_version = "3.12"
strict = false
warn_return_any = false
warn_unused_ignores = false
ignore_missing_imports = true
disallow_untyped_defs = false
check_untyped_defs = false

[[tool.mypy.overrides]]
module = "tests.*"
ignore_errors = true

[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["tests"]