pyproject.toml

[build-system]
requires = ["setuptools>=61.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "memories-dev"
description = "Collective Memory Infrastructure for AGI"
version = "2.0.9"
readme = "README.md"
requires-python = ">=3.9"
license = {file = "LICENSE"}
keywords = ["earth observation", "data processing", "memory management"]
authors = [
    {name = "Memories-dev Team", email = "team@memories-app.com"}
]
maintainers = [
    {name = "Memories-dev", email = "hello@memories.dev"}
]
classifiers = [
    "Development Status :: 5 - Production/Stable",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Operating System :: OS Independent",
    "License :: OSI Approved :: Apache Software License",
    "Topic :: Scientific/Engineering :: GIS",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Natural Language :: English"
]

dependencies = [
    # Core ML/DL
    "torch>=2.2.0",
    "transformers>=4.30.0",
    "diffusers>=0.25.0",
    "langchain>=0.1.0",
    "langchain-community>=0.0.1",
    
    # Data Processing
    "numpy>=1.24.4,<2.0.0",  # Version adjusted based on Python version during install
    "pandas>=2.0.0",  # Version adjusted based on Python version during install
    "duckdb>=0.9.0",
    "pyarrow>=14.0.1",
    "matplotlib>=3.7.0",  # Required for visualization
    
    # GIS/Spatial
    "geopandas>=0.14.0",
    "rasterio>=1.3.8",
    "shapely>=2.0.0,<3.0.0",  # Version adjusted based on Python version during install
    "mercantile>=1.2.1",
    "mapbox-vector-tile>=2.0.1",  # Version adjusted based on Python version during install
    "pyproj>=3.6.1",
    "pystac>=1.8.0",
    "osmnx>=1.9.0",
    "py6s>=1.9.0",
    "geopy>=2.4.1",  # Added for geocoding support
    "folium>=0.15.1",  # Added for map visualization
    "rtree>=1.1.0",  # Added for spatial indexing
    "owslib>=0.31.0",  # Added for WFS/WMS support
    
    # Image Processing
    "pillow>=10.0.0",
    "opencv-python>=4.8.0",
    "albumentations>=1.3.1",
    
    # AI/ML Tools
    "nltk>=3.8.1",
    "faiss-cpu>=1.7.4",
    "sentence-transformers>=2.2.0",
    
    # Data Storage/Processing
    "duckdb>=0.9.0",  # Used for in-memory and persistent storage
    "xarray>=2023.0.0",
    "dask>=2024.1.0",
    
    # Web/API
    "fastapi>=0.109.0",
    "pydantic>=2.6.0",
    "uvicorn>=0.27.0",
    "aiohttp>=3.9.0",
    "requests>=2.31.0",
    
    # Earth Observation
    "planetary-computer>=1.0.0",
    "pystac-client>=0.8.3",
    "earthengine-api>=0.1.390",
    
    # Utilities
    "tqdm>=4.65.0",
    "python-dotenv>=1.0.0",
    "pyyaml>=6.0.1",
    "cryptography>=42.0.0",
    "typing-extensions>=4.9.0",
    "fsspec>=2024.2.0",
    "noise>=1.2.2"
]

[project.optional-dependencies]
py39 = [
    "numpy>=1.24.4,<2.0.0",
    "pandas>=2.0.0,<2.1.0",
    "matplotlib<3.8.0",
    "scikit-learn<1.4.0",
    "albumentations<1.4.0",
    "accelerate<1.4.0",
    "shapely>=1.7.0,<2.0.0",
    "mapbox-vector-tile>=1.2.0,<2.0.0"
]

py310 = [
    "numpy>=1.24.4,<2.0.0",
    "pandas>=2.0.0",
    "matplotlib>=3.7.0",
    "scikit-learn>=1.3.0",
    "shapely>=1.7.0,<2.0.0",
    "mapbox-vector-tile>=1.2.0,<2.0.0"
]

py311 = [
    "numpy>=1.24.4,<2.0.0",
    "pandas>=2.0.0",
    "matplotlib>=3.7.0",
    "scikit-learn>=1.3.0",
    "shapely>=1.7.0,<2.0.0",
    "mapbox-vector-tile>=1.2.0,<2.0.0"
]

py312 = [
    "numpy>=1.26.0,<2.0.0",
    "pandas>=2.2.0",
    "matplotlib>=3.8.0",
    "scikit-learn>=1.3.0",
    "shapely>=1.7.0,<2.0.0",
    "mapbox-vector-tile>=1.2.0,<2.0.0"
]

py313 = [
    "numpy>=1.26.0,<2.0.0",
    "pandas>=2.2.0",
    "matplotlib>=3.8.0",
    "scikit-learn>=1.3.0",
    "shapely>=2.0.0,<3.0.0",
    "mapbox-vector-tile>=2.0.1"
]

# GPU support
gpu = [
    "torch>=2.2.0",  # Base requirement
    "torchvision>=0.17.0",
    "torchaudio>=2.2.0",
    "faiss-gpu==1.7.2",  # Fixed version that's known to work
    "cupy-cuda12x>=12.0.0",  # Updated for CUDA 12.x
    "torch-scatter>=2.1.2",
    "torch-sparse>=0.6.18",
    "torch-cluster>=1.6.3",
    "torch-geometric>=2.4.0"
]

# Development tools
dev = [
    "pytest>=8.3.4",
    "pytest-asyncio>=0.23.5",
    "pytest-cov>=6.0.0",
    "pytest-mock>=3.14.0",
    "pytest-xdist>=3.6.1",
    "pytest-benchmark>=4.0.0",
    "pytest-timeout>=2.2.0",
    "black>=24.1.0",
    "flake8>=7.0.0",
    "mypy>=1.8.0",
    "isort>=5.13.0",
    "pre-commit>=3.6.0"
]

# Documentation tools
docs = [
    "sphinx>=7.2.6",
    "sphinx-rtd-theme>=2.0.0",
    "sphinx-autodoc-typehints>=2.0.1; python_version>='3.12' and python_version<'3.13'",  # Only for Python 3.12
    "nbsphinx>=0.9.3",
    "pandoc>=2.3",
    "typing-extensions>=4.9.0",
    "docutils>=0.20.1",
    "packaging>=23.2",
    "furo>=2024.1.29",
    "sphinx-copybutton>=0.5.2",
    "myst-parser>=2.0.0",
    "platformdirs>=4.2.0",  # Add platformdirs to handle Jupyter warning
    "sphinx-design>=0.5.0",  # Added to fix failing test
    "sphinxcontrib-mermaid>=0.9.2",  # Added to fix failing test
    "sphinx-tabs>=3.4.1",  # Added to fix failing test
    "sphinx-togglebutton>=0.3.2",  # Added to fix failing test
    "sphinx-favicon>=1.0.1",  # Added to fix failing test
    "sphinx-sitemap>=2.5.1",  # Added to fix failing test
    "sphinx-last-updated-by-git>=0.3.6"  # Added to fix failing test
]

test = [
    "pytest>=8.3.4",
    "pytest-asyncio>=0.23.5",
    "pytest-cov>=6.0.0",
    "pytest-mock>=3.14.0",
    "pytest-xdist>=3.6.1",
    "pytest-benchmark>=4.0.0",
]

[project.urls]
Homepage = "https://memories.dev"
Documentation = "https://docs.memories.dev"
Repository = "https://github.com/Vortx-AI/memories-dev.git"
Issues = "https://github.com/Vortx-AI/memories-dev/issues"
Changelog = "https://github.com/Vortx-AI/memories-dev/blob/main/CHANGELOG.md"

[tool.setuptools]
packages = ["memories"]
package-dir = {"memories" = "memories"}

[tool.setuptools.package-data]
memories = [
    "config/*.yaml",
    "data/*.json",
    "models/config/*.json",
    "utils/styles/*.json"
]

[tool.black]
line-length = 100
target-version = ['py39']
include = '\.pyi?$'

[tool.isort]
profile = "black"
multi_line_output = 3
line_length = 100

[tool.mypy]
python_version = "3.9"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = true
check_untyped_defs = true

[tool.pytest.ini_options]
minversion = "8.3"
addopts = "-ra -q --cov=memories"
testpaths = ["tests"]
python_files = ["test_*.py"]
python_classes = ["Test*"]
python_functions = ["test_*"]
markers = ["asyncio: mark test as async"]