Skip to content

Commit b2fb1f9

Browse files
fix: Update test_logging_callback.py to use correct log parameters
1 parent 09295f8 commit b2fb1f9

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

tests/client/test_logging_callback.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ async def test_tool_with_log(
3939
"""Send a log notification to the client."""
4040
await server.get_context().log(
4141
level=level,
42-
message=message,
42+
data=message,
4343
logger_name=logger,
4444
)
4545
return True
@@ -55,9 +55,12 @@ async def test_tool_with_log_extra(
5555
"""Send a log notification to the client with extra fields."""
5656
await server.get_context().log(
5757
level=level,
58-
message=message,
58+
data={
59+
"message": message,
60+
"extra_string": extra_string,
61+
"extra_dict": extra_dict,
62+
},
5963
logger_name=logger,
60-
extra={"extra_string": extra_string, "extra_dict": extra_dict},
6164
)
6265
return True
6366

0 commit comments

Comments
 (0)