Skip to content

Commit 9959e67

Browse files
committed
rename methods to improve clarity and consistency
1 parent 09becf0 commit 9959e67

File tree

3 files changed

+8
-8
lines changed

3 files changed

+8
-8
lines changed

examples/memory/advanced_sqlite_session_example_2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -88,7 +88,7 @@ async def main():
8888

8989
# Soft delete from turn 2 to create a branch point
9090
print("\nSoft deleting from turn 2 onwards to create branch point...")
91-
deleted = await session.soft_delete_from_turn(2)
91+
deleted = await session.deactivate_from_turn(2)
9292
print(f"Deleted: {deleted}")
9393

9494
# Show only active items (turn 1 only)

src/agents/extensions/memory/advanced_sqlite_session.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def _add_structure_sync():
194194
current_turn += 1
195195
self._current_user_turn = current_turn
196196

197-
msg_type = self._classify_simple(item)
197+
msg_type = self._classify_message_type(item)
198198
tool_name = self._extract_tool_name(item)
199199

200200
structure_data.append(
@@ -228,7 +228,7 @@ def _add_structure_sync():
228228
)
229229
# Don't re-raise - structure metadata is supplementary
230230

231-
def _classify_simple(self, item: TResponseInputItem) -> str:
231+
def _classify_message_type(self, item: TResponseInputItem) -> str:
232232
"""Simple classification."""
233233
if isinstance(item, dict):
234234
if item.get("role") == "user":
@@ -328,10 +328,10 @@ def _get_active_items_sync():
328328

329329
return await asyncio.to_thread(_get_active_items_sync)
330330

331-
async def soft_delete_from_turn(self, user_turn_number: int) -> bool:
331+
async def deactivate_from_turn(self, user_turn_number: int) -> bool:
332332
"""Soft delete conversation from a specific user turn onwards."""
333333

334-
def _soft_delete_sync():
334+
def _deactivate_sync():
335335
conn = self._get_connection()
336336
with self._lock if self._is_memory_db else threading.Lock():
337337
with closing(conn.cursor()) as cursor:
@@ -347,7 +347,7 @@ def _soft_delete_sync():
347347
conn.commit()
348348
return affected > 0
349349

350-
return await asyncio.to_thread(_soft_delete_sync)
350+
return await asyncio.to_thread(_deactivate_sync)
351351

352352
async def reactivate_from_turn(self, user_turn_number: int) -> bool:
353353
"""Reactivate soft-deleted conversation from a specific turn."""

tests/extensions/memory/test_advanced_sqlite_session.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -185,7 +185,7 @@ async def test_soft_deletion_functionality(agent: Agent):
185185
assert len(all_items) == 6
186186

187187
# Soft delete from turn 2 onwards
188-
deleted = await session.soft_delete_from_turn(2)
188+
deleted = await session.deactivate_from_turn(2)
189189
assert deleted is True
190190

191191
# Verify only turn 1 items are active
@@ -414,7 +414,7 @@ async def test_empty_session_operations():
414414
assert session_usage is None
415415

416416
# Test soft deletion on empty session
417-
deleted = await session.soft_delete_from_turn(1)
417+
deleted = await session.deactivate_from_turn(1)
418418
assert deleted is False
419419

420420
session.close()

0 commit comments

Comments
 (0)