Skip to content

Commit 9c2a083

Browse files
committed
refactor: improve docstring clarity in MongoDBSession and related tests
1 parent a861452 commit 9c2a083

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

src/agents/extensions/memory/mongodb_session.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ def __init__(
109109
self._client = client
110110
self._owns_client = False
111111

112-
# Pattern B: annotate an externally-supplied client with library metadata.
112+
# Annotate an externally-supplied client with library metadata.
113113
# append_metadata is available in PyMongo >=4.14; guard for older installs.
114114
if hasattr(client, "append_metadata"):
115115
client.append_metadata(_DRIVER_INFO)

tests/extensions/memory/test_mongodb_session.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -645,7 +645,7 @@ async def test_runner_with_session_settings_limit(agent: Agent) -> None:
645645

646646

647647
async def test_injected_client_receives_append_metadata() -> None:
648-
"""Pattern B: append_metadata is called on a caller-supplied client."""
648+
"""Append_metadata is called on a caller-supplied client."""
649649
MongoDBSession._initialized_keys.clear()
650650
MongoDBSession._init_locks.clear()
651651
client = FakeAsyncMongoClient()
@@ -658,7 +658,7 @@ async def test_injected_client_receives_append_metadata() -> None:
658658

659659

660660
async def test_from_uri_passes_driver_info_to_constructor() -> None:
661-
"""Pattern A: driver=_DRIVER_INFO is forwarded to AsyncMongoClient via from_uri."""
661+
"""driver=_DRIVER_INFO is forwarded to AsyncMongoClient via from_uri."""
662662
MongoDBSession._initialized_keys.clear()
663663
MongoDBSession._init_locks.clear()
664664

@@ -679,7 +679,7 @@ def _fake_client(uri: str, **kwargs: Any) -> FakeAsyncMongoClient:
679679

680680

681681
async def test_caller_supplied_driver_info_is_not_overwritten() -> None:
682-
"""Pattern A: a caller-supplied driver kwarg must not be silently replaced."""
682+
"""A caller-supplied driver kwarg must not be silently replaced."""
683683
MongoDBSession._initialized_keys.clear()
684684
MongoDBSession._init_locks.clear()
685685

0 commit comments

Comments
 (0)