import logging

from wandb.sdk.integration_utils.auto_logging import AutologAPI

from .resolver import HuggingFacePipelineRequestResponseResolver

logger = logging.getLogger(__name__)

resolver = HuggingFacePipelineRequestResponseResolver()

autolog = AutologAPI(
    name="transformers",
    symbols=("Pipeline.__call__",),
    resolver=resolver,
    telemetry_feature="hf_pipeline_autolog",
)

autolog.get_latest_id = resolver.get_latest_id