# 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.

"""Rubrics for reward computation.

See RFC 004 for full design: rfcs/004-rubrics.md
"""

from openenv.core.rubrics.base import Rubric
from openenv.core.rubrics.containers import (
    Gate,
    RubricDict,
    RubricList,
    Sequential,
    WeightedSum,
)
from openenv.core.rubrics.llm_judge import LLMJudge
from openenv.core.rubrics.trajectory import (
    ExponentialDiscountingTrajectoryRubric,
    TrajectoryRubric,
)

__all__ = [
    # Base
    "Rubric",
    # Containers
    "Sequential",
    "Gate",
    "WeightedSum",
    "RubricList",
    "RubricDict",
    # Trajectory
    "TrajectoryRubric",
    "ExponentialDiscountingTrajectoryRubric",
    # LLM Judge
    "LLMJudge",
]