# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

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

[project]
name = "openenv-ontology_architect"
version = "0.1.0"
description = "Ontology Architect environment for OpenEnv"
requires-python = ">=3.10"
dependencies = [
    # Core OpenEnv dependencies (required for server functionality)
    # "openenv-core @ git+https://github.com/meta-pytorch/OpenEnv.git@main#subdirectory=src/core",
    "openenv-core>=0.1.0",
    "fastapi>=0.115.0",
    "pydantic>=2.0.0",
    "uvicorn>=0.24.0",
    "requests>=2.31.0",
    "numpy>=1.26.0",
    "scipy>=1.11.0",
    "matplotlib>=3.10.9",
]

[project.optional-dependencies]
dev = [
    "pytest>=8.0.0",
    "pytest-cov>=4.0.0",
]
train = [
    "accelerate>=0.29.0",
    "datasets>=2.18.0",
    "torch>=2.2.0",
    "transformers>=4.40.0",
    "trl>=0.8.0",
]

[project.scripts]
# Server entry point - enables running via: uv run --project . server
# or: python -m ontology_architect.server.app
server = "ontology_architect.server.app:main"
ontology-smoke = "ontology_architect.scripts.smoke:main"
ontology-generate-curriculum = "ontology_architect.scripts.generate_curriculum:main"
ontology-train-sft = "ontology_architect.scripts.train_sft:main"
ontology-train-gro = "ontology_architect.scripts.train_gro:main"
ontology-evaluate = "ontology_architect.scripts.evaluate:main"

[tool.setuptools]
include-package-data = true
packages = ["ontology_architect", "ontology_architect.server", "ontology_architect.scripts"]
package-dir = { "ontology_architect" = ".", "ontology_architect.server" = "server", "ontology_architect.scripts" = "scripts" }

[tool.pytest.ini_options]
norecursedirs = ["artifacts"]