Skip to content

Commit b7cb6db

Browse files
wip: use get_attr for new fields to avoid failures.
1 parent 53650d5 commit b7cb6db

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

  • instrumentation-genai/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2

instrumentation-genai/opentelemetry-instrumentation-openai-v2/src/opentelemetry/instrumentation/openai_v2/chat_wrappers.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,9 @@ def _set_response_service_tier(self, chunk: ChatCompletionChunk) -> None:
6767
if self._self_service_tier:
6868
return
6969

70-
if chunk.service_tier:
71-
self._self_service_tier = chunk.service_tier
70+
service_tier = getattr(chunk, "service_tier", None)
71+
if service_tier:
72+
self._self_service_tier = service_tier
7273

7374
def _build_streaming_response(self, chunk: ChatCompletionChunk) -> None:
7475
if chunk.choices is None:
@@ -98,9 +99,10 @@ def _build_streaming_response(self, chunk: ChatCompletionChunk) -> None:
9899
)
99100

100101
def _set_usage(self, chunk: ChatCompletionChunk) -> None:
101-
if chunk.usage:
102-
self._self_completion_tokens = chunk.usage.completion_tokens
103-
self._self_prompt_tokens = chunk.usage.prompt_tokens
102+
usage = getattr(chunk, "usage", None)
103+
if usage:
104+
self._self_completion_tokens = usage.completion_tokens
105+
self._self_prompt_tokens = usage.prompt_tokens
104106

105107
def _process_chunk(self, chunk: ChatCompletionChunk) -> None:
106108
self._set_response_id(chunk)

0 commit comments

Comments
 (0)