@@ -144,7 +144,7 @@ def test_create_message_with_payloads(
144144 "event" : "request_start" ,
145145 "source" : "client" ,
146146 "method" : "test_method" ,
147- "payload" : '{"method":"tools/call","params":{"_meta":null,"name":"test_method","arguments":{"param":"value"}}}' ,
147+ "payload" : '{"method":"tools/call","params":{"task":null," _meta":null,"name":"test_method","arguments":{"param":"value"}}}' ,
148148 "payload_type" : "CallToolRequest" ,
149149 }
150150 )
@@ -159,7 +159,7 @@ def test_calculate_response_size(self, mock_context: MiddlewareContext[Any]):
159159 "event" : "request_start" ,
160160 "source" : "client" ,
161161 "method" : "test_method" ,
162- "payload_length" : 98 ,
162+ "payload_length" : 110 ,
163163 }
164164 )
165165
@@ -177,8 +177,8 @@ def test_calculate_response_size_with_token_estimation(
177177 "event" : "request_start" ,
178178 "source" : "client" ,
179179 "method" : "test_method" ,
180- "payload_tokens" : 24 ,
181- "payload_length" : 98 ,
180+ "payload_tokens" : 27 ,
181+ "payload_length" : 110 ,
182182 }
183183 )
184184
@@ -303,7 +303,7 @@ async def test_on_message_with_pydantic_types_in_payload(
303303
304304 assert get_log_lines (caplog ) == snapshot (
305305 [
306- '{"event": "request_start", "method": "test_method", "source": "client", "payload": "{\\ "method\\ ":\\ "resources/read\\ ",\\ "params\\ ":{\\ "_meta\\ ":null,\\ "uri\\ ":\\ "test://example/1\\ "}}", "payload_type": "ReadResourceRequest"}' ,
306+ '{"event": "request_start", "method": "test_method", "source": "client", "payload": "{\\ "method\\ ":\\ "resources/read\\ ",\\ "params\\ ":{\\ "task \\ ":null, \\ " _meta\\ ":null,\\ "uri\\ ":\\ "test://example/1\\ "}}", "payload_type": "ReadResourceRequest"}' ,
307307 '{"event": "request_success", "method": "test_method", "source": "client", "duration_ms": 0.02}' ,
308308 ]
309309 )
@@ -365,7 +365,7 @@ def __str__(self) -> str:
365365
366366 assert get_log_lines (caplog ) == snapshot (
367367 [
368- '{"event": "request_start", "method": "test_method", "source": "client", "payload": "{\\ "method\\ ":\\ "tools/call\\ ",\\ "params\\ ":{\\ "_meta\\ ":null,\\ "name\\ ":\\ "test_method\\ ",\\ "arguments\\ ":{\\ "obj\\ ":\\ "NON_SERIALIZABLE\\ "}}}", "payload_type": "CallToolRequest"}' ,
368+ '{"event": "request_start", "method": "test_method", "source": "client", "payload": "{\\ "method\\ ":\\ "tools/call\\ ",\\ "params\\ ":{\\ "task \\ ":null, \\ " _meta\\ ":null,\\ "name\\ ":\\ "test_method\\ ",\\ "arguments\\ ":{\\ "obj\\ ":\\ "NON_SERIALIZABLE\\ "}}}", "payload_type": "CallToolRequest"}' ,
369369 '{"event": "request_success", "method": "test_method", "source": "client", "duration_ms": 0.02}' ,
370370 ]
371371 )
@@ -546,7 +546,7 @@ async def test_logging_middleware_with_payloads(
546546
547547 assert get_log_lines (caplog ) == snapshot (
548548 [
549- 'event=request_start method=tools/call source=client payload={"_meta":null,"name":"simple_operation","arguments":{"data":"payload_test"}} payload_type=CallToolRequestParams' ,
549+ 'event=request_start method=tools/call source=client payload={"task":null," _meta":null,"name":"simple_operation","arguments":{"data":"payload_test"}} payload_type=CallToolRequestParams' ,
550550 "event=request_success method=tools/call source=client duration_ms=0.02" ,
551551 ]
552552 )
@@ -570,7 +570,7 @@ async def test_structured_logging_middleware_produces_json(
570570
571571 assert get_log_lines (caplog ) == snapshot (
572572 [
573- '{"event": "request_start", "method": "tools/call", "source": "client", "payload": "{\\ "_meta\\ ":null,\\ "name\\ ":\\ "simple_operation\\ ",\\ "arguments\\ ":{\\ "data\\ ":\\ "json_test\\ "}}", "payload_type": "CallToolRequestParams"}' ,
573+ '{"event": "request_start", "method": "tools/call", "source": "client", "payload": "{\\ "task \\ ":null, \\ " _meta\\ ":null,\\ "name\\ ":\\ "simple_operation\\ ",\\ "arguments\\ ":{\\ "data\\ ":\\ "json_test\\ "}}", "payload_type": "CallToolRequestParams"}' ,
574574 '{"event": "request_success", "method": "tools/call", "source": "client", "duration_ms": 0.02}' ,
575575 ]
576576 )
@@ -665,6 +665,6 @@ async def test_logging_middleware_custom_configuration(
665665 # Check that our custom logger captured the logs
666666 log_output = log_buffer .getvalue ()
667667 assert log_output == snapshot ("""\
668- event=request_start method=tools/call source=client payload={"_meta":null,"name":"simple_operation","arguments":{"data":"custom_test"}} payload_type=CallToolRequestParams
668+ event=request_start method=tools/call source=client payload={"task":null," _meta":null,"name":"simple_operation","arguments":{"data":"custom_test"}} payload_type=CallToolRequestParams
669669event=request_success method=tools/call source=client duration_ms=0.02
670670""" )
0 commit comments