File tree Expand file tree Collapse file tree 2 files changed +10
-5
lines changed Expand file tree Collapse file tree 2 files changed +10
-5
lines changed Original file line number Diff line number Diff line change @@ -154,7 +154,6 @@ async def __aexit__(
154
154
for exit_stack in self ._session_exit_stacks .values ():
155
155
tg .start_soon (exit_stack .aclose )
156
156
157
-
158
157
@property
159
158
def sessions (self ) -> list [mcp .ClientSession ]:
160
159
"""Returns the list of sessions being managed."""
Original file line number Diff line number Diff line change @@ -451,7 +451,7 @@ async def mock_server():
451
451
# Assert that capabilities are properly set with defaults
452
452
assert received_capabilities is not None
453
453
assert received_capabilities .sampling is None # No custom sampling callback
454
- assert received_capabilities .roots is None # No custom list_roots callback
454
+ assert received_capabilities .roots is None # No custom list_roots callback
455
455
456
456
457
457
@pytest .mark .anyio
@@ -536,8 +536,14 @@ async def mock_server():
536
536
537
537
# Assert that capabilities are properly set with custom callbacks
538
538
assert received_capabilities is not None
539
- assert received_capabilities .sampling is not None # Custom sampling callback provided
539
+ assert (
540
+ received_capabilities .sampling is not None
541
+ ) # Custom sampling callback provided
540
542
assert isinstance (received_capabilities .sampling , types .SamplingCapability )
541
- assert received_capabilities .roots is not None # Custom list_roots callback provided
543
+ assert (
544
+ received_capabilities .roots is not None
545
+ ) # Custom list_roots callback provided
542
546
assert isinstance (received_capabilities .roots , types .RootsCapability )
543
- assert received_capabilities .roots .listChanged is True # Should be True for custom callback
547
+ assert (
548
+ received_capabilities .roots .listChanged is True
549
+ ) # Should be True for custom callback
You can’t perform that action at this time.
0 commit comments