File size: 453 Bytes
9c6594c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
import logging
from wandb.sdk.integration_utils.auto_logging import AutologAPI
from .resolver import CohereRequestResponseResolver
logger = logging.getLogger(__name__)
autolog = AutologAPI(
name="Cohere",
symbols=(
"Client.generate",
"Client.chat",
"Client.classify",
"Client.summarize",
"Client.rerank",
),
resolver=CohereRequestResponseResolver(),
telemetry_feature="cohere_autolog",
)
|