Skip to content

Commit 47270d1

Browse files
committed
Add changelog, fix lint, fix ruff
1 parent e406a18 commit 47270d1

File tree

5 files changed

+15
-20
lines changed

5 files changed

+15
-20
lines changed

CHANGELOG.md

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
2323
([#3567](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3567))
2424
- `opentelemetry-resource-detector-containerid`: make it more quiet on platforms without cgroups
2525
([#3579](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3579))
26+
- `opentelemetry-instrumentation-botocore`: migrate off the deprecated events API to use the logs API
27+
([#3624](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3624))
2628

2729
### Added
2830

instrumentation/opentelemetry-instrumentation-botocore/src/opentelemetry/instrumentation/botocore/__init__.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -199,9 +199,8 @@ def _get_logger(self, extension: _AwsSdkExtension):
199199
"""This is a multiplexer in order to have a logger per extension"""
200200

201201
instrumentation_name = self._get_instrumentation_name(extension)
202-
logger = self._loggers.get(instrumentation_name)
203-
if logger:
204-
return logger
202+
if self._loggers.get(instrumentation_name):
203+
return self._loggers.get(instrumentation_name)
205204

206205
schema_version = extension.event_logger_schema_version()
207206
self._loggers[instrumentation_name] = get_logger(
@@ -287,11 +286,10 @@ def _patched_api_call(self, original_func, instance, args, kwargs):
287286
end_span_on_exit = extension.should_end_span_on_exit()
288287

289288
tracer = self._get_tracer(extension)
290-
logger = self._get_logger(extension)
291289
meter = self._get_meter(extension)
292290
metrics = self._get_metrics(extension, meter)
293291
instrumentor_ctx = _BotocoreInstrumentorContext(
294-
logger=logger,
292+
logger=self._get_logger(extension),
295293
metrics=metrics,
296294
)
297295
with tracer.start_as_current_span(

instrumentation/opentelemetry-instrumentation-botocore/src/opentelemetry/instrumentation/botocore/extensions/bedrock_utils.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -21,9 +21,9 @@
2121

2222
from botocore.eventstream import EventStream, EventStreamError
2323
from wrapt import ObjectProxy
24-
from opentelemetry.context import get_current
25-
from opentelemetry._logs import LogRecord
2624

25+
from opentelemetry._logs import LogRecord
26+
from opentelemetry.context import get_current
2727
from opentelemetry.instrumentation.botocore.environment_variables import (
2828
OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT,
2929
)
@@ -512,14 +512,14 @@ def message_to_event(
512512
event_name="gen_ai.tool.message",
513513
attributes=attributes,
514514
body=tool_body,
515-
context = get_current(),
515+
context=get_current(),
516516
)
517517

518518
yield LogRecord(
519519
event_name=f"gen_ai.{role}.message",
520520
attributes=attributes,
521521
body=body if body else None,
522-
context = get_current(),
522+
context=get_current(),
523523
)
524524

525525

@@ -627,5 +627,5 @@ def to_choice_event(self, **event_kwargs) -> LogRecord:
627627
attributes=attributes,
628628
body=self._to_body_dict(),
629629
**event_kwargs,
630-
context = get_current(),
630+
context=get_current(),
631631
)

instrumentation/opentelemetry-instrumentation-botocore/tests/bedrock_utils.py

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,6 @@
2626
)
2727
from opentelemetry.sdk.metrics._internal.point import ResourceMetrics
2828
from opentelemetry.sdk.trace import ReadableSpan
29-
from opentelemetry.semconv._incubating.attributes import (
30-
event_attributes as EventAttributes,
31-
)
3229
from opentelemetry.semconv._incubating.attributes import (
3330
gen_ai_attributes as GenAIAttributes,
3431
)
@@ -282,7 +279,9 @@ def remove_none_values(body):
282279

283280
def assert_log_parent(log, span):
284281
if span:
285-
assert log.log_record.trace_id == span.get_span_context().trace_id, "{} does not equal {}".format(span.get_span_context().trace_id, log.log_record.trace_id)
282+
assert (
283+
log.log_record.trace_id == span.get_span_context().trace_id
284+
), f"{span.get_span_context().trace_id} does not equal {log.log_record.trace_id}"
286285
assert log.log_record.span_id == span.get_span_context().span_id
287286
assert (
288287
log.log_record.trace_flags == span.get_span_context().trace_flags

instrumentation/opentelemetry-instrumentation-botocore/tests/conftest.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -99,9 +99,7 @@ def vcr_config():
9999

100100

101101
@pytest.fixture(scope="function")
102-
def instrument_no_content(
103-
tracer_provider, logger_provider, meter_provider
104-
):
102+
def instrument_no_content(tracer_provider, logger_provider, meter_provider):
105103
os.environ.update(
106104
{OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT: "False"}
107105
)
@@ -119,9 +117,7 @@ def instrument_no_content(
119117

120118

121119
@pytest.fixture(scope="function")
122-
def instrument_with_content(
123-
tracer_provider, logger_provider, meter_provider
124-
):
120+
def instrument_with_content(tracer_provider, logger_provider, meter_provider):
125121
os.environ.update(
126122
{OTEL_INSTRUMENTATION_GENAI_CAPTURE_MESSAGE_CONTENT: "True"}
127123
)

0 commit comments

Comments
 (0)