"""
Agent tools for the NL-to-SQL agent.

This module exports all tools that the agent can use to explore schema,
resolve values, generate SQL, and execute queries.
"""

from semantic_models.agent.tools.schema_tools import (
    get_attribute_values,
    get_entity_attributes,
    get_relationships,
    list_entities,
)
from semantic_models.agent.tools.resolution_tools import (
    get_glossary_terms,
    resolve_values,
)
from semantic_models.agent.tools.sql_tools import (
    execute_sql,
    generate_sql,
    validate_sql,
)

__all__ = [
    # Schema exploration
    "list_entities",
    "get_entity_attributes",
    "get_attribute_values",
    "get_relationships",
    # Value resolution
    "resolve_values",
    "get_glossary_terms",
    # SQL generation & execution
    "generate_sql",
    "validate_sql",
    "execute_sql",
]
