Skip to content

Commit b8b84af

Browse files
committed
Reformat with ruff.
1 parent b3b1f6d commit b8b84af

File tree

2 files changed

+29
-12
lines changed

2 files changed

+29
-12
lines changed

instrumentation-genai/opentelemetry-instrumentation-google-genai/src/opentelemetry/instrumentation/google_genai/tool_call_wrapper.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -97,15 +97,15 @@ def _record_function_call_arguments(
9797
span.set_attribute(value_attribute, _to_otel_value(value))
9898

9999

100-
def _record_function_call_result(
101-
otel_wrapper, wrapped_function, result
102-
):
100+
def _record_function_call_result(otel_wrapper, wrapped_function, result):
103101
"""Records the details about a function result as span attributes."""
104102
include_values = is_content_recording_enabled()
105103
span = trace.get_current_span()
106104
span.set_attribute("code.function.return.type", type(result).__name__)
107105
if include_values:
108-
span.set_attribute("code.function.return.value", _to_otel_value(result))
106+
span.set_attribute(
107+
"code.function.return.value", _to_otel_value(result)
108+
)
109109

110110

111111
def _wrap_sync_tool_function(

instrumentation-genai/opentelemetry-instrumentation-google-genai/tests/generate_content/test_tool_call_instrumentation.py

Lines changed: 25 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -111,17 +111,27 @@ def somefunction(someparam, otherparam=2):
111111
self.otel.assert_has_span_named("tool_call somefunction")
112112
generated_span = self.otel.get_span_named("tool_call somefunction")
113113
self.assertEqual(
114-
generated_span.attributes["code.function.parameters.someparam.type"], "int"
114+
generated_span.attributes[
115+
"code.function.parameters.someparam.type"
116+
],
117+
"int",
115118
)
116119
self.assertEqual(
117-
generated_span.attributes["code.function.parameters.otherparam.type"],
120+
generated_span.attributes[
121+
"code.function.parameters.otherparam.type"
122+
],
118123
"str",
119124
)
120125
self.assertEqual(
121-
generated_span.attributes["code.function.parameters.someparam.value"], "123"
126+
generated_span.attributes[
127+
"code.function.parameters.someparam.value"
128+
],
129+
"123",
122130
)
123131
self.assertEqual(
124-
generated_span.attributes["code.function.parameters.otherparam.value"],
132+
generated_span.attributes[
133+
"code.function.parameters.otherparam.value"
134+
],
125135
"'abc'",
126136
)
127137

@@ -154,17 +164,24 @@ def somefunction(someparam, otherparam=2):
154164
self.otel.assert_has_span_named("tool_call somefunction")
155165
generated_span = self.otel.get_span_named("tool_call somefunction")
156166
self.assertEqual(
157-
generated_span.attributes["code.function.parameters.someparam.type"], "int"
167+
generated_span.attributes[
168+
"code.function.parameters.someparam.type"
169+
],
170+
"int",
158171
)
159172
self.assertEqual(
160-
generated_span.attributes["code.function.parameters.otherparam.type"],
173+
generated_span.attributes[
174+
"code.function.parameters.otherparam.type"
175+
],
161176
"str",
162177
)
163178
self.assertNotIn(
164-
"code.function.parameters.someparam.value", generated_span.attributes
179+
"code.function.parameters.someparam.value",
180+
generated_span.attributes,
165181
)
166182
self.assertNotIn(
167-
"code.function.parameters.otherparam.value", generated_span.attributes
183+
"code.function.parameters.otherparam.value",
184+
generated_span.attributes,
168185
)
169186

170187
def test_tool_calls_record_return_values_on_span_if_enabled(self):

0 commit comments

Comments
 (0)