pyproject.toml

[build-system]
build-backend = "setuptools.build_meta"
requires = ["setuptools >= 64", "setuptools-scm>=8"]

[tool.setuptools.packages.find]
where = ["src"]
include = ["health_ai", "health_assistant.*"]

[tool.setuptools_scm]
git_describe_command = "git describe --long --first-parent"

[build-system]
requires = ["hatchling", "hatch-fancy-pypi-readme"]
build-backend = "hatchling.build"

[project]
name = "health-assistant-ai"
dynamic = ["readme", "version"]
description = "AI-powered health consultation platform with intelligent symptom analysis, medication guidance, and wellness management."
license = { text = "Apache-2.0" }
requires-python = ">=3.11"
authors = [
  { name = "Health Assistant AI Team" },
  { email = "team@health-assistant-ai.org" },
]
classifiers = [
  "Development Status :: 4 - Beta",
  "Environment :: Console",
  "Environment :: Web Environment",
  "Intended Audience :: Healthcare Industry",
  "Intended Audience :: End Users/Desktop",
  "Intended Audience :: Developers",
  "License :: OSI Approved :: Apache Software License",
  "Operating System :: OS Independent",
  "Programming Language :: Python :: 3",
  "Programming Language :: Python :: 3.11",
  "Programming Language :: Python :: 3.12",
  "Topic :: Scientific/Engineering :: Artificial Intelligence",
  "Topic :: Scientific/Engineering :: Medical Science Apps.",
  "Topic :: Internet :: WWW/HTTP :: Dynamic Content",
  "Topic :: Software Development :: Libraries :: Python Modules",
]
keywords = [
  "health",
  "medical",
  "AI",
  "healthcare",
  "symptom-analysis",
  "medication",
  "wellness",
  "nutrition",
  "fitness",
  "telemedicine",
  "consultation",
  "diagnosis-support",
  "health-monitoring",
  "HIPAA-compliant",
]
dependencies = [
  "aiofiles~=24.1",
  "click~=8.1",
  "cryptography~=43.0",
  "fastapi~=0.115",
  "httpx~=0.27",
  "jinja2~=3.1",
  "jsonschema~=4.23",
  "loguru~=0.7",
  "neo4j~=5.25",
  "numpy~=1.26",
  "overrides~=7.7",
  "pillow~=10.4",
  "psycopg[binary]~=3.2",
  "pymilvus~=2.4",
  "pydantic~=2.9",
  "python-dotenv~=1.0",
  "PyYAML~=6.0",
  "requests~=2.32",
  "retry~=0.9",
  "rich~=13.9",
  "sqlalchemy~=2.0",
  "tabulate~=0.9",
  "typer~=0.12",
  "urllib3~=2.2",
  "uvicorn~=0.31",
  "uvloop~=0.20",
  "scikit-learn~=1.3",
  "pandas~=2.1",
  "matplotlib~=3.8",
  "seaborn~=0.12",
  "nltk~=3.8",
  "spacy~=3.7",
  "transformers~=4.35",
  "torch~=2.1",
  "langchain~=0.1",
  "openai~=1.3",
  "anthropic~=0.7",
]


[project.optional-dependencies]
# Optional dependencies for specialized health functionalities
# Keep sorted!!!
all = ["health-assistant-ai[medical,nutrition,fitness,mental-health,emergency]"] # All health modules
medical = [
  "fhir.resources~=7.0.2",
  "medspacy~=1.0.0",
  "bioregex~=1.0.3",
  "pydicom~=2.4.0",  # Medical imaging
  "nibabel~=5.2.0",  # Neuroimaging
]
nutrition = [
  "nutrient-database~=1.0.0",
  "food-data-central~=0.2.0",
  "calorie-tracker~=2.1.0",
]
fitness = [
  "fitness-tracker~=1.5.0",
  "workout-planner~=2.0.0",
  "exercise-database~=1.2.0",
]
mental-health = [
  "psychology-nlp~=0.8.0",
  "mood-tracker~=1.1.0",
  "mental-wellness~=1.0.0",
]
emergency = [
  "emergency-protocols~=1.0.0",
  "first-aid-db~=2.0.0",
  "crisis-intervention~=1.2.0",
]
databases = [
  "health-assistant-ai[mysql,redis]",
]
mysql = ["health-assistant-mysql"]
redis = ["health-assistant-redis"]
s3 = ["health-assistant-s3"]
telemetry = [
  "health-assistant-opentelemetry",
  "health-assistant-monitoring",
]
security = [
  "cryptography~=43.0.0",
  "bcrypt~=4.1.2",
  "pyjwt~=2.8.0",
]

examples = [
  "health_symptom_analyzer",
  "health_medication_checker",
  "health_nutrition_planner",
  "health_fitness_tracker",
  "health_emergency_response",
  "health_mental_wellness",
  "health_preventive_care",
  "health_telemedicine_bridge",
]

# Optional dependency needed when use_gunicorn is set to true
gunicorn = [
  "gunicorn~=23.0",
]

# Additional health-specific CLI tools
[project.scripts]
health-ai = "health_assistant.cli:main"
symptom-analyzer = "health_assistant.tools.symptom_analyzer:main"
medication-checker = "health_assistant.tools.medication_checker:main"
nutrition-planner = "health_assistant.tools.nutrition_planner:main"
fitness-tracker = "health_assistant.tools.fitness_tracker:main"
emergency-response = "health_assistant.tools.emergency_response:main"
mental-wellness = "health_assistant.tools.mental_wellness:main"
health-monitor = "health_assistant.tools.health_monitor:main"
telemedicine = "health_assistant.tools.telemedicine:main"

[project.urls]
Homepage = "https://github.com/cworld1/health-assistant-ai"
Documentation = "https://health-assistant-ai.readthedocs.io/"
Repository = "https://github.com/cworld1/health-assistant-ai.git"
"Bug Tracker" = "https://github.com/cworld1/health-assistant-ai/issues"
Changelog = "https://github.com/cworld1/health-assistant-ai/blob/main/CHANGELOG.md"
"Medical Disclaimer" = "https://health-assistant-ai.org/disclaimer"
"Privacy Policy" = "https://health-assistant-ai.org/privacy"


[tool.uv]
config-settings = { editable_mode = "compat" }

[tool.uv.sources]
# Health Assistant workspace packages
health-assistant-mysql = { workspace = true }
health-assistant-redis = { workspace = true }
health-assistant-s3 = { workspace = true }
health-assistant-opentelemetry = { workspace = true }
health-assistant-monitoring = { workspace = true }

# Health examples
health_symptom_analyzer = { path = "examples/medical/symptom_analyzer", editable = true }
health_medication_checker = { path = "examples/medical/medication_checker", editable = true }
health_nutrition_planner = { path = "examples/wellness/nutrition_planner", editable = true }
health_fitness_tracker = { path = "examples/wellness/fitness_tracker", editable = true }
health_emergency_response = { path = "examples/emergency/emergency_response", editable = true }
health_mental_wellness = { path = "examples/mental_health/mental_wellness", editable = true }
health_preventive_care = { path = "examples/preventive/preventive_care", editable = true }
health_telemedicine_bridge = { path = "examples/telemedicine/telemedicine_bridge", editable = true }

[tool.uv.workspace]
members = ["packages/*"]
exclude = ["packages/compat"]

[dependency-groups]
# Dependency groups for health assistant development
dev = [
  "asgi-lifespan~=2.1",
  "httpx-sse~=0.4",
  "ipython~=8.31",
  "myst-parser~=4.0",
  "nbsphinx~=0.9",
  "health-assistant-test",
  "sphinx-rtd-theme>=2.0.0",
  "pip>=24.3.1",
  "pre-commit>=4.0,<5.0",
  "pytest_httpserver==1.1.*",
  "pytest-asyncio==0.24.*",
  "pytest-cov~=6.1",
  "pytest-pretty~=1.2.0",
  "pytest~=8.3",
  "python-docx~=1.1.0",
  "ruff~=0.12",
  "setuptools >= 64",
  "setuptools_scm>=8",
  "sphinx~=8.2",
  "sphinx-autoapi>=3.6",
  "sphinx-copybutton>=0.5",
  "sphinx-mermaid",
  "tomlkit~=0.13.2",
  "twine~=6.0",
  "vale~=3.12",
  "yapf==0.43.*",
]

medical = [
  "fhir.resources~=7.0.2",
  "medspacy~=1.0.0",
  "bioregex~=1.0.3",
]

[project.entry-points.'health_assistant.components']
health_agents = "health_assistant.agent.register"
health_authentication = "health_assistant.authentication.register"
health_embedders = "health_assistant.embedder.register"
health_evaluators = "health_assistant.eval.register"
health_llms = "health_assistant.llm.register"
health_object_stores = "health_assistant.object_store.register"
health_observability = "health_assistant.observability.register"
health_retrievers = "health_assistant.retriever.register"
health_tools = "health_assistant.tool.register"

[project.entry-points.'health_assistant.medical_services']
symptom_analyzer = "health_assistant.medical.symptom_analyzer"
medication_checker = "health_assistant.medical.medication_checker"
emergency_response = "health_assistant.medical.emergency_response"
health_monitoring = "health_assistant.medical.health_monitoring"

[project.entry-points.'health_assistant.wellness_services']
nutrition_planner = "health_assistant.wellness.nutrition_planner"
fitness_tracker = "health_assistant.wellness.fitness_tracker"
mental_health = "health_assistant.wellness.mental_health"

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

# List any markers that users would reasonably want to filter by.
# These show up when querying `pytest --markers`
[tool.pytest.ini_options]
markers = [
  "benchmark: Performance benchmarks for health services",
  "e2e: End-to-end health consultation tests",
  "integration: Integration tests for medical databases and APIs",
  "medical: Medical accuracy and safety tests",
  "privacy: Privacy and security compliance tests",
  "slow: Slow tests requiring external medical APIs",
]
filterwarnings = [
  # Health-specific warning filters
  "ignore:^invalid escape sequence:SyntaxWarning",
  "ignore:.*medical.*:DeprecationWarning",
]
testpaths = ["tests", "examples/*/tests", "packages/*/tests"]
asyncio_mode = "auto"
asyncio_default_fixture_loop_scope = "session"


[tool.ruff]
# Set the maximum line length to 120.
line-length = 120
target-version = "py311"

[tool.ruff.lint]

select = [
  # The E, F, W categories are flake8 rules.
  "E",
  "F",
  "W",
  # Enable isort support
  "I",
  # Enable partial pylint support
  "PL",
]

extend-ignore = [
  "PLC0415", # Allow lazy imports inside functions
  # Disable all the "too-*" checks, as they are too strict for our codebase
  "PLR0904", # too-many-public-methods
  "PLR0911", # too-many-return-statements
  "PLR0912", # too-many-branches
  "PLR0913", # too-many-arguments
  "PLR0914", # too-many-locals
  "PLR0915", # too-many-statements
  "PLR0917", # too-many-positional-arguments
  "PLR1702", # too-many-nested-blocks
  "PLR2004", # Magic value used in comparison this happens a lot within unittests
  "PLW0108", # unnecessary-lambda
  "PLW2901", # allow re-assigning a loop variable
]

[tool.ruff.lint.isort]
combine-as-imports = true
default-section = "third-party"
force-single-line = true
known-first-party = ["health_assistant", "health_*", "_utils"]
known-third-party = [
  "biopython",
  "medpy",
  "pydicom",
  "fhir",
  "langchain",
  "openai",
  "anthropic",
  "transformers",
  "torch",
  "scikit-learn",
  "pandas",
  "numpy",
  "redis",
]
order-by-type = true

# ===== yapf Config =====
[tool.yapf]
based_on_style = "pep8"
column_limit = 120
split_all_top_level_comma_separated_values = true
join_multiple_lines = true
indent_dictionary_value = true


# ===== pyright Config =====
[tool.pyright]
deprecateTypingAliases = true
reportInvalidTypeForm = "warning"