Skip to content

Commit 76fdabd

Browse files
authored
replace id with session_id in kernel api functions (#219)
1 parent efa1262 commit 76fdabd

File tree

1 file changed

+10
-2
lines changed
  • src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/kernel

1 file changed

+10
-2
lines changed

src/lmnr/opentelemetry_lib/opentelemetry/instrumentation/kernel/__init__.py

Lines changed: 10 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -196,9 +196,13 @@ def _wrap(
196196
f"{to_wrap.get('class_name')}.{to_wrap.get('method')}",
197197
span_type=to_wrap.get("span_type", "DEFAULT"),
198198
) as span:
199+
input_kv = get_input_from_func_args(wrapped, True, args, kwargs)
200+
if "id" in input_kv:
201+
input_kv["session_id"] = input_kv.get("id")
202+
input_kv.pop("id")
199203
span.set_attribute(
200204
"lmnr.span.input",
201-
json_dumps(get_input_from_func_args(wrapped, True, args, kwargs)),
205+
json_dumps(input_kv),
202206
)
203207
try:
204208
result = wrapped(*args, **kwargs)
@@ -223,9 +227,13 @@ async def _wrap_async(
223227
f"{to_wrap.get('class_name')}.{to_wrap.get('method')}",
224228
span_type=to_wrap.get("span_type", "DEFAULT"),
225229
) as span:
230+
input_kv = get_input_from_func_args(wrapped, True, args, kwargs)
231+
if "id" in input_kv:
232+
input_kv["session_id"] = input_kv.get("id")
233+
input_kv.pop("id")
226234
span.set_attribute(
227235
"lmnr.span.input",
228-
json_dumps(get_input_from_func_args(wrapped, True, args, kwargs)),
236+
json_dumps(input_kv),
229237
)
230238
try:
231239
result = await wrapped(*args, **kwargs)

0 commit comments

Comments
 (0)