Skip to content

Commit a6f2214

Browse files
authored
Merge branch 'main' into fix-urllib-tests-flakyness
2 parents a824874 + 6572081 commit a6f2214

File tree

7 files changed

+194
-25
lines changed

7 files changed

+194
-25
lines changed

CHANGELOG.md

+4-2
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
1515

1616
- `opentelemetry-instrumentation-openai-v2` Update doc for OpenAI Instrumentation to support OpenAI Compatible Platforms
1717
([#3279](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3279))
18-
- `opentelemetry-instrumentation-system-metrics` Add `process` metrics and deprecated `process.runtime` prefixed ones
18+
- `opentelemetry-instrumentation-system-metrics` Add `process` metrics and deprecated `process.runtime` prefixed ones
1919
([#3250](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3250))
2020
- `opentelemetry-instrumentation-botocore` Add support for GenAI user events and lazy initialize tracer
2121
([#3258](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3258))
@@ -41,6 +41,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
4141
([#3247](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3247))
4242
- `opentelemetry-instrumentation-asyncpg` Fix fallback for empty queries.
4343
([#3253](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3253))
44+
- `opentelemetry-instrumentation-requests` always record span status code in duration metric
45+
([#3323](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3323))
4446

4547
## Version 1.30.0/0.51b0 (2025-02-03)
4648

@@ -100,7 +102,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
100102

101103
### Breaking changes
102104

103-
- `opentelemetry-exporter-prometheus-remote-write` updated protobuf required version from 4.21 to 5.26 and regenerated protobufs
105+
- `opentelemetry-exporter-prometheus-remote-write` updated protobuf required version from 4.21 to 5.26 and regenerated protobufs
104106
([#3219](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3219))
105107
- `opentelemetry-instrumentation-sqlalchemy` including sqlcomment in `db.statement` span attribute value is now opt-in
106108
([#3112](https://github.com/open-telemetry/opentelemetry-python-contrib/pull/3112))

gen-requirements.txt

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
-c dev-requirements.txt
22
astor==0.8.1
3-
jinja2==3.1.4
3+
jinja2==3.1.6
44
markupsafe==2.0.1
55
ruff==0.6.9
66
requests

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

+4
Original file line numberDiff line numberDiff line change
@@ -386,6 +386,10 @@ def extract_tool_results(
386386
if not content:
387387
return
388388

389+
# langchain sends content as string with InvokeModel and Anthropic Claude
390+
if isinstance(content, str):
391+
return
392+
389393
# Converse format
390394
tool_results = [
391395
item["toolResult"] for item in content if "toolResult" in item
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,74 @@
1+
interactions:
2+
- request:
3+
body: |-
4+
{
5+
"messages": [
6+
{
7+
"role": "user",
8+
"content": "say this is a test"
9+
}
10+
],
11+
"max_tokens": 10,
12+
"anthropic_version": "bedrock-2023-05-31"
13+
}
14+
headers:
15+
Content-Length:
16+
- '126'
17+
User-Agent:
18+
- Boto3/1.35.56 md/Botocore#1.35.56 ua/2.0 os/linux#6.1.0-1034-oem md/arch#x86_64
19+
lang/python#3.10.12 md/pyimpl#CPython cfg/retry-mode#legacy Botocore/1.35.56
20+
X-Amz-Date:
21+
- 20250306T091535Z
22+
X-Amz-Security-Token:
23+
- test_aws_security_token
24+
X-Amzn-Trace-Id:
25+
- Root=1-fcd0825a-03f328cca8cde3e741cd83b4;Parent=5a8eccf8d7b031e7;Sampled=1
26+
amz-sdk-invocation-id:
27+
- e8a6c9c5-8a45-4ad7-881b-0761d121abc7
28+
amz-sdk-request:
29+
- attempt=1
30+
authorization:
31+
- Bearer test_aws_authorization
32+
method: POST
33+
uri: https://bedrock-runtime.us-east-1.amazonaws.com/model/us.anthropic.claude-3-5-sonnet-20240620-v1%3A0/invoke
34+
response:
35+
body:
36+
string: |-
37+
{
38+
"id": "msg_bdrk_01SjMAZgb8kNMweUzf3moSbU",
39+
"type": "message",
40+
"role": "assistant",
41+
"model": "claude-3-5-sonnet-20240620",
42+
"content": [
43+
{
44+
"type": "text",
45+
"text": "This is a test."
46+
}
47+
],
48+
"stop_reason": "end_turn",
49+
"stop_sequence": null,
50+
"usage": {
51+
"input_tokens": 12,
52+
"output_tokens": 8
53+
}
54+
}
55+
headers:
56+
Connection:
57+
- keep-alive
58+
Content-Type:
59+
- application/json
60+
Date:
61+
- Thu, 06 Mar 2025 09:15:36 GMT
62+
Set-Cookie: test_set_cookie
63+
X-Amzn-Bedrock-Input-Token-Count:
64+
- '12'
65+
X-Amzn-Bedrock-Invocation-Latency:
66+
- '544'
67+
X-Amzn-Bedrock-Output-Token-Count:
68+
- '8'
69+
x-amzn-RequestId:
70+
- e97cb4f9-61fe-4a62-a29a-a582ddd17414
71+
status:
72+
code: 200
73+
message: OK
74+
version: 1

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

+47
Original file line numberDiff line numberDiff line change
@@ -1256,6 +1256,53 @@ def test_invoke_model_with_content(
12561256
assert_message_in_logs(logs[1], "gen_ai.choice", choice_body, span)
12571257

12581258

1259+
@pytest.mark.vcr()
1260+
def test_invoke_model_with_content_user_content_as_string(
1261+
span_exporter,
1262+
log_exporter,
1263+
bedrock_runtime_client,
1264+
instrument_with_content,
1265+
):
1266+
llm_model_value = "us.anthropic.claude-3-5-sonnet-20240620-v1:0"
1267+
max_tokens = 10
1268+
body = json.dumps(
1269+
{
1270+
"messages": [{"role": "user", "content": "say this is a test"}],
1271+
"max_tokens": max_tokens,
1272+
"anthropic_version": "bedrock-2023-05-31",
1273+
}
1274+
)
1275+
response = bedrock_runtime_client.invoke_model(
1276+
body=body,
1277+
modelId=llm_model_value,
1278+
)
1279+
1280+
(span,) = span_exporter.get_finished_spans()
1281+
assert_completion_attributes_from_streaming_body(
1282+
span,
1283+
llm_model_value,
1284+
response,
1285+
"chat",
1286+
request_max_tokens=max_tokens,
1287+
)
1288+
1289+
logs = log_exporter.get_finished_logs()
1290+
assert len(logs) == 2
1291+
user_content = {"content": "say this is a test"}
1292+
assert_message_in_logs(logs[0], "gen_ai.user.message", user_content, span)
1293+
1294+
message = {
1295+
"role": "assistant",
1296+
"content": [{"type": "text", "text": "This is a test."}],
1297+
}
1298+
choice_body = {
1299+
"index": 0,
1300+
"finish_reason": "end_turn",
1301+
"message": message,
1302+
}
1303+
assert_message_in_logs(logs[1], "gen_ai.choice", choice_body, span)
1304+
1305+
12591306
@pytest.mark.parametrize(
12601307
"model_family",
12611308
["amazon.nova", "anthropic.claude"],

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

+33-22
Original file line numberDiff line numberDiff line change
@@ -101,15 +101,14 @@ def response_hook(span, request_obj, response):
101101
_set_http_network_protocol_version,
102102
_set_http_peer_port_client,
103103
_set_http_scheme,
104-
_set_http_status_code,
105104
_set_http_url,
105+
_set_status,
106106
_StabilityMode,
107107
)
108108
from opentelemetry.instrumentation.instrumentor import BaseInstrumentor
109109
from opentelemetry.instrumentation.requests.package import _instruments
110110
from opentelemetry.instrumentation.requests.version import __version__
111111
from opentelemetry.instrumentation.utils import (
112-
http_status_to_status_code,
113112
is_http_instrumentation_enabled,
114113
suppress_http_instrumentation,
115114
)
@@ -126,7 +125,6 @@ def response_hook(span, request_obj, response):
126125
)
127126
from opentelemetry.trace import SpanKind, Tracer, get_tracer
128127
from opentelemetry.trace.span import Span
129-
from opentelemetry.trace.status import StatusCode
130128
from opentelemetry.util.http import (
131129
ExcludeList,
132130
get_excluded_urls,
@@ -142,6 +140,32 @@ def response_hook(span, request_obj, response):
142140
_ResponseHookT = Optional[Callable[[Span, PreparedRequest, Response], None]]
143141

144142

143+
def _set_http_status_code_attribute(
144+
span,
145+
status_code,
146+
metric_attributes=None,
147+
sem_conv_opt_in_mode=_StabilityMode.DEFAULT,
148+
):
149+
status_code_str = str(status_code)
150+
try:
151+
status_code = int(status_code)
152+
except ValueError:
153+
status_code = -1
154+
if metric_attributes is None:
155+
metric_attributes = {}
156+
# When we have durations we should set metrics only once
157+
# Also the decision to include status code on a histogram should
158+
# not be dependent on tracing decisions.
159+
_set_status(
160+
span,
161+
metric_attributes,
162+
status_code,
163+
status_code_str,
164+
server_span=False,
165+
sem_conv_opt_in_mode=sem_conv_opt_in_mode,
166+
)
167+
168+
145169
# pylint: disable=unused-argument
146170
# pylint: disable=R0915
147171
def _instrument(
@@ -269,25 +293,12 @@ def get_or_create_headers():
269293

270294
if isinstance(result, Response):
271295
span_attributes = {}
272-
if span.is_recording():
273-
_set_http_status_code(
274-
span_attributes,
275-
result.status_code,
276-
sem_conv_opt_in_mode,
277-
)
278-
_set_http_status_code(
279-
metric_labels, result.status_code, sem_conv_opt_in_mode
280-
)
281-
status_code = http_status_to_status_code(
282-
result.status_code
283-
)
284-
span.set_status(status_code)
285-
if (
286-
_report_new(sem_conv_opt_in_mode)
287-
and status_code is StatusCode.ERROR
288-
):
289-
span_attributes[ERROR_TYPE] = str(result.status_code)
290-
metric_labels[ERROR_TYPE] = str(result.status_code)
296+
_set_http_status_code_attribute(
297+
span,
298+
result.status_code,
299+
metric_labels,
300+
sem_conv_opt_in_mode,
301+
)
291302

292303
if result.raw is not None:
293304
version = getattr(result.raw, "version", None)

instrumentation/opentelemetry-instrumentation-requests/tests/test_requests_integration.py

+31
Original file line numberDiff line numberDiff line change
@@ -832,3 +832,34 @@ def test_basic_metric_both_semconv(self):
832832
dict(data_point.attributes),
833833
)
834834
self.assertEqual(data_point.count, 1)
835+
836+
def test_basic_metric_non_recording_span(self):
837+
expected_attributes = {
838+
SpanAttributes.HTTP_STATUS_CODE: 200,
839+
SpanAttributes.HTTP_HOST: "examplehost",
840+
SpanAttributes.NET_PEER_PORT: 8000,
841+
SpanAttributes.NET_PEER_NAME: "examplehost",
842+
SpanAttributes.HTTP_METHOD: "GET",
843+
SpanAttributes.HTTP_FLAVOR: "1.1",
844+
SpanAttributes.HTTP_SCHEME: "http",
845+
}
846+
847+
with mock.patch("opentelemetry.trace.INVALID_SPAN") as mock_span:
848+
RequestsInstrumentor().uninstrument()
849+
RequestsInstrumentor().instrument(
850+
tracer_provider=trace.NoOpTracerProvider()
851+
)
852+
mock_span.is_recording.return_value = False
853+
result = self.perform_request(self.URL)
854+
self.assertEqual(result.text, "Hello!")
855+
self.assertFalse(mock_span.is_recording())
856+
self.assertTrue(mock_span.is_recording.called)
857+
self.assertFalse(mock_span.set_attribute.called)
858+
self.assertFalse(mock_span.set_status.called)
859+
metrics = self.get_sorted_metrics()
860+
self.assertEqual(len(metrics), 1)
861+
duration_data_point = metrics[0].data.data_points[0]
862+
self.assertDictEqual(
863+
expected_attributes, dict(duration_data_point.attributes)
864+
)
865+
self.assertEqual(duration_data_point.count, 1)

0 commit comments

Comments
 (0)