Skip to content

Commit baef8ef

Browse files
committed
fix: streamable http mcp params
1 parent efa8b0d commit baef8ef

File tree

1 file changed

+14
-4
lines changed
  • packages/opentelemetry-instrumentation-mcp/opentelemetry/instrumentation/mcp

1 file changed

+14
-4
lines changed

packages/opentelemetry-instrumentation-mcp/opentelemetry/instrumentation/mcp/instrumentation.py

Lines changed: 14 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -130,11 +130,21 @@ async def traced_method(
130130
async with wrapped(*args, **kwargs) as result:
131131
try:
132132
read_stream, write_stream = result
133+
yield InstrumentedStreamReader(
134+
read_stream, tracer
135+
), InstrumentedStreamWriter(write_stream, tracer)
133136
except ValueError:
134-
read_stream, write_stream, _ = result
135-
yield InstrumentedStreamReader(
136-
read_stream, tracer
137-
), InstrumentedStreamWriter(write_stream, tracer)
137+
try:
138+
read_stream, write_stream, get_session_id_callback = result
139+
yield InstrumentedStreamReader(
140+
read_stream, tracer
141+
), InstrumentedStreamWriter(write_stream, tracer), get_session_id_callback
142+
except Exception as e:
143+
logging.warning(f"mcp instrumentation _transport_wrapper exception: {e}")
144+
yield result
145+
except Exception as e:
146+
logging.warning(f"mcp instrumentation transport_wrapper exception: {e}")
147+
yield result
138148

139149
return traced_method
140150

0 commit comments

Comments
 (0)