|
65 | 65 | @dataclass |
66 | 66 | class _SpanState: |
67 | 67 | span: Span |
68 | | - span_context: Context |
| 68 | + context: Context |
69 | 69 | start_time: float |
70 | 70 | request_model: Optional[str] = None |
71 | 71 | system: Optional[str] = None |
@@ -101,7 +101,7 @@ def _message_to_log_record( |
101 | 101 | } |
102 | 102 |
|
103 | 103 | if capture_content: |
104 | | - attributes["gen_ai.input.messages"] = [message._to_part_dict()] |
| 104 | + attributes["gen_ai.input.messages"] = [message._to_semconv_dict()] |
105 | 105 |
|
106 | 106 | return LogRecord( |
107 | 107 | event_name="gen_ai.client.inference.operation.details", |
@@ -247,7 +247,7 @@ def _maybe_set_input_messages( |
247 | 247 | return |
248 | 248 | message_parts: List[OtelMessage] = [] |
249 | 249 | for message in messages: |
250 | | - message_parts.append(message._to_part_dict()) |
| 250 | + message_parts.append(message._to_semconv_dict()) |
251 | 251 | if message_parts: |
252 | 252 | span.set_attribute("gen_ai.input.messages", json.dumps(message_parts)) |
253 | 253 |
|
@@ -390,7 +390,7 @@ def emit(self, invocation: LLMInvocation): |
390 | 390 | request_model = invocation.attributes.get("request_model") |
391 | 391 | span_state = _SpanState( |
392 | 392 | span=span, |
393 | | - span_context=get_current(), |
| 393 | + context=get_current(), |
394 | 394 | request_model=request_model, |
395 | 395 | system=system, |
396 | 396 | start_time=invocation.start_time, |
@@ -462,7 +462,7 @@ def error(self, error: Error, invocation: LLMInvocation): |
462 | 462 |
|
463 | 463 | span_state = _SpanState( |
464 | 464 | span=span, |
465 | | - span_context=get_current(), |
| 465 | + context=get_current(), |
466 | 466 | request_model=request_model, |
467 | 467 | system=system, |
468 | 468 | start_time=invocation.start_time, |
@@ -562,7 +562,7 @@ def emit(self, invocation: LLMInvocation): |
562 | 562 | request_model = invocation.attributes.get("request_model") |
563 | 563 | span_state = _SpanState( |
564 | 564 | span=span, |
565 | | - span_context=get_current(), |
| 565 | + context=get_current(), |
566 | 566 | request_model=request_model, |
567 | 567 | system=system, |
568 | 568 | start_time=invocation.start_time, |
@@ -635,7 +635,7 @@ def error(self, error: Error, invocation: LLMInvocation): |
635 | 635 |
|
636 | 636 | span_state = _SpanState( |
637 | 637 | span=span, |
638 | | - span_context=get_current(), |
| 638 | + context=get_current(), |
639 | 639 | request_model=request_model, |
640 | 640 | system=system, |
641 | 641 | start_time=invocation.start_time, |
|
0 commit comments