Skip to content

Commit 848100d

Browse files
wip: removing redundant truthy checks for required attributes.
1 parent 54a2316 commit 848100d

1 file changed

Lines changed: 13 additions & 17 deletions

File tree

  • instrumentation-genai/opentelemetry-instrumentation-anthropic/src/opentelemetry/instrumentation/anthropic

instrumentation-genai/opentelemetry-instrumentation-anthropic/src/opentelemetry/instrumentation/anthropic/messages_extractors.py

Lines changed: 13 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -162,28 +162,24 @@ def set_invocation_response_attributes(
162162
if message is None:
163163
return
164164

165-
if message.model:
166-
invocation.response_model_name = message.model
167-
168-
if message.id:
169-
invocation.response_id = message.id
165+
invocation.response_model_name = message.model
166+
invocation.response_id = message.id
170167

171168
finish_reason = normalize_finish_reason(message.stop_reason)
172169
if finish_reason:
173170
invocation.finish_reasons = [finish_reason]
174171

175-
if message.usage:
176-
tokens = extract_usage_tokens(message.usage)
177-
invocation.input_tokens = tokens.input_tokens
178-
invocation.output_tokens = tokens.output_tokens
179-
if tokens.cache_creation_input_tokens is not None:
180-
invocation.attributes[GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS] = (
181-
tokens.cache_creation_input_tokens
182-
)
183-
if tokens.cache_read_input_tokens is not None:
184-
invocation.attributes[GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS] = (
185-
tokens.cache_read_input_tokens
186-
)
172+
tokens = extract_usage_tokens(message.usage)
173+
invocation.input_tokens = tokens.input_tokens
174+
invocation.output_tokens = tokens.output_tokens
175+
if tokens.cache_creation_input_tokens is not None:
176+
invocation.attributes[GEN_AI_USAGE_CACHE_CREATION_INPUT_TOKENS] = (
177+
tokens.cache_creation_input_tokens
178+
)
179+
if tokens.cache_read_input_tokens is not None:
180+
invocation.attributes[GEN_AI_USAGE_CACHE_READ_INPUT_TOKENS] = (
181+
tokens.cache_read_input_tokens
182+
)
187183

188184
if capture_content:
189185
invocation.output_messages = get_output_messages_from_message(message)

0 commit comments

Comments
 (0)