From 094b896e139a4bc74c812369de13b8ddd10fb82a Mon Sep 17 00:00:00 2001 From: Yun Kim Date: Wed, 5 Jun 2024 12:42:45 -0400 Subject: [PATCH] Remove anthropic contrib import in llmobs file --- ddtrace/llmobs/_integrations/anthropic.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/ddtrace/llmobs/_integrations/anthropic.py b/ddtrace/llmobs/_integrations/anthropic.py index 5b18a43dd74..b292d22dfc9 100644 --- a/ddtrace/llmobs/_integrations/anthropic.py +++ b/ddtrace/llmobs/_integrations/anthropic.py @@ -3,7 +3,6 @@ from typing import Optional from ddtrace._trace.span import Span -from ddtrace.contrib.anthropic.utils import _get_attr from ddtrace.internal.logger import get_logger from .base import BaseLLMIntegration @@ -38,8 +37,8 @@ def _set_base_span_tags( def record_usage(self, span: Span, usage: Dict[str, Any]) -> None: if not usage: return - input_tokens = _get_attr(usage, "input_tokens", None) - output_tokens = _get_attr(usage, "output_tokens", None) + input_tokens = getattr(usage, "input_tokens", None) + output_tokens = getattr(usage, "output_tokens", None) span.set_metric("anthropic.response.usage.input_tokens", input_tokens) span.set_metric("anthropic.response.usage.output_tokens", output_tokens)