Skip to content

Commit 770f878

Browse files
committed
rename context, update _to_semconv_dict name
1 parent 92cd5c8 commit 770f878

File tree

2 files changed

+8
-8
lines changed

2 files changed

+8
-8
lines changed

util/opentelemetry-util-genai/src/opentelemetry/util/genai/data.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ class Message:
3838
type: str
3939
name: str
4040

41-
def _to_part_dict(self) -> OtelMessage:
41+
def _to_semconv_dict(self) -> OtelMessage:
4242
"""Convert the message to a dictionary suitable for OpenTelemetry semconvs.
4343
4444
Ref: https://github.com/open-telemetry/semantic-conventions/blob/main/docs/registry/attributes/gen-ai.md#gen-ai-input-messages

util/opentelemetry-util-genai/src/opentelemetry/util/genai/emitters.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@
6565
@dataclass
6666
class _SpanState:
6767
span: Span
68-
span_context: Context
68+
context: Context
6969
start_time: float
7070
request_model: Optional[str] = None
7171
system: Optional[str] = None
@@ -101,7 +101,7 @@ def _message_to_log_record(
101101
}
102102

103103
if capture_content:
104-
attributes["gen_ai.input.messages"] = [message._to_part_dict()]
104+
attributes["gen_ai.input.messages"] = [message._to_semconv_dict()]
105105

106106
return LogRecord(
107107
event_name="gen_ai.client.inference.operation.details",
@@ -247,7 +247,7 @@ def _maybe_set_input_messages(
247247
return
248248
message_parts: List[OtelMessage] = []
249249
for message in messages:
250-
message_parts.append(message._to_part_dict())
250+
message_parts.append(message._to_semconv_dict())
251251
if message_parts:
252252
span.set_attribute("gen_ai.input.messages", json.dumps(message_parts))
253253

@@ -390,7 +390,7 @@ def emit(self, invocation: LLMInvocation):
390390
request_model = invocation.attributes.get("request_model")
391391
span_state = _SpanState(
392392
span=span,
393-
span_context=get_current(),
393+
context=get_current(),
394394
request_model=request_model,
395395
system=system,
396396
start_time=invocation.start_time,
@@ -462,7 +462,7 @@ def error(self, error: Error, invocation: LLMInvocation):
462462

463463
span_state = _SpanState(
464464
span=span,
465-
span_context=get_current(),
465+
context=get_current(),
466466
request_model=request_model,
467467
system=system,
468468
start_time=invocation.start_time,
@@ -562,7 +562,7 @@ def emit(self, invocation: LLMInvocation):
562562
request_model = invocation.attributes.get("request_model")
563563
span_state = _SpanState(
564564
span=span,
565-
span_context=get_current(),
565+
context=get_current(),
566566
request_model=request_model,
567567
system=system,
568568
start_time=invocation.start_time,
@@ -635,7 +635,7 @@ def error(self, error: Error, invocation: LLMInvocation):
635635

636636
span_state = _SpanState(
637637
span=span,
638-
span_context=get_current(),
638+
context=get_current(),
639639
request_model=request_model,
640640
system=system,
641641
start_time=invocation.start_time,

0 commit comments

Comments
 (0)