| 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", | |
| ) | |