from schemas.context import FossilContext
from tools.functions import (
    check_keypoints_tool,
    generate_3d_model_tool,
    inspect_specimen_tool,
    record_view_choice_tool,
    remove_background_tool,
    segment_fossil_sam3_tool,
    select_keypoints_tool,
    select_views_tool,
)

# The orchestrator drives the pipeline. It never looks at an image, and it has no
# tool that produces a coordinate or ranks a view — it cannot make either of those
# judgements even if it tries.
orchestrator_tools = [
    inspect_specimen_tool,
    select_views_tool,
    remove_background_tool,
    select_keypoints_tool,
    generate_3d_model_tool,
]

# The vision analyst is the only agent that sees pixels. It selects the keypoints
# per view itself, verifies them, and segments.
analyst_tools = [
    check_keypoints_tool,
    segment_fossil_sam3_tool,
]

view_selector_tools = [record_view_choice_tool]

__all__ = [
    "FossilContext",
    "analyst_tools",
    "check_keypoints_tool",
    "generate_3d_model_tool",
    "inspect_specimen_tool",
    "orchestrator_tools",
    "record_view_choice_tool",
    "remove_background_tool",
    "segment_fossil_sam3_tool",
    "select_keypoints_tool",
    "select_views_tool",
    "view_selector_tools",
]