Skip to content

Commit e92d6c5

Browse files
committed
Fix TypeError: Remove await from background_tasks.add_task calls
The HybridBackgroundTasks.add_task() method returns None and should not be awaited. This was causing production errors with 'TypeError: object NoneType can't be used in 'await' expression'. Fixed in: - agent_memory_server/api.py (3 locations) - agent_memory_server/long_term_memory.py (2 locations) Resolves the 500 Internal Server Error on POST /v1/long-term-memory/
1 parent 68c5138 commit e92d6c5

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

agent_memory_server/api.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -507,7 +507,7 @@ async def put_working_memory(
507507
updated_memory.memories or updated_memory.messages
508508
):
509509
# Promote structured memories from working memory to long-term storage
510-
await background_tasks.add_task(
510+
background_tasks.add_task(
511511
long_term_memory.promote_working_memory_to_long_term,
512512
session_id=session_id,
513513
user_id=updated_memory.user_id,
@@ -596,7 +596,7 @@ async def create_long_term_memory(
596596
# Clear any client-provided persisted_at value
597597
memory.persisted_at = None
598598

599-
await background_tasks.add_task(
599+
background_tasks.add_task(
600600
long_term_memory.index_long_term_memories,
601601
memories=payload.memories,
602602
)
@@ -732,7 +732,7 @@ def _vals(f):
732732
ids = [m.id for m in ranked if m.id]
733733
if ids:
734734
background_tasks = get_background_tasks()
735-
await background_tasks.add_task(long_term_memory.update_last_accessed, ids)
735+
background_tasks.add_task(long_term_memory.update_last_accessed, ids)
736736

737737
raw_results.memories = ranked
738738
return raw_results

agent_memory_server/long_term_memory.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -842,7 +842,7 @@ async def index_long_term_memories(
842842

843843
# Schedule background tasks for topic/entity extraction
844844
for memory in processed_memories:
845-
await background_tasks.add_task(extract_memory_structure, memory)
845+
background_tasks.add_task(extract_memory_structure, memory)
846846

847847
if settings.enable_discrete_memory_extraction:
848848
needs_extraction = [
@@ -853,7 +853,7 @@ async def index_long_term_memories(
853853
# Extract discrete memories from the indexed messages and persist
854854
# them as separate long-term memory records. This process also
855855
# runs deduplication if requested.
856-
await background_tasks.add_task(
856+
background_tasks.add_task(
857857
extract_memories_with_strategy,
858858
memories=needs_extraction,
859859
deduplicate=deduplicate,

0 commit comments

Comments
 (0)